| 123456789101112131415161718192021222324252627282930313233 |
- from collections.abc import Mapping
- from typing import Any
- from core.plugin.entities.request import RequestInvokeEncrypt
- from core.tools.utils.configuration import ProviderConfigEncrypter
- from models.account import Tenant
- class PluginEncrypter:
- @classmethod
- def invoke_encrypt(cls, tenant: Tenant, payload: RequestInvokeEncrypt) -> dict:
- encrypter = ProviderConfigEncrypter(
- tenant_id=tenant.id,
- config=payload.data,
- provider_type=payload.namespace,
- provider_identity=payload.identity,
- )
- if payload.opt == "encrypt":
- return {
- "data": encrypter.encrypt(payload.data),
- }
- elif payload.opt == "decrypt":
- return {
- "data": encrypter.decrypt(payload.data),
- }
- elif payload.opt == "clear":
- return {
- "data": encrypter.delete_tool_credentials_cache(),
- }
- else:
- raise ValueError(f"Invalid opt: {payload.opt}")
|
