use serde_json::{Map, Value};
use crate::error::AppError;
use bamboo_infrastructure::Config;
pub use bamboo_infrastructure::patch::{
deep_merge_json, domains_for_root_patch, effects_for_root_patch, is_masked_api_key,
preserve_masked_provider_api_keys, provider_api_key_intents, sanitize_root_patch,
DomainChanges, PatchEffects, ReloadMode,
};
pub fn sync_provider_api_keys_encrypted_for_patch(
config: &mut Config,
providers: &std::collections::BTreeSet<String>,
) -> Result<(), AppError> {
for name in providers.iter() {
match name.as_str() {
"openai" => {
if let Some(openai) = config.providers.openai.as_mut() {
let api_key = openai.api_key.trim();
openai.api_key_encrypted = if api_key.is_empty() {
None
} else {
Some(
bamboo_infrastructure::encryption::encrypt(api_key).map_err(|e| {
AppError::InternalError(anyhow::anyhow!(
"Failed to encrypt OpenAI api_key: {e}"
))
})?,
)
};
}
}
"anthropic" => {
if let Some(anthropic) = config.providers.anthropic.as_mut() {
let api_key = anthropic.api_key.trim();
anthropic.api_key_encrypted = if api_key.is_empty() {
None
} else {
Some(
bamboo_infrastructure::encryption::encrypt(api_key).map_err(|e| {
AppError::InternalError(anyhow::anyhow!(
"Failed to encrypt Anthropic api_key: {e}"
))
})?,
)
};
}
}
"gemini" => {
if let Some(gemini) = config.providers.gemini.as_mut() {
let api_key = gemini.api_key.trim();
gemini.api_key_encrypted = if api_key.is_empty() {
None
} else {
Some(
bamboo_infrastructure::encryption::encrypt(api_key).map_err(|e| {
AppError::InternalError(anyhow::anyhow!(
"Failed to encrypt Gemini api_key: {e}"
))
})?,
)
};
}
}
_ => {}
}
}
Ok(())
}
pub fn assert_json_object(value: Value) -> Result<Map<String, Value>, AppError> {
match value {
Value::Object(map) => Ok(map),
_ => Err(AppError::BadRequest(
"config.json must be a JSON object".to_string(),
)),
}
}
pub fn build_merged_config(
current: &Config,
patch_obj: Map<String, Value>,
) -> Result<Config, AppError> {
let mut merged = serde_json::to_value(current)
.map_err(|e| AppError::InternalError(anyhow::anyhow!("Failed to serialize config: {e}")))?;
deep_merge_json(&mut merged, Value::Object(patch_obj));
let mut new_config: Config = serde_json::from_value(merged)
.map_err(|e| AppError::BadRequest(format!("Invalid configuration JSON: {e}")))?;
new_config.hydrate_proxy_auth_from_encrypted();
new_config.hydrate_provider_api_keys_from_encrypted();
new_config.hydrate_mcp_secrets_from_encrypted();
new_config.hydrate_env_vars_from_encrypted();
new_config.normalize_tool_settings();
new_config.normalize_skill_settings();
Ok(new_config)
}