# First time — requires private key
privacy-boost login --private-key $PRIVATE_KEY

# Subsequent commands — no private key needed
privacy-boost balance --token $TOKEN
privacy-boost shield --token $TOKEN --amount 1.0 --human

if let Some(session) = sdk.export_session() {
    let json = serde_json::to_string(&session)?;
    std::fs::write("session.json", json)?;
    println!("Session saved");
}

let json = std::fs::read_to_string("session.json")?;
let session: ExportedSession = serde_json::from_str(&json)?;

match sdk.import_session(&session) {
    Ok(true) => println!("Session restored"),
    Ok(false) => println!("Session invalid or expired"),
    Err(e) => println!("Import failed: {:?}", e),
}

pub struct ExportedSession {
    pub wallet_public_key_x: String,
    pub wallet_public_key_y: String,
    pub viewing_key: String,
    pub viewing_public_key_x: String,
    pub viewing_public_key_y: String,
    pub nullifying_key: String,
    pub nullifying_public_key_x: String,
    pub nullifying_public_key_y: String,
    pub mpk: String,
    pub account_id: String,
    pub jwt: String,
    pub jwt_expiry: u64,
    pub wallet_address: String,
}

let config_path = CliConfig::default_config_path()?;
println!("Config at: {:?}", config_path);

// Save config
config.save_to_default()?;

// Load config
let config = CliConfig::load_from_default()?;

# End session, clear saved state
privacy-boost logout

// Full logout
sdk.logout();

// Quick session clear
sdk.clear_session();

chmod 600 session.json

match sdk.import_session(&session) {
    Ok(true) => println!("Session restored"),
    Ok(false) | Err(_) => {
        // Re-authenticate
        sdk.authenticate(&private_key, None, None)?;
    }
}

// Delete session file on logout
sdk.logout();
std::fs::remove_file("session.json").ok();