mirror of
https://github.com/langgenius/dify.git
synced 2025-08-01 13:58:23 +00:00
a4ef900916
Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Yeuoly <admin@srmxy.cn>
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from core.helper.provider_cache import SingletonProviderCredentialsCache
|
|
from core.plugin.entities.request import RequestInvokeEncrypt
|
|
from core.tools.utils.encryption import create_provider_encrypter
|
|
from models.account import Tenant
|
|
|
|
|
|
class PluginEncrypter:
|
|
@classmethod
|
|
def invoke_encrypt(cls, tenant: Tenant, payload: RequestInvokeEncrypt) -> dict:
|
|
encrypter, cache = create_provider_encrypter(
|
|
tenant_id=tenant.id,
|
|
config=payload.config,
|
|
cache=SingletonProviderCredentialsCache(
|
|
tenant_id=tenant.id,
|
|
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":
|
|
cache.delete()
|
|
return {
|
|
"data": {},
|
|
}
|
|
else:
|
|
raise ValueError(f"Invalid opt: {payload.opt}")
|