ojnext/src/oj/api.ts

import { DIFFICULTY } from "utils/constants"
import { getACRate } from "utils/functions"
import http from "utils/http"
import {
  Problem,
  Submission,
  SubmissionListPayload,
  SubmitCodePayload,
} from "utils/types"

function filterResult(result: Problem) {
  const newResult = {
    id: result.id,
    _id: result._id,
    title: result.title,
    difficulty: DIFFICULTY[result.difficulty],
    tags: result.tags,
    submission: result.submission_number,
    rate: getACRate(result.accepted_number, result.submission_number),
    status: "",
    author: result.created_by.username,
    allow_flowchart: result.allow_flowchart,
  }
  if (result.my_status === null || result.my_status === undefined) {
    newResult.status = "not_test"
  } else if (result.my_status === 0) {
    newResult.status = "passed"
  } else {
    newResult.status = "failed"
  }
  return newResult
}

export function getWebsiteConfig() {
  return http.get("website")
}

export async function getProblemList(
  offset = 0,
  limit = 10,
  searchParams: any = {},
) {
  let params: any = {
    paging: true,
    offset,
    limit,
  }
  Object.keys(searchParams).forEach((element) => {
    if (searchParams[element]) {
      params[element] = searchParams[element]
    }
  })
  const res = await http.get("problem", { params })
  return {
    results: res.data.results.map(filterResult),
    total: res.data.total,
  }
}

export function getAuthors(all = false) {
  return http.get("problem/author", {
    params: {
      all: all ? "1" : "0",
    },
  })
}

export function getRandomProblemID() {
  return http.get("pickone")
}

export function getProblem(problemID: string, contestID: string) {
  const endpoint = !!contestID ? "contest/problem" : "problem"
  return http.get(endpoint, {
    params: {
      problem_id: problemID,
      contest_id: contestID,
    },
  })
}

export function getProblemBeatRate(problemID: number) {
  return http.get("problem/beat_count", { params: { problem_id: problemID } })
}

export function getSubmission(id: string) {
  return http.get<Submission>("submission", {
    params: { id },
  })
}

export function submitCode(data: SubmitCodePayload) {
  return http.post("submission", data)
}

export function getSubmissions(params: Partial<SubmissionListPayload>) {
  const endpoint = !!params.contest_id ? "contest_submissions" : "submissions"
  return http.get(endpoint, { params })
}

export function getRankOfProblem(problem_id: string) {
  return http.get("user_problem_rank", { params: { problem_id: problem_id } })
}

export function getTodaySubmissionCount() {
  return http.get("submissions/today_count")
}

export function adminRejudge(id: string) {
  return http.get("admin/submission/rejudge", {
    params: { id },
  })
}

export function getSubmissionStatistics(
  duration: { start: string; end: string },
  problemID?: string,
  username?: string,
) {
  return http.get("admin/submission/statistics", {
    params: {
      ...duration,
      problem_id: problemID,
      username,
    },
  })
}

export function getRank(
  offset: number,
  limit: number,
  n: number,
  username?: string,
) {
  return http.get("user_rank", {
    params: { offset, limit, rule: "acm", username, n },
  })
}

export function getActivityRank(start: string) {
  return http.get("user_activity_rank", {
    params: { start },
  })
}

export function getContestList(query: {
  offset: number
  limit: number
  keyword: string
  status: string
  tag: string
}) {
  return http.get("contests", { params: query })
}

export function getContest(id: string) {
  return http.get("contest", { params: { id } })
}

export function getContestAccess(id: string) {
  return http.get("contest/access", { params: { contest_id: id } })
}

export function checkContestPassword(contestID: string, password: string) {
  return http.post("contest/password", {
    contest_id: contestID,
    password,
  })
}

export async function getContestProblems(contestID: string) {
  const res = await http.get("contest/problem", {
    params: { contest_id: contestID },
  })
  return res.data.map(filterResult)
}

export function getContestRank(
  contestID: string,
  query: { limit: number; offset: number; force_refresh: "1" | "0" },
) {
  return http.get("contest_rank", {
    params: {
      contest_id: contestID,
      ...query,
    },
  })
}

