cli.rs
View Options

	use crate::file;
	use crate::helpers::{self, ColoredString};
	use crate::process::Runner;
	use crate::structs::Args;

	use colored::Colorize;
	use global_placeholders::global;
	use macros_rs::{crashln, string, ternary};
	use psutil::process::{MemoryInfo, Process};
	use serde_json::json;
	use std::env;

	use tabled::{
	settings::{
	object::Rows,
	style::{BorderColor, Style},
	themes::Colorization,
	Color,
	},
	Table, Tabled,
	};

	pub fn get_version(short: bool) -> String {
	return match short {
	true => format!("{} {}", env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION")),
	false => format!("{} ({} {}) [{}]", env!("CARGO_PKG_VERSION"), env!("GIT_HASH"), env!("BUILD_DATE"), env!("PROFILE")),
	};
	}

	pub fn start(name: &Option<String>, args: &Option<Args>) {
	let mut runner = Runner::new();

	match args {
	Some(Args::Id(id)) => {
	println!("{} Applying action restartProcess on ({id})", *helpers::SUCCESS);
	runner.restart(*id, name);

	println!("{} restarted ({id}) ✓", *helpers::SUCCESS);
	list(&string!(""));
	}
	Some(Args::Script(script)) => {
	let name = match name {
	Some(name) => string!(name),
	None => string!(script.split_whitespace().next().unwrap_or_default()),
	};

	println!("{} Creating process with ({name})", *helpers::SUCCESS);
	runner.start(name.clone(), script);

	println!("{} created ({name}) ✓", *helpers::SUCCESS);
	list(&string!(""));
	}
	None => {}
	}
	}

	pub fn stop(id: &usize) {
	println!("{} Applying action stopProcess on ({id})", *helpers::SUCCESS);
	let mut runner = Runner::new();
	runner.stop(*id);
	println!("{} stopped ({id}) ✓", *helpers::SUCCESS);
	list(&string!(""));
	}

	pub fn remove(id: &usize) {
	println!("{} Applying action removeProcess on ({id})", *helpers::SUCCESS);
	let mut runner = Runner::new();
	runner.remove(*id);

	println!("{} removed ({id}) ✓", *helpers::SUCCESS);
	list(&string!(""));
	}

	pub fn info(id: &usize) {
	let runner = Runner::new();
	println!("{:?}", runner.info(*id));
	}

	pub fn logs(id: &usize, lines: &usize) {
	let runner = Runner::new();

	if let Some(item) = runner.info(*id) {
	println!("{}", format!("Showing last {lines} lines for process [{id}] (change the value with --lines option)").yellow());

	let log_error = global!("pmc.logs.error", item.name.as_str());
	let log_out = global!("pmc.logs.out", item.name.as_str());

	file::logs(lines, &log_error, id, "error", &item.name);
	file::logs(lines, &log_out, id, "out", &item.name);
	} else {
	crashln!("{} Process ({id}) not found", *helpers::FAIL);
	}
	}

	pub fn list(format: &String) {
	let runner = Runner::new();
	let mut processes: Vec<ProcessItem> = Vec::new();

	#[derive(Tabled, Debug)]
	struct ProcessItem {
	id: ColoredString,
	name: String,
	pid: String,
	uptime: String,
	status: ColoredString,
	cpu: String,
	mem: String,
	}

	impl serde::Serialize for ProcessItem {
	fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
	let trimmed_json = json!({
	"id": &self.id.0.trim(),
	"name": &self.name.trim(),
	"pid": &self.pid.trim(),
	"uptime": &self.uptime.trim(),
	"status": &self.status.0.trim(),
	"cpu": &self.cpu.trim(),
	"mem": &self.mem.trim(),
	});
	trimmed_json.serialize(serializer)
	}
	}

	if runner.list().is_empty() {
	println!("{} Process table empty", *helpers::SUCCESS);
	} else {
	for (id, item) in runner.list() {
	let mut memory_usage: Option<MemoryInfo> = None;
	let mut cpu_percent: Option<f32> = None;

	if let Ok(mut process) = Process::new(item.pid as u32) {
	memory_usage = process.memory_info().ok();
	cpu_percent = process.cpu_percent().ok();
	}

	let cpu_percent = match cpu_percent {
	Some(percent) => format!("{percent}%"),
	None => string!("0%"),
	};

	let memory_usage = match memory_usage {
	Some(usage) => helpers::format_memory(usage.rss()),
	None => string!("0b"),
	};

	processes.push(ProcessItem {
	id: ColoredString(id.cyan().bold()),
	pid: ternary!(item.running, format!("{} ", item.pid), string!("n/a ")),
	cpu: format!("{cpu_percent} "),
	mem: format!("{memory_usage} "),
	name: format!("{} ", item.name.clone()),
	status: ColoredString(ternary!(item.running, "online ".green().bold(), "stopped ".red().bold())),
	uptime: ternary!(item.running, format!("{} ", helpers::format_duration(item.started)), string!("none ")),
	});
	}

	let mut table = Table::new(&processes);
	table
	.with(Style::rounded().remove_verticals())
	.with(BorderColor::filled(Color::FG_BRIGHT_BLACK))
	.with(Colorization::exact([Color::FG_BRIGHT_CYAN], Rows::first()));

	if let Ok(json) = serde_json::to_string(&processes) {
	match format.as_str() {
	"raw" => println!("{:?}", processes),
	"json" => println!("{json}"),
	_ => println!("{}", table.to_string()),
	};
	};
	}
	}

File Metadata

Mime Type: text/plain
Expires: Mon, Dec 15, 2:40 PM (1 d, 9 h)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 484844
Default Alt Text: cli.rs (5 KB)

cli.rs
No OneTemporary
Actions

cli.rs
View Options

File Metadata

Event Timeline

cli.rsNo OneTemporaryActions

cli.rsView Options

File Metadata

Event Timeline

cli.rs
No OneTemporary
Actions

cli.rs
View Options