probatio 0.3.0 → 0.5.0

package/dist/mutate/precise-lines.mjs

@@ -1,304 +0,0 @@
-// Inspector line samples shared by the node:test collector, the Mocha collector,
-// and a plain-script child. This file has no test hooks and writes no shard.
-import { existsSync, readdirSync, readFileSync, unlinkSync } from "node:fs"
-import { Session as SyncSession } from "node:inspector"
-import { Session } from "node:inspector/promises"
-import { findSourceMap } from "node:module"
-import path from "node:path"
-import { fileURLToPath } from "node:url"
-
-const B64 = new Int32Array(128)
-for (let i = 0; i < 64; i++) B64["ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/".charCodeAt(i)] = i
-
-function decodeSegments(row) {
-  const segments = []
-  let i = 0
-  while (i < row.length) {
-    const fields = []
-    while (i < row.length && row.charCodeAt(i) !== 44) {
-      let value = 0
-      let shift = 0
-      let digit = 0
-      do {
-        digit = B64[row.charCodeAt(i++)]
-        value |= (digit & 31) << shift
-        shift += 5
-      } while (digit & 32)
-      fields.push(value & 1 ? -(value >> 1) : value >> 1)
-    }
-    if (row.charCodeAt(i) === 44) i++
-    if (fields.length > 0) segments.push(fields)
-  }
-  return segments
-}
-
-/**
- * Offsets to source lines, shared by the async sampler (test processes) and the sync one (children).
- * Paths are relative to `root`. node_modules and paths outside root are left out.
- * @param {(rel: string) => boolean} [skipRel]
- * @param {string} [root]
- */
-function lineConverter(skipRel, root = process.cwd()) {
-  const indexCache = new Map()
-
-  function relPath(url) {
-    if (!url || url.startsWith("node:")) return null
-    let abs
-    try {
-      abs = fileURLToPath(url.split("?")[0].split("#")[0])
-    } catch {
-      return null
-    }
-    const rel = path.relative(root, abs).split(path.sep).join("/")
-    if (!rel || rel.startsWith("..") || path.isAbsolute(rel)) return null
-    if (rel.split("/").includes("node_modules")) return null
-    if (skipRel && skipRel(rel)) return null
-    return rel
-  }
-
-  function indexFor(scriptId, text, sm) {
-    if (indexCache.has(scriptId)) return indexCache.get(scriptId)
-    const mapped = sm && sm.payload && typeof sm.payload.mappings === "string" ? linesFromMappings(text, sm.payload.mappings) : null
-    const lines = mapped && hasLine(mapped) ? mapped : linesFromNewlines(text, sm)
-    indexCache.set(scriptId, lines)
-    return lines
-  }
-
-  function hasLine(lines) {
-    for (let i = 0; i < lines.length; i++) if (lines[i] > 0) return true
-    return false
-  }
-
-  function linesFromNewlines(text, sm) {
-    const lines = new Int32Array(text.length + 1)
-    if (!sm) {
-      let line = 1
-      for (let i = 0; i < text.length; i++) {
-        lines[i] = line
-        if (text.charCodeAt(i) === 10) line++
-      }
-      return lines
-    }
-    let genLine = 0
-    let col = 0
-    for (let i = 0; i < text.length; i++) {
-      const entry = sm.findEntry(genLine, col)
-      if (entry && Number.isInteger(entry.originalLine) && entry.originalLine >= 0) lines[i] = entry.originalLine + 1
-      if (text.charCodeAt(i) === 10) {
-        genLine++
-        col = 0
-      } else {
-        col++
-      }
-    }
-    return lines
-  }
-
-  function linesFromMappings(text, mappings) {
-    const lines = new Int32Array(text.length + 1)
-    const starts = [0]
-    for (let i = 0; i < text.length; i++) if (text.charCodeAt(i) === 10) starts.push(i + 1)
-    let carryLine = 0
-    const rows = mappings.split(";")
-    for (let genLine = 0; genLine < rows.length && genLine < starts.length; genLine++) {
-      const start = starts[genLine]
-      const next = genLine + 1 < starts.length ? starts[genLine + 1] : text.length
-      let col = 0
-      let cursor = start
-      let active = -1
-      for (const fields of decodeSegments(rows[genLine])) {
-        col += fields[0]
-        const at = Math.min(next, start + Math.max(0, col))
-        if (active >= 0) {
-          for (let i = cursor; i < at; i++) lines[i] = active + 1
-        }
-        cursor = at
-        if (fields.length >= 4) {
-          carryLine += fields[2]
-          active = carryLine
-        } else {
-          active = -1
-        }
-      }
-      if (active >= 0) {
-        for (let i = cursor; i < next; i++) lines[i] = active + 1
-      }
-    }
-    return lines
-  }
-
-  function applyRanges(text, script, lines) {
-    const ranges = []
-    for (const fn of script.functions || []) {
-      for (const range of fn.ranges || []) ranges.push(range)
-    }
-    ranges.sort((a, b) => a.startOffset - b.startOffset || b.endOffset - a.endOffset)
-    const offsets = new Int32Array(text.length)
-    for (const range of ranges) {
-      const end = Math.min(text.length, range.endOffset)
-      for (let i = Math.max(0, range.startOffset); i < end; i++) offsets[i] = range.count
-    }
-    const counts = new Map()
-    for (let i = 0; i < text.length; i++) {
-      const line = lines[i]
-      if (line <= 0 || offsets[i] <= 0) continue
-      if (offsets[i] > (counts.get(line) || 0)) counts.set(line, offsets[i])
-    }
-    return counts
-  }
-
-  /** Lines with a positive count in one script, or null when the script is not ours. */
-  function scriptLines(script, text) {
-    const file = relPath(script.url)
-    if (!file || !text) return null
-    const counts = applyRanges(text, script, indexFor(script.scriptId, text, findSourceMap(script.url)))
-    const found = []
-    for (const [line, count] of counts) {
-      if (count > 0) found.push({ file, line })
-    }
-    return found
-  }
-
-  return { relPath, scriptLines }
-}
-
-/**
- * @param {(rel: string) => boolean} [skipRel]
- * @returns {Promise<{ positiveLines: () => Promise<Array<{ file: string, line: number }>>, reset: () => Promise<void> }>}
- */
-export async function openLineSampler(skipRel) {
-  const session = new Session()
-  session.connect()
-  await session.post("Profiler.enable")
-  await session.post("Debugger.enable")
-  await session.post("Profiler.startPreciseCoverage", { callCount: true, detailed: true })
-
-  const textCache = new Map()
-  const convert = lineConverter(skipRel)
-
-  async function scriptText(script) {
-    if (textCache.has(script.scriptId)) return textCache.get(script.scriptId)
-    let text = ""
-    try {
-      const got = await session.post("Debugger.getScriptSource", { scriptId: script.scriptId })
-      text = got.scriptSource || ""
-    } catch {
-      text = ""
-    }
-    textCache.set(script.scriptId, text)
-    return text
-  }
-
-  async function positiveLines() {
-    const cov = await session.post("Profiler.takePreciseCoverage")
-    const found = []
-    for (const script of cov.result || []) {
-      if (!convert.relPath(script.url)) continue
-      const lines = convert.scriptLines(script, await scriptText(script))
-      if (lines) found.push(...lines)
-    }
-    return found
-  }
-
-  return {
-    positiveLines,
-    reset: async () => {
-      await session.post("Profiler.takePreciseCoverage")
-    },
-  }
-}
-
-/**
- * The same sampler with synchronous calls, so an `exit` handler can read it.
- * `exit` runs after process.exit(), where a promise never settles.
- * @param {(rel: string) => boolean} [skipRel]
- * @param {string} [root]
- */
-export function openSyncLineSampler(skipRel, root) {
-  const session = new SyncSession()
-  session.connect()
-  const post = (method, params) => {
-    let result
-    let failure = null
-    let done = false
-    session.post(method, params, (err, value) => {
-      failure = err
-      result = value
-      done = true
-    })
-    if (!done) throw new Error(`inspector ${method} did not answer synchronously`)
-    if (failure) throw failure
-    return result
-  }
-  post("Profiler.enable")
-  post("Debugger.enable")
-  post("Profiler.startPreciseCoverage", { callCount: true, detailed: true })
-  const convert = lineConverter(skipRel, root)
-  return {
-    positiveLines() {
-      const cov = post("Profiler.takePreciseCoverage")
-      const found = []
-      for (const script of cov.result || []) {
-        if (!convert.relPath(script.url)) continue
-        let text = ""
-        try {
-          text = post("Debugger.getScriptSource", { scriptId: script.scriptId }).scriptSource || ""
-        } catch {
-          text = ""
-        }
-        const lines = convert.scriptLines(script, text)
-        if (lines) found.push(...lines)
-      }
-      return found
-    },
-  }
-}
-
-/**
- * Called inside a test process that collects a line map. Every node child it starts
- * loads child-lines.mjs through NODE_OPTIONS, with no import in the child's own code.
- * The marker tells child-lines that this process already has test hooks.
- */
-export function exposeChildLines() {
-  globalThis.__probatioTestHooks = true
-  // One dump dir per test process. Test files can run in parallel, and a child's lines
-  // belong to the test in the process that started it, not to a test in another file.
-  const mapPath = process.env.PROBATIO_COVERAGE_MAP
-  if (mapPath) process.env.PROBATIO_CHILD_DIR = childDir(mapPath)
-  const childLines = new URL("./child-lines.mjs", import.meta.url).href
-  const flag = `--import=${childLines}`
-  const current = process.env.NODE_OPTIONS || ""
-  if (!current.includes(flag)) process.env.NODE_OPTIONS = current ? `${current} ${flag}` : flag
-  if (!process.env.PROBATIO_PACKAGE_ROOT) process.env.PROBATIO_PACKAGE_ROOT = process.cwd()
-}
-
-export function childDir(mapPath) {
-  return path.join(`${mapPath}.children`, String(process.pid))
-}
-
-/** Line dumps the children of the current test left. Each one is read once. */
-export function takeChildHits(mapPath) {
-  const dir = childDir(mapPath)
-  if (!existsSync(dir)) return []
-  const hits = []
-  for (const name of readdirSync(dir)) {
-    if (!name.endsWith(".json")) continue
-    const full = path.join(dir, name)
-    try {
-      const parsed = JSON.parse(readFileSync(full, "utf8"))
-      const list = parsed && Array.isArray(parsed.hits) ? parsed.hits : []
-      for (const hit of list) {
-        if (!hit || typeof hit.file !== "string" || !Number.isInteger(hit.line)) continue
-        hits.push({ file: hit.file, line: hit.line })
-      }
-    } catch {
-      // A torn dump is ignored. The parent does not invent a line.
-    }
-    try {
-      unlinkSync(full)
-    } catch {
-      // The next test must not inherit this dump.
-    }
-  }
-  return hits
-}