Add server-side workflows with YAML files, task freeze and lead tools

2026-10-15 21:17:58 -07:00
parent 2f02f5ac3c
commit 2ce10a6c65
19 changed files with 1293 additions and 199 deletions
@@ -340,6 +340,13 @@ INSERT INTO reviews_new SELECT * FROM reviews;
 DROP TABLE reviews;
 ALTER TABLE reviews_new RENAME TO reviews;
 CREATE UNIQUE INDEX reviews_session ON reviews(agent_session_id);
+`}, {"add-workflows", `
+CREATE TABLE workflows (
+	name TEXT PRIMARY KEY,
+	definition TEXT NOT NULL
+);
+ALTER TABLE tasks ADD COLUMN workflow_name TEXT NOT NULL DEFAULT '';
+ALTER TABLE tasks ADD COLUMN workflow_definition TEXT NOT NULL DEFAULT '';
 `}}

 const (
@@ -605,7 +612,18 @@ func (s *Store) GetProject(ctx context.Context, id string) (api.Project, error)
 	return p, err
 }

-func (s *Store) CreateTask(ctx context.Context, t api.Task, actor string) error {
+func (s *Store) CreateTask(ctx context.Context, t api.Task, frozen *FrozenWorkflow, actor string) error {
+	workflowName := ""
+	workflowDefinition := ""
+	if frozen != nil {
+		b, err := json.Marshal(frozen.Definition)
+		if err != nil {
+			return err
+		}
+		workflowName = frozen.Name
+		workflowDefinition = string(b)
+		t.Workflow = &workflowName
+	}
 	todoItem := ""
 	if t.TodoItem != nil {
 		todoItem = *t.TodoItem
@@ -642,9 +660,9 @@ func (s *Store) CreateTask(ctx context.Context, t api.Task, actor string) error
 			return err
 		}
 		_, err = tx.ExecContext(ctx, `
-			INSERT INTO tasks (id, project_id, title, merge_title, branch, worktree_path, base_ref, port_base, color, status, created_by, created_at, todo_item, forked_from, forked_at)
-			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-		`, t.Id, t.ProjectId, t.Title, mergeTitle, t.Branch, t.WorktreePath, t.BaseRef, t.PortBase, t.Color, string(t.Status), t.CreatedBy, formatTime(t.CreatedAt), todoItem, forkedFrom, forkedAt)
+			INSERT INTO tasks (id, project_id, title, merge_title, branch, worktree_path, base_ref, port_base, color, status, created_by, created_at, todo_item, forked_from, forked_at, workflow_name, workflow_definition)
+			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+		`, t.Id, t.ProjectId, t.Title, mergeTitle, t.Branch, t.WorktreePath, t.BaseRef, t.PortBase, t.Color, string(t.Status), t.CreatedBy, formatTime(t.CreatedAt), todoItem, forkedFrom, forkedAt, workflowName, workflowDefinition)
 		if err != nil {
 			return err
 		}
