Remove useless code (#4416)

api/services/enterprise/enterprise_feature_service.py

@@ -1,28 +0,0 @@
-from flask import current_app
-from pydantic import BaseModel
-
-from services.enterprise.enterprise_service import EnterpriseService
-
-
-class EnterpriseFeatureModel(BaseModel):
-    sso_enforced_for_signin: bool = False
-    sso_enforced_for_signin_protocol: str = ''
-
-
-class EnterpriseFeatureService:
-
-    @classmethod
-    def get_enterprise_features(cls) -> EnterpriseFeatureModel:
-        features = EnterpriseFeatureModel()
-
-        if current_app.config['ENTERPRISE_ENABLED']:
-            cls._fulfill_params_from_enterprise(features)
-
-        return features
-
-    @classmethod
-    def _fulfill_params_from_enterprise(cls, features):
-        enterprise_info = EnterpriseService.get_info()
-
-        features.sso_enforced_for_signin = enterprise_info['sso_enforced_for_signin']
-        features.sso_enforced_for_signin_protocol = enterprise_info['sso_enforced_for_signin_protocol']

api/services/enterprise/enterprise_sso_service.py

@@ -1,60 +0,0 @@
-import logging
-
-from models.account import Account, AccountStatus
-from services.account_service import AccountService, TenantService
-from services.enterprise.base import EnterpriseRequest
-
-logger = logging.getLogger(__name__)
-
-
-class EnterpriseSSOService:
-
-    @classmethod
-    def get_sso_saml_login(cls) -> str:
-        return EnterpriseRequest.send_request('GET', '/sso/saml/login')
-
-    @classmethod
-    def post_sso_saml_acs(cls, saml_response: str) -> str:
-        response = EnterpriseRequest.send_request('POST', '/sso/saml/acs', json={'SAMLResponse': saml_response})
-        if 'email' not in response or response['email'] is None:
-            logger.exception(response)
-            raise Exception('Saml response is invalid')
-
-        return cls.login_with_email(response.get('email'))
-
-    @classmethod
-    def get_sso_oidc_login(cls):
-        return EnterpriseRequest.send_request('GET', '/sso/oidc/login')
-
-    @classmethod
-    def get_sso_oidc_callback(cls, args: dict):
-        state_from_query = args['state']
-        code_from_query = args['code']
-        state_from_cookies = args['oidc-state']
-
-        if state_from_cookies != state_from_query:
-            raise Exception('invalid state or code')
-
-        response = EnterpriseRequest.send_request('GET', '/sso/oidc/callback', params={'code': code_from_query})
-        if 'email' not in response or response['email'] is None:
-            logger.exception(response)
-            raise Exception('OIDC response is invalid')
-
-        return cls.login_with_email(response.get('email'))
-
-    @classmethod
-    def login_with_email(cls, email: str) -> str:
-        account = Account.query.filter_by(email=email).first()
-        if account is None:
-            raise Exception('account not found, please contact system admin to invite you to join in a workspace')
-
-        if account.status == AccountStatus.BANNED:
-            raise Exception('account is banned, please contact system admin')
-
-        tenants = TenantService.get_join_tenants(account)
-        if len(tenants) == 0:
-            raise Exception("workspace not found, please contact system admin to invite you to join in a workspace")
-
-        token = AccountService.get_account_jwt_token(account)
-
-        return token

api/services/feature_service.py

@@ -2,6 +2,7 @@ from flask import current_app
 from pydantic import BaseModel
 
 from services.billing_service import BillingService
+from services.enterprise.enterprise_service import EnterpriseService
 
 
 class SubscriptionModel(BaseModel):
@@ -30,6 +31,13 @@ class FeatureModel(BaseModel):
     can_replace_logo: bool = False
 
 
+class SystemFeatureModel(BaseModel):
+    sso_enforced_for_signin: bool = False
+    sso_enforced_for_signin_protocol: str = ''
+    sso_enforced_for_web: bool = False
+    sso_enforced_for_web_protocol: str = ''
+
+
 class FeatureService:
 
     @classmethod
@@ -43,6 +51,15 @@ class FeatureService:
 
         return features
 
+    @classmethod
+    def get_system_features(cls) -> SystemFeatureModel:
+        system_features = SystemFeatureModel()
+
+        if current_app.config['ENTERPRISE_ENABLED']:
+            cls._fulfill_params_from_enterprise(system_features)
+
+        return system_features
+
     @classmethod
     def _fulfill_params_from_env(cls, features: FeatureModel):
         features.can_replace_logo = current_app.config['CAN_REPLACE_LOGO']
@@ -73,3 +90,11 @@ class FeatureService:
         features.docs_processing = billing_info['docs_processing']
         features.can_replace_logo = billing_info['can_replace_logo']
 
+    @classmethod
+    def _fulfill_params_from_enterprise(cls, features):
+        enterprise_info = EnterpriseService.get_info()
+
+        features.sso_enforced_for_signin = enterprise_info['sso_enforced_for_signin']
+        features.sso_enforced_for_signin_protocol = enterprise_info['sso_enforced_for_signin_protocol']
+        features.sso_enforced_for_web = enterprise_info['sso_enforced_for_web']
+        features.sso_enforced_for_web_protocol = enterprise_info['sso_enforced_for_web_protocol']