gitlab_monitor/controllers/manageRobot/index.ts

import db from "../../db"
import service from "../../service"
import { calculateWeeklyRate } from "../../utils/robotTools"
import { getPrevWeekWithYear, getWeekTimeWithYear } from "../../utils/timeTools"

/**
 * 获取新的CI/CD状态
 * @returns {Promise<{ has_new_cicd_count: string, without_new_cicd_count: string }>} - 返回包含有新CI/CD和无新CI/CD的项目数量
 */
const getNewCicdStatus = async () => {
  const fullProjList = await db.project.getFullList()
  const has_new_cicd_count = String(
    fullProjList.filter((item) => {
      return item.has_new_cicd === true
    }).length
  )
  const without_new_cicd_count = String(
    fullProjList.filter((item) => {
      return item.has_new_cicd === false
    }).length
  )
  return {
    has_new_cicd_count,
    without_new_cicd_count,
  }
}

/**
 * 获取统计信息
 * @returns {Promise<Object>} - 返回包含统计信息的对象
 */
const getStatisticsInfo = async () => {
  const curWeekInfo = await db.view.getFullStatisticsByWeek(
    getWeekTimeWithYear()
  )
  const prevWeekInfo = await db.view.getFullStatisticsByWeek(
    getPrevWeekWithYear()
  )
  if (!curWeekInfo || !prevWeekInfo) return {}
  return {
    total_count: String(curWeekInfo?.total_count ?? 0),
    weekly_count_rate: calculateWeeklyRate(
      curWeekInfo?.total_count ?? 0,
      prevWeekInfo?.total_count ?? 0
    ).text,
    duration: String(curWeekInfo?.duration ?? 0),
    weekly_duration_rate: calculateWeeklyRate(
      curWeekInfo?.duration ?? 0,
      prevWeekInfo?.duration ?? 0
    ).text,
    success_rate: String(curWeekInfo?.success_rate),
    weekly_success_rate: calculateWeeklyRate(
      curWeekInfo?.success_rate,
      prevWeekInfo?.success_rate
    ).text,
  }
}

/**
 * 获取项目差异信息
 * @returns {Promise<Array>} - 返回包含项目差异信息的数组
 */
const getProjDiffInfo = async () => {
  const curWeekInfo =
    (await db.view.getProjStatisticsByWeek(getWeekTimeWithYear())) || []
  const prevWeekInfo =
    (await db.view.getProjStatisticsByWeek(getPrevWeekWithYear())) || []

  const group: {
    project_name: string
    project_ref: string
    project_duration: string
    project_duration_rate: string
    percentage: string
  }[] = []

  curWeekInfo.forEach((curWeekProjInfo) => {
    const prevWeekProjInfo = prevWeekInfo.find(
      (info) =>
        info.ref === curWeekProjInfo.ref && info.name === curWeekProjInfo.name
    )
    if (!prevWeekProjInfo) return

    const { text: project_duration_rate, percentage } = calculateWeeklyRate(
      curWeekProjInfo.duration,
      prevWeekProjInfo.duration,
      false
    )

    if (percentage === "0") return

    group.push({
      project_name: curWeekProjInfo.name,
      project_ref: curWeekProjInfo.ref,
      project_duration: String(curWeekProjInfo.duration),
      project_duration_rate,
      percentage,
    })
  })

  // 排序
  group.sort((a, b) => Number(b.percentage) - Number(a.percentage))

  // 取前五个
  return group.slice(0, 5)
}

/**
 * 获取机器人消息
 * @returns {Promise<string>} - 返回机器人消息的JSON字符串
 */
const getRobotMsg = async () =>
  JSON.stringify({
    type: "template",
    data: {
      config: {
        update_multi: true,
      },
      template_id: "ctp_AAyVLS6Q37cL",
      template_variable: {
        ...(await getNewCicdStatus()),
        ...(await getStatisticsInfo()),
        group_table: await getProjDiffInfo(),
      },
    },
  })

/**
 * 通过ChatID发送CI报告
 * @param {string} chat_id - ChatID
 * @returns {Promise<void>}
 */
const sendCIReportByChatId = async (chat_id: string) => {
  await service.message.byChatId(chat_id, await getRobotMsg())
}

/**
 * 通过定时任务发送CI报告
 * @returns {Promise<void>}
 */
const sendCIReportByCron = async () =>
  await service.message.byGroupId("52usf3w8l6z4vs1", await getRobotMsg())

const manageRobot = {
  sendCIReportByChatId,
  sendCIReportByCron,
}

export default manageRobot