89 lines
2.8 KiB
Rust
89 lines
2.8 KiB
Rust
// Copyright 2021 System76 <info@system76.com>
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
use pop_launcher_toolkit::plugins;
|
|
use pop_launcher_toolkit::service;
|
|
|
|
use mimalloc::MiMalloc;
|
|
use tracing::info;
|
|
|
|
#[global_allocator]
|
|
static GLOBAL: MiMalloc = MiMalloc;
|
|
|
|
#[tokio::main(flavor = "current_thread")]
|
|
async fn main() {
|
|
if let Some(plugin) = std::env::args().next() {
|
|
let start = plugin.rfind('/').map(|v| v + 1).unwrap_or(0);
|
|
let cmd = &plugin.as_str()[start..];
|
|
|
|
init_logging(cmd);
|
|
|
|
info!("starting {}", cmd);
|
|
|
|
match cmd {
|
|
"calc" => plugins::calc::main().await,
|
|
"desktop-entries" => plugins::desktop_entries::main().await,
|
|
"find" => plugins::find::main().await,
|
|
"files" => plugins::files::main().await,
|
|
"pop-launcher" => service::main().await,
|
|
"pop-shell" => plugins::pop_shell::main().await,
|
|
"pulse" => plugins::pulse::main().await,
|
|
"recent" => plugins::recent::main().await,
|
|
"scripts" => plugins::scripts::main().await,
|
|
"terminal" => plugins::terminal::main().await,
|
|
"web" => plugins::web::main().await,
|
|
"cosmic-toplevel" => plugins::cosmic_toplevel::main().await,
|
|
unknown => {
|
|
eprintln!("unknown cmd: {}", unknown);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn init_logging(cmd: &str) {
|
|
use tracing_subscriber::prelude::*;
|
|
use tracing_subscriber::{fmt, EnvFilter};
|
|
|
|
let logdir = match dirs::state_dir() {
|
|
Some(dir) => dir.join("pop-launcher/"),
|
|
None => dirs::home_dir()
|
|
.expect("home directory required")
|
|
.join(".cache/pop-launcher"),
|
|
};
|
|
|
|
let _ = std::fs::create_dir_all(&logdir);
|
|
|
|
let logfile = std::fs::OpenOptions::new()
|
|
.create(true)
|
|
.append(true)
|
|
.open(logdir.join([cmd, ".log"].concat().as_str()).as_path());
|
|
|
|
if let Ok(file) = logfile {
|
|
if let Ok(meta) = file.metadata() {
|
|
if meta.len() > 1000 {
|
|
let _ = file.set_len(0);
|
|
}
|
|
}
|
|
|
|
let fmt_layer = tracing_subscriber::fmt::layer()
|
|
.with_target(false)
|
|
.with_timer(fmt::time::ChronoLocal::new("%T".into()))
|
|
.with_writer(file);
|
|
|
|
let filter_layer = EnvFilter::try_from_default_env()
|
|
.or_else(|_| EnvFilter::try_new("warn"))
|
|
.unwrap();
|
|
|
|
let registry = tracing_subscriber::registry()
|
|
.with(fmt_layer)
|
|
.with(filter_layer);
|
|
|
|
// would be nice to implement this tracing issue
|
|
// for journald https://github.com/tokio-rs/tracing/issues/2348
|
|
if let Ok(journald_layer) = tracing_journald::layer() {
|
|
registry.with(journald_layer).init();
|
|
} else {
|
|
registry.init();
|
|
}
|
|
}
|
|
}
|