fix: user's cwd preference

Author: selemondev

.vscode/settings.json

@@ -9,5 +9,7 @@
         "dist": true // set this to false to include "dist" folder in search results
     },
     // Turn off tsc task auto detection since we have the necessary tasks as npm scripts
-    "typescript.tsc.autoDetect": "off"
+    "typescript.tsc.autoDetect": "off",
+    "terminal.integrated.cwd": "${workspaceFolder}/backend",
+    "terminal.integrated.defaultProfile.linux": ""
 }

src/extension.ts

@@ -7,7 +7,7 @@ import {
   getRegistry,
   shadCnDocUrl,
 } from "./utils/registry";
-import { executeCommand } from "./utils/vscode";
+import { executeCommand, getOrChooseCwd } from "./utils/vscode";
 import type { Components, Component } from "./utils/registry";
 
 const commands = {
@@ -24,7 +24,8 @@ export function activate(context: vscode.ExtensionContext) {
 
   const disposables: vscode.Disposable[] = [
     vscode.commands.registerCommand(commands.initCli, async () => {
-      const intCmd = await getInitCmd();
+      const cwd = await getOrChooseCwd();
+      const intCmd = await getInitCmd(cwd);
       executeCommand(intCmd);
     }),
     vscode.commands.registerCommand(commands.addNewComponent, async () => {
@@ -46,8 +47,8 @@ export function activate(context: vscode.ExtensionContext) {
       if (!selectedComponent) {
         return;
       }
-
-      const installCmd = await getInstallCmd([selectedComponent.label]);
+      const cwd = await getOrChooseCwd();
+      const installCmd = await getInstallCmd([selectedComponent.label], cwd);
       executeCommand(installCmd);
     }),
 
@@ -73,8 +74,8 @@ export function activate(context: vscode.ExtensionContext) {
       }
 
       const selectedComponent = selectedComponents.map((component: Component) => component.label);
-
-      const installCmd = await getInstallCmd(selectedComponent);
+      const cwd = await getOrChooseCwd();
+      const installCmd = await getInstallCmd(selectedComponent, cwd);
       executeCommand(installCmd);
     }),
     vscode.commands.registerCommand(commands.gotoComponentDoc, async () => {
@@ -120,4 +121,4 @@ export function activate(context: vscode.ExtensionContext) {
 }
 
 // This method is called when your extension is deactivated
-export function deactivate() {}
+export function deactivate() { }

src/utils/registry.ts

@@ -45,33 +45,33 @@ export const getRegistry = async (): Promise<Components | null> => {
   return components;
 };
 
-export const getInstallCmd = async (components: string[]) => {
+export const getInstallCmd = async (components: string[], cwd: string) => {
   const packageManager = await detectPackageManager();
   const componentStr = components.join(" ");
 
   if (packageManager === "bun") {
-    return `bunx shadcn-vue add ${componentStr}`;
+    return `bunx shadcn-vue add ${componentStr} -c ${cwd}`;
   }
 
   if (packageManager === "pnpm") {
-    return `pnpm dlx shadcn-vue@latest add ${componentStr}`;
+    return `pnpm dlx shadcn-vue@latest add ${componentStr} -c ${cwd}`;
   }
 
-  return `npx shadcn-vue@latest add ${componentStr}`;
+  return `npx shadcn-vue@latest add ${componentStr} -c ${cwd}`;
 };
 
-export const getInitCmd = async () => {
+export const getInitCmd = async (cwd: string) => {
   const packageManager = await detectPackageManager();
 
   if (packageManager === "bun") {
-    return "bunx shadcn-vue init";
+    return `bunx shadcn-vue@latest init -c ${cwd}`;
   }
 
   if (packageManager === "pnpm") {
-    return "pnpm dlx shadcn-vue@latest init";
+    return `pnpm dlx shadcn-vue@latest init -c ${cwd}`;
   }
 
-  return "npx shadcn-vue@latest init";
+  return `npx shadcn-vue@latest init -c ${cwd}`;
 };
 
 export const getComponentDocLink = (component: string) => {

src/utils/vscode.ts

@@ -56,3 +56,42 @@ export const detectPackageManager = async (): Promise<PackageManager> => {
 
   return "npm";
 };
+
+
+export const getOrChooseCwd = async (): Promise<string> => {
+  let cwd = "";
+  const prefix = "${workspaceFolder}/";
+
+  const workspaceFolders = (vscode.workspace.workspaceFolders ?? []).filter(
+    (f) => f.uri.scheme === "file"
+  );
+
+  if (!workspaceFolders.length) { return "./"; }
+
+  const workspacePath = workspaceFolders[0]?.uri.fsPath ?? "";
+  const cwdFromConfig = vscode.workspace
+    .getConfiguration()
+    .get<string>("terminal.integrated.cwd")
+    ?.trim();
+
+  if (cwdFromConfig) {
+    if (cwdFromConfig.startsWith(prefix)) {
+      cwd = cwdFromConfig.slice(prefix.length);
+    }
+    else if (cwdFromConfig.startsWith(workspacePath)) {
+      cwd = cwdFromConfig.replace(new RegExp(`^${workspacePath}/?`), "");
+    } else {
+      cwd = cwdFromConfig;
+    }
+
+    return `${workspacePath}/${cwd}`;
+  }
+
+  const choice = await vscode.window.showQuickPick(
+    workspaceFolders.map((f) => f.name)
+  );
+
+  if (!choice) { return "./"; }
+
+  return workspaceFolders.find((f) => f.name === choice)?.uri.fsPath ?? "./";
+};