gastown/internal/refinery/types.go

// Package refinery provides the merge queue processing agent.
package refinery

import (
	"errors"
	"fmt"
	"time"
)

// State represents the refinery's running state.
type State string

const (
	// StateStopped means the refinery is not running.
	StateStopped State = "stopped"

	// StateRunning means the refinery is actively processing.
	StateRunning State = "running"

	// StatePaused means the refinery is paused (not processing new items).
	StatePaused State = "paused"
)

// Refinery represents a rig's merge queue processor.
type Refinery struct {
	// RigName is the rig this refinery processes.
	RigName string `json:"rig_name"`

	// State is the current running state.
	State State `json:"state"`

	// PID is the process ID if running in background.
	PID int `json:"pid,omitempty"`

	// StartedAt is when the refinery was started.
	StartedAt *time.Time `json:"started_at,omitempty"`

	// CurrentMR is the merge request currently being processed.
	CurrentMR *MergeRequest `json:"current_mr,omitempty"`

	// PendingMRs tracks merge requests that have been submitted.
	// Key is the MR ID.
	PendingMRs map[string]*MergeRequest `json:"pending_mrs,omitempty"`

	// LastMergeAt is when the last successful merge happened.
	LastMergeAt *time.Time `json:"last_merge_at,omitempty"`

	// Stats contains cumulative statistics.
	Stats RefineryStats `json:"stats"`
}

// MergeRequest represents a branch waiting to be merged.
type MergeRequest struct {
	// ID is a unique identifier for this merge request.
	ID string `json:"id"`

	// Branch is the source branch name (e.g., "polecat/Toast/gt-abc").
	Branch string `json:"branch"`

	// Worker is the polecat that created this branch.
	Worker string `json:"worker"`

	// IssueID is the beads issue being worked on.
	IssueID string `json:"issue_id"`

	// SwarmID is the swarm this work belongs to (if any).
	SwarmID string `json:"swarm_id,omitempty"`

	// TargetBranch is where this should merge (usually integration or main).
	TargetBranch string `json:"target_branch"`

	// CreatedAt is when the MR was queued.
	CreatedAt time.Time `json:"created_at"`

	// Status is the current status of the merge request.
	Status MRStatus `json:"status"`

	// CloseReason indicates why the MR was closed (only set when Status=closed).
	CloseReason CloseReason `json:"close_reason,omitempty"`

	// Error contains error details if the MR failed.
	Error string `json:"error,omitempty"`
}

// MRStatus represents the status of a merge request.
// Uses beads-style statuses for consistency with the issue tracking system.
type MRStatus string

const (
	// MROpen means the MR is waiting to be processed or needs rework.
	MROpen MRStatus = "open"

	// MRInProgress means the MR is currently being merged by the Engineer.
	MRInProgress MRStatus = "in_progress"

	// MRClosed means the MR processing is complete (merged, rejected, etc).
	MRClosed MRStatus = "closed"
)

// CloseReason indicates why a merge request was closed.
type CloseReason string

const (
	// CloseReasonMerged means the MR was successfully merged.
	CloseReasonMerged CloseReason = "merged"

	// CloseReasonRejected means the MR was manually rejected.
	CloseReasonRejected CloseReason = "rejected"

	// CloseReasonConflict means the MR had unresolvable conflicts.
	CloseReasonConflict CloseReason = "conflict"

	// CloseReasonSuperseded means the MR was replaced by another.
	CloseReasonSuperseded CloseReason = "superseded"
)


// MergeConfig contains configuration for the merge process.
type MergeConfig struct {
	// RunTests controls whether tests are run after merge.
	// Default: true
	RunTests bool `json:"run_tests"`

	// TestCommand is the command to run for testing.
	// Default: "go test ./..."
	TestCommand string `json:"test_command"`

	// DeleteMergedBranches controls whether merged branches are deleted.
	// Default: true
	DeleteMergedBranches bool `json:"delete_merged_branches"`

	// PushRetryCount is the number of times to retry a failed push.
	// Default: 3
	PushRetryCount int `json:"push_retry_count"`

	// PushRetryDelayMs is the base delay between push retries in milliseconds.
	// Each retry doubles the delay (exponential backoff).
	// Default: 1000
	PushRetryDelayMs int `json:"push_retry_delay_ms"`
}

// DefaultMergeConfig returns the default merge configuration.
func DefaultMergeConfig() MergeConfig {
	return MergeConfig{
		RunTests:             true,
		TestCommand:          "go test ./...",
		DeleteMergedBranches: true,
		PushRetryCount:       3,
		PushRetryDelayMs:     1000,
	}
}

// RefineryStats contains cumulative refinery statistics.
type RefineryStats struct {
	// TotalMerged is the total number of successful merges.
	TotalMerged int `json:"total_merged"`

	// TotalFailed is the total number of failed merges.
	TotalFailed int `json:"total_failed"`

	// TotalSkipped is the total number of skipped MRs.
	TotalSkipped int `json:"total_skipped"`

	// TodayMerged is the number of merges today.
	TodayMerged int `json:"today_merged"`

	// TodayFailed is the number of failures today.
	TodayFailed int `json:"today_failed"`
}

// QueueItem represents an item in the merge queue for display.
type QueueItem struct {
	Position  int       `json:"position"`
	MR        *MergeRequest `json:"mr"`
	Age       string    `json:"age"`
}

