Add support for custom npm registry url

chromaui · Jan 31, 2022 · e1dca0e · e1dca0e
1 parent c9e6479
commit e1dca0e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/bin-src/lib/checkForUpdates.ts b/bin-src/lib/checkForUpdates.ts
@@ -1,3 +1,4 @@
+import execa from 'execa';
 import semver from 'semver';
 import yon from 'yarn-or-npm';
 import { Context } from '../types';
@@ -18,7 +19,10 @@ export default async function checkForUpdates(ctx: Context) {
 
   let latestVersion: string;
   try {
-    const pkgUrl = `https://registry.npmjs.org/${ctx.pkg.name}`;
+    const registryUrl = await execa('npm', ['config', 'get', 'registry'])
+      .then(({ stdout }) => stdout.trim())
+      .catch(() => 'https://registry.npmjs.org/');
+    const pkgUrl = `${registryUrl}${ctx.pkg.name}`;
     // If not fetched within 5 seconds, nevermind.
     const res = await withTimeout(ctx.http.fetch(pkgUrl), 5000);
     const { 'dist-tags': distTags = {} } = (await res.json()) as any;

diff --git a/bin-src/main.test.ts b/bin-src/main.test.ts
@@ -612,9 +612,10 @@ describe('runAll', () => {
     ctx.http.fetch.mockReturnValueOnce({
       json: () => Promise.resolve({ 'dist-tags': { latest: '5.0.0' } }),
     } as any);
+    execa.mockResolvedValue({ stdout: 'https://registry.example.org/' } as any);
     await runAll(ctx);
     expect(ctx.exitCode).toBe(1);
-    expect(ctx.http.fetch).toHaveBeenCalledWith('https://registry.npmjs.org/chromatic');
+    expect(ctx.http.fetch).toHaveBeenCalledWith('https://registry.example.org/chromatic');
     expect(ctx.testLogger.warnings[0]).toMatch('Using outdated package');
   });