aboutsummaryrefslogtreecommitdiff
path: root/tool/src/main.rs
blob: 9ce7cf1574563a81f72c76b5d5cddf37e762c9c8 (plain)
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
/*
    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
    Copyright (C) 2025 metamuffin <metamuffin.org>
*/

use anyhow::anyhow;
use clap::Parser;
use jellybase::{CONF, SECRETS};
use jellyclient::Instance;
use jellycommon::user::CreateSessionParams;
use jellytool::{
    add::add,
    cli::{Action, Args},
    migrate::migrate,
};
use log::info;

fn main() -> anyhow::Result<()> {
    env_logger::builder()
        .filter_level(log::LevelFilter::Info)
        .parse_env("LOG")
        .init();
    let args = Args::parse();

    match args {
        a @ Action::Add { .. } => tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap()
            .block_on(add(a)),
        a @ Action::Migrate { .. } => migrate(a),
        Action::Reimport {
            hostname,
            no_tls,
            no_incremental,
        } => tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .unwrap()
            .block_on(async move {
                let inst = Instance::new(hostname.unwrap_or(CONF.hostname.clone()), !no_tls);
                info!("login");
                let session = inst
                    .login(CreateSessionParams {
                        drop_permissions: None,
                        expire: None,
                        password: SECRETS
                            .admin_password
                            .clone()
                            .ok_or(anyhow!("admin account required"))?,
                        username: CONF
                            .admin_username
                            .clone()
                            .ok_or(anyhow!("admin account required"))?,
                    })
                    .await?;

                session.reimport(!no_incremental).await?;
                Ok(())
            }),
    }
}