/**
 * 把 Python 的编译错误（`statistic_info.err_info`）翻成中文。
 *
 * 学生是中职生，英文报错等于没有报错。好在句式非常集中：2025 秋以来 2700 条
 * Python 编译错误里，前 12 种句子占 91%、前 16 种占 97%，所以这里是一张规则表，
 * 不是通用翻译。第二名「invalid character」（22%）几乎全是中文标点，另有一键修复，
 * 见 chinesePunctuation.ts。
 *
 * 判题机给的原文有两种形状：
 *
 *     File "/judger/run/<随机目录>/solution.py", line 7
 *         print("{:.0f}"."{:.2f}".format(x))
 *                        ^^^^^^^^
 *     SyntaxError: invalid syntax
 *
 *     Sorry: IndentationError: unexpected indent (solution.py, line 3)
 *
 * 第一种的代码行是 Python 去掉**行首空格**之后再垫 4 个空格打出来的，`^` 的列号
 * 按去掉之后的算（traceback 的 `_format_syntax_error`，3.10 起如此）。
 */

export interface PythonErrorExplanation {
  /** 出错的行，1 起 */
  line: number | null
  /** 判题机打出来的那一行代码（去掉了行首空格），用来和编辑器里的对位 */
  sourceLine: string | null
  /** `^` 在 sourceLine 里标的范围，[from, to) */
  caret: { from: number; to: number } | null
  /** 给学生看的中文说明，不含行号 */
  message: string
  /** 报错点名的那个中文标点；有它才出「一键换成英文」 */
  punctuation: string | null
}

// 行首可能有空格：报错前面先打了一条 SyntaxWarning（比如 `3and`）时，File 那行是缩进的
const WHERE_RE = /^\s*File "[^"]*", line (\d+)/m
const SHORT_RE = /^Sorry: (\w+Error): (.*) \([^,()]*, line (\d+)\)\s*$/m

/**
 * 中文空格（U+3000）。写成常量而不是字面量：Oxfmt 会把字符串里的 `\u3000` 转义
 * 还原成原字符，那样源码里就是一个看不见的空白。
 */
const IDEOGRAPHIC_SPACE = String.fromCharCode(0x3000)

/** 中文标点 → 英文。「。」不在表里：它多半是句末多打的，换成「.」照样错，只能删 */
export const PUNCTUATION_MAP: Record<string, string> = {
  '（': '(',
  '）': ')',
  '，': ',',
  '：': ':',
  '；': ';',
  '【': '[',
  '】': ']',
  '｛': '{',
  '｝': '}',
  '！': '!',
  '？': '?',
  '＝': '=',
  '＋': '+',
  '－': '-',
  '＊': '*',
  '／': '/',
  '＜': '<',
  '＞': '>',
  '“': '"',
  '”': '"',
  '‘': "'",
  '’': "'",
  '＂': '"',
  '＃': '#',
  '×': '*',
  '÷': '/',
  [IDEOGRAPHIC_SPACE]: ' ',
}

/**
 * 标点的名字。说明里**只说名字、不拿字形对比**：「：」和「:」、「，」和「,」在正文字体里
 * 几乎一样，「把「：」换成「:」」这句话学生看不出区别。「冒号是中文的，要换成英文的」
 * 看得懂，要换哪一个由卡片里那行代码的标红来指。
 */
export const PUNCTUATION_NAMES: Record<string, string> = {
  '（': '左括号',
  '）': '右括号',
  '，': '逗号',
  '：': '冒号',
  '；': '分号',
  '【': '左方括号',
  '】': '右方括号',
  '｛': '左花括号',
  '｝': '右花括号',
  '！': '感叹号',
  '？': '问号',
  '＝': '等号',
  '＋': '加号',
  '－': '减号',
  '＊': '星号',
  '／': '斜杠',
  '＜': '小于号',
  '＞': '大于号',
  '“': '引号',
  '”': '引号',
  '‘': '单引号',
  '’': '单引号',
  '＂': '引号',
  '＃': '井号',
  [IDEOGRAPHIC_SPACE]: '空格',
}

/**
 * 规则除了报错原文，还能看出错的那一行（判题机打出来的，去掉了行首空格）。同一句报错
 * 学生错法不同，要看那一行才分得开：`expected ':'` 九成是 `else a < 60:`，行尾明明有冒号；
 * 「Maybe you meant '=='」只有 if / while 那行才是真想判断相等。返回 null 就接着往下找。
 */
type Rule = [RegExp, (match: RegExpMatchArray, source: string) => string | null]

