feat: support authorization

lib/cnpm_config.js

@@ -0,0 +1,37 @@
+'use strict';
+const path = require('path');
+const fs = require('fs');
+const config = {};
+
+function createConfigs() {
+  let root;
+  if (process.platform === 'win32') {
+    root = process.env.USERPROFILE || process.env.APPDATA || process.env.TMP || process.env.TEMP;
+  } else {
+    root = process.env.HOME || process.env.TMPDIR || '/tmp';
+  }
+  let userConfig = path.join(root, '.cnpmrc');
+  if (!fs.existsSync(userConfig)) return;
+  let userConfigContent = fs.readFileSync(userConfig).toString();
+  let configs = typeof userConfigContent === 'string' && userConfigContent.split('\n');
+  configs.reduce((pre, next) => {
+    if (typeof next === 'string') {
+      let map = next.split('=');
+      let key = map[0];
+      let value = map[1];
+      if (value === 'true') value = true;
+      if (value === 'false') value = false;
+      pre[key] = value;
+    }
+    return pre;
+  }, config);
+
+}
+
+createConfigs();
+
+module.exports = {
+  get(key) {
+    return config[key];
+  }
+}

lib/get.js

@@ -6,6 +6,8 @@ const destroy = require('destroy');
 const HttpAgent = require('agentkeepalive');
 const HttpsAgent = require('agentkeepalive').HttpsAgent;
 const utils = require('./utils');
+const cnpmConfig = require('./cnpm_config');
+const urlParser = require('url');
 
 module.exports = get;
 
@@ -41,6 +43,14 @@ function* get(url, options, globalOptions) {
       debug('enableProxy: %s', options.proxy);
     }
   }
+  // need auth
+  const registryUrl = cnpmConfig.get('registry');
+  const registryUri = registryUrl && registryUrl.replace(urlParser.parse(registryUrl).protocol, '') || '';
+  const authed = registryUri && url.indexOf(registryUri) !== -1 || false;
+  if (authed || cnpmConfig.get(registryUri + ':always-auth') || cnpmConfig.get('always-auth')) {
+    const authToken = (`${cnpmConfig.get(registryUri + ':username')}:${Buffer.from(cnpmConfig.get(registryUri + ':_password'), 'base64').toString()}`);
+    options.headers['Authorization'] = `Basic ${Buffer.from(authToken).toString('base64')}`;
+  }
   const retry = options.retry || options.retry === 0 ? options.retry : MAX_RETRY;
   options.retry = undefined;
   debug('GET %s with headers: %j', url, options.headers);
@@ -63,7 +73,14 @@ function* get(url, options, globalOptions) {
 
 function* _get(url, options, retry, globalOptions) {
   try {
-    return yield urllib.request(url, options);
+    const result = yield urllib.request(url, options);
+    if (result.data && Buffer.isBuffer(result.data) && result.data.length > 0 &&
+      options.headers.accept &&
+      options.headers.accept.indexOf('application/vnd.npm.install-v1+json') >= 0) {
+      result.data = JSON.parse(result.data);
+      debug('%s support vnd.npm.install-v1+json format, modified', result.data.name, result.data.modified);
+    }
+    return result;
   } catch (err) {
     retry--;
     if (retry > 0) {