From 2ccd38916636664bba85bb0781f0001a258b4443 Mon Sep 17 00:00:00 2001
From: blam <ben@blam.sh>
Date: Tue, 16 Apr 2024 09:16:03 +0200
Subject: [PATCH] chore: starting some more on the script

Signed-off-by: blam <ben@blam.sh>
---
 .../cli-node/src/monorepo/PackageGraph.ts     |  1 +
 packages/cli/src/commands/index.ts            |  7 ++++
 .../cli/src/commands/versions/migrate.test.ts | 37 +++++++++++++++----
 packages/cli/src/commands/versions/migrate.ts | 10 ++++-
 4 files changed, 46 insertions(+), 9 deletions(-)

diff --git a/packages/cli-node/src/monorepo/PackageGraph.ts b/packages/cli-node/src/monorepo/PackageGraph.ts
index cd01ec833f..86d54bf328 100644
--- a/packages/cli-node/src/monorepo/PackageGraph.ts
+++ b/packages/cli-node/src/monorepo/PackageGraph.ts
@@ -136,6 +136,7 @@ export class PackageGraph extends Map<string, PackageGraphNode> {
    * Lists all local packages in a monorepo.
    */
   static async listTargetPackages(): Promise<BackstagePackage[]> {
+    console.log(paths.targetDir);
     const { packages } = await getPackages(paths.targetDir);
     return packages as BackstagePackage[];
   }
diff --git a/packages/cli/src/commands/index.ts b/packages/cli/src/commands/index.ts
index 431bd6cc68..a33394b414 100644
--- a/packages/cli/src/commands/index.ts
+++ b/packages/cli/src/commands/index.ts
@@ -395,6 +395,13 @@ export function registerCommands(program: Command) {
     .description('Check Backstage package versioning')
     .action(lazy(() => import('./versions/lint').then(m => m.default)));
 
+  program
+    .command('versions:migrate')
+    .description(
+      'Migrate any plugins that have been moved to the @backstage-community namespace automatically',
+    )
+    .action(lazy(() => import('./versions/migrate').then(m => m.default)));
+
   // TODO(Rugvip): Deprecate in favor of package variant
   program
     .command('clean')
diff --git a/packages/cli/src/commands/versions/migrate.test.ts b/packages/cli/src/commands/versions/migrate.test.ts
index 1aef6ff307..c822b98e9c 100644
--- a/packages/cli/src/commands/versions/migrate.test.ts
+++ b/packages/cli/src/commands/versions/migrate.test.ts
@@ -20,18 +20,21 @@ import {
 import * as run from '../../lib/run';
 import migrate from './migrate';
 import { withLogCollector } from '@backstage/test-utils';
+import { paths } from '../../lib/paths';
 import fs from 'fs-extra';
 
 let mockDir: MockDirectory;
-jest.mock('../../lib/paths', () => ({
-  paths: {
+jest.mock('@backstage/cli-common', () => ({
+  ...jest.requireActual('@backstage/cli-common'),
+  findPaths: () => ({
     resolveTargetRoot(filename: string) {
       return mockDir.resolve(filename);
     },
     get targetDir() {
+      console.log('calling!');
       return mockDir.path;
     },
-  },
+  }),
 }));
 
 jest.mock('../../lib/run', () => {
@@ -49,10 +52,30 @@ jest.mock('../../lib/versioning/packages', () => {
   };
 });
 
+const REGISTRY_VERSIONS: { [name: string]: string } = {
+  '@backstage/core': '1.0.6',
+  '@backstage/core-api': '1.0.7',
+  '@backstage/theme': '2.0.0',
+  '@backstage-extra/custom': '1.1.0',
+  '@backstage-extra/custom-two': '2.0.0',
+  '@backstage/create-app': '1.0.0',
+};
+
 describe('versions:migrate', () => {
   mockDir = createMockDirectory();
 
-  beforeEach(() => {});
+  beforeEach(() => {
+    mockFetchPackageInfo.mockImplementation(async name => ({
+      name: name,
+      'dist-tags': {
+        latest: REGISTRY_VERSIONS[name],
+      },
+    }));
+  });
+
+  afterEach(() => {
+    jest.resetAllMocks();
+  });
 
   it('should bump to the moved version when the package is moved', async () => {
     mockDir.setContent({
@@ -87,10 +110,10 @@ describe('versions:migrate', () => {
     });
 
     jest.spyOn(run, 'run').mockResolvedValue(undefined);
+    console.log(paths);
+    await migrate({});
 
-    const { log: logs } = await withLogCollector(['log'], async () => {
-      await migrate({});
-    });
+    const { log: logs } = await withLogCollector(['log'], async () => {});
     // expectLogsToMatch(logs, [
     //   'Checking for updates of @backstage/core',
     //   'Checking for updates of @backstage/custom',
diff --git a/packages/cli/src/commands/versions/migrate.ts b/packages/cli/src/commands/versions/migrate.ts
index 2e18db456c..557680d5e3 100644
--- a/packages/cli/src/commands/versions/migrate.ts
+++ b/packages/cli/src/commands/versions/migrate.ts
@@ -17,7 +17,13 @@ import { PackageGraph } from '@backstage/cli-node';
 import { OptionValues } from 'commander';
 
 export default async (_: OptionValues) => {
-  console.log(
-    PackageGraph.fromPackages(await PackageGraph.listTargetPackages()),
+  const packageMap = PackageGraph.fromPackages(
+    await PackageGraph.listTargetPackages(),
   );
+
+  const packagesThatHaveMoved = new Map<string, string>();
+
+  for (const [name, pkg] of packageMap.entries()) {
+    console.log(name, pkg);
+  }
 };