Blame view

node_modules/commoner/lib/grep.js 1.39 KB
ce4c83ff   wxy   初始提交
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
  var assert = require("assert");
  var path = require("path");
  var Q = require("q");
  var fs = require("graceful-fs");
  var util = require("./util");
  var readdir = Q.denodeify(fs.readdir);
  var lstat = Q.denodeify(fs.lstat);
  
  function processDirP(pattern, dir) {
      return readdir(dir).then(function(files) {
          return Q.all(files.map(function(file) {
              file = path.join(dir, file);
              return lstat(file).then(function(stat) {
                  return stat.isDirectory()
                      ? processDirP(pattern, file)
                      : processFileP(pattern, file);
              });
          })).then(function(results) {
              return util.flatten(results);
          });
      });
  }
  
  function processFileP(pattern, file) {
      return util.readFileP(file).then(function(contents) {
          var result = new RegExp(pattern, 'g').exec(contents);
          return result ? [{
              path: file,
              match: result[0]
          }] : [];
      });
  }
  
  module.exports = function(pattern, sourceDir) {
      assert.strictEqual(typeof pattern, "string");
  
      return processDirP(pattern, sourceDir).then(function(results) {
          var pathToMatch = {};
  
          results.forEach(function(result) {
              pathToMatch[path.relative(
                  sourceDir,
                  result.path
              ).split("\\").join("/")] = result.match;
          });
          
          return pathToMatch;
      });
  };