马宇豪
2024-07-16 f591c27b57e2418c9495bc02ae8cfff84d35bc18
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/usr/bin/env node
 
 
'use strict';
 
/*eslint-disable no-console*/
 
 
var fs       = require('fs');
var argparse = require('argparse');
var yaml     = require('..');
 
 
////////////////////////////////////////////////////////////////////////////////
 
 
var cli = new argparse.ArgumentParser({
  prog:     'js-yaml',
  add_help:  true
});
 
cli.add_argument('-v', '--version', {
  action: 'version',
  version: require('../package.json').version
});
 
cli.add_argument('-c', '--compact', {
  help:   'Display errors in compact mode',
  action: 'store_true'
});
 
// deprecated (not needed after we removed output colors)
// option suppressed, but not completely removed for compatibility
cli.add_argument('-j', '--to-json', {
  help:   argparse.SUPPRESS,
  dest:   'json',
  action: 'store_true'
});
 
cli.add_argument('-t', '--trace', {
  help:   'Show stack trace on error',
  action: 'store_true'
});
 
cli.add_argument('file', {
  help:   'File to read, utf-8 encoded without BOM',
  nargs:  '?',
  default: '-'
});
 
 
////////////////////////////////////////////////////////////////////////////////
 
 
var options = cli.parse_args();
 
 
////////////////////////////////////////////////////////////////////////////////
 
function readFile(filename, encoding, callback) {
  if (options.file === '-') {
    // read from stdin
 
    var chunks = [];
 
    process.stdin.on('data', function (chunk) {
      chunks.push(chunk);
    });
 
    process.stdin.on('end', function () {
      return callback(null, Buffer.concat(chunks).toString(encoding));
    });
  } else {
    fs.readFile(filename, encoding, callback);
  }
}
 
readFile(options.file, 'utf8', function (error, input) {
  var output, isYaml;
 
  if (error) {
    if (error.code === 'ENOENT') {
      console.error('File not found: ' + options.file);
      process.exit(2);
    }
 
    console.error(
      options.trace && error.stack ||
      error.message ||
      String(error));
 
    process.exit(1);
  }
 
  try {
    output = JSON.parse(input);
    isYaml = false;
  } catch (err) {
    if (err instanceof SyntaxError) {
      try {
        output = [];
        yaml.loadAll(input, function (doc) { output.push(doc); }, {});
        isYaml = true;
 
        if (output.length === 0) output = null;
        else if (output.length === 1) output = output[0];
 
      } catch (e) {
        if (options.trace && err.stack) console.error(e.stack);
        else console.error(e.toString(options.compact));
 
        process.exit(1);
      }
    } else {
      console.error(
        options.trace && err.stack ||
        err.message ||
        String(err));
 
      process.exit(1);
    }
  }
 
  if (isYaml) console.log(JSON.stringify(output, null, '  '));
  else console.log(yaml.dump(output));
});