[Plugin] plugin -i to install all missing plugins.

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

Author: skygragon

lib/cli.js

@@ -4,7 +4,6 @@ var _ = require('underscore');
 var chalk = require('./chalk');
 var cache = require('./cache');
 var config = require('./config');
-var core = require('./core');
 var h = require('./helper');
 var icon = require('./icon');
 var log = require('./log');
@@ -63,8 +62,7 @@ cli.run = function() {
   initIcon();
   initLogLevel();
 
-  if (Plugin.init(core))
-    Plugin.save();
+  Plugin.init() && Plugin.save();
 
   process.stdout.on('error', function(e) {
     if (e.code === 'EPIPE') process.exit();

lib/commands/plugin.js

@@ -1,13 +1,12 @@
 'use strict';
-var fs = require('fs');
-
 var sprintf = require('sprintf-js').sprintf;
 
 var h = require('../helper');
 var chalk = require('../chalk');
 var config = require('../config');
 var log = require('../log');
 var Plugin = require('../plugin');
+var Queue = require('../queue');
 var session = require('../session');
 
 const cmd = {
@@ -55,59 +54,86 @@ const cmd = {
       .example(chalk.yellow('leetcode plugin company'), 'Show company plugin')
       .example(chalk.yellow('leetcode plugin company -c'), 'Show config of company plugin')
       .example('', '')
+      .example(chalk.yellow('leetcode plugin -i'), 'Install all missing plugins from GtiHub')
       .example(chalk.yellow('leetcode plugin -i company'), 'Install company plugin from GtiHub')
       .example(chalk.yellow('leetcode plugin -d company'), 'Disable company plugin')
       .example(chalk.yellow('leetcode plugin -e company'), 'Enable comapny plugin')
       .example(chalk.yellow('leetcode plugin -D company'), 'Delete company plugin');
   }
 };
 
-function printPlugins(plugins) {
+function print(plugins) {
   log.info(chalk.gray(sprintf(' %6s  %-18s %-15s %s', 'Active', 'Name', 'Version', 'Desc')));
   log.info(chalk.gray('-'.repeat(100)));
 
   plugins = plugins || Plugin.plugins;
   for (let p of plugins)
     log.printf('   %s     %-18s %-15s %s',
-      h.prettyText('', p.enabled), p.name, p.ver, p.desc);
+      h.prettyText('', p.enabled && !p.missing),
+      p.name, p.ver, p.desc);
   Plugin.save();
 }
 
+function install(plugins) {
+  function doTask(plugin, queue, cb) {
+    Plugin.install(plugin.name, function(e, p) {
+      if (!e) {
+        p.enable(plugin.enabled);
+        p.save();
+        p.help();
+      }
+      return cb(e);
+    });
+  }
+
+  const q = new Queue(plugins, {}, doTask);
+  q.run(1, function(e) {
+    if (e) return log.fail(e);
+    Plugin.init();
+    print();
+  });
+}
+
 cmd.handler = function(argv) {
   session.argv = argv;
 
+  let plugins = Plugin.plugins;
   const name = argv.name;
+
   if (argv.install) {
-    Plugin.install(name, function(e, plugin) {
-      if (e) return log.error(e);
-      plugin.help();
-    });
+    if (name) {
+      install([new Plugin(-1, name, 'missing')]);
+    } else {
+      plugins = plugins.filter(x => x.missing);
+      install(plugins);
+    }
     return;
   }
 
-  let plugins = Plugin.plugins;
-  if (name) {
-    plugins = plugins.filter(x => x.name === name);
-  }
+  if (name) plugins = plugins.filter(x => x.name === name);
   if (plugins.length === 0) return log.error('Plugin not found!');
 
   const plugin = plugins[0];
-  const fullpath = h.getPluginFile(plugin.file);
+  if (plugin.missing && (argv.enable || argv.disable))
+    return log.error('Plugin missing, install it first');
 
   if (argv.enable) {
     plugin.enable(true);
-    printPlugins();
+    plugin.save();
+    print();
   } else if (argv.disable) {
     plugin.enable(false);
-    printPlugins();
+    plugin.save();
+    print();
   } else if (argv.delete) {
-    fs.unlink(fullpath, function(e) {
-      if (e) log.error(e.message);
-    });
+    plugin.delete();
+    plugin.save();
+    Plugin.init();
+    print();
   } else if (argv.config) {
     log.info(JSON.stringify(config.plugins[name] || {}, null, 2));
   } else {
-    printPlugins(plugins);
+    print(plugins);
   }
 };
 

lib/plugin.js

@@ -15,8 +15,11 @@ function Plugin(id, name, ver, desc, deps) {
   this.id = id;
   this.name = name;
   this.ver = ver || 'default';
-  this.desc = desc;
+  this.desc = desc || '';
+
   this.enabled = true;
+  this.deleted = false;
+  this.missing = (ver === 'missing');
 
   // only need deps for current platform
   this.deps = _.chain(deps || [])
@@ -45,10 +48,32 @@ Plugin.prototype.enable = function(enabled) {
   const newfile = enabled ? this.file.substr(1) : '.' + this.file;
   try {
     fs.renameSync(h.getPluginFile(this.file), h.getPluginFile(newfile));
-    this.setFile(newfile);
   } catch(e) {
-    log.error(e);
+    log.error(e.message);
+  }
+  this.setFile(newfile);
+};
+
+Plugin.prototype.delete = function() {
+  if (!this.missing) {
+    try {
+      const fullpath = h.getPluginFile(this.file);
+      fs.unlinkSync(fullpath);
+    } catch(e) {
+      return log.error(e.message);
+    }
   }
+  this.deleted = true;
+};
+
+Plugin.prototype.save = function() {
+  const data = cache.get(h.KEYS.plugins) || {};
+
+  if (this.deleted) delete data[this.name];
+  else if (this.missing) return;
+  else data[this.name] = this.enabled;
+
+  cache.set(h.KEYS.plugins, data);
 };
 
 Plugin.prototype.install = function(cb) {
@@ -68,6 +93,10 @@ Plugin.prototype.help = function() {};
 Plugin.plugins = [];
 
 Plugin.init = function(head) {
+  log.debug('initializing all plugins');
+  head = head || require('./core');
+
+  // 1. check installed plugins
   let plugins = [];
   for (let f of h.getCodeDirData('lib/plugins')) {
     const p = f.data;
@@ -95,8 +124,20 @@ Plugin.init = function(head) {
     last.setNext(p);
     last = p;
   }
-  Plugin.plugins = plugins;
-  return true;
+
+  // 2. check saved plugins
+  const missings = [];
+  const data = cache.get(h.KEYS.plugins) || {};
+  for (let k of _.keys(data)) {
+    if (plugins.find(x => x.name === k)) continue;
+    const p = new Plugin(-1, k, 'missing');
+    p.enabled = data[k];
+    missings.push(p);
+  }
+  log.debug('missing plugins: ' + missings.length);
+
+  Plugin.plugins = plugins.concat(missings);
+  return missings.length === 0;
 };
 
 Plugin.copy = function(src, cb) {
@@ -141,11 +182,7 @@ Plugin.install = function(name, cb) {
 };
 
 Plugin.save = function() {
-  const data = Plugin.plugins
-    .map(x => {
-      return {name: x.name, enabled: x.enabled}
-    });
-  cache.set(h.KEYS.plugins, data);
+  for (let p of this.plugins) p.save();
 };
 
 module.exports = Plugin;