Environment variables (#13)

* Support environment variables

* testcase

* debug ci config

* debug ci config

* debug

* skip testcase of environment

* update appveyor config

Author: gucong3000

.gitignore

@@ -1,12 +1,18 @@
+# Created by https://www.gitignore.io/api/node
+
+### Node ###
 # Logs
 logs
 *.log
 npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
 
 # Runtime data
 pids
 *.pid
 *.seed
+*.pid.lock
 
 # Directory for instrumented libs generated by jscoverage/JSCover
 lib-cov
@@ -20,23 +26,42 @@ coverage
 # Grunt intermediate storage (http://gruntjs.com/creating-plugins#storing-task-files)
 .grunt
 
+# Bower dependency directory (https://bower.io/)
+bower_components
+
 # node-waf configuration
 .lock-wscript
 
 # Compiled binary addons (http://nodejs.org/api/addons.html)
 build/Release
 
 # Dependency directories
-node_modules
-jspm_packages
+node_modules/
+jspm_packages/
+
+# Typescript v1 declaration files
+typings/
 
 # Optional npm cache directory
 .npm
 
+# Optional eslint cache
+.eslintcache
+
 # Optional REPL history
 .node_repl_history
 
-# Proj
-dist
+# Output of 'npm pack'
+*.tgz
+
+# Yarn Integrity file
+.yarn-integrity
+
+# dotenv environment variables file
+.env
+
+
+# End of https://www.gitignore.io/api/node
 
 .npmrc
+dist

.travis.yml

@@ -1,4 +1,5 @@
 language: node_js
+dist: trusty
 node_js:
   - "stable"
   - "4"

appveyor.yml

@@ -1,21 +1,20 @@
-# Test against this version of Node.js
+version: "{build}"
+skip_branch_with_pr: true
+skip_tags: true
+build: off
+
 environment:
   matrix:
-    # node.js
-    - nodejs_version: "stable"
-    - nodejs_version: "4"
+    - nodejs_version: stable
+    - nodejs_version: 4
 
 # Install scripts. (runs after repo cloning)
 install:
-  # Get the latest stable version of Node.js or io.js
+  # install Node.js
   - ps: Install-Product node $env:nodejs_version
   # install modules
   - npm install
 
-# Post-install test scripts.
+# to run your custom scripts instead of automatic tests
 test_script:
-  # run tests
   - npm test
-
-# Don't actually build.
-build: off

lib/cli.js

@@ -2,6 +2,8 @@
 
 'use strict';
 const npmrc = require('./npmrc');
+const env = require('./env');
 const path = require('path');
 
 npmrc(process.argv, path.join(process.cwd(), '.npmrc')).catch(console.error);
+env(process.argv).catch(console.error);

lib/env.js

@@ -0,0 +1,100 @@
+'use strict';
+const path = require('path');
+const os = require('os');
+const exec = require('mz/child_process').execFile;
+const fs = require('fs-extra');
+const config = require('./config');
+
+
+function envForWin(argv) {
+	const MIRRORS = config(argv);
+
+	// 将环境变量转化为数组
+	let reg = Object.keys(MIRRORS).sort().map(key => {
+		if (MIRRORS[key]) {
+			return `"${ key.toUpperCase() }"="${ MIRRORS[key] }"`;
+		}
+	}).filter(Boolean);
+
+	// 环境变量与预期完全一致时，什么都不做
+	if (!reg.length) {
+		return Promise.resolve();
+	}
+
+	// 将数组拼接为windows的reg文件格式
+	reg.unshift(
+		'Windows Registry Editor Version 5.00',
+		'',
+		'[HKEY_LOCAL_MACHINE\\SYSTEM\\CurrentControlSet\\Control\\Session Manager\\Environment]'
+	);
+
+	reg = reg.join('\r\n');
+	// reg文件保存路径
+	const regFilePath = path.join(__dirname, '../dist/env.reg');
+
+	// 写入reg文件
+	return fs.outputFile(regFilePath, reg).then(() => {
+		var regedit = path.join(process.env.windir, 'regedit.exe');
+		var args = ['//S', regFilePath];
+		// 尝试直接使用windows的注册表编辑器导入reg文件
+		return exec(regedit, args).catch(function() {
+			// 权限不足时，尝试在命令行中调用注册表编辑器
+			return exec(regedit, args, {
+				shell: true
+			});
+		}).catch(function(err) {
+			// 权限不足时抛出异常
+			err.code = 'EACCES';
+			throw err;
+		});
+	});
+}
+
+function envInProfile(argv) {
+	const MIRRORS = config(argv);
+
+	// 环境变量信息转换为shell脚本
+	let sh = Object.keys(MIRRORS).sort().map(key => {
+		if (MIRRORS[key]) {
+			return `export ${ key.toUpperCase() }=${ MIRRORS[key] }`;
+		}
+	}).filter(Boolean);
+
+	// 环境变量与预期完全一致时，什么都不做
+	if (!sh.length) {
+		return Promise.resolve();
+	}
+
+	sh.unshift('\n\n# Created by mirror-config-china');
+	sh.push('# End of mirror-config-china\n');
+
+	const home = os.homedir();
+	let shFilePath;
+	[
+		'.bashrc',
+		'.bash_profile',
+		'.zshrc'
+	].some(file => {
+		file = path.join(home, file);
+		if (fs.existsSync(file)) {
+			shFilePath = file;
+			return shFilePath;
+		}
+	});
+
+	return fs.readFile(shFilePath, 'utf8').then(content => {
+		const newContent = content
+			.replace(/\n+# Created by mirror-config-china\n(export\s+.*?\n)*# End of mirror-config-china\n+/, '\n')
+			.trimRight() + sh.join('\n');
+
+		if (newContent !== content) {
+			fs.writeFile(shFilePath, newContent);
+		}
+	});
+}
+
+if (process.platform === 'win32') {
+	module.exports = envForWin;
+} else {
+	module.exports = envInProfile;
+}

package.json

@@ -34,5 +34,5 @@
     "pretest": "eslint lib/**/*.js *.js",
     "test": "mocha"
   },
-  "version": "2.0.1"
+  "version": "2.1.0"
 }

test/index.js

@@ -3,8 +3,16 @@ var describe = require('mocha').describe;
 var it = require('mocha').it;
 var assert = require('assert');
 
-describe('env', function() {
+describe.skip('environment variables', function() {
+	it('electron', function() {
+		assert.ok(process.env.ELECTRON_MIRROR);
+	});
+	it('node', function() {
+		assert.ok(process.env.NODEJS_ORG_MIRROR);
+	});
+});
 
+describe('npm config', function() {
 	it('electron', function() {
 		assert.ok(process.env.npm_config_electron_mirror);
 	});