sway-groups-cli 0.2.0

Command-line tool for managing sway workspaces in named groups.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! swayg CLI - Sway workspace groups management.

mod commands;

use anyhow::Result as AnyResult;
use clap::Parser;
use directories::ProjectDirs;
use std::path::PathBuf;
use tracing::info;
use tracing_appender::rolling::{RollingFileAppender, Rotation};
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt, EnvFilter};

use sway_groups_core::db::DatabaseManager;
use sway_groups_core::services::{GroupService, NavigationService, WorkspaceService, WaybarSyncService};
use sway_groups_core::sway::SwayIpcClient;

fn get_data_dir() -> PathBuf {
    if let Some(proj_dirs) = ProjectDirs::from("com", "swayg", "swayg") {
        let data_dir = proj_dirs.data_dir();
        std::fs::create_dir_all(data_dir).ok();
        data_dir.to_path_buf()
    } else {
        PathBuf::from(".")
    }
}

fn get_db_path() -> PathBuf {
    get_data_dir().join("swayg.db")
}

#[tokio::main]
async fn main() -> AnyResult<()> {
    let cli = commands::Cli::parse();

    let level = if cli.verbose { "debug" } else { "info" };

    let db_path = cli.db.clone().unwrap_or_else(get_db_path);
    let data_dir = db_path.parent().map(|p| p.to_path_buf()).unwrap_or_else(get_data_dir);
    std::fs::create_dir_all(&data_dir).ok();

    let file_appender = RollingFileAppender::new(Rotation::DAILY, data_dir, "swayg");
    tracing_subscriber::registry()
        .with(tracing_subscriber::fmt::layer().with_writer(std::io::stderr))
        .with(tracing_subscriber::fmt::layer().with_writer(file_appender))
        .with(EnvFilter::from_default_env()
            .add_directive(format!("swayg={}", level).parse()?)
            .add_directive(format!("sway_groups_core={}", level).parse()?))
        .init();

    info!("Using database at: {}", db_path.display());

    let config = match &cli.config {
        Some(path) => sway_groups_config::SwaygConfig::load_from(path)?,
        None => sway_groups_config::SwaygConfig::load()?,
    };

    let db: DatabaseManager = DatabaseManager::new(db_path.clone()).await?;
    let ipc_client = SwayIpcClient::new()?;
    let group_service = GroupService::with_config(db.clone(), ipc_client.clone(), &config);
    let workspace_service = WorkspaceService::with_config(db.clone(), ipc_client.clone(), &config);
    let waybar_sync = WaybarSyncService::with_config(db.clone(), ipc_client.clone(), &config);
    let nav_service = NavigationService::new(db.clone(), ipc_client.clone());

    commands::run(cli, &group_service, &workspace_service, &waybar_sync, &nav_service, &ipc_client, db_path).await?;

    Ok(())
}