Merge pull request #16 from aibtcdev/feat/frontend-endpoints

Add endpoints needed by front-end

Author: whoabuddy

src/database/helpers/agents.ts

@@ -0,0 +1,146 @@
+import { D1Orm } from 'd1-orm';
+import { userAgentsModel, userTasksModel, UserAgentsTable } from '../models';
+
+/** AGENT MANAGEMENT */
+
+interface AgentResult {
+    success: boolean;
+    error?: string;
+    agent?: UserAgentsTable;
+    agents?: UserAgentsTable[];
+}
+
+/**
+ * Get all agents for a specific crew
+ * @param orm The D1Orm instance from durable object class
+ * @param crewId The ID of the crew to get agents for
+ * @returns Promise containing array of agents and metadata
+ * @throws Error if database query fails
+ */
+export async function getAgents(orm: D1Orm, crewId: number): Promise<AgentResult> {
+    try {
+        userAgentsModel.SetOrm(orm);
+        const result = await userAgentsModel.All({
+            where: {
+                crew_id: crewId
+            },
+            orderBy: [
+                {
+                    column: 'created_at',
+                    descending: true,
+                },
+            ],
+        });
+        return {
+            agents: result.results as unknown as UserAgentsTable[],
+            success: true
+        };
+    } catch (error) {
+        console.error(`Error in getAgents: ${error instanceof Error ? error.message : String(error)}`);
+        return {
+            success: false,
+            error: `Failed to get agents: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+
+/**
+ * Create a new agent
+ * @param orm The D1Orm instance from durable object class
+ * @param agentData The agent configuration data
+ * @returns Promise containing the created agent or error details
+ * @throws Error if database insertion fails
+ */
+export async function createAgent(
+    orm: D1Orm, 
+    agentData: Omit<UserAgentsTable, 'id' | 'created_at' | 'updated_at'>
+): Promise<AgentResult> {
+    try {
+        userAgentsModel.SetOrm(orm);
+        const agent = await userAgentsModel.InsertOne(agentData);
+        return {
+            agent: agent as unknown as UserAgentsTable,
+            success: true
+        };
+    } catch (error) {
+        console.error(`Error in createAgent: ${error instanceof Error ? error.message : String(error)}`);
+        return {
+            success: false,
+            error: `Failed to create agent: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+
+/**
+ * Update an existing agent's configuration
+ * @param orm The D1Orm instance from durable object class
+ * @param agentId The ID of the agent to update
+ * @param updates The fields to update on the agent
+ * @returns Promise containing the update result or error details
+ * @throws Error if database update fails
+ */
+export async function updateAgent(
+    orm: D1Orm,
+    agentId: number,
+    updates: Partial<Omit<UserAgentsTable, 'id' | 'created_at' | 'updated_at' | 'profile_id' | 'crew_id'>>
+): Promise<AgentResult> {
+    try {
+        userAgentsModel.SetOrm(orm);
+        await userAgentsModel.Update({
+            where: {
+                id: agentId
+            },
+            data: updates
+        });
+        return {
+            success: true
+        };
+    } catch (error) {
+        console.error(`Error in updateAgent: ${error instanceof Error ? error.message : String(error)}`);
+        return {
+            success: false,
+            error: `Failed to update agent: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+
+/**
+ * Delete an agent and all associated tasks
+ * @param orm The D1Orm instance from durable object class
+ * @param agentId The ID of the agent to delete
+ * @returns Promise containing the deletion result or error details
+ * @throws Error if database deletion fails
+ */
+export async function deleteAgent(orm: D1Orm, agentId: number): Promise<{
+    success: boolean;
+    error?: string;
+}> {
+    try {
+        userTasksModel.SetOrm(orm);
+        userAgentsModel.SetOrm(orm);
+        
+        // First delete all tasks associated with this agent
+        await userTasksModel.Delete({
+            where: {
+                agent_id: agentId
+            }
+        });
+        
+        // Then delete the agent
+        await userAgentsModel.Delete({
+            where: {
+                id: agentId
+            }
+        });
+        
+        return {
+            success: true
+        };
+    } catch (error) {
+        console.error(`Error in deleteAgent: ${error instanceof Error ? error.message : String(error)}`);
+        return {
+            success: false,
+            error: `Failed to delete agent: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}