Adding logs for cloud

tembo-cli/src/cmd/logs.rs

@@ -0,0 +1,108 @@
+use crate::apply::{get_instance_id, get_instance_settings};
+use crate::cli::context::{get_current_context, Environment, Profile};
+use anyhow::Result;
+use clap::Args;
+use reqwest::blocking::Client;
+use reqwest::header::{HeaderMap, HeaderValue, AUTHORIZATION};
+use serde::{Deserialize, Serialize};
+use temboclient::apis::configuration::Configuration;
+
+#[derive(Args)]
+pub struct LogsCommand {
+    #[clap(short, long)]
+    pub verbose: bool,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+struct LogStream {
+    app: String,
+    container: String,
+    pod: String,
+    stream: String,
+    tembo_instance_id: String,
+    tembo_organization_id: String,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+struct LogEntry {
+    stream: LogStream,
+    values: Vec<Vec<String>>,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+struct LogResult {
+    resultType: String,
+    result: Vec<LogEntry>,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+struct LogData {
+    status: String,
+    data: LogResult,
+}
+
+fn beautify_logs(json_data: &str) -> Result<()> {
+    let log_data: LogData = serde_json::from_str(json_data)?;
+
+    for entry in log_data.data.result {
+        //println!("\nApp: {}, Container: {}, Pod: {}", entry.stream.app, entry.stream.container, entry.stream.pod);
+        for value in entry.values {
+            let log_message = &value[1];
+            println!("\n{}", log_message);
+        }
+    }
+
+    Ok(())
+}
+
+pub fn execute(verbose: bool) -> Result<()> {
+    let env = get_current_context()?;
+    let org_id = env.org_id.clone().unwrap_or_default();
+    let profile = env.selected_profile.clone().unwrap();
+
+    let config = Configuration {
+        base_path: profile.tembo_host,
+        bearer_access_token: Some(profile.tembo_access_token.clone()),
+        ..Default::default()
+    };
+
+    let instance_settings = get_instance_settings(None, None)?;
+
+    let client = Client::new();
+    let mut headers = HeaderMap::new();
+    headers.insert("X-Scope-OrgID", HeaderValue::from_str(&org_id)?);
+    headers.insert(
+        AUTHORIZATION,
+        HeaderValue::from_str(&format!("Bearer {}", profile.tembo_access_token))?,
+    );
+
+    for (_key, value) in instance_settings.iter() {
+        let instance_id_option =
+            get_instance_id(value.instance_name.clone(), &config, env.clone())?;
+
+        let instance_id = if let Some(id) = instance_id_option {
+            id
+        } else {
+            eprintln!("Instance ID not found for {}", value.instance_name);
+            continue;
+        };
+
+        let query = format!("{{tembo_instance_id=\"{}\"}}", instance_id);
+        let url = "https://api.data-1.use1.tembo.io/loki/api/v1/query_range";
+
+        let response = client
+            .get(url)
+            .headers(headers.clone())
+            .query(&[("query", &query)])
+            .send()?;
+
+        if response.status().is_success() {
+            let response_body = response.text()?;
+            beautify_logs(&response_body)?;
+        } else {
+            eprintln!("Error: {:?}", response.status());
+        }
+    }
+
+    Ok(())
+}

tembo-cli/src/cmd/mod.rs

@@ -2,4 +2,5 @@ pub mod apply;
 pub mod context;
 pub mod delete;
 pub mod init;
+pub mod logs;
 pub mod validate;

tembo-cli/src/main.rs

@@ -1,10 +1,11 @@
 use crate::cmd::delete::DeleteCommand;
 use crate::cmd::validate::ValidateCommand;
-use crate::cmd::{apply, context, delete, init, validate};
+use crate::cmd::{apply, context, delete, init, logs, validate};
 use clap::{crate_authors, crate_version, Args, Parser, Subcommand};
 use cmd::apply::ApplyCommand;
 use cmd::context::{ContextCommand, ContextSubCommand};
 use cmd::init::InitCommand;
+use cmd::logs::LogsCommand;
 
 mod cli;
 mod cmd;
@@ -28,6 +29,7 @@ enum SubCommands {
     Apply(ApplyCommand),
     Validate(ValidateCommand),
     Delete(DeleteCommand),
+    Logs(LogsCommand),
 }
 
 #[derive(Args)]
@@ -62,6 +64,9 @@ fn main() -> Result<(), anyhow::Error> {
         SubCommands::Validate(_validate_cmd) => {
             validate::execute(app.global_opts.verbose)?;
         }
+        SubCommands::Logs(_logs_cmd) => {
+            logs::execute(app.global_opts.verbose);
+        }
         SubCommands::Delete(_delete_cmd) => {
             delete::execute()?;
         }