From a751336b84510fb3105d66463256972841e31ad2 Mon Sep 17 00:00:00 2001
From: p-sw <shinwoo.park@psw.kr>
Date: Fri, 7 Jun 2024 00:11:25 +0900
Subject: [PATCH] feat(cli): add 'add' command

This commit introduces a new command 'add' to the CLI. The 'add' command allows users to add a component to the project with options for custom configurations, and it performs validation checks to ensure the component exists in the registry before installing.
---
 packages/cli/src/commands/add.ts | 94 ++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 packages/cli/src/commands/add.ts

diff --git a/packages/cli/src/commands/add.ts b/packages/cli/src/commands/add.ts
new file mode 100644
index 0000000..3fe0dfc
--- /dev/null
+++ b/packages/cli/src/commands/add.ts
@@ -0,0 +1,94 @@
+import {Args, Command, Flags} from '@oclif/core'
+import {loadConfig, validateConfig} from '../helpers/config.js'
+import {existsSync} from 'node:fs'
+import {mkdir, writeFile} from 'node:fs/promises'
+import {join} from 'node:path'
+import {getAvailableComponentNames, getComponentURL, getRegistry} from '../helpers/registry.js'
+import ora from 'ora'
+
+export default class Add extends Command {
+  static override args = {
+    name: Args.string({description: 'name of component to install'}),
+  }
+
+  static override description = 'Add a component to the project.'
+
+  static override examples = ['<%= config.bin %> <%= command.id %>']
+
+  static override flags = {
+    force: Flags.boolean({char: 'f', description: 'override the existing file'}),
+    // WARNING: forceShared could break your components!
+    forceShared: Flags.boolean({char: 'F', description: 'override the existing shared.ts and update it to latest'}),
+    config: Flags.string({char: 'p', description: 'path to config'}),
+    shared: Flags.string({char: 's', description: 'place for installation of shared.ts'}),
+    components: Flags.string({char: 'c', description: 'place for installation of components'}),
+  }
+
+  public async run(): Promise<void> {
+    const {args, flags} = await this.parse(Add)
+
+    if (!args.name) {
+      this.error('No component name provided. Please provide name of component you want to be installed.')
+    }
+
+    const resolvedConfig = await validateConfig(this.log, await loadConfig(flags.config))
+    const componentFolder = join(process.cwd(), resolvedConfig.paths.components)
+    const sharedFile = join(process.cwd(), resolvedConfig.paths.shared)
+
+    if (!existsSync(componentFolder)) {
+      await mkdir(componentFolder, {recursive: true})
+    }
+
+    const loadRegistryOra = ora('Fetching registry...').start()
+    const unsafeRegistry = await getRegistry()
+    if (!unsafeRegistry.ok) {
+      loadRegistryOra.fail(unsafeRegistry.message)
+      return
+    }
+    const registry = unsafeRegistry.registry
+    const componentNames = await getAvailableComponentNames(registry)
+    loadRegistryOra.succeed(`Successfully fetched registry! (${componentNames.length} components)`)
+
+    if (!componentNames.includes(args.name)) {
+      this.error(`Component with name ${args.name} does not exists in registry. Please provide correct name.`)
+    }
+
+    const sharedFileOra = ora('Installing shared module...').start()
+    if (!existsSync(sharedFile) || flags.forceShared) {
+      const sharedFileContentResponse = await fetch(registry.shared)
+      if (!sharedFileContentResponse.ok) {
+        sharedFileOra.fail(
+          `Error while fetching shared module content: ${sharedFileContentResponse.status} ${sharedFileContentResponse.statusText}`,
+        )
+        return
+      }
+      const sharedFileContent = await sharedFileContentResponse.text()
+      await writeFile(sharedFile, sharedFileContent)
+      sharedFileOra.succeed('Shared module is successfully installed!')
+    } else {
+      sharedFileOra.succeed('Shared module is already installed!')
+    }
+
+    const componentFileOra = ora(`Installing ${args.name} component...`).start()
+    const componentFile = join(componentFolder, registry.components[args.name])
+    if (existsSync(componentFile) && !flags.force) {
+      componentFileOra.succeed(`Component is already installed! (${componentFile})`)
+    } else {
+      const componentFileContentResponse = await fetch(await getComponentURL(registry, args.name))
+      if (!componentFileContentResponse.ok) {
+        componentFileOra.fail(
+          `Error while fetching component file content: ${componentFileContentResponse.status} ${componentFileContentResponse.statusText}`,
+        )
+        return
+      }
+      const componentFileContent = (await componentFileContentResponse.text()).replaceAll(
+        /import\s+{[^}]*}\s+from\s+"..\/shared"/g,
+        (match) => match.replace(/..\/shared/, resolvedConfig.import.shared),
+      )
+      await writeFile(componentFile, componentFileContent)
+      componentFileOra.succeed('Component is successfully installed!')
+    }
+
+    this.log('Now you can import the component.')
+  }
+}