use azure_sdk_core::errors::AzureError;
use azure_sdk_cosmos::clients::DefaultCosmosUri;
use azure_sdk_cosmos::prelude::*;
use azure_sdk_cosmos::AuthorizationToken;

pub fn initialize() -> Result<CosmosStruct<'static, DefaultCosmosUri>, AzureError> {
    let account = std::env::var("COSMOS_ACCOUNT").expect("Set env variable COSMOS_ACCOUNT first!");
    let key =
        std::env::var("COSMOS_MASTER_KEY").expect("Set env variable COSMOS_MASTER_KEY first!");

    let authorization_token = AuthorizationToken::new_master(&key)?;
    let client = ClientBuilder::new(account, authorization_token)?;

    Ok(client)
}