pulse-cli-divakar 1.0.1 → 1.0.4

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pulse-cli-divakar",
-  "version": "1.0.1",
-  "main": "src/index.js",
+  "version": "1.0.4",
+  "main": "./src/cli/main.js",
   "bin": {
     "pulse": "./src/cli/main.js"
   },
@@ -47,9 +47,12 @@
     "prisma": "^7.10.1"
   },
    "files": [
-    "src/cli",
-     "src/lib",
-     "src/config"
+  "src/cli",
+  "src/service",
+  "src/lib/token.js",
+  "src/config/agent-config.js",
+  "src/config/google.config.js",
+  "src/config/tool.config.js"
   ]
   
 }

package/src/cli/chat/chat-with-aiAgent.js

@@ -16,7 +16,6 @@ import { markedTerminal } from "marked-terminal";
 import { AIService } from "../ai/google-service.js";
 import { ChatService } from "../../service/chat.service.js";
 import { getStoredToken } from "../../lib/token.js";
-import { prisma } from "../../lib/db.js";
 import { getUserFromToken, saveMessage } from "./utils/chataiutils.js";
 import { generateApplication } from "../../config/agent-config.js";
 

package/src/cli/chat/chat-with-aitool.js

@@ -7,7 +7,7 @@ import {markedTerminal} from "marked-terminal"
 import { AIService } from "../ai/google-service.js";
 import { ChatService } from "../../service/chat.service.js";
 import { getStoredToken } from "../../lib/token.js";
-import {prisma} from "../../lib/db.js"
+
 
 import {availableTools,getEnabledToolNames,getEnabledTools,enableTools,resetTools} from "../../config/tool.config.js"
 import { displayMessages, getUserFromToken ,updateConversationTitle,saveMessage} from "./utils/chataiutils.js";

package/src/cli/chat/utils/chataiutils.js

@@ -1,40 +1,61 @@
 import yoctoSpinner from "yocto-spinner";
-import { getStoredToken } from "../../../lib/token.js";
-import { prisma } from "../../../lib/db.js";
 import { AIService } from "../../ai/google-service.js";
 import { ChatService } from "../../../service/chat.service.js";
 import { isCancel, text } from "@clack/prompts";
 import chalk from "chalk";
 import boxen from "boxen";
 import { marked } from "marked";
