Add local config file: ~/.lcconfig

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

Author: skygragon

lib/cache.js

@@ -1,7 +1,9 @@
 var fs = require('fs');
 
+var h = require('./helper');
+
 // try to save cache in user home.
-var BASE_DIR = (process.env.HOME || process.env.USERPROFILE) + '/.lc/';
+var BASE_DIR = h.getHomeDir() + '/.lc/';
 
 function getFullPath(k) {
   if (!fs.existsSync(BASE_DIR))

lib/config.js

@@ -1,4 +1,9 @@
-module.exports = {
+var _ = require('underscore');
+var fs = require('fs');
+
+var h = require('./helper');
+
+var defaultConfig = {
   // usually you don't wanna change those
   BASE_URL:     'https://leetcode.com',
   LOGIN_URL:    'https://leetcode.com/accounts/login/',
@@ -9,5 +14,20 @@ module.exports = {
 
   // but you will want change these
   LANG:      'cpp', // avail: [c,cpp,csharp,golang,java,javascript,python,ruby,swift]
-  USE_COLOR: false
+  USE_COLOR: true
 };
+
+function initConfig() {
+  var config = defaultConfig;
+
+  // check local config: ~/.lcconfig
+  var f = h.getHomeDir() + '/.lcconfig';
+  if (fs.existsSync(f)) {
+    var localConfig = JSON.parse(fs.readFileSync(f));
+    _.extend(defaultConfig, localConfig);
+  }
+
+  return config;
+}
+
+module.exports = initConfig();

lib/helper.js

@@ -70,6 +70,10 @@ h.getFilename = function(fullpath) {
   return path.basename(fullpath, path.extname(fullpath));
 };
 
+h.getHomeDir = function() {
+  return process.env.HOME || process.env.USERPROFILE;
+};
+
 h.readStdin = function() {
   // FIXME: not work for win32
   return fs.readFileSync('/dev/stdin').toString();