From e053c105f4a27fa29b76aea11a2a209c38664f49 Mon Sep 17 00:00:00 2001
From: Alper Altay <alper.altay@lego.com>
Date: Tue, 25 Jun 2024 12:38:10 +0200
Subject: [PATCH] chore: breaking out into function

Signed-off-by: Alper Altay <alper.altay@lego.com>
---
 .../src/lib/new/factories/backendModule.ts    | 28 +++-------------
 .../src/lib/new/factories/backendPlugin.ts    | 28 +++-------------
 packages/cli/src/lib/tasks.ts                 | 32 +++++++++++++++++++
 3 files changed, 40 insertions(+), 48 deletions(-)

diff --git a/packages/cli/src/lib/new/factories/backendModule.ts b/packages/cli/src/lib/new/factories/backendModule.ts
index e0c56b34ae..8cfa7f1e00 100644
--- a/packages/cli/src/lib/new/factories/backendModule.ts
+++ b/packages/cli/src/lib/new/factories/backendModule.ts
@@ -20,7 +20,7 @@ import camelCase from 'lodash/camelCase';
 import { paths } from '../../paths';
 import { addCodeownersEntry, getCodeownersFilePath } from '../../codeowners';
 import { CreateContext, createFactory } from '../types';
-import { addPackageDependency, Task } from '../../tasks';
+import { addPackageDependency, addToBackend, Task } from '../../tasks';
 import {
   moduleIdIdPrompt,
   ownerPrompt,
@@ -91,29 +91,9 @@ export const backendModule = createFactory<Options>({
       });
     }
 
-    await Task.forItem('backend', 'adding module', async () => {
-      const backendFilePath = paths.resolveTargetRoot(
-        'packages/backend/src/index.ts',
-      );
-      if (!(await fs.pathExists(backendFilePath))) {
-        return;
-      }
-
-      const content = await fs.readFile(backendFilePath, 'utf8');
-      const lines = content.split('\n');
-      const backendAddLine = `backend.add(import('${name}'));`;
-
-      const backendStartIndex = lines.findIndex(line =>
-        line.match(/backend.start/),
-      );
-
-      if (backendStartIndex !== -1) {
-        const [indentation] = lines[backendStartIndex].match(/^\s*/) ?? [];
-        lines.splice(backendStartIndex, 0, indentation + backendAddLine);
-
-        const newContent = lines.join('\n');
-        await fs.writeFile(backendFilePath, newContent, 'utf8');
-      }
+    await addToBackend(name, {
+      defaultVersion: ctx.defaultVersion,
+      type: 'module',
     });
 
     if (options.owner) {
diff --git a/packages/cli/src/lib/new/factories/backendPlugin.ts b/packages/cli/src/lib/new/factories/backendPlugin.ts
index c5cd043132..d7bf33b033 100644
--- a/packages/cli/src/lib/new/factories/backendPlugin.ts
+++ b/packages/cli/src/lib/new/factories/backendPlugin.ts
@@ -20,7 +20,7 @@ import camelCase from 'lodash/camelCase';
 import { paths } from '../../paths';
 import { addCodeownersEntry, getCodeownersFilePath } from '../../codeowners';
 import { CreateContext, createFactory } from '../types';
-import { addPackageDependency, Task } from '../../tasks';
+import { addPackageDependency, addToBackend, Task } from '../../tasks';
 import { ownerPrompt, pluginIdPrompt } from './common/prompts';
 import { executePluginPackageTemplate } from './common/tasks';
 import { resolvePackageName } from './common/util';
@@ -79,29 +79,9 @@ export const backendPlugin = createFactory<Options>({
         );
       });
 
-      await Task.forItem('backend', 'adding plugin', async () => {
-        const backendFilePath = paths.resolveTargetRoot(
-          'packages/backend/src/index.ts',
-        );
-        if (!(await fs.pathExists(backendFilePath))) {
-          return;
-        }
-
-        const content = await fs.readFile(backendFilePath, 'utf8');
-        const lines = content.split('\n');
-        const backendAddLine = `backend.add(import('${name}'));`;
-
-        const backendStartIndex = lines.findIndex(line =>
-          line.match(/backend.start/),
-        );
-
-        if (backendStartIndex !== -1) {
-          const [indentation] = lines[backendStartIndex].match(/^\s*/)!;
-          lines.splice(backendStartIndex, 0, `${indentation}${backendAddLine}`);
-
-          const newContent = lines.join('\n');
-          await fs.writeFile(backendFilePath, newContent, 'utf8');
-        }
+      await addToBackend(name, {
+        defaultVersion: ctx.defaultVersion,
+        type: 'plugin',
       });
     }
 
diff --git a/packages/cli/src/lib/tasks.ts b/packages/cli/src/lib/tasks.ts
index 851f962c62..84362d7856 100644
--- a/packages/cli/src/lib/tasks.ts
+++ b/packages/cli/src/lib/tasks.ts
@@ -196,3 +196,35 @@ export async function addPackageDependency(
     throw new Error(`Failed to add package dependencies, ${error}`);
   }
 }
+
+export async function addToBackend(
+  name: string,
+  ctx: { defaultVersion: string; type: 'plugin' | 'module' },
+) {
+  if (await fs.pathExists(paths.resolveTargetRoot('packages/backend'))) {
+    await Task.forItem('backend', `adding ${ctx.type}`, async () => {
+      const backendFilePath = paths.resolveTargetRoot(
+        'packages/backend/src/index.ts',
+      );
+      if (!(await fs.pathExists(backendFilePath))) {
+        return;
+      }
+
+      const content = await fs.readFile(backendFilePath, 'utf8');
+      const lines = content.split('\n');
+      const backendAddLine = `backend.add(import('${name}'));`;
+
+      const backendStartIndex = lines.findIndex(line =>
+        line.match(/backend.start/),
+      );
+
+      if (backendStartIndex !== -1) {
+        const [indentation] = lines[backendStartIndex].match(/^\s*/)!;
+        lines.splice(backendStartIndex, 0, `${indentation}${backendAddLine}`);
+
+        const newContent = lines.join('\n');
+        await fs.writeFile(backendFilePath, newContent, 'utf8');
+      }
+    });
+  }
+}