ds-01 0.1.2 → 0.1.4

package/.turbo/turbo-build.log

@@ -1,3 +1,3 @@
 [?9001h[?1004h]0;C:\WINDOWS\system32\cmd.exe[?25h[?25l
-> @repo/cli@0.0.1 build C:\Web-dev 2.0\Personal Projects\ds01\packages\cli
+> ds-01@0.1.3 build C:\Web-dev 2.0\Personal Projects\ds01\packages\cli
 > tsc[?25h[?9001l[?1004l

package/dist/commands/add.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare const add: Command;
+//# sourceMappingURL=add.d.ts.map

package/dist/commands/add.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"add.d.ts","sourceRoot":"","sources":["../../src/commands/add.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAUpC,eAAO,MAAM,GAAG,SAiEZ,CAAC"}

package/dist/commands/add.js

@@ -0,0 +1,63 @@
+import { Command } from "commander";
+import chalk from "chalk";
+import ora from "ora";
+import fs from "fs";
+import path from "path";
+import { execSync } from "child_process";
+import { getToken } from "../utils/config.js";
+const API_BASE_URL = process.env.DS01_API_URL || "https://ds-01.vercel.app"; // Point to localhost for testing
+export const add = new Command()
+    .name("add")
+    .description("Add a component from DS01 to your project")
+    .argument("<component>", "The name of the component (e.g., premium-section)")
+    .action(async (componentName) => {
+    const cwd = process.cwd();
+    const configPath = path.join(cwd, "ds01.config.json");
+    if (!fs.existsSync(configPath)) {
+        console.log(chalk.red("✖ ds01.config.json not found."));
+        console.log(`Run ${chalk.cyan("npx @arpit2023/ds01 init")} first.`);
+        process.exit(1);
+    }
+    const config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+    const token = getToken();
+    const spinner = ora(`Fetching <${componentName} /> from registry...`).start();
+    try {
+        const response = await fetch(`${API_BASE_URL}/api/registry/${componentName}`, {
+            headers: { Authorization: `Bearer ${token || ""}` },
+        });
+        if (!response.ok) {
+            throw new Error(`Component not found or server error (${response.status})`);
+        }
+        const componentData = await response.json();
+        // Create a dedicated folder for the component inside the user's project
+        const targetDir = path.join(cwd, config.componentsPath, componentName);
+        if (!fs.existsSync(targetDir)) {
+            fs.mkdirSync(targetDir, { recursive: true });
+        }
+        // Write all the pure files!
+        for (const file of componentData.files) {
+            const filePath = path.join(targetDir, file.name);
+            fs.writeFileSync(filePath, file.content);
+        }
+        spinner.succeed(chalk.green(`Downloaded ${componentData.files.length} files for ${componentName}`));
+        // Auto-install dependencies like framer-motion
+        if (componentData.dependencies && componentData.dependencies.length > 0) {
+            const depsToInstall = componentData.dependencies.join(" ");
+            const installSpinner = ora(`Installing dependencies: ${chalk.cyan(depsToInstall)}...`).start();
+            try {
+                execSync(`npm install ${depsToInstall}`, { stdio: "ignore" });
+                installSpinner.succeed(chalk.green("Dependencies installed successfully."));
+            }
+            catch (error) {
+                installSpinner.fail(chalk.red("Failed to auto-install dependencies. Please install manually."));
+            }
+        }
+        console.log("\n" + chalk.green.bold(`✔ Component <${componentName} /> is ready!`));
+        console.log(chalk.gray(`Import it: import { Section } from "@/${config.componentsPath}/${componentName}/Section"`));
+    }
+    catch (error) {
+        spinner.fail(chalk.red(`Failed to add component.`));
+        console.error(chalk.red(error.message));
+        process.exit(1);
+    }
+});

package/dist/commands/init.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare const init: Command;
+//# sourceMappingURL=init.d.ts.map

package/dist/commands/init.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../../src/commands/init.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAMpC,eAAO,MAAM,IAAI,SA4Fb,CAAC"}

package/dist/commands/init.js

@@ -0,0 +1,86 @@
+import { Command } from "commander";
+import fs from "fs";
+import path from "path";
+import { intro, outro, spinner, text, isCancel, cancel } from "@clack/prompts";
+import chalk from "chalk";
+export const init = new Command()
+    .name("init")
+    .description("Initialize DS01 in your React/Next.js TypeScript project")
+    .action(async () => {
+    intro(chalk.bgCyan.black.bold(" DS01 Setup Wizard "));
+    const s = spinner();
+    s.start("Checking project environment...");
+    const cwd = process.cwd();
+    const packageJsonPath = path.join(cwd, "package.json");
+    const tsconfigPath = path.join(cwd, "tsconfig.json");
+    // 1. Check for package.json
+    if (!fs.existsSync(packageJsonPath)) {
+        s.stop("Environment check failed.");
+        cancel("No package.json found. Please run this command in the root of your project.");
+        process.exit(1);
+    }
+    const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, "utf-8"));
+    const deps = { ...packageJson.dependencies, ...packageJson.devDependencies };
+    // 2. Check strictly for React or Next.js
+    if (!deps.react && !deps.next) {
+        s.stop("Environment check failed.");
+        cancel("DS01 strictly supports React and Next.js environments. Please run this in a supported project.");
+        process.exit(1);
+    }
+    // 3. Check strictly for TypeScript
+    if (!fs.existsSync(tsconfigPath)) {
+        s.stop("Environment check failed.");
+        cancel("DS01 requires TypeScript. No tsconfig.json found in the project root.");
+        process.exit(1);
+    }
+    s.stop(chalk.green("✔ React and TypeScript detected!"));
+    // 4. Ask the user where their global CSS is located
+    const cssPath = await text({
+        message: "Where is your global CSS file?",
+        placeholder: "app/globals.css",
+        initialValue: "app/globals.css",
+    });
+    if (isCancel(cssPath)) {
+        cancel("Setup cancelled.");
+        process.exit(0);
+    }
+    // 5. Inject Dark Mode / Glassmorphic CSS variables
+    const fullCssPath = path.join(cwd, cssPath);
+    if (fs.existsSync(fullCssPath)) {
+        const cssContent = fs.readFileSync(fullCssPath, "utf-8");
+        if (!cssContent.includes("/* ds01-theme-start */")) {
+            const variables = `
+/* ds01-theme-start */
+@layer base {
+  :root {
+    --ds-background: #0a0a0a; /* Deep charcoal */
+    --ds-foreground: #ffffff;
+    --ds-glass-border: rgba(255, 255, 255, 0.08);
+    --ds-glass-blur: blur(16px);
+  }
+}
+/* ds01-theme-end */
+`;
+            fs.appendFileSync(fullCssPath, variables);
+            console.log(chalk.green(`✔ Injected design tokens into ${cssPath}`));
+        }
+        else {
+            console.log(chalk.gray(`- Design tokens already exist in ${cssPath}`));
+        }
+    }
+    else {
+        console.log(chalk.yellow(`⚠ Could not find ${cssPath}. You may need to add the CSS variables manually.`));
+    }
+    // 6. Create the local configuration file so the CLI remembers this project
+    const configPath = path.join(cwd, "ds01.config.json");
+    const configData = {
+        $schema: "https://ds-01.vercel.app/schema.json",
+        style: "default",
+        typescript: true,
+        cssPath: cssPath,
+        componentsPath: "components/ui" // This is where we will inject components later
+    };
+    fs.writeFileSync(configPath, JSON.stringify(configData, null, 2));
+    console.log(chalk.green(`✔ Created ds01.config.json`));
+    outro(chalk.green.bold("DS01 successfully initialized! Ready to add components."));
+});

package/dist/commands/login.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"login.d.ts","sourceRoot":"","sources":["../../src/commands/login.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAgBpC,eAAO,MAAM,KAAK,SAgGd,CAAC"}
+{"version":3,"file":"login.d.ts","sourceRoot":"","sources":["../../src/commands/login.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAWpC,eAAO,MAAM,KAAK,SAgFd,CAAC"}

package/dist/commands/login.js

@@ -3,13 +3,9 @@ import ora from "ora";
 import chalk from "chalk";
 import open from "open";
 import machineIdPkg from "node-machine-id";
-import fs from "fs";
-import path from "path";
-import os from "os";
+import { saveToken } from "../utils/config.js"; // <-- Imported our new wallet helper!
 // ⚠️ In production, replace with your actual deployed Vercel domain
 const API_BASE_URL = process.env.DS01_API_URL || "https://ds-01.vercel.app";
-const CONFIG_DIR = path.join(os.homedir(), ".ds01");
-const AUTH_FILE = path.join(CONFIG_DIR, "auth.json");
 const { machineIdSync } = machineIdPkg;
 export const login = new Command()
     .name("login")
@@ -25,7 +21,7 @@ export const login = new Command()
             headers: {
                 "Content-Type": "application/json",
             },
-            body: JSON.stringify({ machineId: hardwareId }), // <-- Send the machine ID to the server
+            body: JSON.stringify({ machineId: hardwareId }),
         });
         if (!initResponse.ok) {
             throw new Error("Failed to reach DS01 authorization server.");
@@ -41,9 +37,8 @@ export const login = new Command()
         spinner.start("Waiting for web approval...");
         // 4. Poll the Token Endpoint
         let token = null;
-        const pollInterval = (interval || 2) * 1000;
+        const pollInterval = (interval || 5) * 1000; // Updated to match our 5s secure interval
         while (!token) {
-            // Wait before polling again
             await new Promise((resolve) => setTimeout(resolve, pollInterval));
             const tokenResponse = await fetch(`${API_BASE_URL}/api/auth/device/token`, {
                 method: "POST",
@@ -55,21 +50,14 @@ export const login = new Command()
                 token = tokenData.accessToken;
             }
             else if (tokenData.error !== "authorization_pending") {
-                // If error is anything OTHER than pending, fail out.
                 spinner.fail(chalk.red("Authorization failed or expired."));
                 console.error(chalk.red(`Reason: ${tokenData.error}`));
                 process.exit(1);
             }
         }
-        // 5. Save Token & Machine ID to ~/.ds01/auth.json
-        if (!fs.existsSync(CONFIG_DIR)) {
-            fs.mkdirSync(CONFIG_DIR, { recursive: true });
-        }
-        fs.writeFileSync(AUTH_FILE, JSON.stringify({
-            token,
-            machineId: hardwareId,
-            updatedAt: new Date().toISOString(),
-        }, null, 2));
+        // 5. Save Token securely using our new config utility
+        // We pass the token and hardwareId into the wallet.
+        saveToken(token, hardwareId);
         spinner.succeed(chalk.green.bold("Terminal paired successfully!"));
         console.log(chalk.gray(`Session token bound to hardware signature and saved locally.`));
     }

package/dist/commands/logout.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare const logout: Command;
+//# sourceMappingURL=logout.d.ts.map

package/dist/commands/logout.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logout.d.ts","sourceRoot":"","sources":["../../src/commands/logout.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAKpC,eAAO,MAAM,MAAM,SAuBf,CAAC"}

package/dist/commands/logout.js

@@ -0,0 +1,25 @@
+import { Command } from "commander";
+import chalk from "chalk";
+import ora from "ora";
+import { deleteToken, getToken } from "../utils/config.js";
+export const logout = new Command()
+    .name("logout")
+    .description("Log out of your DS01 account and clear local credentials")
+    .action(async () => {
+    const token = getToken();
+    if (!token) {
+        console.log(chalk.yellow("You are already logged out."));
+        return;
+    }
+    const spinner = ora("Logging out...").start();
+    try {
+        // Throw away the digital wallet / token
+        deleteToken();
+        spinner.succeed(chalk.green("Successfully logged out."));
+        console.log(chalk.gray("Your local credentials have been cleared."));
+    }
+    catch (error) {
+        spinner.fail(chalk.red("Failed to log out cleanly."));
+        console.error(error);
+    }
+});

package/dist/commands/whoami.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare const whoami: Command;
+//# sourceMappingURL=whoami.d.ts.map

package/dist/commands/whoami.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"whoami.d.ts","sourceRoot":"","sources":["../../src/commands/whoami.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAIpC,eAAO,MAAM,MAAM,SAef,CAAC"}

package/dist/commands/whoami.js

@@ -0,0 +1,17 @@
+import { Command } from "commander";
+import chalk from "chalk";
+import { getToken } from "../utils/config.js";
+export const whoami = new Command()
+    .name("whoami")
+    .description("Check your current authentication status")
+    .action(async () => {
+    const token = getToken();
+    if (!token) {
+        console.log(chalk.red("✖ You are not logged in."));
+        console.log(`Run ${chalk.cyan("npx @arpit2023/ds01 login")} to authenticate.`);
+        process.exit(1);
+    }
+    // For now, we just verify the token is saved. 
+    console.log(chalk.green("✔ You are securely logged in!"));
+    console.log(chalk.gray(`Token found ending in: ...${token.slice(-6)}`));
+});

package/dist/index.js

@@ -1,13 +1,18 @@
 #!/usr/bin/env node
 import { Command } from "commander";
 import { login } from "./commands/login.js";
-// Note: Add .js extension if you have module resolution issues, or keep it without depending on your tsconfig
+import { whoami } from "./commands/whoami.js";
+import { logout } from "./commands/logout.js";
+import { init } from "./commands/init.js"; // <-- Imported the new init command!
 const program = new Command();
 program
     .name("ds01")
     .description("DS01 Custom Component Registry CLI")
     .version("1.0.0");
-// Register the login command
+// Register the commands
 program.addCommand(login);
+program.addCommand(whoami);
+program.addCommand(logout);
+program.addCommand(init); // <-- Registered it here!
 // Parse the arguments from the terminal
 program.parse(process.argv);

package/dist/utils/config.d.ts

@@ -0,0 +1,4 @@
+export declare function saveToken(token: string, machineId: string): void;
+export declare function getToken(): string | null;
+export declare function deleteToken(): void;
+//# sourceMappingURL=config.d.ts.map

package/dist/utils/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/utils/config.ts"],"names":[],"mappings":"AAQA,wBAAgB,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,QAczD;AAED,wBAAgB,QAAQ,IAAI,MAAM,GAAG,IAAI,CAWxC;AAED,wBAAgB,WAAW,SAI1B"}

package/dist/utils/config.js

@@ -0,0 +1,37 @@
+import fs from "fs";
+import path from "path";
+import os from "os";
+// We use the same path you originally had for consistency
+const configDir = path.join(os.homedir(), ".ds01");
+const credentialsPath = path.join(configDir, "auth.json");
+export function saveToken(token, machineId) {
+    if (!fs.existsSync(configDir)) {
+        fs.mkdirSync(configDir, { recursive: true });
+    }
+    const payload = {
+        token,
+        machineId,
+        updatedAt: new Date().toISOString(),
+    };
+    fs.writeFileSync(credentialsPath, JSON.stringify(payload, null, 2), {
+        mode: 0o600,
+    });
+}
+export function getToken() {
+    if (fs.existsSync(credentialsPath)) {
+        try {
+            const data = fs.readFileSync(credentialsPath, "utf-8");
+            const parsed = JSON.parse(data);
+            return parsed.token || null;
+        }
+        catch {
+            return null;
+        }
+    }
+    return null;
+}
+export function deleteToken() {
+    if (fs.existsSync(credentialsPath)) {
+        fs.unlinkSync(credentialsPath);
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ds-01",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Make your site best with DS01",
   "type": "module",
   "bin": {
@@ -13,14 +13,15 @@
     "access": "public"
   },
   "dependencies": {
-    "chalk": "^5",
+    "@clack/prompts": "^1.7.0",
+    "chalk": "^5.6.2",
     "commander": "^14",
     "node-machine-id": "^1",
     "open": "^10",
     "ora": "^8"
   },
   "devDependencies": {
-    "@types/node": "^24",
+    "@types/node": "^24.13.3",
     "tsup": "^8.5.1",
     "typescript": "^5"
   }

package/src/commands/add.ts

@@ -0,0 +1,76 @@
+import { Command } from "commander";
+import chalk from "chalk";
+import ora from "ora";
+import fs from "fs";
+import path from "path";
+import { execSync } from "child_process";
+import { getToken } from "../utils/config.js";
+
+const API_BASE_URL = process.env.DS01_API_URL || "https://ds-01.vercel.app"; // Point to localhost for testing
+
+export const add = new Command()
+  .name("add")
+  .description("Add a component from DS01 to your project")
+  .argument("<component>", "The name of the component (e.g., premium-section)")
+  .action(async (componentName: string) => {
+    const cwd = process.cwd();
+    const configPath = path.join(cwd, "ds01.config.json");
+
+    if (!fs.existsSync(configPath)) {
+      console.log(chalk.red("✖ ds01.config.json not found."));
+      console.log(`Run ${chalk.cyan("npx @arpit2023/ds01 init")} first.`);
+      process.exit(1);
+    }
+
+    const config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+    const token = getToken();
+
+    const spinner = ora(`Fetching <${componentName} /> from registry...`).start();
+
+    try {
+      const response = await fetch(`${API_BASE_URL}/api/registry/${componentName}`, {
+        headers: { Authorization: `Bearer ${token || ""}` },
+      });
+
+      if (!response.ok) {
+        throw new Error(`Component not found or server error (${response.status})`);
+      }
+
+      const componentData = await response.json();
+
+      // Create a dedicated folder for the component inside the user's project
+      const targetDir = path.join(cwd, config.componentsPath, componentName);
+      if (!fs.existsSync(targetDir)) {
+        fs.mkdirSync(targetDir, { recursive: true });
+      }
+
+      // Write all the pure files!
+      for (const file of componentData.files) {
+        const filePath = path.join(targetDir, file.name);
+        fs.writeFileSync(filePath, file.content);
+      }
+      
+      spinner.succeed(chalk.green(`Downloaded ${componentData.files.length} files for ${componentName}`));
+
+      // Auto-install dependencies like framer-motion
+      if (componentData.dependencies && componentData.dependencies.length > 0) {
+        const depsToInstall = componentData.dependencies.join(" ");
+        const installSpinner = ora(`Installing dependencies: ${chalk.cyan(depsToInstall)}...`).start();
+        
+        try {
+          execSync(`npm install ${depsToInstall}`, { stdio: "ignore" });
+          installSpinner.succeed(chalk.green("Dependencies installed successfully."));
+        } catch (error) {
+          installSpinner.fail(chalk.red("Failed to auto-install dependencies. Please install manually."));
+        }
+      }
+
+      console.log("\n" + chalk.green.bold(`✔ Component <${componentName} /> is ready!`));
+      console.log(chalk.gray(`Import it: import { Section } from "@/${config.componentsPath}/${componentName}/Section"`));
+
+    } catch (error: any) {
+      spinner.fail(chalk.red(`Failed to add component.`));
+      console.error(chalk.red(error.message));
+      process.exit(1);
+    }
+  });

package/src/commands/init.ts

@@ -0,0 +1,99 @@
+import { Command } from "commander";
+import fs from "fs";
+import path from "path";
+import { intro, outro, spinner, text, isCancel, cancel } from "@clack/prompts";
+import chalk from "chalk";
+
+export const init = new Command()
+  .name("init")
+  .description("Initialize DS01 in your React/Next.js TypeScript project")
+  .action(async () => {
+    intro(chalk.bgCyan.black.bold(" DS01 Setup Wizard "));
+
+    const s = spinner();
+    s.start("Checking project environment...");
+
+    const cwd = process.cwd();
+    const packageJsonPath = path.join(cwd, "package.json");
+    const tsconfigPath = path.join(cwd, "tsconfig.json");
+
+    // 1. Check for package.json
+    if (!fs.existsSync(packageJsonPath)) {
+      s.stop("Environment check failed.");
+      cancel("No package.json found. Please run this command in the root of your project.");
+      process.exit(1);
+    }
+
+    const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, "utf-8"));
+    const deps = { ...packageJson.dependencies, ...packageJson.devDependencies };
+
+    // 2. Check strictly for React or Next.js
+    if (!deps.react && !deps.next) {
+      s.stop("Environment check failed.");
+      cancel("DS01 strictly supports React and Next.js environments. Please run this in a supported project.");
+      process.exit(1);
+    }
+
+    // 3. Check strictly for TypeScript
+    if (!fs.existsSync(tsconfigPath)) {
+      s.stop("Environment check failed.");
+      cancel("DS01 requires TypeScript. No tsconfig.json found in the project root.");
+      process.exit(1);
+    }
+
+    s.stop(chalk.green("✔ React and TypeScript detected!"));
+
+    // 4. Ask the user where their global CSS is located
+    const cssPath = await text({
+      message: "Where is your global CSS file?",
+      placeholder: "app/globals.css",
+      initialValue: "app/globals.css",
+    });
+
+    if (isCancel(cssPath)) {
+      cancel("Setup cancelled.");
+      process.exit(0);
+    }
+
+    // 5. Inject Dark Mode / Glassmorphic CSS variables
+    const fullCssPath = path.join(cwd, cssPath as string);
+    if (fs.existsSync(fullCssPath)) {
+      const cssContent = fs.readFileSync(fullCssPath, "utf-8");
+      
+      if (!cssContent.includes("/* ds01-theme-start */")) {
+        const variables = `
+/* ds01-theme-start */
+@layer base {
+  :root {
+    --ds-background: #0a0a0a; /* Deep charcoal */
+    --ds-foreground: #ffffff;
+    --ds-glass-border: rgba(255, 255, 255, 0.08);
+    --ds-glass-blur: blur(16px);
+  }
+}
+/* ds01-theme-end */
+`;
+        fs.appendFileSync(fullCssPath, variables);
+        console.log(chalk.green(`✔ Injected design tokens into ${cssPath}`));
+      } else {
+        console.log(chalk.gray(`- Design tokens already exist in ${cssPath}`));
+      }
+    } else {
+      console.log(chalk.yellow(`⚠ Could not find ${cssPath}. You may need to add the CSS variables manually.`));
+    }
+
+    // 6. Create the local configuration file so the CLI remembers this project
+    const configPath = path.join(cwd, "ds01.config.json");
+    const configData = {
+      $schema: "https://ds-01.vercel.app/schema.json",
+      style: "default",
+      typescript: true,
+      cssPath: cssPath,
+      componentsPath: "components/ui" // This is where we will inject components later
+    };
+    
+    fs.writeFileSync(configPath, JSON.stringify(configData, null, 2));
+    console.log(chalk.green(`✔ Created ds01.config.json`));
+
+    outro(chalk.green.bold("DS01 successfully initialized! Ready to add components."));
+  });

package/src/commands/login.ts

@@ -3,15 +3,10 @@ import ora from "ora";
 import chalk from "chalk";
 import open from "open";
 import machineIdPkg from "node-machine-id";
-
-import fs from "fs";
-import path from "path";
-import os from "os";
+import { saveToken } from "../utils/config.js"; // <-- Imported our new wallet helper!
 
 // ⚠️ In production, replace with your actual deployed Vercel domain
 const API_BASE_URL = process.env.DS01_API_URL || "https://ds-01.vercel.app";
-const CONFIG_DIR = path.join(os.homedir(), ".ds01");
-const AUTH_FILE = path.join(CONFIG_DIR, "auth.json");
 const { machineIdSync } = machineIdPkg;
 
 export const login = new Command()
@@ -30,7 +25,7 @@ export const login = new Command()
         headers: {
           "Content-Type": "application/json",
         },
-        body: JSON.stringify({ machineId: hardwareId }), // <-- Send the machine ID to the server
+        body: JSON.stringify({ machineId: hardwareId }), 
       });
 
       if (!initResponse.ok) {
@@ -54,10 +49,9 @@ export const login = new Command()
 
       // 4. Poll the Token Endpoint
       let token = null;
-      const pollInterval = (interval || 2) * 1000;
+      const pollInterval = (interval || 5) * 1000; // Updated to match our 5s secure interval
 
       while (!token) {
-        // Wait before polling again
         await new Promise((resolve) => setTimeout(resolve, pollInterval));
 
         const tokenResponse = await fetch(
@@ -74,30 +68,15 @@ export const login = new Command()
         if (tokenResponse.ok && tokenData.accessToken) {
           token = tokenData.accessToken;
         } else if (tokenData.error !== "authorization_pending") {
-          // If error is anything OTHER than pending, fail out.
           spinner.fail(chalk.red("Authorization failed or expired."));
           console.error(chalk.red(`Reason: ${tokenData.error}`));
           process.exit(1);
         }
       }
 
-      // 5. Save Token & Machine ID to ~/.ds01/auth.json
-      if (!fs.existsSync(CONFIG_DIR)) {
-        fs.mkdirSync(CONFIG_DIR, { recursive: true });
-      }
-
-      fs.writeFileSync(
-        AUTH_FILE,
-        JSON.stringify(
-          {
-            token,
-            machineId: hardwareId,
-            updatedAt: new Date().toISOString(),
-          },
-          null,
-          2,
-        ),
-      );
+      // 5. Save Token securely using our new config utility
+      // We pass the token and hardwareId into the wallet.
+      saveToken(token, hardwareId);
 
       spinner.succeed(chalk.green.bold("Terminal paired successfully!"));
       console.log(
@@ -110,4 +89,4 @@ export const login = new Command()
       console.error(chalk.red(error.message));
       process.exit(1);
     }
-  });
+  });

