chore(repo): reinitialize repository

2026-03-18 11:29:54 +08:00
commit 24871e213a
288 changed files with 44369 additions and 0 deletions
@@ -0,0 +1,120 @@
+package workflowrun
+
+import (
+	"context"
+	"fmt"
+
+	"inbox/internal/app/runtimeconfig"
+	"inbox/internal/base/timeutil"
+	"inbox/internal/domain/workflow"
+)
+
+type Repository interface {
+	CreateWorkflowRun(ctx context.Context, value workflow.Run) (workflow.Run, error)
+	GetWorkflowRun(ctx context.Context, runID string) (workflow.Run, error)
+	UpdateWorkflowRun(ctx context.Context, value workflow.Run) (workflow.Run, error)
+	ListWorkflowRunsByTopic(ctx context.Context, topicID string) ([]workflow.Run, error)
+	ListWorkflowRunLogs(ctx context.Context, runID string, afterSeq int) ([]workflow.RunLog, error)
+	AppendWorkflowRunLog(ctx context.Context, value workflow.RunLog) (workflow.RunLog, error)
+}
+
+type RuntimeResolver interface {
+	ResolveRole(ctx context.Context, workspaceID, roleName string) (runtimeconfig.ResolvedRole, error)
+}
+
+type Service struct {
+	repo     Repository
+	resolver RuntimeResolver
+	clock    timeutil.Clock
+}
+
+type Patch struct {
+	Status         *workflow.RunStatus
+	ReplyMessageID *string
+	ExitCode       *int
+	CompletedAt    *string
+	ErrorMessage   *string
+	CommandJSON    *string
+}
+
+func NewService(repo Repository, resolver RuntimeResolver, clock timeutil.Clock) *Service {
+	if clock == nil {
+		clock = timeutil.SystemClock{}
+	}
+	return &Service{
+		repo:     repo,
+		resolver: resolver,
+		clock:    clock,
+	}
+}
+
+func (s *Service) Start(ctx context.Context, run workflow.Run) (workflow.Run, error) {
+	resolved, err := s.resolver.ResolveRole(ctx, run.WorkspaceID, run.RoleName)
+	if err != nil {
+		return workflow.Run{}, fmt.Errorf("resolve runtime config for %s: %w", run.RoleName, err)
+	}
+	snapshot, err := resolved.SnapshotJSON()
+	if err != nil {
+		return workflow.Run{}, err
+	}
+	run.ConfigSnapshotJSON = snapshot
+	if run.Status == "" {
+		run.Status = workflow.RunStatusRunning
+	}
+	return s.repo.CreateWorkflowRun(ctx, run)
+}
+
+func (s *Service) ListByTopic(ctx context.Context, topicID string) ([]workflow.Run, error) {
+	return s.repo.ListWorkflowRunsByTopic(ctx, topicID)
+}
+
+func (s *Service) Get(ctx context.Context, runID string) (workflow.Run, error) {
+	return s.repo.GetWorkflowRun(ctx, runID)
+}
+
+func (s *Service) Update(ctx context.Context, value workflow.Run) (workflow.Run, error) {
+	if value.Status != workflow.RunStatusRunning && value.CompletedAt == "" {
+		value.CompletedAt = timeutil.FormatRFC3339(s.clock.Now())
+	}
+	return s.repo.UpdateWorkflowRun(ctx, value)
+}
+
+func (s *Service) Patch(ctx context.Context, runID string, patch Patch) (workflow.Run, error) {
+	current, err := s.repo.GetWorkflowRun(ctx, runID)
+	if err != nil {
+		return workflow.Run{}, err
+	}
+	if patch.Status != nil {
+		current.Status = *patch.Status
+	}
+	if patch.ReplyMessageID != nil {
+		current.ReplyMessageID = *patch.ReplyMessageID
+	}
+	if patch.ExitCode != nil {
+		current.ExitCode = *patch.ExitCode
+	}
+	if patch.CompletedAt != nil {
+		current.CompletedAt = *patch.CompletedAt
+	}
+	if patch.ErrorMessage != nil {
+		current.ErrorMessage = *patch.ErrorMessage
+	}
+	if patch.CommandJSON != nil {
+		current.CommandJSON = *patch.CommandJSON
+	}
+	return s.Update(ctx, current)
+}
+
+func (s *Service) ListLogs(ctx context.Context, runID string, afterSeq int) ([]workflow.RunLog, error) {
+	if _, err := s.repo.GetWorkflowRun(ctx, runID); err != nil {
+		return nil, err
+	}
+	return s.repo.ListWorkflowRunLogs(ctx, runID, afterSeq)
+}
+
+func (s *Service) AppendLog(ctx context.Context, value workflow.RunLog) (workflow.RunLog, error) {
+	if _, err := s.repo.GetWorkflowRun(ctx, value.RunID); err != nil {
+		return workflow.RunLog{}, err
+	}
+	return s.repo.AppendWorkflowRunLog(ctx, value)
+}