from argparse import Action
from typing import Dict, List

from facefusion.types import CapabilitySet, CapabilityStore, Scope

CAPABILITY_STORE : CapabilityStore =\
{
	'api': {},
	'cli': {},
	'sys': {}
}


def get_api_capability_set() -> Dict[str, CapabilitySet]:
	return CAPABILITY_STORE.get('api')


def get_cli_capability_set() -> Dict[str, CapabilitySet]:
	return CAPABILITY_STORE.get('cli')


def get_sys_capability_set() -> Dict[str, CapabilitySet]:
	return CAPABILITY_STORE.get('sys')


def get_api_arguments() -> List[str]:
	return list(get_api_capability_set().keys())


def get_cli_arguments() -> List[str]:
	return list(get_cli_capability_set().keys())


def get_sys_arguments() -> List[str]:
	return list(get_sys_capability_set().keys())


def register_capability_set(actions : List[Action], scopes : List[Scope]) -> None:
	for action in actions:
		value : CapabilitySet =\
		{
			'default': action.default
		}

		if action.choices:
			value['choices'] = list(action.choices)

		for scope in scopes:
			if scope == 'api':
				CAPABILITY_STORE['api'][action.dest] = value
			if scope == 'cli':
				CAPABILITY_STORE['cli'][action.dest] = value
			if scope == 'sys':
				CAPABILITY_STORE['sys'][action.dest] = value