bare-operating-system/packages/bare-os-seeder/kernel/bin/dhtctl

/* BARE_OS_BIN_API 1.0.0 — bump when staged /bin script semantics change (see developer guide). */
/** Shared helpers for drive-resident /bin scripts (prepended before each command). */
function bareStdin(ctx) {
  return typeof ctx.shellStdin === 'string' ? ctx.shellStdin : ''
}

/** @param {number} mode @param {'file' | 'directory' | 'symlink'} type */
function bareFormatModeString(mode, type) {
  const typeChar = type === 'directory' ? 'd' : type === 'symlink' ? 'l' : '-'
  const perm = mode & 0o777
  const r = (bit) => (perm & bit ? 'r' : '-')
  const w = (bit) => (perm & bit ? 'w' : '-')
  const x = (bit) => (perm & bit ? 'x' : '-')
  return (
    typeChar +
    r(0o400) +
    w(0o200) +
    x(0o100) +
    r(0o040) +
    w(0o020) +
    x(0o010) +
    r(0o004) +
    w(0o002) +
    x(0o001)
  )
}

/** @param {number} mtimeMs @param {number} [nowMs] */
function bareFormatLsMtime(mtimeMs, nowMs) {
  const now = nowMs != null ? nowMs : Date.now()
  const d = new Date(mtimeMs)
  const months = [
    'Jan',
    'Feb',
    'Mar',
    'Apr',
    'May',
    'Jun',
    'Jul',
    'Aug',
    'Sep',
    'Oct',
    'Nov',
    'Dec'
  ]
  const mon = months[d.getMonth()]
  const day = String(d.getDate()).padStart(2, ' ')
  const sixMo = 180 * 24 * 3600 * 1000
  if (Math.abs(now - mtimeMs) > sixMo) {
    const yr = String(d.getFullYear()).padStart(4, ' ')
    return mon + ' ' + day + '  ' + yr
  }
  const hh = String(d.getHours()).padStart(2, '0')
  const mm = String(d.getMinutes()).padStart(2, '0')
  return mon + ' ' + day + ' ' + hh + ':' + mm
}

/** @param {number} size */
function barePosixBlocks(size) {
  return Math.ceil(Number(size) / 512) || 0
}

/**
 * Raw stdout for NUL/binary when **`process.stdout.write`** is missing.
 * If **`ctx.bareOsBinWrite(Uint8Array|string)`** is set (tests / host), use it.
 * @param {Record<string, unknown>} ctx
 * @param {string | Uint8Array} chunk
 * @returns {boolean}
 */
function bareOsEmitRaw(ctx, chunk) {
  if (typeof ctx.bareOsBinWrite === 'function') {
    const b4 = ctx.b4a
    const u8 =
      typeof chunk === 'string'
        ? b4 && typeof b4.from === 'function'
          ? b4.from(chunk)
          : new TextEncoder().encode(chunk)
        : chunk
    ctx.bareOsBinWrite(u8 instanceof Uint8Array ? u8 : new Uint8Array(u8))
    return true
  }
  const w = globalThis.process?.stdout?.write
  if (typeof w === 'function') {
    w.call(globalThis.process.stdout, chunk)
    return true
  }
  return false
}

async function run(ctx, argv) {
  const argv0 = argv[0] || 'dhtctl'
  const parsed = bareP2pParseCommonFlags(argv.slice(1))
  const args = parsed.rest
  const opt = parsed.opt
  if (opt.help || args.length === 0) {
    ctx.console.log(
      bareP2pHelpText(
        argv0,
        'HyperDHT operator utility over /proc and P2P envelopes.',
        argv0 +
          ' status | lookup TOPIC | announce TOPIC | connect PEER_KEY | firewall allow|deny [PEER_KEY] | export-key',
        [
          'status --summary',
          'lookup deadbeef',
          'firewall deny 0123abcd',
          'announce bare-os.swarm.control --yes'
        ],
        ['dhtscan', 'swarmdoctor', 'peerctl']
      )
    )
    if (args.length === 0) ctx.exitCode = 1
    return
  }
  const sub = String(args[0] || '').trim()
  if (sub === 'status') {
    const dht = await bareP2pReadProcJson(ctx, '/proc/bare_os/dht_status.json')
    const scan = await bareP2pReadProcJson(ctx, '/proc/bare_os/dht_scan.json')
    const out = {
      schema: 1,
      health: dht.health || 'unknown',
      firewalled: dht.firewalled ?? null,
      peers: scan.peers ?? null,
      bootstrap: dht.bootstrap || []
    }
    bareP2pPrint(ctx, opt.summary ? { health: out.health, peers: out.peers } : out, opt)
    return
  }
  if (sub === 'export-key') {
    const ident = await bareP2pReadProcJson(ctx, '/proc/bare_os/identity.json')
    const key = ident.dhtPublicKey || ident.devicePublicKey || ''
    if (!key) {
      bareP2pError(
        ctx,
        argv0,
        'no DHT public key found in /proc/bare_os/identity.json',
        argv0 + ' status',
        opt
      )
      return
    }
    bareP2pPrint(ctx, { schema: 1, dhtPublicKey: key }, opt)
    return
  }
  if (
    sub === 'lookup' ||
    sub === 'announce' ||
    sub === 'connect' ||
    sub === 'firewall'
  ) {
    let action = sub
    let payload = {}
    if (sub === 'firewall') {
      const mode = String(args[1] || '').trim().toLowerCase()
      if (mode !== 'allow' && mode !== 'deny') {
        bareP2pError(ctx, argv0, 'firewall requires allow|deny', argv0 + ' firewall allow [PEER_KEY]', opt)
        return
      }
      action = 'firewall.' + mode
      payload = { peerKey: String(args[2] || '').trim() }
    } else {
      payload = { value: String(args[1] || '').trim() }
    }
    if (!opt.yes && (sub === 'announce' || sub === 'connect' || sub === 'firewall')) {
      bareP2pError(ctx, argv0, 'confirmation required (pass --yes)', argv0 + ' ' + sub + ' ... --yes', opt)
      return
    }
    if (opt.dryRun) {
      bareP2pPrint(ctx, { ok: true, dryRun: true, action, payload }, opt)
      return
    }
    const r = bareP2pSend(ctx, 'peerctl', 'dhtctl', {
      requestId: bareP2pId('dht'),
      action,
      payload
    })
    if (r && r.ok === false) {
      bareP2pError(ctx, argv0, String(r.reason || 'send failed'), 'swarmdoctor', opt)
      return
    }
    bareP2pPrint(ctx, { ok: true, action, payload }, opt)
    return
  }
  const sug = bareP2pSuggestSubcommand(sub, [
    'status',
    'lookup',
    'announce',
    'connect',
    'firewall',
    'export-key'
  ])
  bareP2pError(
    ctx,
    argv0,
    'unsupported subcommand' + (sug ? ' (did you mean ' + sug + '?)' : ''),
    argv0 + ' --help',
    opt
  )
}