// State transition errors.
var (
	// ErrInvalidTransition is returned when a state transition is not allowed.
	ErrInvalidTransition = errors.New("invalid state transition")

	// ErrClosedImmutable is returned when attempting to change a closed MR.
	ErrClosedImmutable = errors.New("closed merge requests are immutable")
)

// ValidateTransition checks if a state transition from -> to is valid.
//
// Valid transitions:
//   - open → in_progress (Engineer claims MR)
//   - in_progress → closed (merge success or rejection)
//   - in_progress → open (failure, reassign to worker)
//   - open → closed (manual rejection)
//
// Invalid:
//   - closed → anything (immutable once closed)
func ValidateTransition(from, to MRStatus) error {
	// Same state is always valid (no-op)
	if from == to {
		return nil
	}

	// Closed is immutable - cannot transition to anything else
	if from == MRClosed {
		return fmt.Errorf("%w: cannot change status from closed", ErrClosedImmutable)
	}

	// Check valid transitions
	switch from {
	case MROpen:
		// open → in_progress: Engineer claims MR
		// open → closed: manual rejection
		if to == MRInProgress || to == MRClosed {
			return nil
		}
	case MRInProgress:
		// in_progress → closed: merge success or rejection
		// in_progress → open: failure, reassign to worker
		if to == MRClosed || to == MROpen {
			return nil
		}
	}

	return fmt.Errorf("%w: %s → %s is not allowed", ErrInvalidTransition, from, to)
}

// SetStatus updates the MR status after validating the transition.
// Returns an error if the transition is not allowed.
func (mr *MergeRequest) SetStatus(newStatus MRStatus) error {
	if err := ValidateTransition(mr.Status, newStatus); err != nil {
		return err
	}
	mr.Status = newStatus
	return nil
}

// Close closes the MR with the given reason after validating the transition.
// Returns an error if the MR cannot be closed from its current state.
// Once closed, an MR cannot be closed again (even with a different reason).
func (mr *MergeRequest) Close(reason CloseReason) error {
	// Closed MRs are immutable - cannot be closed again
	if mr.Status == MRClosed {
		return fmt.Errorf("%w: MR is already closed", ErrClosedImmutable)
	}
	if err := ValidateTransition(mr.Status, MRClosed); err != nil {
		return err
	}
	mr.Status = MRClosed
	mr.CloseReason = reason
	return nil
}

// Reopen reopens a failed MR (transitions from in_progress back to open).
// Returns an error if the transition is not allowed.
func (mr *MergeRequest) Reopen() error {
	if mr.Status != MRInProgress {
		return fmt.Errorf("%w: can only reopen from in_progress, current status is %s",
			ErrInvalidTransition, mr.Status)
	}
	mr.Status = MROpen
	mr.CloseReason = "" // Clear any previous close reason
	return nil
}

// Claim transitions the MR from open to in_progress (Engineer claims it).
// Returns an error if the transition is not allowed.
func (mr *MergeRequest) Claim() error {
	if mr.Status != MROpen {
		return fmt.Errorf("%w: can only claim from open, current status is %s",
			ErrInvalidTransition, mr.Status)
	}
	mr.Status = MRInProgress
	return nil
}

// IsClosed returns true if the MR is in a closed state.
func (mr *MergeRequest) IsClosed() bool {
	return mr.Status == MRClosed
}

// FailureType categorizes merge failures for appropriate handling.
type FailureType string

const (
	// FailureNone indicates no failure (success).
	FailureNone FailureType = ""

	// FailureConflict indicates merge conflicts with target branch.
	FailureConflict FailureType = "conflict"

	// FailureTestsFail indicates tests failed after merge.
	FailureTestsFail FailureType = "tests_fail"

	// FailureBuildFail indicates build failed after merge.
	FailureBuildFail FailureType = "build_fail"

	// FailureFlakyTest indicates a potentially flaky test failure (may retry).
	FailureFlakyTest FailureType = "flaky_test"

	// FailurePushFail indicates push to remote failed.
	FailurePushFail FailureType = "push_fail"

	// FailureFetch indicates fetch of source branch failed.
	FailureFetch FailureType = "fetch_fail"

	// FailureCheckout indicates checkout of target branch failed.
	FailureCheckout FailureType = "checkout_fail"
)

// FailureLabel returns the beads label for this failure type.
func (f FailureType) FailureLabel() string {
	switch f {
	case FailureConflict:
		return "needs-rebase"
	case FailureTestsFail, FailureBuildFail, FailureFlakyTest:
		return "needs-fix"
	case FailurePushFail:
		return "needs-retry"
	default:
		return ""
	}
}

// ShouldAssignToWorker returns true if this failure should be assigned back to the worker.
func (f FailureType) ShouldAssignToWorker() bool {
	switch f {
	case FailureConflict, FailureTestsFail, FailureBuildFail, FailureFlakyTest:
		return true
	default:
		return false
	}
}

// IsOpen returns true if the MR is in an open state (waiting for processing).
func (mr *MergeRequest) IsOpen() bool {
	return mr.Status == MROpen
}

// IsInProgress returns true if the MR is currently being processed.
func (mr *MergeRequest) IsInProgress() bool {
	return mr.Status == MRInProgress
}