package/src/commands/logout.ts

@@ -0,0 +1,29 @@
+import { Command } from "commander";
+import chalk from "chalk";
+import ora from "ora";
+import { deleteToken, getToken } from "../utils/config.js";
+
+export const logout = new Command()
+  .name("logout")
+  .description("Log out of your DS01 account and clear local credentials")
+  .action(async () => {
+    const token = getToken();
+
+    if (!token) {
+      console.log(chalk.yellow("You are already logged out."));
+      return;
+    }
+
+    const spinner = ora("Logging out...").start();
+
+    try {
+      // Throw away the digital wallet / token
+      deleteToken();
+      
+      spinner.succeed(chalk.green("Successfully logged out."));
+      console.log(chalk.gray("Your local credentials have been cleared."));
+    } catch (error) {
+      spinner.fail(chalk.red("Failed to log out cleanly."));
+      console.error(error);
+    }
+  });

package/src/commands/whoami.ts

@@ -0,0 +1,20 @@
+import { Command } from "commander";
+import chalk from "chalk";
+import { getToken } from "../utils/config.js";
+
+export const whoami = new Command()
+  .name("whoami")
+  .description("Check your current authentication status")
+  .action(async () => {
+    const token = getToken();
+
+    if (!token) {
+      console.log(chalk.red("✖ You are not logged in."));
+      console.log(`Run ${chalk.cyan("npx @arpit2023/ds01 login")} to authenticate.`);
+      process.exit(1);
+    }
+
+    // For now, we just verify the token is saved. 
+    console.log(chalk.green("✔ You are securely logged in!"));
+    console.log(chalk.gray(`Token found ending in: ...${token.slice(-6)}`));
+  });

