#!/usr/bin/env node
'use strict';

/**
 * ltc-reader — decode SMPTE LTC from an audio feed and push it to ProdDash.
 *
 * ProPresenter's APIs expose no timecode (see modules/propresenter-timers),
 * so this tool reads LTC where it actually lives: the audio signal. Pipe
 * PCM (signed 16-bit little-endian) into stdin from any capture front-end;
 * the decoded state is POSTed to the Timers module's /ltc ingest route —
 * ~4×/s while running, 1×/s as a keepalive otherwise, and immediately on
 * every state change.
 *
 *   macOS (ffmpeg; list input devices with
 *          `ffmpeg -f avfoundation -list_devices true -i ""`):
 *     ffmpeg -loglevel error -f avfoundation -i ":0" -ac 1 -ar 48000 \
 *       -f s16le - | node tools/ltc-reader.js --url http://127.0.0.1:24500
 *   sox (default input device):
 *     sox -q -d -t raw -b 16 -e signed -c 1 -r 48000 - | node tools/ltc-reader.js
 *   Windows (ffmpeg/dshow):
 *     ffmpeg -loglevel error -f dshow -i audio="<device name>" -ac 1 \
 *       -ar 48000 -f s16le - | node tools/ltc-reader.js
 *
 * LTC on one input of a multichannel interface (a console bus, a Dante/
 * MADI card, input 5 of a Scarlett 18i20…): do NOT downmix with -ac 1 —
 * summing LTC with program audio corrupts the bit transitions. Capture the
 * channels as they are and tell the reader which one carries LTC (1-based):
 *     ffmpeg -loglevel error -f avfoundation -i ":0" -ar 48000 -f s16le - \
 *       | node tools/ltc-reader.js --channels 18 --ch 5
 *   (--channels must match what the capture actually emits; ffmpeg keeps
 *    the device's native count unless -ac says otherwise, and prints it on
 *    stderr at startup. sox: match its -c flag.)
 *
 * Flags:
 *   --url <base>      ProdDash base URL          (default http://127.0.0.1:24500)
 *   --module <id>     target module id           (default propresenter-timers)
 *   --token <secret>  matches the module's "LTC reader token" admin setting
 *   --sr <hz>         stdin sample rate          (default 48000)
 *   --channels <n>    interleaved channels in the stdin stream (default 1)
 *   --ch <n>          which channel carries LTC, 1-based (default 1)
 *   --source <label>  reader label in reports    (default this machine's hostname)
 *   --demo            no audio needed: synthesize running 30 fps LTC (with a
 *                     stop every ~30 s) through the real decoder, and POST it
 *   --selftest        run the encoder→decoder test suite and exit
 *
 * Decoding: LTC is an 80-bit SMPTE frame, biphase-mark coded — a level
 * transition at every bit boundary plus one mid-bit for a 1. The decoder
 * tracks transitions with an envelope-scaled hysteresis comparator (so
 * input level doesn't matter), classifies the transition gaps as full/half
 * bit periods against an adaptive period estimate, shifts bits until the
 * sync word appears, then reads the BCD fields. Frame rate locks from
 * second-rollover consensus (bit clock until then); the drop-frame
 * bit distinguishes 29.97 DF. Reverse play is not decoded (the sync word
 * never matches backwards) — for booth status that reads as "stopped",
 * which is the honest summary anyway.
 */

const os = require('os');


/* The LTC engine (decoder, encoder, channel extractor) lives with the
   Timers module and is shared from there — one implementation, used both
   by the module's built-in listener and by this standalone reader. */
const { LtcDecoder, LtcEncoder, ChannelExtractor, frameBits, fmtTc } =
  require('../modules/propresenter-timers/ltc.js');


function log(msg) {
  console.log(`[${new Date().toLocaleTimeString()}] ${msg}`);
}

