batteries/lib/file-ext.js

var fs = require('fs')
  , ArrayExt = require('./array-ext').ArrayExt
  , LineEmitter = require('./line-emitter').LineEmitter
  , ext = require('./ext')
  , constants
  , ENOENT
  ;

try {
    constants = require('constants')
} catch (e) {
    constants = process
}
ENOENT = constants.ENOENT

exports.extend = function(obj) {
    ext.extend(obj || fs, FileExt);
}

var FileExt = exports.FileExt = {

    eachLine: function(f, optionsOrLineFn, endFn) {
        var lineFn, hasLineFn, hasEndFn;
        if (typeof optionsOrLineFn === 'object') {
          lineFn = optionsOrLineFn.line;
          endFn = optionsOrLineFn.end;
        }
        else if (typeof optionsOrLineFn === 'function') {
          lineFn = optionsOrLineFn;
        }
        hasLineFn = typeof lineFn == 'function';
        hasEndFn = typeof endFn == 'function';
        if (!hasLineFn && !hasEndFn) throw new Error('bad arguments');
        var le = new LineEmitter(f);
        if (hasLineFn) le.on('line', lineFn);
        if (hasEndFn) le.on('end', endFn);
    }

  , exists: function(f) {
        try {
            fs.statSync(f)
            return true
        } catch (e) {
            if (e.errno === ENOENT) return false
            throw e
        }
    }

  , grep: function(regex, f, callback) {
        if (!callback) throw new Error('grep requires a callback');
        var results = [];
        FileExt.eachLine(f,
            { line: function(line) { if (line.match(regex)) results.push(line); }
            , end: callback(results)
            });
    }

  , home: function(user, callback) {
        // user is optional so the first param may be a callback
        if (typeof user === 'function') {
            callback = user;
            user = null;
        }
        if (user && callback && user !== process.env['USER']) {
            FileExt.grep(new RegExp('^' + user + ':'), '/etc/passwd', function(line) {
                callback(line && line.split(':')[4]);
            });
        }
        else if (user)
            throw new Error('home requires a callback with user');
        else if (callback)
            callback(process.env['HOME']);
        else
            return process.env['HOME'];
    }

  , readLines: function(f, cb) {
        var lines = [];
        FileExt.eachLine(f, { line: function(line) { lines.push(line); }
                            , end: function() { cb(lines); }
                            });
    }

};

// isDirectory, isFile, isSymbolicLink, etc.
var s = fs.statSync(__dirname);
Object.keys(Object.getPrototypeOf(s)).forEach(function(k) {
    if (k.match(/^is/) && typeof s[k] === 'function') {
        FileExt[k] = function(f, cb) {
            if (cb) {
                fs.stat(f, function(err, stat) {
                    cb(err, err ? null : stat[k]());
                });
            } else {
                return fs.statSync(f)[k]();
            }
        }
    }
});