package/src/index.ts

@@ -2,7 +2,9 @@
 
 import { Command } from "commander";
 import { login } from "./commands/login.js";
-// Note: Add .js extension if you have module resolution issues, or keep it without depending on your tsconfig
+import { whoami } from "./commands/whoami.js"; 
+import { logout } from "./commands/logout.js"; 
+import { init } from "./commands/init.js"; // <-- Imported the new init command!
 
 const program = new Command();
 
@@ -11,8 +13,11 @@ program
   .description("DS01 Custom Component Registry CLI")
   .version("1.0.0");
 
-// Register the login command
+// Register the commands
 program.addCommand(login);
+program.addCommand(whoami); 
+program.addCommand(logout); 
+program.addCommand(init); // <-- Registered it here!
 
 // Parse the arguments from the terminal
-program.parse(process.argv);
+program.parse(process.argv);

package/src/utils/config.ts

@@ -0,0 +1,42 @@
+import fs from "fs";
+import path from "path";
+import os from "os";
+
+// We use the same path you originally had for consistency
+const configDir = path.join(os.homedir(), ".ds01");
+const credentialsPath = path.join(configDir, "auth.json");
+
+export function saveToken(token: string, machineId: string) {
+  if (!fs.existsSync(configDir)) {
+    fs.mkdirSync(configDir, { recursive: true });
+  }
+
+  const payload = {
+    token,
+    machineId,
+    updatedAt: new Date().toISOString(),
+  };
+
+  fs.writeFileSync(credentialsPath, JSON.stringify(payload, null, 2), {
+    mode: 0o600, 
+  });
+}
+
+export function getToken(): string | null {
+  if (fs.existsSync(credentialsPath)) {
+    try {
+      const data = fs.readFileSync(credentialsPath, "utf-8");
+      const parsed = JSON.parse(data);
+      return parsed.token || null;
+    } catch {
+      return null;
+    }
+  }
+  return null;
+}
+
+export function deleteToken() {
+  if (fs.existsSync(credentialsPath)) {
+    fs.unlinkSync(credentialsPath);
+  }
+}