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

use std::sync::{RwLock, Arc};
use std::time::Duration;
use crate::AsciiValue;
use super::{Credentials, UpdatableToken};

#[derive(Debug, Clone)]
/// An automaitc updatable token.
/// Updates every 11 hours by run command `yc iam create-token`.
/// To use that you need [Yandex Cloud CLI](https://cloud.yandex.ru/docs/cli/operations/install-cli) installed
pub struct Cli {
    token: Arc<RwLock<AsciiValue>>,
}

impl Credentials for Cli {
    fn token(&self) -> AsciiValue {
        self.token.read().unwrap().clone()
    }
}

impl Into<UpdatableToken> for Cli {
    fn into(self) -> UpdatableToken {
        let Self{ token} = self;
        UpdatableToken { token }
    }
}

impl Cli {
    pub async fn new() -> Self {
        let token = Self::create_token().await;
        let token = Arc::new(RwLock::new(token));
        let update_me = Arc::downgrade(&token);
        tokio::spawn(async move {
            let mut timer = tokio::time::interval(Duration::from_secs(60*60*11));
            loop {
                timer.tick().await;
                if let Some(update_me) = update_me.upgrade() {
                    let token = Self::create_token().await;
                    *update_me.write().unwrap() = token;
                } else {
                    break;
                }
            }
        });
        Self {token}
    }
    async fn create_token() -> AsciiValue {
        let out = tokio::process::Command::new("yc").arg("iam").arg("create-token").output().await.expect("cannot run `yc iam create-token`");
        let stdout = out.stdout.as_slice();
        let stdout = &stdout[0..stdout.len() - 1];
        AsciiValue::try_from(stdout).unwrap()
    }
}