fix: refactor

Author: 0xi4o

packages/server/src/enterprise/controllers/login-method.controller.ts

@@ -16,6 +16,12 @@ import { decrypt } from '../utils/encryption.util'
 export class LoginMethodController {
     public async create(req: Request, res: Response, next: NextFunction) {
         try {
+            if (
+                getRunningExpressApp().identityManager.getPlatformType() === Platform.CLOUD ||
+                getRunningExpressApp().identityManager.getPlatformType() === Platform.OPEN_SOURCE
+            ) {
+                throw new InternalFlowiseError(StatusCodes.BAD_GATEWAY, GeneralErrorMessage.FORBIDDEN)
+            }
             const loginMethodService = new LoginMethodService()
             const loginMethod = await loginMethodService.createLoginMethod(req.body)
             return res.status(StatusCodes.CREATED).json(loginMethod)
@@ -66,8 +72,11 @@ export class LoginMethodController {
         try {
             queryRunner = getRunningExpressApp().AppDataSource.createQueryRunner()
             await queryRunner.connect()
-            if (getRunningExpressApp().identityManager.getPlatformType() === Platform.CLOUD) {
-                throw new InternalFlowiseError(StatusCodes.BAD_REQUEST, GeneralErrorMessage.FORBIDDEN)
+            if (
+                getRunningExpressApp().identityManager.getPlatformType() === Platform.CLOUD ||
+                getRunningExpressApp().identityManager.getPlatformType() === Platform.OPEN_SOURCE
+            ) {
+                throw new InternalFlowiseError(StatusCodes.FORBIDDEN, GeneralErrorMessage.FORBIDDEN)
             }
             const query = req.query as Partial<LoginMethod>
             const loginMethodService = new LoginMethodService()
@@ -106,8 +115,11 @@ export class LoginMethodController {
     public async update(req: Request, res: Response, next: NextFunction) {
         try {
             const loginMethodService = new LoginMethodService()
-            if (getRunningExpressApp().identityManager.getPlatformType() === Platform.CLOUD) {
-                throw new InternalFlowiseError(StatusCodes.BAD_REQUEST, GeneralErrorMessage.FORBIDDEN)
+            if (
+                getRunningExpressApp().identityManager.getPlatformType() === Platform.CLOUD ||
+                getRunningExpressApp().identityManager.getPlatformType() === Platform.OPEN_SOURCE
+            ) {
+                throw new InternalFlowiseError(StatusCodes.FORBIDDEN, GeneralErrorMessage.FORBIDDEN)
             }
             const loginMethod = await loginMethodService.createOrUpdateConfig(req.body)
             if (loginMethod?.status === 'OK' && loginMethod?.organizationId) {

packages/server/src/enterprise/services/login-method.service.ts

@@ -79,9 +79,6 @@ export class LoginMethodService {
         try {
             queryRunner = this.dataSource.createQueryRunner()
             await queryRunner.connect()
-            if (getRunningExpressApp().identityManager.getPlatformType() === Platform.CLOUD) {
-                throw new InternalFlowiseError(StatusCodes.BAD_REQUEST, GeneralErrorMessage.FORBIDDEN)
-            }
             const createdBy = await this.userService.readUserById(data.createdBy, queryRunner)
             if (!createdBy) throw new InternalFlowiseError(StatusCodes.NOT_FOUND, UserErrorMessage.USER_NOT_FOUND)
             const organization = await this.organizationService.readOrganizationById(data.organizationId, queryRunner)