111 lines
4.6 KiB
Python
111 lines
4.6 KiB
Python
|
from sqlalchemy.orm import Session
|
||
|
|
from sqlalchemy.exc import IntegrityError
|
||
|
|
from fastapi import HTTPException
|
||
|
|
from typing import List
|
||
|
|
import uuid
|
||
|
|
from app.models.auth.module_model import Module
|
||
|
|
from app.models.auth.module_environment_model import ModuleEnvironment
|
||
|
|
from app.schemas.auth.module_environment_schema import EnvironmentCreate, EnvironmentUpdate
|
||
|
|
|
||
|
|
class ModuleEnvironmentService:
|
||
|
|
@staticmethod
|
||
|
|
def list_environments(db: Session, module_id: str) -> List[ModuleEnvironment]:
|
||
|
|
module = db.query(Module).filter(Module.id == uuid.UUID(module_id)).first()
|
||
|
|
if not module:
|
||
|
|
raise HTTPException(status_code=404, detail="Module not found")
|
||
|
|
|
||
|
|
return db.query(ModuleEnvironment).filter(
|
||
|
|
ModuleEnvironment.module_id == module.id
|
||
|
|
).order_by(ModuleEnvironment.is_default.desc(), ModuleEnvironment.slug).all()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def create_environment(db: Session, module_id: str, env_data: EnvironmentCreate) -> ModuleEnvironment:
|
||
|
|
module = db.query(Module).filter(Module.id == uuid.UUID(module_id)).first()
|
||
|
|
if not module:
|
||
|
|
raise HTTPException(status_code=404, detail="Module not found")
|
||
|
|
|
||
|
|
if env_data.is_default:
|
||
|
|
db.query(ModuleEnvironment).filter(
|
||
|
|
ModuleEnvironment.module_id == module.id,
|
||
|
|
ModuleEnvironment.is_default == True
|
||
|
|
).update({"is_default": False})
|
||
|
|
|
||
|
|
try:
|
||
|
|
environment = ModuleEnvironment(
|
||
|
|
module_id=module.id,
|
||
|
|
slug=env_data.slug,
|
||
|
|
frontend_base_url=env_data.frontend_base_url,
|
||
|
|
backend_base_url=env_data.backend_base_url,
|
||
|
|
sso_entry_path=env_data.sso_entry_path,
|
||
|
|
permission_sync_endpoint=env_data.permission_sync_endpoint,
|
||
|
|
sso_exchange_endpoint=env_data.sso_exchange_endpoint,
|
||
|
|
trust_type=env_data.trust_type,
|
||
|
|
trust_credentials=env_data.trust_credentials,
|
||
|
|
is_default=env_data.is_default,
|
||
|
|
is_active=env_data.is_active
|
||
|
|
)
|
||
|
|
db.add(environment)
|
||
|
|
db.commit()
|
||
|
|
db.refresh(environment)
|
||
|
|
return environment
|
||
|
|
except IntegrityError:
|
||
|
|
db.rollback()
|
||
|
|
raise HTTPException(status_code=409, detail="Environment slug already exists for this module")
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def update_environment(db: Session, module_id: str, env_id: str, env_data: EnvironmentUpdate) -> ModuleEnvironment:
|
||
|
|
environment = db.query(ModuleEnvironment).filter(
|
||
|
|
ModuleEnvironment.id == uuid.UUID(env_id),
|
||
|
|
ModuleEnvironment.module_id == uuid.UUID(module_id)
|
||
|
|
).first()
|
||
|
|
|
||
|
|
if not environment:
|
||
|
|
raise HTTPException(status_code=404, detail="Environment not found")
|
||
|
|
|
||
|
|
if env_data.is_default and env_data.is_default != environment.is_default:
|
||
|
|
db.query(ModuleEnvironment).filter(
|
||
|
|
ModuleEnvironment.module_id == environment.module_id,
|
||
|
|
ModuleEnvironment.id != environment.id,
|
||
|
|
ModuleEnvironment.is_default == True
|
||
|
|
).update({"is_default": False})
|
||
|
|
|
||
|
|
update_data = env_data.model_dump(exclude_unset=True)
|
||
|
|
for key, value in update_data.items():
|
||
|
|
setattr(environment, key, value)
|
||
|
|
|
||
|
|
db.commit()
|
||
|
|
db.refresh(environment)
|
||
|
|
return environment
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def set_default_environment(db: Session, module_id: str, env_id: str):
|
||
|
|
environment = db.query(ModuleEnvironment).filter(
|
||
|
|
ModuleEnvironment.id == uuid.UUID(env_id),
|
||
|
|
ModuleEnvironment.module_id == uuid.UUID(module_id)
|
||
|
|
).first()
|
||
|
|
|
||
|
|
if not environment:
|
||
|
|
raise HTTPException(status_code=404, detail="Environment not found")
|
||
|
|
|
||
|
|
db.query(ModuleEnvironment).filter(
|
||
|
|
ModuleEnvironment.module_id == environment.module_id
|
||
|
|
).update({"is_default": False})
|
||
|
|
|
||
|
|
environment.is_default = True
|
||
|
|
db.commit()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def delete_environment(db: Session, module_id: str, env_id: str):
|
||
|
|
environment = db.query(ModuleEnvironment).filter(
|
||
|
|
ModuleEnvironment.id == uuid.UUID(env_id),
|
||
|
|
ModuleEnvironment.module_id == uuid.UUID(module_id)
|
||
|
|
).first()
|
||
|
|
|
||
|
|
if not environment:
|
||
|
|
raise HTTPException(status_code=404, detail="Environment not found")
|
||
|
|
|
||
|
|
if environment.is_default:
|
||
|
|
raise HTTPException(status_code=400, detail="Cannot delete default environment. Set another environment as default first.")
|
||
|
|
|
||
|
|
db.delete(environment)
|
||
|
|
db.commit()
|