134 lines
4.4 KiB
Python
134 lines
4.4 KiB
Python
"""Core CLI commands for env-pro."""
|
|
|
|
import click
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
|
|
@click.command()
|
|
def init():
|
|
"""Initialize env-pro in the current directory."""
|
|
from env_pro.core.profile import init_project
|
|
|
|
profiles_dir = init_project()
|
|
click.echo(f"Initialized env-pro in {profiles_dir}")
|
|
click.echo("Created 'default' profile. Use 'env-pro profile create' to create more.")
|
|
|
|
|
|
@click.command()
|
|
@click.argument("key")
|
|
@click.argument("value", required=False)
|
|
@click.option("--encrypt/--no-encrypt", default=False, help="Encrypt the value")
|
|
@click.option("--interactive", "-i", is_flag=True, help="Prompt for value")
|
|
def add(key: str, value: Optional[str], encrypt: bool, interactive: bool):
|
|
"""Add or update an environment variable."""
|
|
from env_pro.core.profile import get_active_profile, get_profile_vars, set_profile_var
|
|
from env_pro.core.encryption import encrypt_value, generate_key, get_key_from_keyring
|
|
|
|
profile = get_active_profile() or "default"
|
|
|
|
if interactive and value is None:
|
|
value = click.prompt(f"Enter value for {key}", hide_input=encrypt)
|
|
|
|
if value is None:
|
|
raise click.ClickException("Value is required (use argument or --interactive)")
|
|
|
|
if encrypt:
|
|
keyring_key = get_key_from_keyring()
|
|
if keyring_key is None:
|
|
keyring_key = generate_key()
|
|
from env_pro.core.encryption import store_key_in_keyring
|
|
store_key_in_keyring(keyring_key)
|
|
click.echo("Generated new encryption key")
|
|
|
|
value = encrypt_value(value, keyring_key)
|
|
click.echo(f"Encrypted value for {key}")
|
|
|
|
set_profile_var(profile, key, value)
|
|
click.echo(f"Set {key} in profile '{profile}'")
|
|
|
|
|
|
@click.command()
|
|
@click.argument("key")
|
|
@click.argument("value")
|
|
@click.option("--encrypt/--no-encrypt", default=False, help="Encrypt the value")
|
|
def set(key: str, value: str, encrypt: bool):
|
|
"""Set an environment variable (alias for add)."""
|
|
from env_pro.core.profile import get_active_profile, set_profile_var
|
|
from env_pro.core.encryption import encrypt_value, generate_key, get_key_from_keyring
|
|
|
|
profile = get_active_profile() or "default"
|
|
|
|
if encrypt:
|
|
keyring_key = get_key_from_keyring()
|
|
if keyring_key is None:
|
|
keyring_key = generate_key()
|
|
from env_pro.core.encryption import store_key_in_keyring
|
|
store_key_in_keyring(keyring_key)
|
|
click.echo("Generated new encryption key")
|
|
|
|
value = encrypt_value(value, keyring_key)
|
|
click.echo(f"Encrypted value for {key}")
|
|
|
|
set_profile_var(profile, key, value)
|
|
click.echo(f"Set {key} in profile '{profile}'")
|
|
|
|
|
|
@click.command()
|
|
@click.argument("key")
|
|
@click.option("--decrypt/--no-decrypt", default=False, help="Decrypt the value")
|
|
def get(key: str, decrypt: bool):
|
|
"""Get an environment variable."""
|
|
from env_pro.core.profile import get_active_profile, get_profile_vars
|
|
from env_pro.core.encryption import decrypt_value, get_key_from_keyring
|
|
|
|
profile = get_active_profile() or "default"
|
|
vars = get_profile_vars(profile)
|
|
|
|
if key not in vars:
|
|
raise click.ClickException(f"Key '{key}' not found in profile '{profile}'")
|
|
|
|
value = vars[key]
|
|
|
|
if decrypt:
|
|
keyring_key = get_key_from_keyring()
|
|
if keyring_key is None:
|
|
raise click.ClickException("No encryption key found. Run 'env-pro key generate' first.")
|
|
|
|
try:
|
|
value = decrypt_value(value, keyring_key)
|
|
except Exception as e:
|
|
raise click.ClickException(f"Failed to decrypt: {e}")
|
|
|
|
click.echo(value)
|
|
|
|
|
|
@click.command()
|
|
def list():
|
|
"""List all environment variables."""
|
|
from env_pro.core.profile import get_active_profile, get_profile_vars
|
|
|
|
profile = get_active_profile() or "default"
|
|
vars = get_profile_vars(profile)
|
|
|
|
if not vars:
|
|
click.echo(f"No variables in profile '{profile}'")
|
|
return
|
|
|
|
click.echo(f"Variables in profile '{profile}':")
|
|
for key, value in vars.items():
|
|
if len(value) > 50:
|
|
value = value[:47] + "..."
|
|
click.echo(f" {key}={value}")
|
|
|
|
|
|
@click.command()
|
|
@click.argument("key")
|
|
def delete(key: str):
|
|
"""Delete an environment variable."""
|
|
from env_pro.core.profile import get_active_profile, delete_profile_var
|
|
|
|
profile = get_active_profile() or "default"
|
|
delete_profile_var(profile, key)
|
|
click.echo(f"Deleted {key} from profile '{profile}'")
|