Lines
0 %
Functions
use anyhow::Context;
use chrono::{Duration, Utc};
use clap::{CommandFactory, Parser};
use clap_complete::{Shell, generate};
use roowho2_lib::{proto::WhodStatusUpdate, server::varlink_api::VarlinkRwhodClientProxy};
/// Show host status of local machines.
///
/// `ruptime` gives a status line like uptime for each machine on the local network;
/// these are formed from packets broadcast by each host on the network once a minute.
/// Machines for which no status report has been received for 11 minutes are shown as being down.
#[derive(Debug, Parser)]
#[command(
author = "Programvareverkstedet <projects@pvv.ntnu.no>",
about,
version
)]
pub struct Args {
/// Users idle an hour or more are not counted unless the `-a` flag is given.
#[arg(long, short)]
all: bool,
/// Sort by load average.
#[arg(long, short, conflicts_with = "time", conflicts_with = "users")]
load: bool,
/// Reverses the sort order.
reverse: bool,
/// Sort by uptime.
#[arg(long, short, conflicts_with = "load", conflicts_with = "users")]
time: bool,
/// Sort by number of users.
#[arg(long, short, conflicts_with = "load", conflicts_with = "time")]
users: bool,
/// Print the output with the old formatting
old: bool,
/// Output in JSON format
json: bool,
/// Generate shell completion scripts for the specified shell
/// and print them to stdout.
#[arg(long, value_enum, hide = true)]
completions: Option<Shell>,
}
#[tokio::main]
async fn main() -> anyhow::Result<()> {
let args = Args::parse();
if let Some(shell) = args.completions {
generate(
shell,
&mut Args::command(),
"ruptime",
&mut std::io::stdout(),
);
return Ok(());
let mut conn = zlink::unix::connect("/run/roowho2/roowho2.varlink")
.await
.expect("Failed to connect to rwhod server");
let mut reply = conn
.ruptime()
.context("Failed to send rwho request")?
.map_err(|e| anyhow::anyhow!("Server returned an error for rwho request: {:?}", e))?;
sort_entries(&mut reply, args.load, args.time, args.users, args.reverse);
if args.json {
println!("{}", serde_json::to_string_pretty(&reply).unwrap());
// } else if args.old {
// for entry in &reply {
// let line = old_format_machine_entry(args.all, entry);
// println!("{}", line);
// }
} else {
for entry in &reply {
let line = old_format_machine_entry(args.all, entry);
println!("{}", line);
Ok(())
fn sort_entries(
entries: &mut [WhodStatusUpdate],
sort_by_load: bool,
sort_by_time: bool,
sort_by_users: bool,
) {
entries.sort_by(|entry1, entry2| {
let ordering = if sort_by_load {
let load1 = entry1.load_average.0 + entry1.load_average.1 + entry1.load_average.2;
let load2 = entry2.load_average.0 + entry2.load_average.1 + entry2.load_average.2;
load1
.partial_cmp(&load2)
.unwrap_or(std::cmp::Ordering::Equal)
} else if sort_by_time {
let uptime1 = Utc::now() - entry1.sendtime;
let uptime2 = Utc::now() - entry2.sendtime;
uptime1.cmp(&uptime2)
} else if sort_by_users {
let users1 = entry1.users.len();
let users2 = entry2.users.len();
users1.cmp(&users2)
entry1.hostname.cmp(&entry2.hostname)
};
if reverse {
ordering.reverse()
ordering
});
fn old_format_machine_entry(all: bool, entry: &WhodStatusUpdate) -> String {
let time_since_last_ping = Utc::now() - entry.sendtime;
let is_up = time_since_last_ping <= Duration::minutes(11);
let uptime = Utc::now() - entry.boot_time;
let days = uptime.num_days();
let hours = uptime.num_hours() % 24;
let minutes = uptime.num_minutes() % 60;
let uptime_str = if days > 0 {
format!("{:3}+{:02}:{:02}", days, hours, minutes)
} else if uptime.num_seconds() < 0 || days > 999 {
" ??:??".to_string()
format!(" {:2}:{:02}", hours, minutes)
let user_count = if all {
entry.users.len()
entry
.users
.iter()
.filter(|user| user.idle_time < Duration::hours(1))
.count()
format!(
"{:<12.12} {} {}, {:4} user{} load {:>4.2}, {:>4.2}, {:>4.2}",
entry.hostname,
if is_up { "up" } else { "down" },
uptime_str,
user_count,
if user_count == 1 { ", " } else { "s," },
entry.load_average.0 as f32 / 100.0,
entry.load_average.1 as f32 / 100.0,
entry.load_average.2 as f32 / 100.0,
)