const fs = require('fs');
const path = require('path');

global.chrome = {
  storage: {
    local: {
      get: async (keys) => {
        const store = {};
        if (typeof keys === 'string') keys = [keys];
        for (const key of keys) {
          store[key] = global.__chromeStore?.[key] ?? undefined;
        }
        return store;
      },
      set: async (items) => {
        global.__chromeStore = { ...global.__chromeStore, ...items };
      }
    }
  }
};

global.__chromeStore = {};

const moduleCache = {};

async function loadModule(filePath) {
  const absPath = path.resolve(filePath);
  console.log('loadModule:', absPath);
  if (moduleCache[absPath]) {
    console.log('  cache hit, returning:', Object.keys(moduleCache[absPath]));
    return moduleCache[absPath];
  }

  const content = fs.readFileSync(absPath, 'utf-8');
  const dir = path.dirname(absPath);

  const exportedNames = [];
  let modified = content;

  modified = modified.replace(
    /import\s*\{([^}]+)\}\s*from\s*['"]([^'"]+)['"];?/g,
    (match, imports, source) => {
      const cleanSource = source.replace(/\.js$/, '');
      return `const { ${imports} } = require('${cleanSource}');`;
    }
  );

  modified = modified.replace(/export\s+async\s+function\s+(\w+)/g, (match, name) => {
    exportedNames.push(name);
    return `async function ${name}`;
  });
  modified = modified.replace(/export\s+function\s+(\w+)/g, (match, name) => {
    exportedNames.push(name);
    return `function ${name}`;
  });
  modified = modified.replace(/export\s+const\s+(\w+)/g, (match, name) => {
    exportedNames.push(name);
    return `const ${name}`;
  });

  modified = modified.replace(
    /export\s*\{([^}]+)\};?/g,
    (match, names) => {
      const assignments = names.split(',').map(n => n.trim()).filter(Boolean).map(n => {
        const [orig, alias] = n.split(' as ').map(s => s.trim());
        exportedNames.push(alias || orig);
        return `${alias || orig}: ${orig}`;
      });
      return `module.exports = { ${assignments.join(', ')} };`;
    }
  );

  if (!modified.includes('module.exports') && exportedNames.length > 0) {
    const uniqueExports = [...new Set(exportedNames)];
    const exportsStr = uniqueExports.map(n => `${n}: ${n}`).join(', ');
    modified += `\nmodule.exports = { ${exportsStr} };\n`;
  }

  const customRequire = (id) => {
    console.log('  require called:', id);
    if (id.startsWith('.')) {
      const base = id.endsWith('.js') ? id : id + '.js';
      const resolved = path.resolve(dir, base);
      console.log('  resolved:', resolved);
      const result = loadModule(resolved);
      console.log('  require result keys:', Object.keys(result || {}));
      return result;
    }
    return require(id);
  };

  const moduleObj = { exports: {} };
  const fn = new Function('module', 'exports', 'require', 'console', 'crypto', 'global', modified);
  fn(moduleObj, moduleObj.exports, customRequire, console, globalThis.crypto, global);

  console.log('  exports:', Object.keys(moduleObj.exports));
  moduleCache[absPath] = moduleObj.exports;
  return moduleObj.exports;
}

async function main() {
  const buildDir = '/home/node/.openclaw/workspace/agents/bunbun/builds/agent-police-department-claude-code-agent-governan';

  const hashMod = await loadModule(path.join(buildDir, 'lib/hash.js'));
  console.log('hashMod keys:', Object.keys(hashMod));
  console.log('hashChain type:', typeof hashMod.hashChain);

  const auditMod = await loadModule(path.join(buildDir, 'lib/audit.js'));
  console.log('auditMod keys:', Object.keys(auditMod));

  // Try calling exportAuditLog
  const { addEvent, exportAuditLog } = auditMod;
  await addEvent({ id: 'e1', type: 'tool_use', timestamp: new Date().toISOString() });
  try {
    const result = await exportAuditLog();
    console.log('SUCCESS! integrity:', result.integrity);
  } catch (err) {
    console.log('FAILED:', err.message);
  }
}

main().catch(console.error);