Blame view

node_modules/shelljs/src/uniq.js 2.47 KB
2a09d1a4   liuqimichale   添加宜春 天水 宣化
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
  var common = require('./common');
  var fs = require('fs');
  
  // add c spaces to the left of str
  function lpad(c, str) {
    var res = '' + str;
    if (res.length < c) {
      res = Array((c - res.length) + 1).join(' ') + res;
    }
    return res;
  }
  
  common.register('uniq', _uniq, {
    canReceivePipe: true,
    cmdOptions: {
      'i': 'ignoreCase',
      'c': 'count',
      'd': 'duplicates',
    },
  });
  
  //@
  //@ ### uniq([options,] [input, [output]])
  //@ Available options:
  //@
  //@ + `-i`: Ignore case while comparing
  //@ + `-c`: Prefix lines by the number of occurrences
  //@ + `-d`: Only print duplicate lines, one for each group of identical lines
  //@
  //@ Examples:
  //@
  //@ ```javascript
  //@ uniq('foo.txt');
  //@ uniq('-i', 'foo.txt');
  //@ uniq('-cd', 'foo.txt', 'bar.txt');
  //@ ```
  //@
  //@ Filter adjacent matching lines from input
  function _uniq(options, input, output) {
    // Check if this is coming from a pipe
    var pipe = common.readFromPipe();
  
    if (!pipe) {
      if (!input) common.error('no input given');
  
      if (!fs.existsSync(input)) {
        common.error(input + ': No such file or directory');
      } else if (fs.statSync(input).isDirectory()) {
        common.error("error reading '" + input + "'");
      }
    }
    if (output && fs.existsSync(output) && fs.statSync(output).isDirectory()) {
      common.error(output + ': Is a directory');
    }
  
    var lines = (input ? fs.readFileSync(input, 'utf8') : pipe).
                trimRight().
                split(/\r*\n/);
  
    var compare = function (a, b) {
      return options.ignoreCase ?
             a.toLocaleLowerCase().localeCompare(b.toLocaleLowerCase()) :
             a.localeCompare(b);
    };
    var uniqed = lines.reduceRight(function (res, e) {
      // Perform uniq -c on the input
      if (res.length === 0) {
        return [{ count: 1, ln: e }];
      } else if (compare(res[0].ln, e) === 0) {
        return [{ count: res[0].count + 1, ln: e }].concat(res.slice(1));
      } else {
        return [{ count: 1, ln: e }].concat(res);
      }
    }, []).filter(function (obj) {
                   // Do we want only duplicated objects?
      return options.duplicates ? obj.count > 1 : true;
    }).map(function (obj) {
                   // Are we tracking the counts of each line?
      return (options.count ? (lpad(7, obj.count) + ' ') : '') + obj.ln;
    }).join('\n') + '\n';
  
    if (output) {
      (new common.ShellString(uniqed)).to(output);
      // if uniq writes to output, nothing is passed to the next command in the pipeline (if any)
      return '';
    } else {
      return uniqed;
    }
  }
  
  module.exports = _uniq;