from sqlalchemy.orm import Session
from typing import List
from app.models.auth.module_model import Module
import uuid
from app.services.auth.tenant_module_service import TenantModuleService
from app.schemas.auth.tenant_module_schema import TenantModuleCreate, TenantModuleUpdate, TenantModuleResponse

class TenantModuleController:
    @staticmethod
    def list_tenant_modules(db: Session, tenant_id: str) -> List[TenantModuleResponse]:
        results = TenantModuleService.list_tenant_modules(db, tenant_id)
        
        response_list = []
        for tm, mod in results:
            response_list.append(TenantModuleResponse(
                id=str(tm.id),
                tenant_id=str(tm.tenant_id),
                module_id=str(tm.module_id),
                module_name=mod.module_name,
                module_icon_url=mod.icon_url,
                assigned_environment_slug=tm.assigned_environment_slug or "prod",
                is_active=tm.is_active,
                module_config=tm.module_config,
                created_at=tm.created_at
            ))
        return response_list

    @staticmethod
    def assign_module(db: Session, tenant_id: str, assignment_data: TenantModuleCreate) -> TenantModuleResponse:
        tm = TenantModuleService.assign_module(db, tenant_id, assignment_data)
        
        module = db.query(Module).filter(Module.id == tm.module_id).first()
        
        return TenantModuleResponse(
            id=str(tm.id),
            tenant_id=str(tm.tenant_id),
            module_id=str(tm.module_id),
            module_name=module.module_name if module else "Unknown",
            module_icon_url=module.icon_url if module else None,
            assigned_environment_slug=tm.assigned_environment_slug,
            is_active=tm.is_active,
            module_config=tm.module_config,
            created_at=tm.created_at
        )

    @staticmethod
    def update_assignment(db: Session, tenant_id: str, tenant_module_id: str, update_data: TenantModuleUpdate) -> TenantModuleResponse:
        tm = TenantModuleService.update_assignment(db, tenant_id, tenant_module_id, update_data)
        
        module = db.query(Module).filter(Module.id == tm.module_id).first()

        return TenantModuleResponse(
            id=str(tm.id),
            tenant_id=str(tm.tenant_id),
            module_id=str(tm.module_id),
            module_name=module.module_name if module else "Unknown",
            module_icon_url=module.icon_url if module else None,
            assigned_environment_slug=tm.assigned_environment_slug,
            is_active=tm.is_active,
            module_config=tm.module_config,
            created_at=tm.created_at
        )

    @staticmethod
    def remove_assignment(db: Session, tenant_id: str, tenant_module_id: str):
        TenantModuleService.remove_assignment(db, tenant_id, tenant_module_id)
        return {"message": "Module removed from tenant successfully"}