[Plugin] support enable/disable.

Signed-off-by: Eric Wang <[email protected]>

Author: skygragon

lib/commands/plugin.js

@@ -1,7 +1,6 @@
 var fs = require('fs');
 var path = require('path');
 
-var _ = require('underscore');
 var sprintf = require('sprintf-js').sprintf;
 
 var h = require('../helper');
@@ -19,9 +18,21 @@ var cmd = {
       describe: 'Install plugin',
       default:  false
     },
-    delete: {
+    enable: {
+      alias:    'e',
+      type:     'boolean',
+      describe: 'Enable plugin',
+      default:  false
+    },
+    disable: {
       alias:    'd',
       type:     'boolean',
+      describe: 'Disable plugin',
+      default:  false
+    },
+    delete: {
+      alias:    'D',
+      type:     'boolean',
       describe: 'Delete plugin',
       default:  false
     }
@@ -36,32 +47,58 @@ cmd.handler = function(argv) {
     if (!name || !fs.existsSync(name))
       return log.error('Plugin not found!');
 
-    var newName = path.resolve(__dirname, '../plugins/' + path.basename(name));
+    var newName = path.join(Plugin.dir, path.basename(name));
     fs.createReadStream(name).pipe(fs.createWriteStream(newName));
     return;
   }
 
-  if (argv.delete) {
-    var f = _.find(h.getDirData(['lib', 'plugins']), function(f) {
-      return f.data !== null && f.data.name === name;
+  var plugins = Plugin.plugins;
+  if (name) {
+    plugins = plugins.filter(function(p) {
+      return p.name === name;
     });
-    if (!f) return log.error('Plugin not found!');
+  }
 
-    fs.unlink(f.fullpath, function(e) {
+  if (!argv.enable && !argv.disable && !argv.delete) {
+    plugins.forEach(function(p) {
+      log.info(sprintf('%s %-20s %-15s %s',
+            h.prettyText('', p.enabled), p.name, p.ver, p.desc));
+    });
+    return;
+  }
+
+  if (plugins.length === 0)
+    return log.error('Plugin not found!');
+  var plugin = plugins[0];
+  var oldname = Plugin.fullpath(plugin.file);
+  var newname;
+
+  if (argv.enable) {
+    if (plugin.file[0] !== '.') return;
+    newname = Plugin.fullpath(plugin.file.substr(1));
+
+    fs.rename(oldname, newname, function(e) {
       if (e) log.error(e.message);
     });
     return;
   }
 
-  var plugins = Plugin.plugins;
-  if (name) {
-    plugins = plugins.filter(function(p) {
-      return p.name === name;
+  if (argv.disable) {
+    if (plugin.file[0] === '.') return;
+    newname = Plugin.fullpath('.' + plugin.file);
+
+    fs.rename(oldname, newname, function(e) {
+      if (e) log.error(e.message);
+    });
+    return;
+  }
+
+  if (argv.delete) {
+    fs.unlink(oldname, function(e) {
+      if (e) log.error(e.message);
     });
+    return;
   }
-  plugins.forEach(function(p) {
-    log.info(sprintf('%-20s %-15s %s', p.name, p.ver, p.desc));
-  });
 };
 
 module.exports = cmd;

lib/helper.js

@@ -140,11 +140,11 @@ h.getDirData = function(paths) {
   paths.unshift(__dirname);
   var dir = path.join.apply(path, paths);
 
-  return _.map(fs.readdirSync(dir), function(filename) {
-    var fullpath = path.join(dir, filename);
-    var ext = path.extname(filename);
+  return _.map(fs.readdirSync(dir), function(file) {
+    var fullpath = path.join(dir, file);
+    var ext = path.extname(file);
 
-    var name = path.basename(filename, ext);
+    var name = path.basename(file, ext);
     var data = null;
 
     switch (ext) {
@@ -157,7 +157,7 @@ h.getDirData = function(paths) {
       default:
         break;
     }
-    return {name: name, data: data, fullpath: fullpath};
+    return {name: name, data: data, file: file};
   });
 };
 

lib/plugin.js

@@ -1,3 +1,5 @@
+var path = require('path');
+
 var _ = require('underscore');
 
 var h = require('./helper');
@@ -9,6 +11,7 @@ function Plugin(id, name, ver, desc) {
   this.name = name;
   this.ver = ver || 'default';
   this.desc = desc;
+  this.enabled = true;
 }
 
 Plugin.prototype.init = function() {
@@ -23,15 +26,25 @@ Plugin.prototype.setNext = function(next) {
 Plugin.plugins = [];
 
 Plugin.init = function(head) {
+  Plugin.dir = path.resolve(__dirname, '../lib/plugins/');
+
   var plugins = [];
   h.getDirData(['lib', 'plugins']).forEach(function(f) {
     var p = f.data;
     if (!p) return;
 
+    p.file = f.file;
+    if (f.name[0] === '.') p.enabled = false;
+
     log.trace('found plugin: ' + p.name + '=' + p.ver);
-    p.init();
+    if (p.enabled) {
+      p.init();
+      log.trace('inited plugin: ' + p.name);
+    } else {
+      log.trace('skipped plugin: ' + p.name);
+    }
+
     plugins.push(p);
-    log.trace('inited plugin: ' + p.name);
   });
 
   // chain the plugins together
@@ -42,10 +55,15 @@ Plugin.init = function(head) {
 
   var last = head;
   plugins.forEach(function(p) {
+    if (!p.enabled) return;
     last.setNext(p);
     last = p;
   });
   Plugin.plugins = plugins;
 };
 
+Plugin.fullpath = function(filename) {
+  return path.join(Plugin.dir, filename);
+};
+
 module.exports = Plugin;