Merge pull request #23 from aibtcdev/feat/more-endpoints

Add requested endpoints

Author: whoabuddy

src/database/aibtcdev.sql

@@ -303,6 +303,7 @@ CREATE TABLE user_crew_execution_steps (
 CREATE INDEX idx_execution_steps_created_at ON user_crew_execution_steps(created_at); 
 CREATE INDEX idx_execution_steps_profile_id ON user_crew_execution_steps(profile_id);
 CREATE INDEX idx_execution_steps_crew_id ON user_crew_execution_steps(crew_id);
+CREATE INDEX idx_execution_steps_profile_id ON user_crew_execution_steps(profile_id);
 CREATE INDEX idx_execution_steps_execution_id ON user_crew_execution_steps(execution_id);
 CREATE INDEX idx_execution_steps_type ON user_crew_execution_steps(step_type);
 

src/database/helpers/crews.ts

@@ -238,3 +238,124 @@ export async function deleteCrew(orm: D1Orm, crewId: number): Promise<CrewResult
         };
     }
 }
+
+/**
+ * Get all crews for a profile
+ * @param orm The D1Orm instance from durable object class
+ * @param address The Stacks address for the user's profile
+ * @returns Promise containing array of crews or error details
+ * @throws Error if database query fails
+ */
+/**
+ * Get all execution steps for a specific execution
+ * @param orm The D1Orm instance from durable object class
+ * @param executionId The ID of the execution to get steps for
+ * @returns Promise containing array of execution steps or error details
+ */
+export async function getExecutionSteps(orm: D1Orm, executionId: number): Promise<CrewResult> {
+    try {
+        userCrewExecutionStepsModel.SetOrm(orm);
+        const result = await userCrewExecutionStepsModel.All({
+            where: {
+                execution_id: executionId
+            },
+            orderBy: [
+                {
+                    column: 'created_at',
+                    ascending: true,
+                },
+            ],
+        });
+        return {
+            steps: result.results as unknown as UserCrewExecutionStepsTable[],
+            success: true
+        };
+    } catch (error) {
+        console.error(`Error in getExecutionSteps: ${error instanceof Error ? error.message : String(error)}`);
+        return {
+            steps: [],
+            success: false,
+            error: `Failed to get execution steps: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+
+/**
+ * Create a new execution step
+ * @param orm The D1Orm instance from durable object class
+ * @param stepData The step data to create
+ * @returns Promise containing the created step or error details
+ */
+export async function createExecutionStep(
+    orm: D1Orm, 
+    stepData: Omit<UserCrewExecutionStepsTable, 'id' | 'created_at' | 'updated_at'>
+): Promise<CrewResult> {
+    try {
+        userCrewExecutionStepsModel.SetOrm(orm);
+        const step = await userCrewExecutionStepsModel.InsertOne(stepData);
+        return {
+            step: step as unknown as UserCrewExecutionStepsTable,
+            success: true
+        };
+    } catch (error) {
+        console.error(`Error in createExecutionStep: ${error instanceof Error ? error.message : String(error)}`);
+        return {
+            success: false,
+            error: `Failed to create execution step: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+
+/**
+ * Delete all execution steps for a specific execution
+ * @param orm The D1Orm instance from durable object class
+ * @param executionId The ID of the execution to delete steps for
+ * @returns Promise containing the deletion result or error details
+ */
+export async function deleteExecutionSteps(orm: D1Orm, executionId: number): Promise<CrewResult> {
+    try {
+        userCrewExecutionStepsModel.SetOrm(orm);
+        await userCrewExecutionStepsModel.Delete({
+            where: {
+                execution_id: executionId
+            }
+        });
+        return {
+            success: true
+        };
+    } catch (error) {
+        console.error(`Error in deleteExecutionSteps: ${error instanceof Error ? error.message : String(error)}`);
+        return {
+            success: false,
+            error: `Failed to delete execution steps: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+
+export async function getCrewsByProfile(orm: D1Orm, address: string): Promise<CrewResult> {
+    try {
+        userCrewsModel.SetOrm(orm);
+        const result = await userCrewsModel.All({
+            where: {
+                profile_id: address
+            },
+            orderBy: [
+                {
+                    column: 'created_at',
+                    descending: true,
+                },
+            ],
+        });
+        return {
+            crews: result.results as unknown as UserCrewsTable[],
+            success: true
+        };
+    } catch (error) {
+        console.error(`Error in getCrewsByProfile: ${error instanceof Error ? error.message : String(error)}`);
+        return {
+            crews: [],
+            success: false,
+            error: `Failed to get crews for profile: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}

src/database/models/UserCrewExecutionSteps.ts

@@ -10,6 +10,7 @@ export const userCrewExecutionStepsModel = new Model(
 	{
 		id: { type: DataTypes.INTEGER, notNull: true },
 		created_at: { type: DataTypes.STRING },
+		profile_id: { type: DataTypes.STRING, notNull: true },
 		crew_id: { type: DataTypes.INTEGER, notNull: true },
 		execution_id: { type: DataTypes.INTEGER, notNull: true },
 		step_type: { type: DataTypes.STRING, notNull: true },

src/database/models/UserCrons.ts

@@ -13,7 +13,7 @@ export const userCronsModel = new Model(
 		updated_at: { type: DataTypes.STRING },
 		profile_id: { type: DataTypes.STRING, notNull: true },
 		crew_id: { type: DataTypes.INTEGER, notNull: true },
-		cron_enabled: { type: DataTypes.INTEGER, notNull: true },
+		cron_enabled: { type: DataTypes.BOOLEAN, notNull: true },
 		cron_interval: { type: DataTypes.STRING, notNull: true },
 		cron_input: { type: DataTypes.STRING, notNull: true },
 	}

src/database/models/XBotTweets.ts

@@ -18,7 +18,7 @@ export const xBotTweetsModel = new Model(
         tweet_created_at: { type: DataTypes.STRING },
         tweet_updated_at: { type: DataTypes.STRING },
         tweet_body: { type: DataTypes.STRING },
-        is_bot_response: { type: DataTypes.INTEGER },
+        is_bot_response: { type: DataTypes.BOOLEAN },
     }
 );