openharness

// The registry the build bakes in as `__DSH_REGISTRY__`: every built-in package folder under
// `store/<agents|viewers>/<name>` as its entry, then `store/registry/<owner>/<name>.json` for packages
// that live elsewhere. Mirrors `storeEntry`, `readStoreDir` and `readRegistryDir` in
// src/dsh/registry.ts (the runtime dev fallback), kept in plain ESM so both build scripts can import
// it without a TypeScript step; src/dsh/store.spec.ts holds the two to the same answer.
import { readdirSync, readFileSync, statSync } from 'fs'
import { join } from 'path'
import { fileURLToPath } from 'url'

const engineSource = readFileSync(new URL('../../src/engines/types.ts', import.meta.url), 'utf8')
const engineList = engineSource.match(/export const ENGINES = \[([\s\S]*?)\] as const/)
if (!engineList) throw new Error('Cannot read the canonical engine list')
const processEngines = [...engineList[1].matchAll(/'([^']+)'/g)].map(match => match[1]).filter(engine => engine !== 'terminal')

const HARNESS_MONOREPO = 'https://github.com/autonomous-ai/openharness'

export function storeEntry(path, manifest, facts) {
  const entry = { id: manifest.id }
  if (manifest.kind !== undefined) entry.kind = manifest.kind
  for (const key of ['name', 'category', 'author', 'description']) if (manifest[key] !== undefined) entry[key] = manifest[key]
  Object.assign(entry, { repo: HARNESS_MONOREPO, ref: 'main', path })
  for (const key of ['homepage', 'upstream', 'license', 'tagline', 'screenshots', 'examples']) if (facts[key] !== undefined) entry[key] = facts[key]
  if (manifest.engine !== undefined) entry.engine = manifest.engine
  if (manifest.engine !== undefined) entry.engines = manifest.kind === 'viewer' || manifest.engine === 'terminal'
    ? [] : [manifest.engine, ...processEngines.filter(engine => engine !== manifest.engine)]
  if (typeof manifest.viewer?.use === 'string') entry.viewerUse = manifest.viewer.use
  entry.tier = manifest.viewer ? 2 : manifest.verdict ? 1 : 0
  entry.verified = true
  return entry
}

function readStoreDir(storeDir, strict) {
  const out = []
  for (const plural of ['agents', 'viewers']) {
    let names
    try { names = readdirSync(join(storeDir, plural)).sort() } catch { continue }
    for (const name of names) {
      const dir = join(storeDir, plural, name)
      if (!statSync(dir).isDirectory() || name.startsWith('.')) continue
      let manifest
      try { manifest = JSON.parse(readFileSync(join(dir, 'harness.json'), 'utf8')) } catch (error) { if (strict) throw error; continue }
      if (strict && (manifest.spec !== 1 || manifest.id !== `autonomous/${name}` || (manifest.kind ?? 'agent') !== (plural === 'agents' ? 'agent' : 'viewer'))) throw new Error(`Invalid package identity: ${dir}`)
      let facts = {}
      try { facts = JSON.parse(readFileSync(join(dir, 'store.json'), 'utf8')) } catch (error) { if (strict) throw error; facts = {} }
      // `"listed": false` unlists a package: its code stays in the repo, checked like any other, and it
      // is left out of the registry and the published catalog. Delete the flag to list it again.
      if (facts.listed === false) continue
      out.push(storeEntry(`store/${plural}/${name}`, manifest, facts))
    }
  }
  return out
}

function readRegistryDir(root) {
  const out = []
  let owners
  try { owners = readdirSync(root) } catch { return out }
  for (const owner of owners) {
    const ownerDir = join(root, owner)
    let files
    try {
      if (!statSync(ownerDir).isDirectory()) continue
      files = readdirSync(ownerDir)
    } catch { continue }
    for (const file of files) {
      if (!file.endsWith('.json')) continue
      // Unlike the runtime's dev reader, which skips it, a malformed entry fails the build: say which.
      const path = join(ownerDir, file)
      try { out.push(JSON.parse(readFileSync(path, 'utf8'))) } catch (error) { throw new Error(`${path}: ${error.message}`) }
    }
  }
  return out
}

/** `storeDir` is the repo's `store/` folder, as a path or a file URL. */
export function readDshRegistry(storeDir, { strict = false } = {}) {
  const root = storeDir instanceof URL ? fileURLToPath(storeDir) : storeDir
  return [...readStoreDir(root, strict), ...readRegistryDir(join(root, 'registry'))]
}