fix(08): fix AI key status detection and enable saving to secrets.yaml
This commit is contained in:
@@ -72,6 +72,74 @@ class ConfigManager:
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
@staticmethod
|
||||
def get_secrets_path():
|
||||
"""Returns the absolute path to secrets.yaml."""
|
||||
base_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
project_root = os.path.dirname(base_dir)
|
||||
return os.path.join(project_root, "config", "secrets.yaml")
|
||||
|
||||
@staticmethod
|
||||
def update_secrets(updates: dict) -> dict:
|
||||
"""Update secrets.yaml with new values (flattened structure)."""
|
||||
path = ConfigManager.get_secrets_path()
|
||||
|
||||
current_secrets = {}
|
||||
if os.path.exists(path):
|
||||
try:
|
||||
with open(path, 'r', encoding='utf-8') as f:
|
||||
current_secrets = yaml.safe_load(f) or {}
|
||||
except Exception as e:
|
||||
log.error(f"❌ Failed to read {path}: {e}")
|
||||
|
||||
# Update flattened secrets
|
||||
current_secrets.update(updates)
|
||||
|
||||
try:
|
||||
with open(path, 'w', encoding='utf-8') as f:
|
||||
# Header for clarity
|
||||
f.write("# TFM aInventory - Managed Secrets (Updated via Admin UI)\n")
|
||||
yaml.dump(current_secrets, f, default_flow_style=False, sort_keys=True)
|
||||
log.info(f"✅ Updated {path} with new secrets.")
|
||||
|
||||
# Reload the global config
|
||||
load_config()
|
||||
return get_config()
|
||||
except Exception as e:
|
||||
log.error(f"❌ Failed to write {path}: {e}")
|
||||
raise
|
||||
|
||||
@staticmethod
|
||||
def update_keys(updates: dict) -> dict:
|
||||
"""
|
||||
Intelligent key update: routes sensitive AI keys to secrets.yaml
|
||||
and others to backend.yaml.
|
||||
"""
|
||||
secrets_updates = {}
|
||||
backend_updates = {}
|
||||
|
||||
sensitive_keys = ["GEMINI_API_KEY", "CLAUDE_API_KEY", "JWT_SECRET_KEY", "LDAP_PASSWORD"]
|
||||
|
||||
for key, val in updates.items():
|
||||
if key in sensitive_keys:
|
||||
secrets_updates[key] = val
|
||||
else:
|
||||
# Non-sensitive or complex nested settings
|
||||
backend_updates[key] = val
|
||||
|
||||
if secrets_updates:
|
||||
ConfigManager.update_secrets(secrets_updates)
|
||||
|
||||
if backend_updates:
|
||||
ConfigManager.update_config(backend_updates)
|
||||
|
||||
return get_config()
|
||||
|
||||
@staticmethod
|
||||
def get_config() -> dict:
|
||||
"""Return the current global configuration."""
|
||||
return get_config()
|
||||
|
||||
@staticmethod
|
||||
def get_masked_key(key_name: str):
|
||||
"""Returns a masked version of a configuration value or env var."""
|
||||
@@ -113,6 +181,10 @@ def update_config(updates: dict) -> dict:
|
||||
"""Update backend.yaml with new values and return updated config."""
|
||||
return ConfigManager.update_config(updates)
|
||||
|
||||
def get_config() -> dict:
|
||||
"""Return the current global configuration."""
|
||||
return ConfigManager.get_config()
|
||||
|
||||
def validate_config_file() -> bool:
|
||||
"""Validate backend.yaml syntax and required fields."""
|
||||
return ConfigManager.validate_config_file()
|
||||
|
||||
Reference in New Issue
Block a user