+import { requireAuth } from "../../../lib/token.js";
+import { URL } from "../../commands/auth/login.js";
 
 const aiService=new AIService()
 const chatService=new ChatService()
 
 export async function getUserFromToken() {
-    const token=await getStoredToken();
-    if(!token.access_token){
-        throw new Error("Not authenticated. Please run 'pulse login' first");
-    }
-    const spinner=yoctoSpinner({text:"Authenticating..."});
-    spinner.start()
+    const token = await requireAuth();
+
+    const spinner = yoctoSpinner({
+        text: "Authenticating..."
+    });
+
+    spinner.start();
+
+    try {
+        const serverUrl = URL.replace(/\/+$/, "");
 
-    const user=await prisma.user.findFirst({
-        where:{
-            sessions:{
-                some:{token:token.access_token},
+        const response = await fetch(`${serverUrl}/api/cli/me`, {
+            headers: {
+                Authorization: `Bearer ${token.access_token}`,
+            },
+        });
+
+        if (!response.ok) {
+            spinner.error("Authentication failed");
+
+            if (response.status === 401) {
+                throw new Error("Session invalid or expired. Please login again.");
             }
+
+            throw new Error(`Failed to fetch user: HTTP ${response.status}`);
         }
-    })
-    if(!user){
-        spinner.error("User not found")
-        throw new Error("User not found.Please login again")
+
+        const user = await response.json();
+
+        if (!user?.id) {
+            spinner.error("User not found");
+            throw new Error("User not found. Please login again.");
+        }
+
+        spinner.success(`Welcome back, ${user.name}!`);
+        return user;
+    } catch (error) {
+        if (spinner.isSpinning) {
+            spinner.error("Failed to authenticate");
+        }
+        throw error;
     }
-    spinner.success(`Welcome back, ${user.name}!`)
-    return user;
 }
 
-
 export async function getAIResponse(conversationId) {
     const spinner = yoctoSpinner({ 
         text: "AI is thinking...", 

package/src/cli/commands/ai/wakeup.js

@@ -1,85 +1,89 @@
-import chalk from "chalk"
-import { Command } from "commander"
-import yoctoSpinner from "yocto-spinner"
-import { getStoredToken } from "../../../lib/token.js"
-import { select } from "@clack/prompts"
-import { prisma } from "../../../lib/db.js"
+import chalk from "chalk";
+import { Command } from "commander";
+import yoctoSpinner from "yocto-spinner";
+import { getStoredToken, requireAuth } from "../../../lib/token.js";
+import { select } from "@clack/prompts";
+import { URL } from "../auth/login.js";
 
+const wakeUpAction = async () => {
+  const token = await requireAuth();
+  if (!token?.access_token) {
+    console.log(chalk.red("Not Authenticated.Please login"));
+    return;
+  }
 
+  const spinner = yoctoSpinner({ text: "Fetching user information..." });
+  spinner.start();
 
-const wakeUpAction=async ()=>{
-    const token=await getStoredToken()
-    if(!token?.access_token){
-        console.log(chalk.red("Not Authenticated.Please login"))
-        return;
-    }
+  const serverUrl = URL.replace(/\/+$/, "");
 
-    const spinner=yoctoSpinner({text:"Fetching user information..."})
-    spinner.start()
+  const response = await fetch(`${serverUrl}/api/cli/me`, {
+    headers: {
+      Authorization: `Bearer ${token.access_token}`,
+    },
+  });
 
-    const user=await prisma.user.findFirst({
-        where:{
-        sessions:{
-            some:{
-            token:token.access_token,
-            }
-        }
-        },
-        select:{
-        id:true,
-        name:true,
-        email:true,
-        image:true
-        }
-    })
+  if (!response.ok) {
+    spinner.stop();
 
-    if(!user){
-        console.log(chalk.red("User not Found."))
-        return;
+    if (response.status === 401) {
+      console.log(chalk.red("Session invalid or expired. Please login again."));
+    } else {
+      console.log(chalk.red(`Failed to fetch user: HTTP ${response.status}`));
     }
 
-    spinner.stop()
-    console.log(chalk.green(`Welcome back, ${user.name}!\n`))
+    return;
+  }
 
-    const choice=await select({
-        message:"Select an Option:",
-        options:[
-            {
-                value:"chat",
-                label:"Chat",
-                hint:"Simple chat with AI"
-            },
-            {
-                value:"tool",
-                label:"Tool Calling",
-                hint:"Chat with tools.(Google Search,Code Execution)"
-            },
-            {
-                value:"agent",
-                label:"Agentic Mode",
-                hint:"Advanced AI agent."
-            }
-        ]
-    })
+  const user = await response.json();
+  if (!user?.id) {
+    spinner.stop();
+    console.log(chalk.red("User not found."));
+    return;
+  }
 
-    switch(choice){
-        case "chat":
-              const { startChat } = await import("../../chat/chat-with-ai.js");
-            await startChat("chat");
-            break;
-        case "tool":
-             const { startToolChat } = await import("../../chat/chat-with-aitool.js");
-            
-            await startToolChat();
-            break;
-        case "agent":
-            const { startAgentChat } = await import("../../chat/chat-with-aiAgent.js");
-            await startAgentChat();
-            break;
-    }
-}
+  spinner.stop();
+  console.log(chalk.green(`Welcome back, ${user.name}!\n`));
+
+  const choice = await select({
+    message: "Select an Option:",
+    options: [
+      {
+        value: "chat",
+        label: "Chat",
+        hint: "Simple chat with AI",
+      },
+      {
+        value: "tool",
+        label: "Tool Calling",
+        hint: "Chat with tools.(Google Search,Code Execution)",
+      },
+      {
+        value: "agent",
+        label: "Agentic Mode",
+        hint: "Advanced AI agent.",
+      },
+    ],
+  });
+
+  switch (choice) {
+    case "chat":
+      const { startChat } = await import("../../chat/chat-with-ai.js");
+      await startChat("chat");
+      break;
+    case "tool":
+      const { startToolChat } = await import("../../chat/chat-with-aitool.js");
 
+      await startToolChat();
+      break;
+    case "agent":
+      const { startAgentChat } =
+        await import("../../chat/chat-with-aiAgent.js");
+      await startAgentChat();
+      break;
+  }
+};
 
-export const wakeUp=new Command("wakeup")
-.description("Wake up the AI.")
-.action(wakeUpAction)
+export const wakeUp = new Command("wakeup")
+  .description("Wake up the AI.")
+  .action(wakeUpAction);

package/src/cli/commands/auth/login.js

@@ -13,7 +13,6 @@ import yoctoSpinner from "yocto-spinner";
 import * as z from "zod/v4";
 import dotenv from "dotenv";
 
-import { prisma } from "../../../lib/db.js";
 import { clearStoredToken, getStoredToken, isTokenExpired, requireAuth, storeToken } from "../../../lib/token.js";
 import { pollForToken } from "../utils/polling.js";
 
@@ -42,8 +41,8 @@ export async function loginAction() {
 
   intro(chalk.bold("🔏 Auth CLI Login."));
 
-  const existingToken = getStoredToken();
-  const expired = isTokenExpired();
+  const existingToken =await  getStoredToken();
+  const expired = await isTokenExpired();
 
   if (existingToken && !expired) {
     const shouldReAuth = await confirm({

package/src/cli/commands/auth/whoami.js

@@ -1,32 +1,38 @@
 import chalk from "chalk";
-import { prisma } from "../../../lib/db.js";
+
 import { requireAuth } from "../../../lib/token.js";
 import { Command } from "commander";
 import { URL } from "./login.js";
 
-export async function whoamiAction() {
+export async function whoamiAction(options) {
   const token=await requireAuth();
   if(!token?.access_token){
     console.log("No access token found.Please login");
     process.exit(1);
   }
 
-  const user=await prisma.user.findFirst({
-    where:{
-      sessions:{
-        some:{
-          token:token.access_token,
-        }
-      }
-    },
-    select:{
-      id:true,
-      name:true,
-      email:true,
-      image:true
-    }
-
-  })
+ 
+const serverUrl = (options.serverUrl || URL).replace(/\/+$/, "");
+
+const response = await fetch(`${serverUrl}/api/cli/me`, {
+  headers: {
+    Authorization: `Bearer ${token.access_token}`,
+  },
+});
+
+if (!response.ok) {
+  console.log(
+    chalk.red(
+      response.status === 401
+        ? "Session invalid or expired. Please login again."
+        : `Failed to fetch user: HTTP ${response.status}`
+    )
+  );
+  return;
+}
+
+const user = await response.json();
+
 
   console.log(
     chalk.bold.greenBright(`\n 👤User: ${user.name}

package/src/cli/commands/config/config.js

@@ -0,0 +1,72 @@
+
+import {
+  cancel,
+  intro,
+  isCancel,
+  outro,
+  password,
+} from "@clack/prompts";
+import chalk from "chalk";
+import { Command } from "commander";
+import fs from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+
+const CONFIG_DIR = path.join(os.homedir(), ".pulse");
+const CONFIG_FILE = path.join(CONFIG_DIR, "config.json");
+
+export async function configAction() {
+  intro(chalk.bold("⚙️ Pulse Configuration"));
+
+  const apiKey = await password({
+    message: "Enter your Google Gemini API key:",
+    validate(value) {
+      if (!value?.trim()) return "API key cannot be empty.";
+    },
+  });
+
+  if (isCancel(apiKey)) {
+    cancel("Configuration cancelled.");
+    return;
+  }
+
+  try {
+    await fs.mkdir(CONFIG_DIR, { recursive: true });
+
+    let existingConfig = {};
+    try {
+      existingConfig = JSON.parse(
+        await fs.readFile(CONFIG_FILE, "utf8"),
+      );
+    } catch {
+      // No existing config file yet.
+    }
+
+    await fs.writeFile(
+      CONFIG_FILE,
+      JSON.stringify(
+        {
+          ...existingConfig,
+          googleApiKey: apiKey.trim(),
+        },
+        null,
+        2,
+      ),
+      { encoding: "utf8", mode: 0o600 },
+    );
+
+    outro(chalk.green("Google API key saved successfully."));
+    console.log(chalk.gray(`Config location: ${CONFIG_FILE}`));
+    console.log(chalk.gray("You can now run Pulse from any folder."));
+  } catch (error) {
+    console.error(
+      chalk.red("Could not save configuration:"),
+      error.message,
+    );
+    process.exitCode = 1;
+  }
+}
+
+export const config = new Command("config")
+  .description("Configure your Google Gemini API key")
+  .action(configAction);

package/src/cli/main.js

@@ -9,6 +9,7 @@ import { login} from "./commands/auth/login.js"
 import { logout } from "./commands/auth/logout.js"
 import { whoami } from "./commands/auth/whoami.js"
 import { wakeUp } from "./commands/ai/wakeup.js"
+import {config} from "./commands/config/config.js"
 
 
 dotenv.config()
@@ -26,12 +27,13 @@ async function main() {
     console.log(chalk.magenta("A Cli Based AI application.\n"))
 
     const program=new Command("pulse")
-    program.version("0.0.1")
+    program.version("1.0.4")
     .description("Pulse is a Cli based application")
     .addCommand(login)
     .addCommand(logout)
     .addCommand(whoami)
     .addCommand(wakeUp)
+    .addCommand(config)
     
 
     program.action(()=>{

package/src/config/google.config.js

@@ -1,8 +1,23 @@
-import dotenv from "dotenv"
-dotenv.config()
+import dotenv from "dotenv";
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
 
-export const config={
-    googleApiKey:process.env.GOOGLE_GENERATIVE_AI_API_KEY,
-    model:process.env.ORBITAL_MODEL || "gemini-2.5-flash"
+dotenv.config();
 
-}
+const configPath = path.join(os.homedir(), ".pulse", "config.json");
+
+let savedConfig = {};
+
+try {
+  savedConfig = JSON.parse(fs.readFileSync(configPath, "utf8"));
+} catch {
+  // Configuration has not been created yet.
+}
+
+export const config = {
+  googleApiKey:
+    process.env.GOOGLE_GENERATIVE_AI_API_KEY || savedConfig.googleApiKey,
+
+  model: process.env.ORBITAL_MODEL || savedConfig.model || "gemini-2.5-flash",
+};

package/src/service/chat.service.js

@@ -0,0 +1,120 @@
+import { URL } from "../cli/commands/auth/login.js";
+import { getStoredToken } from "../lib/token.js";
+
+
+
+export class ChatService{
+    async request(endpoint, options = {}) {
+    const token = await getStoredToken();
+
+    if (!token?.access_token) {
+        throw new Error("Not authenticated. Please run 'pulse login' first.");
+    }
+
+    const response = await fetch(
+        `${URL.replace(/\/+$/, "")}/api/cli${endpoint}`,
+        {
+            ...options,
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${token.access_token}`,
+                ...options.headers,
+            },
+        }
+    );
+
+    if (!response.ok) {
+        const result = await response.json().catch(() => ({}));
+        throw new Error(result.error || `API request failed: HTTP ${response.status}`);
+    }
+
+    if (response.status === 204) return null;
+
+    return response.json();
+}
+    async createConversation(userId, mode = "chat", title = null) {
+    return this.request("/conversations", {
+        method: "POST",
+        body: JSON.stringify({
+            mode,
+            title: title || `New ${mode} conversation`,
+        }),
+    });
+}
+
+async getOrCreateConversation(userId, conversationId = null, mode = "chat") {
+    if (conversationId) {
+        return this.request(
+            `/conversations/${encodeURIComponent(conversationId)}`
+        );
+    }
+
+    return this.createConversation(userId, mode);
+}
+
+    // user messages add to convo like {role:user ,message:"jhfdks"}
+   async addMessage(conversationId, role, content) {
+    const contentStr =
+        typeof content === "string" ? content : JSON.stringify(content);
+
+    return this.request(
+        `/conversations/${encodeURIComponent(conversationId)}/messages`,
+        {
+            method: "POST",
+            body: JSON.stringify({
+                role,
+                content: contentStr,
+            }),
+        }
+    );
+}
+
+
+  async getMessages(conversationId) {
+    const messages = await this.request(
+        `/conversations/${encodeURIComponent(conversationId)}/messages`
+    );
+
+    return messages.map((msg) => ({
+        ...msg,
+        content: this.parseContent(msg.content),
+    }));
+}
+    async getUserConversation(userId) {
+    return this.request("/conversations");
+}
+
+  async deleteConversation(conversationId, userId) {
+    return this.request(
+        `/conversations/${encodeURIComponent(conversationId)}`,
+        {
+            method: "DELETE",
+        }
+    );
+}
+   async updateTitle(conversationId, title) {
+    return this.request(
+        `/conversations/${encodeURIComponent(conversationId)}`,
+        {
+            method: "PATCH",
+            body: JSON.stringify({ title }),
+        }
+    );
+}
+
+    parseContent(content){
+        try {
+            return JSON.parse(content)
+        } catch (error) {
+            return content;
+        }
+    }
+
+    formatMessageForAI(messages){
+        return messages.map((msg)=>({
+            role:msg.role,
+            content:typeof msg.content==="string" ?  msg.content : JSON.stringify(msg.content)
+        }))
+    }
+
+}

package/src/index.js

@@ -1,66 +0,0 @@
-
-import "dotenv/config";
-
-import express from "express";
-import cors from "cors";
-import { toNodeHandler, fromNodeHeaders } from "better-auth/node";
-import { auth } from "./lib/auth.js";
-
-const app = express();
-
-const frontendUrl =
-  process.env.CLIENT_URL || "http://localhost:3000";
-
-const allowedOrigins = [
-  "http://localhost:3000",
-  frontendUrl,
-].filter(Boolean);
-
-app.set("trust proxy", 1);
-
-app.use(
-  cors({
-    origin: allowedOrigins,
-    methods: ["GET", "POST", "PUT", "PATCH", "DELETE", "OPTIONS"],
-    credentials: true,
-  })
-);
-
-// Better Auth handler must come before express.json().
-app.all("/api/auth/*splat", toNodeHandler(auth));
-
-app.use(express.json());
-
-app.get("/api/me", async (req, res, next) => {
-  try {
-    const session = await auth.api.getSession({
-      headers: fromNodeHeaders(req.headers),
-    });
-
-    return res.json(session);
-  } catch (error) {
-    next(error);
-  }
-});
-
-app.get("/device", (req, res) => {
-  const userCode = req.query.user_code;
-
-  const destination = new URL("/device", frontendUrl);
-
-  if (typeof userCode === "string") {
-    destination.searchParams.set("user_code", userCode);
-  }
-
-  return res.redirect(destination.toString());
-});
-
-app.get("/", (_req, res) => {
-  res.status(200).send("Pulse API is running");
-});
-
-const port = process.env.PORT || 3004;
-
-app.listen(port, "0.0.0.0", () => {
-  console.log(`Pulse API listening on port ${port}`);
-});

package/src/lib/auth.js

@@ -1,29 +0,0 @@
-import { betterAuth } from "better-auth";
-import { prismaAdapter } from "better-auth/adapters/prisma";
-import { deviceAuthorization } from "better-auth/plugins"; 
-
-import { prisma } from "./db.js";
-
-export const auth = betterAuth({
-  baseURL: process.env.BETTER_AUTH_URL,
-  database: prismaAdapter(prisma, {
-    provider: "postgresql",
-  }),
-  basePath:"/api/auth",
-  trustedOrigins:["http://localhost:3000",
-     process.env.CLIENT_URL,
-  ],
-  socialProviders: {
-    github: {
-      clientId: process.env.GITHUB_CLIENT_ID,
-      clientSecret: process.env.GITHUB_CLIENT_SECRET,
-    },
-  },
-  plugins: [
-    deviceAuthorization({ 
-      verificationUri: "/device",
-      expiresIn:"30m",
-      interval:"5s" 
-    }), 
-  ],
-});

package/src/lib/db.js

@@ -1,9 +0,0 @@
-import "dotenv/config";
-import { PrismaNeon } from "@prisma/adapter-neon";
-import { PrismaClient } from "../generated/prisma/client.js";
-
-const adapter = new PrismaNeon({
-  connectionString: process.env.DATABASE_URL,
-});
-
-export const prisma = new PrismaClient({ adapter });