use async_trait::async_trait;
use tracing::{Level, event};
use crate::commands::cli_command::{CliCommand, PRINT_TARGET};
use iggy_common::Client;
use super::common::{ContextConfig, ContextManager, validate_transport};
pub struct CreateContextCmd {
context_name: String,
context_config: ContextConfig,
}
impl CreateContextCmd {
pub fn new(context_name: String, context_config: ContextConfig) -> Self {
Self {
context_name,
context_config,
}
}
}
#[async_trait]
impl CliCommand for CreateContextCmd {
fn explain(&self) -> String {
let context_name = &self.context_name;
format!("create context {context_name}")
}
fn login_required(&self) -> bool {
false
}
fn connection_required(&self) -> bool {
false
}
async fn execute_cmd(&mut self, _client: &dyn Client) -> anyhow::Result<(), anyhow::Error> {
if let Some(ref transport) = self.context_config.iggy.transport {
validate_transport(transport)?;
}
let mut context_mgr = ContextManager::default();
context_mgr
.create_context(&self.context_name, self.context_config.clone())
.await?;
event!(target: PRINT_TARGET, Level::INFO, "context '{}' created successfully", self.context_name);
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn should_return_explain_message() {
let cmd = CreateContextCmd::new("production".to_string(), ContextConfig::default());
assert_eq!(cmd.explain(), "create context production");
}
#[test]
fn should_not_require_login() {
let cmd = CreateContextCmd::new("test".to_string(), ContextConfig::default());
assert!(!cmd.login_required());
}
#[test]
fn should_not_require_connection() {
let cmd = CreateContextCmd::new("test".to_string(), ContextConfig::default());
assert!(!cmd.connection_required());
}
}