104 lines
3.0 KiB
Rust
104 lines
3.0 KiB
Rust
use colored::Colorize;
|
|
use discord_rich_presence::{
|
|
activity::{self, Activity},
|
|
DiscordIpc, DiscordIpcClient,
|
|
};
|
|
use std::{
|
|
time::{SystemTime, UNIX_EPOCH},
|
|
vec,
|
|
};
|
|
|
|
use crate::cli::Cli;
|
|
|
|
pub fn run(args: Cli) {
|
|
let state = args.state;
|
|
let details = args.details;
|
|
let large_image = args.large_image;
|
|
let large_text = args.large_text;
|
|
let small_image = args.small_image;
|
|
let small_text = args.small_text;
|
|
let button_1_text = args.button_1_text;
|
|
let button_1_url = args.button_1_url;
|
|
let button_2_text = args.button_2_text;
|
|
let button_2_url = args.button_2_url;
|
|
let enable_timer = args.enable_time;
|
|
|
|
let mut client = DiscordIpcClient::new(&args.client_id).expect("failed to create client");
|
|
|
|
let activity = Activity::new();
|
|
|
|
client
|
|
.connect()
|
|
.expect("failed to connect to Discord, please try again or relaunch Discord app");
|
|
|
|
println!("{} {}", "details :".cyan(), details.yellow());
|
|
|
|
let mut activity: Activity = activity.details(&details);
|
|
|
|
if state != "none" {
|
|
println!("{} {}", "state :".cyan(), state.yellow());
|
|
|
|
activity = activity.state(&state);
|
|
}
|
|
|
|
let mut assets = activity::Assets::new();
|
|
|
|
if large_image != "none" {
|
|
println!("{} {}", "large image :".cyan(), large_image.yellow());
|
|
|
|
assets = assets.large_image(&large_image);
|
|
}
|
|
|
|
if large_text != "none" {
|
|
println!("{} {}", "large image text :".cyan(), large_text.yellow());
|
|
|
|
assets = assets.large_text(&large_text);
|
|
}
|
|
|
|
if small_image != "none" {
|
|
println!("{} {}", "small image :".cyan(), small_image.yellow());
|
|
|
|
assets = assets.small_image(&small_image);
|
|
}
|
|
|
|
if small_text != "none" {
|
|
println!("{} {}", "small image text :".cyan(), small_text.yellow());
|
|
|
|
assets = assets.small_text(&small_text);
|
|
}
|
|
|
|
activity = activity.assets(assets);
|
|
|
|
if button_1_text != "none" && button_1_url != "none" {
|
|
println!("{} {}", "button 1 text :".cyan(), button_1_text.yellow());
|
|
println!("{} {}", "button 1 url :".cyan(), button_1_url.yellow());
|
|
|
|
activity = activity.buttons(vec![activity::Button::new(&button_1_text, &button_1_url)]);
|
|
}
|
|
|
|
if button_1_text != "none"
|
|
&& button_1_url != "none"
|
|
&& button_2_text != "none"
|
|
&& button_2_url != "none"
|
|
{
|
|
println!("{} {}", "button 2 text :".cyan(), button_2_text.yellow());
|
|
println!("{} {}", "button 2 url :".cyan(), button_2_url.yellow());
|
|
|
|
activity = activity.buttons(vec![
|
|
activity::Button::new(&button_1_text, &button_1_url),
|
|
activity::Button::new(&button_2_text, &button_2_url),
|
|
]);
|
|
}
|
|
|
|
if enable_timer {
|
|
let time_unix = SystemTime::now()
|
|
.duration_since(UNIX_EPOCH)
|
|
.unwrap()
|
|
.as_secs() as i64;
|
|
|
|
activity = activity.timestamps(activity::Timestamps::new().start(time_unix))
|
|
}
|
|
|
|
client.set_activity(activity).expect("client set activity");
|
|
}
|