Track start time in clients
This commit is contained in:
parent
a6571a51c1
commit
c71b2dbf22
5 changed files with 50 additions and 10 deletions
|
|
@ -1,7 +1,6 @@
|
|||
use std::{fmt::Debug, sync::Arc};
|
||||
|
||||
use euphoxide_client::{MultiClient, MultiClientConfig, MultiClientEvent, ServerConfig};
|
||||
use jiff::Timestamp;
|
||||
use log::error;
|
||||
use tokio::sync::mpsc;
|
||||
|
||||
|
|
@ -12,7 +11,6 @@ pub struct Bot<E = euphoxide::Error> {
|
|||
pub server_config: ServerConfig,
|
||||
pub commands: Arc<Commands<E>>,
|
||||
pub clients: MultiClient,
|
||||
pub start_time: Timestamp,
|
||||
}
|
||||
|
||||
impl Bot {
|
||||
|
|
@ -37,10 +35,10 @@ impl<E> Bot<E> {
|
|||
server_config,
|
||||
commands: Arc::new(commands),
|
||||
clients: MultiClient::new_with_config(clients_config, event_tx),
|
||||
start_time: Timestamp::now(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<E> Bot<E>
|
||||
where
|
||||
E: Debug + 'static,
|
||||
|
|
@ -61,7 +59,6 @@ impl<E> Clone for Bot<E> {
|
|||
server_config: self.server_config.clone(),
|
||||
commands: self.commands.clone(),
|
||||
clients: self.clients.clone(),
|
||||
start_time: self.start_time,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -62,8 +62,14 @@ pub trait HasStartTime {
|
|||
}
|
||||
|
||||
impl Uptime {
|
||||
fn formulate_reply<E>(&self, ctx: &Context, bot: &Bot<E>, connected: bool) -> String {
|
||||
let start = bot.start_time;
|
||||
fn formulate_reply<E>(
|
||||
&self,
|
||||
ctx: &Context,
|
||||
bot: &Bot<E>,
|
||||
joined: bool,
|
||||
connected: bool,
|
||||
) -> String {
|
||||
let start = bot.clients.start_time();
|
||||
let now = Timestamp::now();
|
||||
|
||||
let mut reply = format!(
|
||||
|
|
@ -72,6 +78,15 @@ impl Uptime {
|
|||
format_relative_time(start - now),
|
||||
);
|
||||
|
||||
if joined {
|
||||
let since = ctx.client.start_time();
|
||||
reply.push_str(&format!(
|
||||
", present since {} ({})",
|
||||
format_time(since),
|
||||
format_relative_time(since - now),
|
||||
));
|
||||
}
|
||||
|
||||
if connected {
|
||||
let since = ctx.joined.since;
|
||||
reply.push_str(&format!(
|
||||
|
|
@ -98,7 +113,7 @@ where
|
|||
bot: &Bot<E>,
|
||||
) -> Result<Propagate, E> {
|
||||
if arg.trim().is_empty() {
|
||||
let reply = self.formulate_reply(ctx, bot, false);
|
||||
let reply = self.formulate_reply(ctx, bot, false, false);
|
||||
ctx.reply_only(msg.id, reply).await?;
|
||||
Ok(Propagate::No)
|
||||
} else {
|
||||
|
|
@ -111,6 +126,9 @@ where
|
|||
#[cfg(feature = "clap")]
|
||||
#[derive(Parser)]
|
||||
pub struct UptimeArgs {
|
||||
/// Show how long the bot has been in this room.
|
||||
#[arg(long, short)]
|
||||
pub present: bool,
|
||||
/// Show how long the bot has been connected without interruption.
|
||||
#[arg(long, short)]
|
||||
pub connected: bool,
|
||||
|
|
@ -131,7 +149,7 @@ where
|
|||
ctx: &Context,
|
||||
bot: &Bot<E>,
|
||||
) -> Result<Propagate, E> {
|
||||
let reply = self.formulate_reply(ctx, bot, args.connected);
|
||||
let reply = self.formulate_reply(ctx, bot, args.present, args.connected);
|
||||
ctx.reply_only(msg.id, reply).await?;
|
||||
Ok(Propagate::No)
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue