Blame view

node_modules/shelljs/src/sed.js 2.2 KB
aaac7fed   liuqimichale   add
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
  var common = require('./common');
  var fs = require('fs');
  
  common.register('sed', _sed, {
    globStart: 3, // don't glob-expand regexes
    canReceivePipe: true,
    cmdOptions: {
      'i': 'inplace',
    },
  });
  
  //@
  //@ ### sed([options,] search_regex, replacement, file [, file ...])
  //@ ### sed([options,] search_regex, replacement, file_array)
  //@ Available options:
  //@
  //@ + `-i`: Replace contents of 'file' in-place. _Note that no backups will be created!_
  //@
  //@ Examples:
  //@
  //@ ```javascript
  //@ sed('-i', 'PROGRAM_VERSION', 'v0.1.3', 'source.js');
  //@ sed(/.*DELETE_THIS_LINE.*\n/, '', 'source.js');
  //@ ```
  //@
  //@ Reads an input string from `files` and performs a JavaScript `replace()` on the input
  //@ using the given search regex and replacement string or function. Returns the new string after replacement.
  //@
  //@ Note:
  //@
  //@ Like unix `sed`, ShellJS `sed` supports capture groups. Capture groups are specified
  //@ using the `$n` syntax:
  //@
  //@ ```javascript
  //@ sed(/(\w+)\s(\w+)/, '$2, $1', 'file.txt');
  //@ ```
  function _sed(options, regex, replacement, files) {
    // Check if this is coming from a pipe
    var pipe = common.readFromPipe();
  
    if (typeof replacement !== 'string' && typeof replacement !== 'function') {
      if (typeof replacement === 'number') {
        replacement = replacement.toString(); // fallback
      } else {
        common.error('invalid replacement string');
      }
    }
  
    // Convert all search strings to RegExp
    if (typeof regex === 'string') {
      regex = RegExp(regex);
    }
  
    if (!files && !pipe) {
      common.error('no files given');
    }
  
    files = [].slice.call(arguments, 3);
  
    if (pipe) {
      files.unshift('-');
    }
  
    var sed = [];
    files.forEach(function (file) {
      if (!fs.existsSync(file) && file !== '-') {
        common.error('no such file or directory: ' + file, 2, { continue: true });
        return;
      }
  
      var contents = file === '-' ? pipe : fs.readFileSync(file, 'utf8');
      var lines = contents.split(/\r*\n/);
      var result = lines.map(function (line) {
        return line.replace(regex, replacement);
      }).join('\n');
  
      sed.push(result);
  
      if (options.inplace) {
        fs.writeFileSync(file, result, 'utf8');
      }
    });
  
    return sed.join('\n');
  }
  module.exports = _sed;