/** 去掉引号里的内容，只看代码本身的符号 */
function stripStrings(source: string) {
  return source.replace(/(["'])(?:\\.|(?!\1).)*\1/g, '""')
}

/** 括号外面的逗号：`if a > 9, a < 100:` 里有，`if x in (1, 2):` 里没有 */
function hasTopLevelComma(source: string) {
  let depth = 0
  for (const ch of stripStrings(source)) {
    if ('([{'.includes(ch)) depth++
    else if (')]}'.includes(ch)) depth--
    else if (ch === ',' && depth === 0) return true
  }
  return false
}

const CONDITION_LINE = /^(if|elif|while)\b/

const RULES: Rule[] = [
  [
    // u：emoji 是两个 UTF-16 单元，不带 u 的 (.) 对不上
    /^invalid character '(.)'/u,
    ([, char]) => {
      // 数学符号不是「中文的」，是写法不对：代码里乘除用 * 和 /
      if (char === '×') return '这里用了乘号「×」，代码里的乘法要写成星号「*」。'
      if (char === '÷') return '这里用了除号「÷」，代码里的除法要写成斜杠「/」。'
      if (char === '√') return '代码里没有根号「√」：开平方写成 x ** 0.5，或者 import math 之后用 math.sqrt(x)。'
      if (char === '²' || char === '³') return `代码里没有上标的「${char}」：平方写成 x ** 2，立方写成 x ** 3。`
      if (char === '≤') return '代码里没有「≤」：小于等于要写成小于号加等号 <=。'
      if (char === '≥') return '代码里没有「≥」：大于等于要写成大于号加等号 >=。'
      if (char === '≠') return '代码里没有「≠」：不等于要写成感叹号加等号 !=。'
      if (char === '、') return '代码里不用顿号「、」：几样东西之间要用英文逗号隔开。写代码的时候先切换到英文输入法。'
      const name = PUNCTUATION_NAMES[char!]
      if (name) {
        return `这里的${name}是中文输入法打出来的，要换成英文的${name}。写代码的时候先切换到英文输入法。`
      }
      if (char === '。') return '这里多了一个中文句号「。」，代码里不用句号，删掉它。'
      return `这里有一个不能出现在代码里的字符「${char}」，多半是中文输入法打出来的，删掉后用英文输入法重新打。`
    },
  ],
  [/^invalid non-printable character/, () => '这里有一个看不见的特殊字符，多半是从别处复制过来的。把这一行删掉，自己重新打一遍。'],
  [
    /^expected ':'/,
    (_, source) =>
      /^else\b\s*[^\s:]/.test(source)
        ? 'else 后面不能写条件，它的意思是「上面的条件都不满足」。还要再判断一个条件，就用 elif 条件:。'
        : '这一行末尾少了英文冒号「:」。if、elif、else、for、while、def 这些行的最后都要有冒号。',
  ],
  [
    /^expected an indented block after '(\w+)' statement on line (\d+)/,
    ([, keyword, owner]) => `这一行要往里缩进（按一下 Tab 键）：它是第 ${owner} 行那个 ${keyword} 要管的代码。`,
  ],
  [/^expected an indented block/, () => '这一行要往里缩进（按一下 Tab 键）：上一行以冒号结尾，它下面的代码都要缩进。'],
  [/^unexpected indent/, () => '这一行前面多了空格。只有 if、for 这类以冒号结尾的行，它下面的代码才要往里缩进。'],
  [/^unindent does not match any outer indentation level/, () => '这一行前面的空格数和上面对不齐。同一层的代码，前面的空格要一样多，建议统一用 Tab 键缩进。'],
  [/^inconsistent use of tabs and spaces/, () => '这一行的缩进混用了 Tab 键和空格键，统一只用 Tab 键缩进。'],
  [
    /^invalid syntax\. Perhaps you forgot a comma\?/,
    // 抽样一半以上是拼文字漏了 +：print(a+"是"a+"岁")
    () => '这里的几样东西之间少了连接的符号：要分开输出，用英文逗号「,」隔开；要把文字拼在一起，用加号「+」。',
  ],
  [
    /Maybe you meant '==' instead of '='|Maybe you meant '==' or ':=' instead of '='|perhaps you meant "=="/,
    (_, source) => {
      // 只有条件里才是真想判断相等；别的行是想赋值，写法错了
      if (CONDITION_LINE.test(source)) return '判断「是不是相等」要用两个等号「==」。一个等号「=」的意思是「把右边的值存进左边的变量」。'
      if (/^\w+\s*=[^=].*,\s*\w+\s*=[^=]/.test(stripStrings(source))) return '一行里给几个变量赋值，要写成 a, b = 96, 92，或者分成几行，一行写一个。'
      if (/^\w+\s*\(/.test(source)) return '括号里不能直接写等号「=」。要把等号显示出来，就把它放进引号里，比如 print(a, "+", b, "=", a + b)。'
      return null
    },
  ],
  [/^cannot assign to /, () => '等号「=」的左边只能是一个变量名，意思是「把右边算出来的值存进这个变量」。比如要写成 c = a + b，不能写成 a + b = c。'],
  [/^unterminated (triple-quoted )?(f-)?string literal/, () => '这一行的引号没有配对。字符串要用一对英文引号括起来，开头一个、结尾一个。'],
  [/^'(.)' was never closed/, ([, open]) => `这里的「${open}」没有配对的另一半。数一数左右括号是不是一样多。`],
  [/^unmatched '(.)'/, ([, close]) => `这里多了一个「${close}」，前面没有和它配对的括号。`],
  [
    /^closing parenthesis '(.)' does not match opening parenthesis '(.)'/,
    ([, close, open]) => `括号配错了：「${open}」要用和它配对的括号来关，不能用「${close}」。`,
  ],
  [/^Missing parentheses in call to '(\w+)'/, ([, name]) => `${name} 后面要加括号，写成 ${name}(...)。`],
  [/^invalid decimal literal/, () => '数字后面紧跟着字母了。比如「2a」要写成「2*a」；变量名也不能用数字开头。'],
  [/^unexpected character after line continuation character/, () => '这里的反斜杠「\\」用错了。除法要用「/」，「\\」只能出现在引号里面。'],
  [/^'(break|continue)' (not properly in|outside) loop/, ([, keyword]) => `${keyword} 只能写在 for 或 while 循环里面（要缩进在循环的下面）。`],
  [/^'return' outside function/, () => 'return 只能写在 def 定义的函数里面（要缩进在 def 的下面）。'],
  [/^f-string: (expecting '}'|single '}' is not allowed)/, () => 'f 字符串里的花括号「{ }」没有配对，每个「{」都要有一个「}」。'],
  [/^f-string: valid expression required/, () => 'f 字符串的花括号「{ }」里要写变量名或算式，不能空着。'],
  [/^leading zeros in decimal integer literals/, () => '整数不能用 0 开头，比如「07」要写成「7」。'],
  [
    /^invalid syntax/,
    (_, source) => {
      if (/^elif\s*:/.test(source)) return 'elif 后面要写条件，比如 elif a > 60:。不需要条件的话就用 else:。'
      if (/^else\s+if\b/.test(source)) return 'Python 里「否则如果」要写成 elif，不是 else if。'
      if (/^(esle|eles|els|elese|esif|elsif|elseif|eilf|elfi)\b/.test(source)) return '这一行开头的关键字拼错了：「否则」是 else，「否则如果」是 elif。'
      const code = stripStrings(source)
      if (/=<|=>/.test(code)) return '小于等于要写成 <=，大于等于要写成 >=，等号放在后面。'
      if (CONDITION_LINE.test(source) && hasTopLevelComma(source))
        return '条件之间不能用逗号连：两个都要满足用 and，满足一个就行用 or，比如 if a > 9 and a < 100:。'
      return null
    },
  ],
  [/^invalid syntax/, () => '标出来的地方写法不对。对照教程检查这一行：括号、引号、冒号、运算符有没有写错、漏写或多写。'],
]

function translate(message: string, source: string) {
  for (const [pattern, render] of RULES) {
    const match = message.match(pattern)
    const text = match && render(match, source)
    if (text) return text
  }
  return null
}

/**
 * 选了 Python 却交了 C 代码：全库 310 条，报错各式各样（多半是 invalid syntax），
 * 挨个解释毫无意义，该说的只有一句
 */
function looksLikeC(code: string) {
  if (/^\s*#\s*include\b|\bint\s+main\s*\(/m.test(code)) return true
  // 只凭 printf 不算：Python 代码里把 print 手误成 printf 的不少
  return /\b(printf|scanf)\s*\(|\bcout\s*<</.test(code) && !/\b(print|input)\s*\(/.test(code)
}

/**
 * 翻不出来（判题机自己的报错、没见过的句式）就返回 null，界面照旧显示原文。
 * `code` 是学生的代码，用来认出「交的其实是 C」
 */
export function explainPythonCompileError(errInfo: string, code = ''): PythonErrorExplanation | null {
  const text = errInfo.trim()

  const short = text.match(SHORT_RE)
  if (short) {
    const [, , detail, line] = short
    const message = translate(detail!, '')
    if (!message) return null
    return {
      line: Number(line),
      sourceLine: null,
      caret: null,
      message,
      punctuation: null,
    }
  }

  const where = text.match(WHERE_RE)
  const lines = text.split('\n')
  const last = lines[lines.length - 1]!
  const detail = last.replace(/^\w+Error: /, '')
  if (detail === last) return null

  let sourceLine: string | null = null
  let caret: PythonErrorExplanation['caret'] = null
  if (where) {
    const index = lines.findIndex((line) => WHERE_RE.test(line))
    const code = lines[index + 1]
    const marks = lines[index + 2]
    if (code?.startsWith('    ')) sourceLine = code.slice(4)
    if (sourceLine !== null && marks?.startsWith('    ')) {
      const from = marks.indexOf('^')
      if (from >= 4) caret = { from: from - 4, to: marks.lastIndexOf('^') - 4 + 1 }
    }
  }

  const message = looksLikeC(code) ? '这是 C 语言的代码，可提交时选的语言是 Python。把语言换成 C 再交。' : translate(detail, sourceLine?.trim() ?? '')
  if (!message) return null

  const char = detail.match(/^invalid character '(.)'/u)?.[1]
  return {
    line: where ? Number(where[1]) : null,
    sourceLine,
    caret,
    message,
    punctuation: char && PUNCTUATION_MAP[char] ? char : null,
  }
}