mirror of
https://github.com/langgenius/dify.git
synced 2026-02-22 19:15:47 +08:00
a46c9238fa
- Add OAuth authorization URL generation API endpoint - Implement OAuth callback handler for credential storage - Support both system-level and tenant-level OAuth clients - Add trigger provider credential encryption utilities - Refactor trigger entities into separate modules - Update trigger provider service with OAuth client management - Add credential cache for trigger providers 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
50 lines
2.1 KiB
Python
50 lines
2.1 KiB
Python
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()),
|
|
),
|
|
)
|