@@ -1230,7 +1248,7 @@ func (s *Store) write(ctx context.Context, fn func(tx *sql.Tx, emit emitFunc) er

 const projectColumns = `id, name, repo_path, default_branch, remote, created_by, created_at`

-const taskColumns = `id, project_id, title, merge_title, branch, worktree_path, base_ref, port_base, color, status, created_by, created_at, closed_at, merge_commit, todo_item, forked_from, forked_at, merge_queued_at`
+const taskColumns = `id, project_id, title, merge_title, branch, worktree_path, base_ref, port_base, color, status, created_by, created_at, closed_at, merge_commit, todo_item, forked_from, forked_at, merge_queued_at, workflow_name`

 const sessionColumns = `id, coalesce(task_id, ''), owner, vendor, model, vendor_session_id, state, role, created_by, created_at, forked_from, forked_at`

@@ -1259,10 +1277,14 @@ func scanTask(row scanner) (api.Task, error) {
 	forkedFrom := ""
 	forkedAt := 0
 	mergeQueuedAt := ""
-	err := row.Scan(&t.Id, &t.ProjectId, &t.Title, &t.MergeTitle, &t.Branch, &t.WorktreePath, &t.BaseRef, &t.PortBase, &t.Color, &status, &t.CreatedBy, &createdAt, &closedAt, &mergeCommit, &todoItem, &forkedFrom, &forkedAt, &mergeQueuedAt)
+	workflowName := ""
+	err := row.Scan(&t.Id, &t.ProjectId, &t.Title, &t.MergeTitle, &t.Branch, &t.WorktreePath, &t.BaseRef, &t.PortBase, &t.Color, &status, &t.CreatedBy, &createdAt, &closedAt, &mergeCommit, &todoItem, &forkedFrom, &forkedAt, &mergeQueuedAt, &workflowName)
 	if err != nil {
 		return t, err
 	}
+	if workflowName != "" {
+		t.Workflow = &workflowName
+	}
 	if t.MergeTitle == "" {
 		t.MergeTitle = t.Title
 	}
@@ -0,0 +1,107 @@
+package store
+
+import (
+	"context"
+	"database/sql"
+	"encoding/json"
+	"errors"
+	"fmt"
+
+	"g.fc.run/misc/byre/internal/api"
+)
+
+type FrozenWorkflow struct {
+	Name       string
+	Definition api.WorkflowDefinition
+}
+
+func (s *Store) ListWorkflows(ctx context.Context) ([]api.WorkflowSummary, error) {
+	rows, err := s.db.QueryContext(ctx, `SELECT name, coalesce(json_extract(definition, '$.description'), '') FROM workflows ORDER BY name`)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	workflows := []api.WorkflowSummary{}
+	for rows.Next() {
+		w := api.WorkflowSummary{}
+		err = rows.Scan(&w.Name, &w.Description)
+		if err != nil {
+			return nil, err
+		}
+		workflows = append(workflows, w)
+	}
+	return workflows, rows.Err()
+}
+
+func (s *Store) GetWorkflow(ctx context.Context, name string) (api.WorkflowDefinition, error) {
+	def := ""
+	err := s.db.QueryRowContext(ctx, `SELECT definition FROM workflows WHERE name = ?`, name).Scan(&def)
+	if errors.Is(err, sql.ErrNoRows) {
+		return api.WorkflowDefinition{}, fmt.Errorf("%w: no workflow %q", ErrNotFound, name)
+	}
+	if err != nil {
+		return api.WorkflowDefinition{}, err
+	}
+	w := api.WorkflowDefinition{}
+	err = json.Unmarshal([]byte(def), &w)
+	return w, err
+}
+
+func (s *Store) PutWorkflow(ctx context.Context, name string, def api.WorkflowDefinition, actor string) error {
+	b, err := json.Marshal(def)
+	if err != nil {
+		return err
+	}
+	return s.write(ctx, func(tx *sql.Tx, emit emitFunc) error {
+		_, err := tx.ExecContext(ctx, `INSERT INTO workflows (name, definition) VALUES (?, ?)
+			ON CONFLICT (name) DO UPDATE SET definition = excluded.definition`, name, string(b))
+		if err != nil {
+			return err
+		}
+		return emit(api.Event{Kind: "workflow.updated", Actor: actor, Payload: map[string]any{"name": name}})
+	})
+}
+
+func (s *Store) DeleteWorkflow(ctx context.Context, name, actor string) error {
+	return s.write(ctx, func(tx *sql.Tx, emit emitFunc) error {
+		res, err := tx.ExecContext(ctx, `DELETE FROM workflows WHERE name = ?`, name)
+		if err != nil {
+			return err
+		}
+		n, err := res.RowsAffected()
+		if err != nil {
+			return err
+		}
+		if n == 0 {
+			return fmt.Errorf("%w: no workflow %q", ErrNotFound, name)
+		}
+		return emit(api.Event{Kind: "workflow.deleted", Actor: actor, Payload: map[string]any{"name": name}})
+	})
+}
+
+func (s *Store) FreezeWorkflow(ctx context.Context, name string) (FrozenWorkflow, error) {
+	def, err := s.GetWorkflow(ctx, name)
+	if err != nil {
+		return FrozenWorkflow{}, err
+	}
+	return FrozenWorkflow{Name: name, Definition: def}, nil
+}
+
+func (s *Store) TaskWorkflow(ctx context.Context, taskID string) (FrozenWorkflow, error) {
+	name := ""
+	def := ""
+	err := s.db.QueryRowContext(ctx, `SELECT workflow_name, workflow_definition FROM tasks WHERE id = ?`, taskID).Scan(&name, &def)
+	if errors.Is(err, sql.ErrNoRows) {
+		return FrozenWorkflow{}, fmt.Errorf("%w: no task %s", ErrNotFound, taskID)
+	}
+	if err != nil {
+		return FrozenWorkflow{}, err
+	}
+	if name == "" {
+		return FrozenWorkflow{}, fmt.Errorf("%w: task %s has no workflow", ErrNotFound, taskID)
+	}
+	f := FrozenWorkflow{Name: name}
+	err = json.Unmarshal([]byte(def), &f.Definition)
+	return f, err
+}