mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 11:42:29 +08:00
136 lines
4.9 KiB
Python
136 lines
4.9 KiB
Python
import os
|
|
from functools import wraps
|
|
|
|
from flask import request
|
|
from flask_restful import Resource, reqparse
|
|
from werkzeug.exceptions import NotFound, Unauthorized
|
|
|
|
from constants.languages import supported_language
|
|
from controllers.console import api
|
|
from controllers.console.wraps import only_edition_cloud
|
|
from extensions.ext_database import db
|
|
from models.model import App, InstalledApp, RecommendedApp
|
|
|
|
|
|
def admin_required(view):
|
|
@wraps(view)
|
|
def decorated(*args, **kwargs):
|
|
if not os.getenv("ADMIN_API_KEY"):
|
|
raise Unauthorized("API key is invalid.")
|
|
|
|
auth_header = request.headers.get("Authorization")
|
|
if auth_header is None:
|
|
raise Unauthorized("Authorization header is missing.")
|
|
|
|
if " " not in auth_header:
|
|
raise Unauthorized("Invalid Authorization header format. Expected 'Bearer <api-key>' format.")
|
|
|
|
auth_scheme, auth_token = auth_header.split(None, 1)
|
|
auth_scheme = auth_scheme.lower()
|
|
|
|
if auth_scheme != "bearer":
|
|
raise Unauthorized("Invalid Authorization header format. Expected 'Bearer <api-key>' format.")
|
|
|
|
if os.getenv("ADMIN_API_KEY") != auth_token:
|
|
raise Unauthorized("API key is invalid.")
|
|
|
|
return view(*args, **kwargs)
|
|
|
|
return decorated
|
|
|
|
|
|
class InsertExploreAppListApi(Resource):
|
|
@only_edition_cloud
|
|
@admin_required
|
|
def post(self):
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("app_id", type=str, required=True, nullable=False, location="json")
|
|
parser.add_argument("desc", type=str, location="json")
|
|
parser.add_argument("copyright", type=str, location="json")
|
|
parser.add_argument("privacy_policy", type=str, location="json")
|
|
parser.add_argument("custom_disclaimer", type=str, location="json")
|
|
parser.add_argument("language", type=supported_language, required=True, nullable=False, location="json")
|
|
parser.add_argument("category", type=str, required=True, nullable=False, location="json")
|
|
parser.add_argument("position", type=int, required=True, nullable=False, location="json")
|
|
args = parser.parse_args()
|
|
|
|
app = App.query.filter(App.id == args["app_id"]).first()
|
|
if not app:
|
|
raise NotFound(f'App \'{args["app_id"]}\' is not found')
|
|
|
|
site = app.site
|
|
if not site:
|
|
desc = args["desc"] or ""
|
|
copy_right = args["copyright"] or ""
|
|
privacy_policy = args["privacy_policy"] or ""
|
|
custom_disclaimer = args["custom_disclaimer"] or ""
|
|
else:
|
|
desc = site.description or args["desc"] or ""
|
|
copy_right = site.copyright or args["copyright"] or ""
|
|
privacy_policy = site.privacy_policy or args["privacy_policy"] or ""
|
|
custom_disclaimer = site.custom_disclaimer or args["custom_disclaimer"] or ""
|
|
|
|
recommended_app = RecommendedApp.query.filter(RecommendedApp.app_id == args["app_id"]).first()
|
|
|
|
if not recommended_app:
|
|
recommended_app = RecommendedApp(
|
|
app_id=app.id,
|
|
description=desc,
|
|
copyright=copy_right,
|
|
privacy_policy=privacy_policy,
|
|
custom_disclaimer=custom_disclaimer,
|
|
language=args["language"],
|
|
category=args["category"],
|
|
position=args["position"],
|
|
)
|
|
|
|
db.session.add(recommended_app)
|
|
|
|
app.is_public = True
|
|
db.session.commit()
|
|
|
|
return {"result": "success"}, 201
|
|
else:
|
|
recommended_app.description = desc
|
|
recommended_app.copyright = copy_right
|
|
recommended_app.privacy_policy = privacy_policy
|
|
recommended_app.custom_disclaimer = custom_disclaimer
|
|
recommended_app.language = args["language"]
|
|
recommended_app.category = args["category"]
|
|
recommended_app.position = args["position"]
|
|
|
|
app.is_public = True
|
|
|
|
db.session.commit()
|
|
|
|
return {"result": "success"}, 200
|
|
|
|
|
|
class InsertExploreAppApi(Resource):
|
|
@only_edition_cloud
|
|
@admin_required
|
|
def delete(self, app_id):
|
|
recommended_app = RecommendedApp.query.filter(RecommendedApp.app_id == str(app_id)).first()
|
|
if not recommended_app:
|
|
return {"result": "success"}, 204
|
|
|
|
app = App.query.filter(App.id == recommended_app.app_id).first()
|
|
if app:
|
|
app.is_public = False
|
|
|
|
installed_apps = InstalledApp.query.filter(
|
|
InstalledApp.app_id == recommended_app.app_id, InstalledApp.tenant_id != InstalledApp.app_owner_tenant_id
|
|
).all()
|
|
|
|
for installed_app in installed_apps:
|
|
db.session.delete(installed_app)
|
|
|
|
db.session.delete(recommended_app)
|
|
db.session.commit()
|
|
|
|
return {"result": "success"}, 204
|
|
|
|
|
|
api.add_resource(InsertExploreAppListApi, "/admin/insert-explore-apps")
|
|
api.add_resource(InsertExploreAppApi, "/admin/insert-explore-apps/<uuid:app_id>")
|