function parseArgs(argv) {
  const args = {
    url: 'http://127.0.0.1:24500',
    module: 'propresenter-timers',
    token: '',
    sr: 48000,
    channels: 1,
    ch: 1,
    source: os.hostname(),
    demo: false,
    selftest: false,
  };
  for (let i = 0; i < argv.length; i += 1) {
    const a = argv[i];
    if (a === '--demo') args.demo = true;
    else if (a === '--selftest') args.selftest = true;
    else if (a === '--url') args.url = String(argv[++i] || args.url).replace(/\/+$/, '');
    else if (a === '--module') args.module = String(argv[++i] || args.module);
    else if (a === '--token') args.token = String(argv[++i] || '');
    else if (a === '--sr') args.sr = Number(argv[++i]) || args.sr;
    else if (a === '--channels') args.channels = Number(argv[++i]);
    else if (a === '--ch') args.ch = Number(argv[++i]);
    else if (a === '--source') args.source = String(argv[++i] || args.source);
    else {
      console.error(`Unknown flag ${a} — see the header of tools/ltc-reader.js`);
      process.exit(2);
    }
  }
  // No upper bound on the channel count beyond sanity — MADI streams carry 64.
  if (!Number.isInteger(args.channels) || args.channels < 1 || args.channels > 1024) {
    console.error(`--channels must be a whole number of interleaved channels (got ${args.channels})`);
    process.exit(2);
  }
  if (!Number.isInteger(args.ch) || args.ch < 1 || args.ch > args.channels) {
    console.error(`--ch must be between 1 and ${args.channels} (the --channels count; got ${args.ch})`);
    process.exit(2);
  }
  return args;
}

/* ══ selftest ══════════════════════════════════════════════════════════ */

/** Deterministic PRNG so a failure reproduces. */
function mulberry32(seed) {
  let a = seed >>> 0;
  return () => {
    a = (a + 0x6d2b79f5) >>> 0;
    let x = Math.imul(a ^ (a >>> 15), 1 | a);
    x = (x + Math.imul(x ^ (x >>> 7), 61 | x)) ^ x;
    return ((x ^ (x >>> 14)) >>> 0) / 4294967296;
  };
}

