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

function run_spec(dirname) {
  fs.readdirSync(dirname).forEach(filename => {
    if (filename.endsWith('.js') && filename !== 'jsfmt.spec.js') {
      const path = dirname + '/' + filename;

      const RUN_AST_TESTS =
        // true || // UNCOMMENT!
        false;

      if (!RUN_AST_TESTS) {
        const source = read(path);
        const output = prettyprint(path);
        test(filename, () => {
          expect(source + '~'.repeat(80) + '\n' + output).toMatchSnapshot();
        });
      }

      if (RUN_AST_TESTS) {
        test(path + ' parse', () => {
          const file = read(dirname + '/' + filename);
          const ast = parse(file);
          const ppfile = pp(file);
          const ppast = parse(ppfile);
          expect(ast).toEqual(ppast);
        });
      }

    }
  });
}
global.run_spec = run_spec;

function stripLocation(ast) {
  if (Array.isArray(ast)) {
    return ast.map(e => stripLocation(e));
  }
  if (typeof ast === 'object') {
    const newObj = {};
    for (var key in ast) {
      if (key === 'loc' || key === 'range' || key === 'raw' || key === 'comments') {
        continue;
      }
      newObj[key] = stripLocation(ast[key]);
    }
    return newObj;
  }
  return ast;
}

function parse(string) {
  const flowParser = require('flow-parser');
  return stripLocation(flowParser.parse(string));
}

function prettyprint(path) {
  const result = child_process.spawnSync(
    './bin/jscodefmt',
    [path],
  );
  return (
    result.stdout.toString() ||
    result.stderr.toString()
      .replace(new RegExp(process.cwd(), 'g'), '')
  );
}

function pp(string) {
  const tmp = 'tmp' + Math.random() + '.js';
  fs.writeFileSync(tmp, string);
  const result = prettyprint(tmp);
  fs.unlinkSync(tmp);
  return result;
}

function read(filename) {
  return fs.readFileSync(filename, 'utf8');
}