export function uploadAvatar(file: File) {
  const form = new window.FormData()
  form.append("image", file)
  return http.post("upload_avatar", form, {
    headers: { "content-type": "multipart/form-data" },
  })
}

export function updateProfile(data: { real_name: string; mood: string }) {
  return http.put("profile", data)
}

export function getAnnouncementList(offset = 0, limit = 10) {
  return http.get("announcement", { params: { limit, offset } })
}

export function getAnnouncement(id: number) {
  return http.get("announcement", { params: { id } })
}

export function createMessage(data: {
  recipient: number
  message: string
  submission: string
}) {
  return http.post("message", data)
}

export function getMessageList(offset = 0, limit = 10) {
  return http.get("message", { params: { limit, offset } })
}

export function createComment(data: {
  problem_id: number
  description_rating: number
  difficulty_rating: number
  comprehensive_rating: number
  content: string
}) {
  return http.post("comment", data)
}

export function getComment(problemID: number) {
  return http.get("comment", { params: { problem_id: problemID } })
}

export function getCommentStatistics(problemID: number) {
  return http.get("comment/statistics", { params: { problem_id: problemID } })
}

// TODO: 这个API有问题
export function refreshUserProblemDisplayIds() {
  return http.get("profile/fresh_display_id")
}

export function getMetrics(userid: number) {
  return http.get("metrics", { params: { userid } })
}

export function getTutorial(id: number) {
  return http.get("tutorial", { params: { id } })
}

export function getTutorials() {
  return http.get("tutorials")
}

export function getAIDetailData(start: string, end: string) {
  return http.get("ai/detail", { params: { start, end } })
}

export function getAIDurationData(end: string, duration: string) {
  return http.get("ai/duration", { params: { end, duration } })
}

export function getAIHeatmapData() {
  return http.get("ai/heatmap")
}

// ==================== 流程图相关API ====================

export function submitFlowchart(data: {
  problem_id: number
  mermaid_code: string
  flowchart_data: any // 这个是压缩之后的，元数据太长了
}) {
  return http.post("flowchart/submission", data)
}

export function getFlowchartSubmission(id: string) {
  return http.get("flowchart/submission", {
    params: { id },
  })
}

export function getFlowchartSubmissions(params: {
  username?: string
  problem_id?: string
  myself?: string
  offset?: number
  limit?: number
}) {
  return http.get("flowchart/submissions", { params })
}

export function retryFlowchartSubmission(submissionId: string) {
  return http.post("flowchart/submission/retry", {
    submission_id: submissionId,
  })
}

export function getCurrentProblemFlowchartSubmission(problemId: number) {
  return http.get("flowchart/submission/current", {
    params: { problem_id: problemId },
  })
}

// ==================== 题单相关API ====================

export function getProblemSetList(
  offset = 0,
  limit = 10,
  keyword = "",
  difficulty = "",
  status = "",
) {
  return http.get("problemset", {
    params: {
      offset,
      limit,
      keyword,
      difficulty,
      status,
    },
  })
}

export function getProblemSetDetail(id: number) {
  return http.get(`problemset/${id}`)
}

export function getProblemSetProblems(problemSetId: number) {
  return http.get(`problemset/${problemSetId}/problems`)
}

export function joinProblemSet(problemSetId: number) {
  return http.post("problemset/progress", {
    problemset_id: problemSetId,
  })
}

export function updateProblemSetProgress(
  problemSetId: number,
  problemId: number,
  submissionId: string,
) {
  return http.put("problemset/progress", {
    problemset_id: problemSetId,
    problem_id: problemId,
    submission_id: submissionId,
  })
}

// 获取用户徽章列表
export function getUserBadges() {
  return http.get("user/badges")
}

// 获取题单徽章列表
export function getProblemSetBadges(problemSetId: number) {
  return http.get(`problemset/${problemSetId}/badges`)
}

// 获取题单用户进度列表
export function getProblemSetUserProgress(problemSetId: number) {
  return http.get(`problemset/${problemSetId}/users_progress`)
}