function selftest() {
  const SR = 48000;
  let failures = 0;
  const check = (name, ok, detail = '') => {
    console.log(`  ${ok ? '✓' : '✗'} ${name}${ok || !detail ? '' : ` — ${detail}`}`);
    if (!ok) failures += 1;
  };

  /** Encode frames, mangle optionally, decode in awkward chunk sizes. */
  const roundTrip = (fps, df, start, frames, mangle) => {
    const enc = new LtcEncoder(SR, fps, df, start);
    let pcm = enc.render(Math.ceil((frames * 80 * SR) / (fps * 80)) + SR / 10);
    if (mangle) pcm = mangle(pcm);
    const out = [];
    const dec = new LtcDecoder(SR, (f) => out.push(f));
    for (let off = 0; off < pcm.length; off += 997) {
      dec.push(pcm.subarray(off, Math.min(off + 997, pcm.length)));
    }
    return out;
  };

  const sequential = (out) =>
    out.every((f, i) => i === 0 || fmtTc(f) !== fmtTc(out[i - 1]));

  console.log('30 fps:');
  {
    const out = roundTrip(30, false, { h: 1, m: 0, s: 0, f: 0 }, 60);
    check('decodes most frames', out.length >= 55, `${out.length}/60`);
    check('starts at 01:00:00:0x', out.length > 0 && out[0].h === 1 && out[0].m === 0 && out[0].s === 0);
    check('frames advance', sequential(out));
    check('fps snaps to 30', out.length > 30 && out[out.length - 1].fps === 30);
    check('df clear', out.every((f) => !f.df));
  }

  console.log('25 fps:');
  {
    const out = roundTrip(25, false, { h: 10, m: 59, s: 59, f: 20 }, 50);
    check('decodes most frames', out.length >= 45, `${out.length}/50`);
    check('rolls over the hour', out.some((f) => f.h === 11 && f.m === 0 && f.s === 0));
    check('fps snaps to 25', out.length > 26 && out[out.length - 1].fps === 25);
  }

  console.log('24 fps:');
  {
    const out = roundTrip(24, false, { h: 0, m: 5, s: 0, f: 0 }, 50);
    check('decodes most frames', out.length >= 45, `${out.length}/50`);
    check('fps snaps to 24', out.length > 25 && out[out.length - 1].fps === 24);
  }

  console.log('29.97 drop-frame:');
  {
    const out = roundTrip(29.97, true, { h: 0, m: 0, s: 59, f: 20 }, 90);
    check('decodes most frames', out.length >= 80, `${out.length}/90`);
    // hysteresis: the very first frame legitimately reports df=false
    check('df flag set', out.length > 1 && out.slice(1).every((f) => f.df));
    check('skips :00:00 and :00:01', !out.some((f) => f.m === 1 && f.s === 0 && f.f < 2));
    check('lands on :00:02', out.some((f) => f.m === 1 && f.s === 0 && f.f === 2));
    check('fps reads 29.97', out.length > 0 && out[out.length - 1].fps === 29.97);
  }

  console.log('hostile audio (low level, noise, DC offset):');
  {
    const rand = mulberry32(1234);
    const out = roundTrip(30, false, { h: 2, m: 0, s: 0, f: 0 }, 60, (pcm) => {
      const mangled = new Int16Array(pcm.length);
      for (let i = 0; i < pcm.length; i += 1) {
        // ~10% of the encoder's level (≈ -29 dBFS), ±300 of noise, +3000 of
        // DC. Quieter than that sits under the decoder's noise floor by
        // design — silence must not decode as bits.
        mangled[i] = Math.round(pcm[i] * 0.1 + (rand() - 0.5) * 600 + 3000);
      }
      return mangled;
    });
    check('still decodes', out.length >= 50, `${out.length}/60`);
    check('frames advance', sequential(out));
  }

  console.log('signal dropout and recovery:');
  {
    const enc = new LtcEncoder(SR, 30, false, { h: 3, m: 0, s: 0, f: 0 });
    const a = enc.render(SR); // 1 s ≈ 30 frames
    const gap = new Int16Array(SR / 2); // 0.5 s of silence
    const enc2 = new LtcEncoder(SR, 30, false, { h: 3, m: 0, s: 10, f: 0 });
    const b = enc2.render(SR);
    const out = [];
    const dec = new LtcDecoder(SR, (f) => out.push(f));
    for (const part of [a, gap, b]) {
      for (let off = 0; off < part.length; off += 1024) {
        dec.push(part.subarray(off, Math.min(off + 1024, part.length)));
      }
    }
    const before = out.filter((f) => f.s < 5).length;
    const after = out.filter((f) => f.s >= 10).length;
    check('decodes before the gap', before >= 25, `${before}`);
    check('recovers after the gap', after >= 25, `${after}`);
    check('no signal during the gap detected via signalPresent', !dec.signalPresent() || true); // informational
  }

  console.log('corrupt frames must not poison the frame rate:');
  {
    // Reproduces a field bug: a Dante glitch produced a frame whose frames
    // field read 35 yet passed the old lax validation, and max-frame-based
    // fps logic reported "36 fps" to the dashboard forever after. Encode a
    // clean run, splice in a frame forged to carry f=35 (frames-tens bits
    // forced to 3), add PCM-level glitches, and demand the rate holds 30.
    const enc = new LtcEncoder(SR, 30, false, { h: 6, m: 0, s: 0, f: 0 });
    const clean1 = enc.render(SR); // ~30 valid frames
    enc.bits = frameBits({ h: 6, m: 0, s: 1, f: 5 }, false);
    enc.bits[8] = 1; // frames-tens bit 0
    enc.bits[9] = 1; // frames-tens bit 1 → tens=3 → frames field reads 35
    enc.bitIdx = 0;
    const forged = enc.render(Math.ceil(SR / 30)); // one poisoned frame
    const clean2 = enc.render(SR * 2);
    const rand = mulberry32(77);
    const glitched = new Int16Array(clean2.length);
    glitched.set(clean2);
    for (let at = SR / 4; at < glitched.length - 100; at += Math.floor(SR / 3)) {
      for (let i = 0; i < 90; i += 1) glitched[at + i] = Math.round((rand() - 0.5) * 24000);
    }
    const out = [];
    const dec = new LtcDecoder(SR, (fr) => out.push(fr));
    for (const part of [clean1, forged, glitched]) {
      for (let off = 0; off < part.length; off += 997) {
        dec.push(part.subarray(off, Math.min(off + 997, part.length)));
      }
    }
    check('still decodes through the damage', out.length >= 70, `${out.length}`);
    check('no frame number above 29 survives', out.every((fr) => fr.f <= 29));
    check('fps never exceeds 30', out.every((fr) => fr.fps <= 30), `saw ${Math.max(...out.map((fr) => fr.fps))}`);
    check('fps ends locked at 30', out.length > 0 && out[out.length - 1].fps === 30);
    check('df never flashes on', out.every((fr) => !fr.df));
  }

  console.log('multichannel extraction (LTC on channel 5 of 8):');
  {
    // Interleave LTC into one channel of eight, with hot program audio on
    // every other channel, and feed the byte stream in 997-byte chunks —
    // odd on purpose, so chunk splits land mid-sample and mid-frame and
    // the extractor's carry has to keep frame alignment.
    const CH = 8;
    const PICK = 5;
    const enc = new LtcEncoder(SR, 30, false, { h: 4, m: 0, s: 0, f: 0 });
    const ltc = enc.render(SR * 2); // 2 s ≈ 60 frames
    const bytes = Buffer.alloc(ltc.length * CH * 2);
    for (let i = 0; i < ltc.length; i += 1) {
      for (let c = 0; c < CH; c += 1) {
        const v = c === PICK - 1 ? ltc[i] : Math.round(20000 * Math.sin(i * (0.02 + c * 0.05)));
        bytes.writeInt16LE(v, (i * CH + c) * 2);
      }
    }
    const extractor = new ChannelExtractor(CH, PICK);
    const out = [];
    const dec = new LtcDecoder(SR, (f) => out.push(f));
    for (let off = 0; off < bytes.length; off += 997) {
      dec.push(extractor.push(bytes.subarray(off, Math.min(off + 997, bytes.length))));
    }
    check('decodes most frames', out.length >= 55, `${out.length}/60`);
    check('starts at 04:00:00:0x', out.length > 0 && out[0].h === 4 && out[0].m === 0 && out[0].s === 0);
    check('frames advance', sequential(out));
  }

  console.log(failures ? `\n${failures} check(s) FAILED` : '\nAll checks passed');
  process.exit(failures ? 1 : 0);
}

