import { GitlabProjectModel } from "../../db/gitlabProject/index."
import { Context } from "../../types"

/**
 * 刷新项目的Gitlab Event
 * @param param0
 * @param project
 * @returns
 */
const refreshProjectHooks = async (
  { gitlabService }: Context,
  project: GitlabProjectModel
) => {
  gitlabService.setProjectId(project.projectId)
  const eventList = await gitlabService.hook.getList()
  const HOOK_URL =
    Bun.env.NODE_ENV === "production"
      ? "https://lark-egg.ai.xiaomi.com/gitlab"
      : "https://lark-egg-preview.ai.xiaomi.com/gitlab"
  // 找到目标Event
  const targetEvent = eventList.find((event) => event.url === HOOK_URL)

  // 清除Event
  if (!project.openCICDNotify && !project.openMRSummary) {
    if (!targetEvent) return
    await gitlabService.hook.delete(targetEvent.id)
    return
  }

  // 添加Event
  if (targetEvent) return
  await gitlabService.hook.add({
    url: HOOK_URL,
    push_events: false,
    job_events: true,
    merge_requests_events: true,
  })
}

const openFunc = async (
  ctx: Context,
  projectId: number,
  func: "openCICDNotify" | "openMRSummary"
) => {
  const { logger, larkBody, larkService, larkCard, db } = ctx
  const cardGender = larkCard.child("gitlabAgent")
  if (!projectId) {
    logger.error("项目ID格式错误", { projectId })
    await larkService.message.replyCard(
      larkBody.messageId,
      cardGender.genErrorCard("项目ID格式错误，请检查项目ID是否正确")
    )
    return
  }
  const project = await db.gitlabProject.getAndCreate(
    projectId,
    ctx.gitlabService
  )
  if (!project) {
    logger.error("项目不存在", { projectId })
    await larkService.message.replyCard(
      larkBody.messageId,
      cardGender.genErrorCard(
        "该项目不存在，请检查项目ID是否正确，或确认 @itp-cr-agent 是否具有对项目的Maintainer权限"
      )
    )
    return
  }
  const newProj = await db.gitlabProject.update(project.id, {
    [func]: true,
  })
  if (!newProj) {
    logger.error("更新项目失败", { projectId })
    await larkService.message.replyCard(
      larkBody.messageId,
      cardGender.genErrorCard("操作失败，请稍后再试")
    )
    return
  }
  await refreshProjectHooks(ctx, newProj)
  await larkService.message.replyCard(
    larkBody.messageId,
    cardGender.genSuccessCard(
      `已开启${func === "openCICDNotify" ? "CI/CD成功提醒" : "MR自动总结"}`
    )
  )
}

const closeFunc = async (
  ctx: Context,
  projectId: number,
  func: "openCICDNotify" | "openMRSummary"
) => {
  const { logger, larkBody, larkService, larkCard, db } = ctx
  const cardGender = larkCard.child("gitlabAgent")
  if (!projectId) {
    logger.error("项目ID格式错误", { projectId })
    await larkService.message.replyCard(
      larkBody.messageId,
      cardGender.genErrorCard("项目ID格式错误，请检查项目ID是否正确")
    )
    return
  }
  const project = await db.gitlabProject.getByProjectId(projectId)
  if (!project) {
    logger.error("项目不存在", { projectId })
    await larkService.message.replyCard(
      larkBody.messageId,
      cardGender.genSuccessCard(
        `已关闭${func === "openCICDNotify" ? "CI/CD成功提醒" : "MR自动总结"}`
      )
    )
    return
  }
  const newProj = await db.gitlabProject.update(project.id, {
    [func]: false,
  })
  if (!newProj) {
    logger.error("更新项目失败", { projectId })
    await larkService.message.replyCard(
      larkBody.messageId,
      cardGender.genErrorCard("操作失败，请稍后再试")
    )
    return
  }
  await refreshProjectHooks(ctx, newProj)
  await larkService.message.replyCard(
    larkBody.messageId,
    cardGender.genSuccessCard(
      `已关闭${func === "openCICDNotify" ? "CI/CD成功提醒" : "MR自动总结"}`
    )
  )
}

const register = {
  openCICDNotify: async (ctx: Context, projectId: number) =>
    openFunc(ctx, projectId, "openCICDNotify"),
  closeCICDNotify: async (ctx: Context, projectId: number) =>
    closeFunc(ctx, projectId, "openCICDNotify"),
  openMRSummary: async (ctx: Context, projectId: number) =>
    openFunc(ctx, projectId, "openMRSummary"),
  closeMRSummary: async (ctx: Context, projectId: number) =>
    closeFunc(ctx, projectId, "openMRSummary"),
}

export default register