import type { Market } from "@/src/domain/types";

export type ParsedStockRef = { code: string; market: Market };

export function parseStockRef(raw: string): ParsedStockRef | null {
  const s = raw.trim();
  if (!s) return null;

  // allow forms:
  // 600519
  // SH:600519
  // SZ:000001
  // BJ:xxxxxx
  const m = s.match(/^(?:(SH|SZ|BJ)[:：])?(\d{6})$/i);
  if (!m) return null;

  const market = (m[1]?.toUpperCase() as Market | undefined) ?? guessMarket(m[2]!);
  return { market, code: m[2]! };
}

function guessMarket(code: string): Market {
  if (code.startsWith("6")) return "SH";
  if (code.startsWith("8") || code.startsWith("9")) return "BJ";
  return "SZ";
}

export function normalizeImportedPool(text: string): ParsedStockRef[] {
  const lines = text
    .split(/\r?\n/)
    .map((l) => l.trim())
    .filter(Boolean);

  const out: ParsedStockRef[] = [];
  const seen = new Set<string>();

  for (const line of lines) {
    const ref = parseStockRef(line);
    if (!ref) continue;
    const key = `${ref.market}:${ref.code}`;
    if (seen.has(key)) continue;
    seen.add(key);
    out.push(ref);
  }

  return out;
}