a78339a040
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: -LAN- <laipz8200@outlook.com>
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):
|
|
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}")
|