Added commands to manage groups

Author: RijulTP

groupmanagement.go

@@ -0,0 +1,26 @@
+package main
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/jackc/pgx/v5/pgxpool"
+)
+
+// DeleteGroupAndResults deletes a group by its ID and all associated results
+func DeleteGroupAndResults(ctx context.Context, db *pgxpool.Pool, groupID int) error {
+	// Delete associated results first
+	res1, err := db.Exec(ctx, `DELETE FROM dprompts_results WHERE group_id = $1`, groupID)
+	if err != nil {
+		return fmt.Errorf("failed to delete results: %w", err)
+	}
+
+	// Delete the group itself
+	res2, err := db.Exec(ctx, `DELETE FROM dprompt_groups WHERE id = $1`, groupID)
+	if err != nil {
+		return fmt.Errorf("failed to delete group: %w", err)
+	}
+
+	fmt.Printf("Deleted %d results and %d group(s) for group ID %d\n", res1.RowsAffected(), res2.RowsAffected(), groupID)
+	return nil
+}

main.go

@@ -22,7 +22,8 @@ func main() {
 
 	totalGroups := flag.Bool("total-groups", false, "Display total number of groups (view mode)")
 	groupID := flag.Int("group", 0, "Display results for a specific group ID (view mode)")
-	
+	deleteGroupID := flag.Int("delete-group-id", 0, "Delete a specific group ID and all its associated results")
+
 	n := flag.Int("n", 10, "Number of results to display (view mode)")
 
 	queueN := flag.Int("queue-n", 10, "Number of queued jobs to display (for view action)")
@@ -47,6 +48,14 @@ func main() {
 		RunClient(ctx, driver, *argsJSON, *metadataJSON, *bulkFile, dbPool)
 	case "worker":
 		RunWorker(ctx, driver, cancel, dbPool)
+	case "delete-group":
+		if *deleteGroupID == 0 {
+			log.Fatal().Msg("Please provide --delete-group-id")
+		}
+		if err := DeleteGroupAndResults(ctx, dbPool, *deleteGroupID); err != nil {
+			log.Fatal().Err(err).Msg("Failed to delete group and results")
+		}
+		log.Info().Int("group_id", *deleteGroupID).Msg("Deleted group and associated results")
 	case "view":
 		if *totalGroups {
 			if err := viewTotalGroups(ctx, dbPool); err != nil {