Merge pull request #5844 from BitGo/DX-1201.add-prepareRelease-tests

feat(scripts): add tests for prepare-release script

Author: OttoAllmendinger

scripts/prepare-release.ts

@@ -14,6 +14,8 @@ let lernaModules: string[] = [];
 let lernaModuleLocations: string[] = [];
 let TARGET_SCOPE = '@bitgo-beta';
 let filesChanged = 0;
+// Default to __dirname/.. but allow override via environment variable
+const ROOT_DIR = process.env.BITGO_PREPARE_RELEASE_ROOT_DIR || path.join(__dirname, '..');
 
 async function setLernaModules(): Promise<void> {
   const modules = await getLernaModules();
@@ -23,7 +25,7 @@ async function setLernaModules(): Promise<void> {
 
 function replacePackageScopes() {
   // replace all @bitgo packages & source code with alternate SCOPE
-  const filePaths = [...walk(path.join(__dirname, '../', 'modules')), ...walk(path.join(__dirname, '../', 'webpack'))];
+  const filePaths = [...walk(path.join(ROOT_DIR, 'modules')), ...walk(path.join(ROOT_DIR, 'webpack'))];
   filePaths.forEach((file) => {
     filesChanged += changeScopeInFile(file, lernaModules, TARGET_SCOPE);
   });
@@ -128,6 +130,7 @@ function getArgs() {
     TARGET_SCOPE = split[1] || TARGET_SCOPE;
   }
   console.log(`Preparing to re-target to ${TARGET_SCOPE}`);
+  console.log(`Using root directory: ${ROOT_DIR}`);
 }
 
 async function main(preid?: string) {

scripts/prepareRelease/distTags.ts

@@ -1,4 +1,4 @@
-import { readFileSync } from 'fs';
+import { readFileSync, writeFileSync, existsSync } from 'fs';
 import * as path from 'path';
 
 export type DistTags = Record<string, string>;
@@ -16,8 +16,24 @@ export async function getDistTags(packageName: string): Promise<DistTags> {
   return response.json();
 }
 
+// Add this function to read from cache
+export function getDistTagsCache(): string | undefined {
+  return process.env.BITGO_PREPARE_RELEASE_CACHE_DIST_TAGS;
+}
+
 export async function getDistTagsForModuleNames(moduleNames: string[]): Promise<Map<string, DistTags>> {
-  return new Map(
+  const cachePath = getDistTagsCache();
+
+  // If cache path is set and file exists, read from cache
+  if (cachePath && existsSync(cachePath)) {
+    console.log(`Reading dist tags from cache: ${cachePath}`);
+    const cacheContent = readFileSync(cachePath, { encoding: 'utf-8' });
+    const cachedTags = JSON.parse(cacheContent);
+    return new Map(Object.entries(cachedTags));
+  }
+
+  // Otherwise fetch from npm
+  const tagsMap = new Map(
     (
       await Promise.all(
         moduleNames.map(async (moduleName): Promise<[string, DistTags][]> => {
@@ -38,6 +54,19 @@ export async function getDistTagsForModuleNames(moduleNames: string[]): Promise<
       )
     ).flat()
   );
+
+  // If cache path is set but file doesn't exist, write to cache
+  if (cachePath) {
+    console.log(`Writing dist tags to cache: ${cachePath}`);
+    const cacheDir = path.dirname(cachePath);
+    if (!existsSync(cacheDir)) {
+      const { mkdirSync } = require('fs');
+      mkdirSync(cacheDir, { recursive: true });
+    }
+    writeFileSync(cachePath, JSON.stringify(Object.fromEntries(tagsMap), null, 2), { encoding: 'utf-8' });
+  }
+
+  return tagsMap;
 }
 
 export async function getDistTagsForModuleLocations(moduleLocations: string[]): Promise<(DistTags | undefined)[]> {