bootstrap-npm.js 3.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. #!/usr/bin/env node
  2. /**
  3. * 安装 npm 到 nodejs/node/node_modules(当 node 自带 npm 缺失时使用)
  4. */
  5. const fs = require('fs');
  6. const path = require('path');
  7. const https = require('https');
  8. const zlib = require('zlib');
  9. const nodeDir = path.join(__dirname, 'node');
  10. const nodeModulesDir = path.join(nodeDir, 'node_modules');
  11. const npmDir = path.join(nodeModulesDir, 'npm');
  12. if (fs.existsSync(path.join(npmDir, 'bin', 'npm-cli.js'))) {
  13. console.log('[OK] npm already present in node_modules');
  14. process.exit(0);
  15. }
  16. function get(url) {
  17. return new Promise((resolve, reject) => {
  18. const req = https.get(url, { headers: { 'User-Agent': 'Node/bootstrap-npm' } }, (res) => {
  19. if ([301, 302, 307, 308].includes(res.statusCode) && res.headers.location) {
  20. const loc = res.headers.location;
  21. return get(loc.startsWith('http') ? loc : new URL(loc, url).href).then(resolve).catch(reject);
  22. }
  23. const chunks = [];
  24. res.on('data', (c) => chunks.push(c));
  25. res.on('end', () => resolve(Buffer.concat(chunks)));
  26. res.on('error', reject);
  27. });
  28. req.on('error', reject);
  29. });
  30. }
  31. function extractTar(buffer, outDir) {
  32. let offset = 0;
  33. while (offset + 512 <= buffer.length) {
  34. const header = buffer.slice(offset, offset + 512);
  35. if (header.every(b => b === 0)) break;
  36. const name = header.slice(0, 100).toString('utf8').replace(/\0/g, '');
  37. const size = parseInt(header.slice(124, 136).toString('utf8').trim(), 8) || 0;
  38. const typeflag = (header[156] && header[156] !== 0) ? String.fromCharCode(header[156]) : '0';
  39. offset += 512;
  40. const content = buffer.slice(offset, offset + size);
  41. offset += Math.ceil(size / 512) * 512;
  42. if (!name || name.includes('..')) continue;
  43. const dest = path.join(outDir, name);
  44. if (typeflag === '5' || name.endsWith('/')) {
  45. if (!fs.existsSync(dest)) fs.mkdirSync(dest, { recursive: true });
  46. } else if (size >= 0) {
  47. const dir = path.dirname(dest);
  48. if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
  49. fs.writeFileSync(dest, content);
  50. }
  51. }
  52. }
  53. async function main() {
  54. if (!fs.existsSync(path.join(nodeDir, 'node.exe'))) {
  55. console.error('[X] nodejs/node/node.exe not found');
  56. process.exit(1);
  57. }
  58. console.log('Fetching npm metadata from registry...');
  59. const meta = JSON.parse((await get('https://registry.npmjs.org/npm/latest')).toString());
  60. const version = meta.version;
  61. const tarball = meta.dist?.tarball || `https://registry.npmjs.org/npm/-/npm-${version}.tgz`;
  62. console.log('Downloading npm@' + version + '...');
  63. const tgz = await get(tarball);
  64. if (tgz.length < 1000 && tgz.toString().includes('<!')) {
  65. throw new Error('Registry returned HTML instead of tarball.');
  66. }
  67. if (tgz[0] !== 0x1f || tgz[1] !== 0x8b) {
  68. throw new Error('Downloaded file is not gzip.');
  69. }
  70. if (!fs.existsSync(nodeModulesDir)) fs.mkdirSync(nodeModulesDir, { recursive: true });
  71. const extractDir = path.join(nodeDir, 'npm-extract');
  72. if (fs.existsSync(extractDir)) { try { fs.rmSync(extractDir, { recursive: true }); } catch (_) {} }
  73. fs.mkdirSync(extractDir, { recursive: true });
  74. const tar = zlib.gunzipSync(tgz);
  75. extractTar(tar, extractDir);
  76. const pkgDir = path.join(extractDir, 'package');
  77. if (fs.existsSync(npmDir)) fs.rmSync(npmDir, { recursive: true });
  78. fs.renameSync(fs.existsSync(pkgDir) ? pkgDir : path.join(extractDir, fs.readdirSync(extractDir)[0]), npmDir);
  79. try { fs.rmSync(extractDir, { recursive: true }); } catch (_) {}
  80. console.log('[OK] npm installed to node_modules');
  81. }
  82. main().catch((err) => { console.error(err); process.exit(1); });