acorn.js 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. import {basename} from "path"
  2. import {readFileSync as readFile} from "fs"
  3. import * as acorn from "acorn"
  4. let infile, forceFile, silent = false, compact = false, tokenize = false
  5. const options = {}
  6. function help(status) {
  7. const print = (status == 0) ? console.log : console.error
  8. print("usage: " + basename(process.argv[1]) + " [--ecma3|--ecma5|--ecma6|--ecma7|...|--ecma2015|--ecma2016|...]")
  9. print(" [--tokenize] [--locations] [---allow-hash-bang] [--compact] [--silent] [--module] [--help] [--] [infile]")
  10. process.exit(status)
  11. }
  12. for (let i = 2; i < process.argv.length; ++i) {
  13. const arg = process.argv[i]
  14. if ((arg == "-" || arg[0] != "-") && !infile) infile = arg
  15. else if (arg == "--" && !infile && i + 2 == process.argv.length) forceFile = infile = process.argv[++i]
  16. else if (arg == "--locations") options.locations = true
  17. else if (arg == "--allow-hash-bang") options.allowHashBang = true
  18. else if (arg == "--silent") silent = true
  19. else if (arg == "--compact") compact = true
  20. else if (arg == "--help") help(0)
  21. else if (arg == "--tokenize") tokenize = true
  22. else if (arg == "--module") options.sourceType = 'module'
  23. else {
  24. let match = arg.match(/^--ecma(\d+)$/)
  25. if (match)
  26. options.ecmaVersion = +match[1]
  27. else
  28. help(1)
  29. }
  30. }
  31. function run(code) {
  32. let result
  33. if (!tokenize) {
  34. try { result = acorn.parse(code, options) }
  35. catch(e) { console.error(e.message); process.exit(1) }
  36. } else {
  37. result = []
  38. let tokenizer = acorn.tokenizer(code, options), token
  39. while (true) {
  40. try { token = tokenizer.getToken() }
  41. catch(e) { console.error(e.message); process.exit(1) }
  42. result.push(token)
  43. if (token.type == acorn.tokTypes.eof) break
  44. }
  45. }
  46. if (!silent) console.log(JSON.stringify(result, null, compact ? null : 2))
  47. }
  48. if (forceFile || infile && infile != "-") {
  49. run(readFile(infile, "utf8"))
  50. } else {
  51. let code = ""
  52. process.stdin.resume()
  53. process.stdin.on("data", chunk => code += chunk)
  54. process.stdin.on("end", () => run(code))
  55. }