smbcloud-cli 0.4.7

smbCloud command line interface.
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
68
69
70
71
72
73

use {
    crate::{
        cli::CommandResult,
        client,
        token::{get_smb_token::get_smb_token, smb_token_file_path::smb_token_file_path},
        ui::{fail_message, fail_symbol, prompt, succeed_message, succeed_symbol},
    },
    anyhow::{anyhow, Result},
    smbcloud_auth::logout::logout,
    smbcloud_network::environment::Environment,
    spinners::Spinner,
    std::fs,
};

pub async fn process_logout(env: Environment) -> Result<CommandResult> {
    if let Some(token_path) = smb_token_file_path(env) {
        // In --ci mode this defaults to "yes": running `smb --ci logout` is an
        // explicit request, so proceed without prompting.
        let confirm = prompt::confirm("Do you want to logout? y/n", true)?;
        if !confirm {
            return Ok(CommandResult {
                spinner: Spinner::new(
                    spinners::Spinners::SimpleDotsScrolling,
                    succeed_message("Cancelled."),
                ),
                symbol: succeed_symbol(),
                msg: succeed_message("Cancelled."),
            });
        }

        let spinner = Spinner::new(
            spinners::Spinners::SimpleDotsScrolling,
            succeed_message("Logging you out"),
        );

        match do_process_logout(env).await {
            Ok(_) => {
                fs::remove_file(token_path)?;
                Ok(CommandResult {
                    spinner,
                    symbol: succeed_symbol(),
                    msg: succeed_message("You are logged out!"),
                })
            }
            Err(e) => Err(anyhow!("{e}")),
        }
    } else {
        Ok(CommandResult {
            spinner: Spinner::new(
                spinners::Spinners::SimpleDotsScrolling,
                succeed_message("Loading"),
            ),
            symbol: fail_symbol(),
            msg: fail_message("You are not logged in. Please login first."),
        })
    }
}

async fn do_process_logout(env: Environment) -> Result<()> {
    let token = get_smb_token(env)?;
    match logout(env, client(), token).await {
        Ok(_) => Ok(()),
        Err(e) => {
            // A 401 means the session is already expired on the server.
            // Treat this as success — the session is gone either way.
            if e.to_string().contains("Unauthorized") {
                Ok(())
            } else {
                Err(anyhow!("{e}"))
            }
        }
    }
}