feat(commit): finish set question range tag in file and get code in file by range tag

Author: wh131462

common/utils/file/getFilePathById.js

@@ -0,0 +1,14 @@
+import fs from 'node:fs'
+
+/**
+ * 根据id获取文件的具体路径
+ * @param id
+ * @param baseDir
+ * @returns {string|undefined|string[]}
+ */
+export const getFilePathById = (id, baseDir = process.cwd()) => {
+  const dir = fs.readdirSync(baseDir)
+  const files = dir.filter((o) => o.startsWith(id + '.'))
+  if (files.length > 1) return files
+  return files?.[0]
+}

common/utils/file/getRootPath.js

@@ -1,4 +1,9 @@
 import path from 'node:path'
 import { __dirname } from '#common/utils/file/getDirname.js'
+import { currentEnv } from '#common/utils/etc/checkEnv.js'
+import { PackageName } from '#common/constants/question.const.js'
 
-export const rootPath = path.dirname(path.dirname(path.dirname(__dirname)))
+export const rootPath =
+  currentEnv() === 'project'
+    ? path.dirname(path.dirname(path.dirname(__dirname)))
+    : path.join(path.dirname(__dirname), PackageName)

common/utils/question-handler/code.js

@@ -0,0 +1,64 @@
+import { getQuestionCodeList } from '#common/utils/question-getter/getQuestionCodeList.js'
+import {
+  getLangByExtension,
+  setLineComment
+} from '#common/utils/question-handler/questionLanguage.js'
+import { DefaultLang } from '#common/constants/question.const.js'
+import fs from 'node:fs'
+import path from 'node:path'
+import { parseFilePath } from '#common/utils/file/parseFilePath.js'
+
+/**
+ * 获取代码
+ * @param slug
+ * @param lang
+ * @returns {Promise<*>}
+ */
+export async function code(slug, lang) {
+  const list = await getQuestionCodeList(slug)
+  return list.find((o) => o.langSlug === lang)?.code
+}
+/**
+ * 获取支持的代码语言
+ * @param slug
+ * @returns {Promise<string[]>}
+ */
+export async function getSupportCode(slug) {
+  const list = await getQuestionCodeList(slug)
+  return list.map((code) => code?.langSlug)
+}
+
+/**
+ * 生成有范围块的代码
+ * @param lang
+ * @param code
+ * @returns {*|string}
+ */
+export const getCodeRange = (lang, code) => {
+  if (!code) {
+    return setLineComment(
+      lang,
+      `!important: 此题目没有当前语言[${lang}]的代码模板!`
+    )
+  }
+  return (
+    setLineComment(lang, '@QUESTION_START') +
+    code +
+    '\n' +
+    setLineComment(lang, '@QUESTION_END')
+  )
+}
+/**
+ * 获取文件中的代码部分
+ */
+export const getCodeInFile = (filePath) => {
+  const lang = getLangByExtension(filePath)?.lang ?? DefaultLang
+  const data = fs.readFileSync(filePath, 'utf-8')
+  const startTag = setLineComment(lang, '@QUESTION_START')
+  const endTag = setLineComment(lang, '@QUESTION_END')
+  const rangeReg = new RegExp(startTag + '.*' + endTag, 'ms')
+  const rangeTagReg = new RegExp(`(${startTag}|${endTag})+`, 'mg')
+  const match = data.match(rangeReg)
+  if (!match) return null
+  return match[0]?.replace(rangeTagReg, '')
+}

common/utils/question-handler/createQuestion.js

@@ -2,7 +2,7 @@ import fs from 'node:fs'
 import path from 'node:path'
 import { fulfillQuestion } from '#common/utils/question-handler/fulfillQuestion.js'
 import { getQuestionFileExtension } from '#common/utils/question-handler/questionLanguage.js'
-import { getSupportCode } from '#common/utils/question-handler/getCode.js'
+import { getSupportCode } from '#common/utils/question-handler/code.js'
 
 /**
  * 创建问题

common/utils/question-handler/fulfillQuestion.js

@@ -9,6 +9,7 @@ import {
   setBlockComment,
   setLineComment
 } from '#common/utils/question-handler/questionLanguage.js'
+import { getCodeRange } from '#common/utils/question-handler/code.js'
 
 /**
  * @typedef {object} Question
@@ -37,14 +38,7 @@ export function generateTemplateContent(question) {
   return template
     .replace('@Title', setBlockComment(lang, title + describe))
     .replace('@Describe', '')
-    .replace(
-      '@Function',
-      code ??
-        setLineComment(
-          lang,
-          `!important: 此题目没有当前语言[${lang}]的代码模板!`
-        )
-    )
+    .replace('@Function', getCodeRange(lang, code))
     .replace('@TestCase', getTestCase(question))
     .replace('@Console', getConsoleText(question))
 }

common/utils/question-handler/getCode.js

@@ -1,5 +1,5 @@
 import { getQuestionDetailJson } from '#resources/headers/questionDetailJson.js'
-import { getCode } from '#common/utils/question-handler/getCode.js'
+import { code } from '#common/utils/question-handler/code.js'
 import { getQuestionLanguage } from '#common/utils/question-handler/questionLanguage.js'
 import { graphql } from '#common/utils/http/graphql.js'
 
@@ -14,7 +14,7 @@ export async function getQuestionDetail(slug, extra = {}) {
   const questionDetail = await graphql(getQuestionDetailJson(slug))
   const detail = questionDetail.data.question
   const curLang = await getQuestionLanguage()
-  const code = await getCode(slug, curLang)
+  const code = await code(slug, curLang)
   return {
     id: detail?.questionId,
     slug,

common/utils/question-handler/getQuestionDetail.js

@@ -210,6 +210,7 @@ export const LANGUAGES = [
     lineComment: '--'
   }
 ]
+
 /**
  * 设置编程主语言
  */
@@ -241,6 +242,17 @@ export function getQuestionFileExtension(lang = DefaultLang) {
   return detail?.extension
 }
 
+/**
+ * 通过后缀获取语言配置
+ * @param extensionLike
+ * @returns {*}
+ */
+export function getLangByExtension(extensionLike) {
+  const reg = /\.[0-9a-zA-Z_]+$/im
+  const match = extensionLike.match(reg)
+  let extension = match === null ? '.' + extensionLike : match[0]
+  return LANGUAGES?.find((o) => o.extension === extension)
+}
 /**
  * 获取行注释
  * @param lang