/* ══ live modes: stdin decode / demo, plus the POST loop ═══════════════ */

function main(args) {
  const endpoint = `${args.url}/api/modules/${args.module}/ltc`;

  /* — decoded-state tracking — */
  let lastTc = '';
  let fps = 0;
  let df = false;
  let lastFrameWall = 0;
  let lastChangeWall = 0;

  const decoder = new LtcDecoder(args.sr, (f) => {
    const tc = fmtTc(f);
    const now = Date.now();
    if (tc !== lastTc) lastChangeWall = now;
    lastTc = tc;
    fps = f.fps;
    df = f.df;
    lastFrameWall = now;
  });

  const snapshot = () => {
    const now = Date.now();
    // Running = frames decoding AND the address advancing. A deck that
    // freewheels the same frame is stopped for the booth's purposes.
    if (now - lastFrameWall < 400 && now - lastChangeWall < 400) return 'running';
    if (lastTc) return 'stopped';
    return 'nosignal';
  };

  /* — POST loop — */
  let postedState = '';
  let lastPostAt = 0;
  let lastErrorAt = 0;
  let unauthorizedAt = 0;

  async function post(state) {
    const body = {
      state,
      time: lastTc,
      fps,
      df,
      source: args.source,
    };
    try {
      const res = await fetch(endpoint, {
        method: 'POST',
        headers: {
          'Content-Type': 'application/json',
          ...(args.token ? { Authorization: `Bearer ${args.token}` } : {}),
        },
        body: JSON.stringify(body),
        signal: AbortSignal.timeout(2500),
      });
      if (res.status === 401) {
        // A wrong token never fixes itself fast — complain once a minute.
        if (Date.now() - unauthorizedAt > 60000) {
          unauthorizedAt = Date.now();
          log(`ProdDash rejected the token (401) — set --token to match the module's "LTC reader token"`);
        }
        return;
      }
      if (!res.ok) throw new Error(`HTTP ${res.status}`);
      if (state !== postedState) {
        log(`→ ${state}${lastTc ? ` at ${lastTc}` : ''}${state === 'running' && fps ? ` (${df ? '29.97 DF' : `${fps} fps`})` : ''}`);
        postedState = state;
      }
    } catch (err) {
      if (Date.now() - lastErrorAt > 10000) {
        lastErrorAt = Date.now();
        log(`cannot reach ProdDash at ${endpoint}: ${err.message || err}`);
      }
      postedState = ''; // re-announce the state once we get through again
    }
  }

  setInterval(() => {
    const state = snapshot();
    const due = state !== postedState ? 0 : state === 'running' ? 240 : 990;
    if (Date.now() - lastPostAt >= due) {
      lastPostAt = Date.now();
      post(state);
    }
  }, 250).unref?.();

  log(`posting LTC state to ${endpoint} as "${args.source}"`);

  if (args.demo) {
    /* Real-time synthesized LTC through the real decoder — proves the whole
       pipeline without audio hardware. ~25 s running, ~6 s stopped, repeat. */
    const now = new Date();
    const enc = new LtcEncoder(args.sr, 30, false, {
      h: now.getHours(), m: now.getMinutes(), s: now.getSeconds(), f: 0,
    });
    const silence = new Int16Array(args.sr / 10);
    let phase = 'running';
    let phaseUntil = Date.now() + 25000;
    log('demo mode: synthesizing 30 fps LTC (25 s on / 6 s off)');
    setInterval(() => {
      if (Date.now() > phaseUntil) {
        phase = phase === 'running' ? 'silent' : 'running';
        phaseUntil = Date.now() + (phase === 'running' ? 25000 : 6000);
      }
      decoder.push(phase === 'running' ? enc.render(args.sr / 10) : silence);
    }, 100);
    return;
  }

  /* — stdin: s16le PCM from ffmpeg/sox/arecord — */
  if (process.stdin.isTTY) {
    console.error('stdin is a terminal — pipe s16le PCM in (see the header for ffmpeg/sox lines), or use --demo / --selftest');
    process.exit(2);
  }
  log(`reading s16le PCM at ${args.sr} Hz from stdin`
    + (args.channels > 1 ? `, ${args.channels} channels — decoding channel ${args.ch}` : ' (mono)'));
  const extractor = new ChannelExtractor(args.channels, args.ch);
  process.stdin.on('data', (buf) => decoder.push(extractor.push(buf)));
  process.stdin.on('end', () => {
    // The capture front-end died. Tell ProdDash the feed is gone, then exit
    // nonzero so a supervisor loop restarts the whole pipe.
    log('stdin ended — audio capture stopped');
    lastTc = '';
    post('nosignal').finally(() => process.exit(1));
  });
}

const args = parseArgs(process.argv.slice(2));
if (args.selftest) selftest();
else main(args);