config.py

# -*- coding: utf-8 -*-

# This file is part of the CloudBlue Connect connect-cli.
# Copyright (c) 2025 CloudBlue. All rights reserved.

import json
import os
from dataclasses import dataclass

from click import ClickException, make_pass_decorator
from connect.client import ConnectClient

from connect.cli.core.constants import DEFAULT_ENDPOINT
from connect.cli.core.http import RequestLogger


@dataclass
class Account:
    id: str
    name: str
    api_key: str
    endpoint: str
    client: ConnectClient

    def is_vendor(self):
        return self.id.startswith('VA-')

    def is_provider(self):
        return self.id.startswith('PA-')


class Config(object):
    def __init__(self):
        self._config_path = None
        self._active = None
        self._accounts = {}

    def add_account(self, id, name, api_key, endpoint=DEFAULT_ENDPOINT):
        client = ConnectClient(
            api_key,
            endpoint=endpoint,
            use_specs=False,
            max_retries=3,
            logger=RequestLogger(),
        )
        self._accounts[id] = Account(id, name, api_key, endpoint, client)
        if not self._active:
            self._active = self._accounts[id]

    @property
    def active(self):
        return self._active

    @property
    def accounts(self):
        return self._accounts

    def activate(self, id):
        account = self._accounts.get(id)
        if account:
            self._active = account
            return
        raise ClickException(f'The account identified by {id} does not exist.')

    def exists(self, id):
        return id in self._accounts

    def remove_account(self, id):
        if id in self._accounts:
            account = self._accounts[id]
            del self._accounts[id]
            if self._active.id == id:
                if self._accounts:
                    self._active = list(self._accounts.values())[0]
                else:
                    self._active = None
            return account
        raise ClickException(f'The account identified by {id} does not exist.')

    def load(self, config_dir):
        self._config_path = os.path.join(config_dir, 'config.json')
        if not os.path.isfile(self._config_path):
            return

        with open(self._config_path, 'r') as f:
            data = json.load(f)
            active_account_id = data['active']
            for account_data in data['accounts']:
                account = Account(
                    **account_data,
                    client=ConnectClient(
                        account_data['api_key'],
                        endpoint=account_data['endpoint'],
                        use_specs=False,
                        max_retries=3,
                        logger=RequestLogger(),
                    ),
                )
                self._accounts[account.id] = account
                if account.id == active_account_id:
                    self._active = account

    def store(self):
        with open(self._config_path, 'w') as f:
            accounts = [
                {
                    'id': account.id,
                    'name': account.name,
                    'api_key': account.api_key,
                    'endpoint': account.endpoint,
                }
                for account in self._accounts.values()
            ]
            f.write(
                json.dumps(
                    {
                        'active': self._active.id if self._active else '',
                        'accounts': accounts,
                    },
                    sort_keys=True,
                    indent=4,
                ),
            )

    def validate(self):
        if not (self._accounts and self._active):
            raise ClickException(
                'connect-cli is not properly configured.\n'
                'You must configure at least a Connect account. To do so please execute:\n\n'
                'ccli account add API_KEY\n\n',
            )


pass_config = make_pass_decorator(Config, ensure=True)