from core.helper.provider_cache import TriggerProviderCredentialCache, TriggerProviderOAuthClientCache
from core.plugin.entities.plugin_daemon import CredentialType
from core.tools.utils.encryption import ProviderConfigCache, ProviderConfigEncrypter, create_provider_encrypter
from core.trigger.entities.api_entities import TriggerProviderCredentialApiEntity
from core.trigger.provider import PluginTriggerProviderController
from models.trigger import TriggerProvider


def create_trigger_provider_encrypter_for_credential(
    tenant_id: str,
    controller: PluginTriggerProviderController,
    credential: TriggerProvider | TriggerProviderCredentialApiEntity,
) -> tuple[ProviderConfigEncrypter, ProviderConfigCache]:
    return create_provider_encrypter(
        tenant_id=tenant_id,
        config=controller.get_credential_schema_config(credential.credential_type),
        cache=TriggerProviderCredentialCache(
            tenant_id=tenant_id,
            provider_id=str(controller.get_provider_id()),
            credential_id=credential.id,
        ),
    )


def create_trigger_provider_encrypter(
    tenant_id: str, controller: PluginTriggerProviderController, credential_id: str, credential_type: CredentialType
) -> tuple[ProviderConfigEncrypter, ProviderConfigCache]:
    return create_provider_encrypter(
        tenant_id=tenant_id,
        config=controller.get_credential_schema_config(credential_type),
        cache=TriggerProviderCredentialCache(
            tenant_id=tenant_id,
            provider_id=str(controller.get_provider_id()),
            credential_id=credential_id,
        ),
    )


def create_trigger_provider_oauth_encrypter(
    tenant_id: str, controller: PluginTriggerProviderController
) -> tuple[ProviderConfigEncrypter, ProviderConfigCache]:
    return create_provider_encrypter(
        tenant_id=tenant_id,
        config=[x.to_basic_provider_config() for x in controller.get_oauth_client_schema()],
        cache=TriggerProviderOAuthClientCache(
            tenant_id=tenant_id,
            provider_id=str(controller.get_provider_id()),
        ),
    )