ytop/src/app.rs

use num_rational::Ratio;

use crate::args::Args;
use crate::colorscheme::Colorscheme;
use crate::widgets::*;

pub struct App<'a, 'b> {
	pub help_menu: HelpMenu<'a>,
	pub statusbar: Option<Statusbar<'a>>,
	pub widgets: Widgets<'a, 'b>,
}

pub struct Widgets<'a, 'b> {
	pub battery: Option<BatteryWidget<'a>>,
	pub cpu: CpuWidget<'a>,
	pub disk: Option<DiskWidget<'a>>,
	pub mem: MemWidget<'a>,
	pub net: Option<NetWidget<'a, 'b>>,
	pub proc: ProcWidget<'a>,
	pub temp: Option<TempWidget<'a>>,
}

pub fn setup_app<'a, 'b>(
	args: &'b Args,
	update_ratio: Ratio<u64>,
	colorscheme: &'a Colorscheme,
	program_name: &str,
) -> App<'a, 'b> {
	let cpu = CpuWidget::new(colorscheme, update_ratio, args.average_cpu, args.per_cpu);
	let mem = MemWidget::new(colorscheme, update_ratio);
	let proc = ProcWidget::new(colorscheme);
	let help_menu = HelpMenu::new(colorscheme);

	let (battery, disk, net, temp) = if args.minimal {
		(None, None, None, None)
	} else {
		(
			if args.battery {
				Some(BatteryWidget::new(colorscheme))
			} else {
				None
			},
			Some(DiskWidget::new(colorscheme)),
			Some(NetWidget::new(colorscheme, &args.interfaces)),
			Some(TempWidget::new(colorscheme, args.fahrenheit)),
		)
	};

	let statusbar = if args.statusbar {
		Some(Statusbar::new(colorscheme, program_name))
	} else {
		None
	};

	App {
		help_menu,
		statusbar,
		widgets: Widgets {
			battery,
			cpu,
			disk,
			mem,
			net,
			proc,
			temp,
		},
	}
}