mirror of
https://github.com/YFGaia/dify-plus.git
synced 2026-06-12 18:11:42 +08:00
npc0-hue
55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
from Crypto.Util.Padding import unpad
|
|
from Crypto.Cipher import Blowfish
|
|
from configs import dify_config
|
|
from .engine import db
|
|
import base64
|
|
|
|
class SystemIntegrationClassify:
|
|
SYSTEM_INTEGRATION_DINGTALK = 1 # 钉钉
|
|
SYSTEM_INTEGRATION_WEIXIN = 2 # 微信
|
|
SYSTEM_INTEGRATION_FEI_SU = 3 # 飞书
|
|
SYSTEM_INTEGRATION_OAUTH_TWO = 4 # OAuth2
|
|
|
|
|
|
class SystemIntegrationExtend(db.Model):
|
|
__tablename__ = "system_integration_extend"
|
|
__table_args__ = (
|
|
db.PrimaryKeyConstraint("id", name="system_integration_joins_pkey"),
|
|
db.Index("system_integration_joins_classify_idx", "classify"),
|
|
)
|
|
id = db.Column(db.BigInteger, db.Sequence("system_integration_id_sequence"), primary_key=True, autoincrement=True)
|
|
classify = db.Column(db.Integer, nullable=False, server_default=db.text("1"))
|
|
status = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
|
|
corp_id = db.Column(db.String(120), nullable=True)
|
|
agent_id = db.Column(db.String(120), nullable=True)
|
|
app_id = db.Column(db.String(120), nullable=True)
|
|
app_key = db.Column(db.String(120), nullable=True)
|
|
app_secret = db.Column(db.Text, nullable=True)
|
|
config = db.Column(db.Text, nullable=True)
|
|
|
|
def decodeSecret(self):
|
|
if len(self.app_secret) == 0:
|
|
return ""
|
|
# Decode the base64 encoded text
|
|
ciphertext = base64.b64decode(self.app_secret)
|
|
|
|
# Ensure the text length is sufficient
|
|
if len(ciphertext) < Blowfish.block_size:
|
|
raise ValueError("Invalid ciphertext")
|
|
|
|
# Extract the initialization vector (IV) from the beginning of the ciphertext
|
|
iv = ciphertext[:Blowfish.block_size]
|
|
ciphertext = ciphertext[Blowfish.block_size:]
|
|
|
|
# Create the cipher object and decrypt the plaintext
|
|
cipher = Blowfish.new(dify_config.SECRET_KEY.encode('utf-8'), Blowfish.MODE_CBC, iv)
|
|
plaintext = cipher.decrypt(ciphertext)
|
|
|
|
# Unpad the plaintext using PKCS7 unpadding
|
|
try:
|
|
plaintext = unpad(plaintext, Blowfish.block_size)
|
|
except ValueError as e:
|
|
raise ValueError("Invalid padding") from e
|
|
|
|
return plaintext.decode('utf-8')
|