// Python 语义的小工具：原仓库是 Python，几处字符串 / 正则行为与 JS 不同，照 Python 的来。

/** len(s)：按 Unicode 码点数，不按 UTF-16 码元。 */
export function cpLen(s: string): number {
  let n = 0;
  for (const _ of s) n++;
  return n;
}

/** s[:n]：按码点切。 */
export function cpSlice(s: string, n: number): string {
  if (s.length <= n) return s; // 码元数不超过 n，码点数更不会超过
  return Array.from(s).slice(0, n).join("");
}

/** str.splitlines()（不保留行尾）。 */
export function splitlines(s: string): string[] {
  if (s === "") return [];
  const parts = s.split(/\r\n|[\n\r\v\f\x1c\x1d\x1e\x85\u2028\u2029]/);
  if (parts.length > 0 && parts[parts.length - 1] === "") parts.pop();
  return parts;
}

/** round(x, n)：结果只用于展示，四舍五入到 n 位即可（不追究 Python 的银行家舍入）。 */
export function pyRound(x: number, n: number): number {
  const f = 10 ** n;
  return Math.round(x * f) / f;
}

// Python 的 \w 是 Unicode 的（字母 L*、数字 N*、下划线，不含组合符 M*）；JS 的 \b 只认 ASCII。把 \b 换成按 Unicode 判的边界。
const W = "[\\p{L}\\p{N}_]";
const UNICODE_BOUNDARY = `(?:(?<=${W})(?!${W})|(?<!${W})(?=${W}))`;

/**
 * 把一条 Python re 写法的正则编成 JS RegExp：
 * - 开头的内联标志 `(?m)` / `(?i)` / `(?s)` 挪成 JS 的 flags；
 * - 字符类以外的 `\b` 换成 Unicode 边界；
 * - 一律带 u 标志（\p{…} 要它）。
 * 本仓库用到的写法都在这个范围里（见 data/behavioral_signatures.ts 与 detectors/identity.ts）。
 */
export function pyRegex(pattern: string, flags = ""): RegExp {
  let p = pattern;
  const fl = new Set(flags.split(""));
  for (;;) {
    const m = /^\(\?([imsx]+)\)/.exec(p);
    if (!m) break;
    for (const c of m[1]!) {
      if (c === "x") throw new Error("verbose regex flag is not supported");
      fl.add(c);
    }
    p = p.slice(m[0].length);
  }
  let out = "";
  let inClass = false;
  for (let i = 0; i < p.length; i++) {
    const c = p[i]!;
    if (c === "\\") {
      const next = p[i + 1] ?? "";
      if (!inClass && next === "b") {
        out += UNICODE_BOUNDARY;
      } else {
        out += c + next;
      }
      i++;
      continue;
    }
    if (c === "[" && !inClass) inClass = true;
    else if (c === "]" && inClass) inClass = false;
    out += c;
  }
  fl.add("u");
  return new RegExp(out, [...fl].join(""));
}

/** re.search(pattern, text, flags)：只要真假。 */
export function pySearch(pattern: string, text: string, flags = ""): boolean {
  return pyRegex(pattern, flags).test(text);
}

/** repr(v)：只给报告里的样例用，形状接近 Python 即可。 */
export function pyRepr(v: unknown): string {
  if (v === null || v === undefined) return "None";
  if (typeof v === "string") {
    const quote = v.includes("'") && !v.includes('"') ? '"' : "'";
    const body = v
      .replaceAll("\\", "\\\\")
      .replaceAll("\n", "\\n")
      .replaceAll("\r", "\\r")
      .replaceAll("\t", "\\t");
    return quote + (quote === "'" ? body.replaceAll("'", "\\'") : body) + quote;
  }
  if (typeof v === "boolean") return v ? "True" : "False";
  if (typeof v === "number") return Number.isInteger(v) ? String(v) : String(v);
  try {
    return JSON.stringify(v);
  } catch {
    return String(v);
  }
}

/** isinstance(v, int) and not isinstance(v, bool)。 */
export function isInt(v: unknown): v is number {
  return typeof v === "number" && Number.isInteger(v);
}

export function asObj(v: unknown): Record<string, unknown> {
  return v && typeof v === "object" && !Array.isArray(v) ? (v as Record<string, unknown>) : {};
}

/** 各个检测器里的 _join_text：把 content 里所有 text 块拼起来。 */
export function joinText(content: unknown): string {
  if (!Array.isArray(content)) return "";
  const parts: string[] = [];
  for (const b of content) {
    if (b && typeof b === "object" && (b as Record<string, unknown>)["type"] === "text") {
      const t = (b as Record<string, unknown>)["text"];
      if (typeof t === "string") parts.push(t);
    }
  }
  return parts.join("");
}

/** statistics.mean / pstdev。 */
export function mean(xs: readonly number[]): number {
  return xs.reduce((a, b) => a + b, 0) / xs.length;
}

export function pstdev(xs: readonly number[]): number {
  const mu = mean(xs);
  return Math.sqrt(xs.reduce((a, x) => a + (x - mu) ** 2, 0) / xs.length);
}