import { LarkService } from "@egg/net-tool"

import { APP_MAP } from "../../constant/config"
import { Context } from "../../types"

/**
 * 处理登录请求
 * @param req
 * @returns
 */
const manageLogin = async (ctx: Context) => {
  const { req, genResp, logger, requestId } = ctx
  logger.info("micro app login")
  const url = new URL(req.url)
  const code = url.searchParams.get("code")
  const appName = url.searchParams.get("app_name")
  if (!code) {
    return genResp.badRequest("code not found")
  }
  if (!appName) {
    return genResp.badRequest("app_name not found")
  }
  // 获取 app info
  const appInfo = APP_MAP[appName]
  if (!appInfo) {
    return genResp.badRequest("app not found")
  }

  const larkService = new LarkService({
    appId: appInfo.appId,
    appSecret: appInfo.appSecret,
    requestId,
  })

  const {
    code: resCode,
    data,
    message,
  } = await larkService.user.code2Login(code)

  logger.debug(`get user session: ${JSON.stringify(data)}`)

  if (resCode !== 0) {
    return genResp.serverError(message)
  }

  return genResp.ok(data)
}

/**
 * 处理批量获取用户信息请求
 * @param req
 * @returns
 */
const manageBatchUser = async (ctx: Context) => {
  const { body, genResp, logger, requestId } = ctx
  logger.info("batch get user info")
  if (!body) return genResp.badRequest("req body is empty")
  const { user_ids, user_id_type, app_name } = body
  if (!user_ids) {
    return genResp.badRequest("user_ids not found")
  }
  if (!user_id_type) {
    return genResp.badRequest("user_id_type not found")
  }
  if (!app_name) {
    return genResp.badRequest("app_name not found")
  }
  // 获取 app info
  const appInfo = APP_MAP[app_name]
  if (!appInfo) {
    return genResp.badRequest("app not found")
  }

  const larkService = new LarkService({
    appId: appInfo.appId,
    appSecret: appInfo.appSecret,
    requestId,
  })

  const { code, data, message } = await larkService.user.batchGet(
    user_ids,
    user_id_type
  )

  logger.debug(`batch get user info: ${JSON.stringify(data)}`)

  if (code !== 0) {
    return genResp.serverError(message)
  }

  return genResp.ok(data)
}

/**
 * 处理小程序请求
 * @param req
 * @returns
 */
export const manageMicroAppReq = async (ctx: Context) => {
  const path = ctx.path.child("/micro_app")
  // 处理登录请求
  if (path.exact("/login")) {
    return manageLogin(ctx)
  }
  // 处理批量获取用户信息请求
  if (path.exact("/batch_user")) {
    return manageBatchUser(ctx)
  }
  return ctx.genResp.ok()
}