import json
from typing import Literal

from fastapi import FastAPI
from starlette.responses import JSONResponse

from app.general.ModuleConfigModule import ModuleConfigModule
from app.general.general_utils.sql_service import sql_service
from app.utils.db_utils import AsyncSessionDep


def add_general_route(app: FastAPI):
  types = ['list', 'item', 'insert', 'update', 'delete', 'batch_insert', 'batch_update']
  for type in types:
    @app.post("/general/{module}/" + type)
    async def func(module: str, query_param: dict, session: AsyncSessionDep, type=type):
      try:
        result = await perform_general_operation(session=session, module=module, data=query_param, debug_data=[], type=type)
        if "error" in result:
          return JSONResponse(content=result, status_code=500)
        else:
          return result
      finally:
        await session.close()


async def perform_general_operation(
  session: AsyncSessionDep,
  module: str,
  data: dict,
  debug_data=None,
  type: Literal['list', 'item', 'insert', 'update', 'delete', 'batch_insert', 'batch_update'] = 'query'
):
  if debug_data is None:
    debug_data = []

  module_query_result = await get_module_config(session, module, debug_data)

  if 'error' in module_query_result:
    return module_query_result

  module_config = module_query_result.get('result', {"": ""}).get('moduleConfig', None)

  if module_config is not None:
    module_config = json.loads(module_config)

  if module_config is None:
    return {
      "error": f"Can't find module config match: {module}",
      "body": module_query_result,
    }

  if type == 'list':
    return await sql_service.list(session, data, module_config, debug_data)
  elif type == 'item':
    return await sql_service.item(session, data, module_config, debug_data)
  elif type == 'insert':
    return await sql_service.insert(session, data, module_config, debug_data)
  elif type == 'update':
    return await sql_service.update(session, data, module_config, debug_data)
  elif type == 'batch_insert':
    return await sql_service.batch_insert(session, data, module_config, debug_data)
  elif type == 'batch_update':
    return await sql_service.batch_update(session, data, module_config, debug_data)
  elif type == 'delete':
    return await sql_service.delete(session, data, module_config, debug_data)
  else:
    return {"error": f"Can't handle operation type: {type}"}


async def get_module_config(session: AsyncSessionDep, module: str, debug_data=None):
  return await sql_service.item(session, {"code": module}, ModuleConfigModule, debug_data)