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

use rustc_serialize::base64::{ToBase64, STANDARD};

use goauth::auth::{JwtClaims, Token};
use goauth::credentials::Credentials;
use goauth::get_token_with_creds;
use goauth::scopes::Scope;
use smpl_jwt::Jwt;
use std::str::FromStr;

use error::BTErr;

pub fn encode_str(str: &str) -> Vec<u8> {
    let mut v = Vec::new();
    v.extend_from_slice(str.as_bytes().to_base64(STANDARD).as_bytes());
    v
}

pub fn get_auth_token(c: &str, fp: bool) -> Result<Token, BTErr> {
    let credentials = if fp {
        Credentials::from_file(c)?
    } else {
        Credentials::from_str(c)?
    };

    let claims = JwtClaims::new(
        credentials.iss(),
        &Scope::CloudPlatform,
        credentials.token_uri(),
        None,
        Some(60),
    );
    let jwt = Jwt::new(claims, credentials.rsa_key()?, None);
    Ok(get_token_with_creds(&jwt, &credentials)?)
}

pub fn row_key_from_str(str: &str) -> Vec<u8> {
    let mut row_key = Vec::new();
    row_key.extend_from_slice(str.as_bytes());
    row_key
}