tihao-ai/claude-code-tihao-sourcing/scripts/generate-pgy-schema.js

#!/usr/bin/env node
const fs = require('fs');
const path = require('path');
const { SOURCES, DIMENSIONS, CORE_FIELDS, INDEXES, buildDdl, buildQueryMap } = require('../mcp/src/features/tihao-sourcing/pgy-schema');

function main() {
  const args = parseArgs(process.argv.slice(2));
  const outputDir = path.resolve(args.output || path.join(__dirname, '..', 'docs'));
  fs.mkdirSync(outputDir, { recursive: true });

  const files = {
    schema: path.join(outputDir, 'pgy-field-schema.json'),
    ddl: path.join(outputDir, 'pgy-collection-ddl.txt'),
    queryMap: path.join(outputDir, 'pgy-sourcing-query-map.md')
  };

  fs.writeFileSync(files.schema, JSON.stringify({
    collection: 'pgy_blogger',
    generatedAt: new Date().toISOString(),
    sources: SOURCES,
    dimensions: DIMENSIONS,
    coreFields: CORE_FIELDS,
    indexes: INDEXES
  }, null, 2), 'utf8');
  fs.writeFileSync(files.ddl, buildDdl(), 'utf8');
  fs.writeFileSync(files.queryMap, buildQueryMap(), 'utf8');

  console.log(JSON.stringify({
    outputDir,
    coreFieldCount: Object.keys(CORE_FIELDS).length,
    dimensionCount: Object.keys(DIMENSIONS).length,
    indexCount: INDEXES.length,
    files
  }, null, 2));
}

function parseArgs(argv) {
  const args = {};
  for (let index = 0; index < argv.length; index += 1) {
    const raw = argv[index];
    if (!raw.startsWith('--')) continue;
    const key = raw.slice(2).replace(/-([a-z])/g, (_, char) => char.toUpperCase());
    const next = argv[index + 1];
    if (!next || next.startsWith('--')) args[key] = true;
    else {
      args[key] = next;
      index += 1;
    }
  }
  return args;
}

if (require.main === module) {
  try {
    main();
  } catch (error) {
    console.error(error && error.stack ? error.stack : String(error));
    process.exit(1);
  }
}

module.exports = { main };