kandev/apps/backend/internal/task/dto/dto.go

package dto

import (
	"time"

	"github.com/kandev/kandev/internal/task/models"
	"github.com/kandev/kandev/internal/task/service"
	v1 "github.com/kandev/kandev/pkg/api/v1"
)

type BoardDTO struct {
	ID          string    `json:"id"`
	WorkspaceID string    `json:"workspace_id"`
	Name        string    `json:"name"`
	Description *string   `json:"description,omitempty"`
	CreatedAt   time.Time `json:"created_at"`
	UpdatedAt   time.Time `json:"updated_at"`
}

type WorkspaceDTO struct {
	ID                    string    `json:"id"`
	Name                  string    `json:"name"`
	Description           *string   `json:"description,omitempty"`
	OwnerID               string    `json:"owner_id"`
	DefaultExecutorID     *string   `json:"default_executor_id,omitempty"`
	DefaultEnvironmentID  *string   `json:"default_environment_id,omitempty"`
	DefaultAgentProfileID *string   `json:"default_agent_profile_id,omitempty"`
	CreatedAt             time.Time `json:"created_at"`
	UpdatedAt             time.Time `json:"updated_at"`
}

type ColumnDTO struct {
	ID        string       `json:"id"`
	BoardID   string       `json:"board_id"`
	Name      string       `json:"name"`
	Position  int          `json:"position"`
	State     v1.TaskState `json:"state"`
	Color     string       `json:"color"`
	CreatedAt time.Time    `json:"created_at"`
	UpdatedAt time.Time    `json:"updated_at"`
}

type RepositoryDTO struct {
	ID             string    `json:"id"`
	WorkspaceID    string    `json:"workspace_id"`
	Name           string    `json:"name"`
	SourceType     string    `json:"source_type"`
	LocalPath      string    `json:"local_path,omitempty"`
	Provider       string    `json:"provider,omitempty"`
	ProviderRepoID string    `json:"provider_repo_id,omitempty"`
	ProviderOwner  string    `json:"provider_owner,omitempty"`
	ProviderName   string    `json:"provider_name,omitempty"`
	DefaultBranch  string    `json:"default_branch,omitempty"`
	SetupScript    string    `json:"setup_script,omitempty"`
	CleanupScript  string    `json:"cleanup_script,omitempty"`
	CreatedAt      time.Time `json:"created_at"`
	UpdatedAt      time.Time `json:"updated_at"`
}

type RepositoryScriptDTO struct {
	ID           string    `json:"id"`
	RepositoryID string    `json:"repository_id"`
	Name         string    `json:"name"`
	Command      string    `json:"command"`
	Position     int       `json:"position"`
	CreatedAt    time.Time `json:"created_at"`
	UpdatedAt    time.Time `json:"updated_at"`
}

type ExecutorDTO struct {
	ID        string                `json:"id"`
	Name      string                `json:"name"`
	Type      models.ExecutorType   `json:"type"`
	Status    models.ExecutorStatus `json:"status"`
	IsSystem  bool                  `json:"is_system"`
	Config    map[string]string     `json:"config,omitempty"`
	CreatedAt time.Time             `json:"created_at"`
	UpdatedAt time.Time             `json:"updated_at"`
}

type EnvironmentDTO struct {
	ID           string                 `json:"id"`
	Name         string                 `json:"name"`
	Kind         models.EnvironmentKind `json:"kind"`
	IsSystem     bool                   `json:"is_system"`
	WorktreeRoot string                 `json:"worktree_root,omitempty"`
	ImageTag     string                 `json:"image_tag,omitempty"`
	Dockerfile   string                 `json:"dockerfile,omitempty"`
	BuildConfig  map[string]string      `json:"build_config,omitempty"`
	CreatedAt    time.Time              `json:"created_at"`
	UpdatedAt    time.Time              `json:"updated_at"`
}

type TaskDTO struct {
	ID              string                   `json:"id"`
	WorkspaceID     string                   `json:"workspace_id"`
	BoardID         string                   `json:"board_id"`
	ColumnID        string                   `json:"column_id"`
	Title           string                   `json:"title"`
	Description     string                   `json:"description"`
	State           v1.TaskState             `json:"state"`
	Priority        int                      `json:"priority"`
	Repositories    []TaskRepositoryDTO      `json:"repositories,omitempty"`
	AssignedAgentID *string                  `json:"assigned_agent_id,omitempty"`
	Position        int                      `json:"position"`
	CreatedAt       time.Time                `json:"created_at"`
	UpdatedAt       time.Time                `json:"updated_at"`
	Metadata        map[string]interface{}   `json:"metadata,omitempty"`
	// Worktree information (populated if worktree exists for this task)
	WorktreePath   *string `json:"worktree_path,omitempty"`
	WorktreeBranch *string `json:"worktree_branch,omitempty"`
}

type TaskRepositoryDTO struct {
	ID           string                 `json:"id"`
	TaskID       string                 `json:"task_id"`
	RepositoryID string                 `json:"repository_id"`
	BaseBranch   string                 `json:"base_branch"`
	Position     int                    `json:"position"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
	CreatedAt    time.Time              `json:"created_at"`
	UpdatedAt    time.Time              `json:"updated_at"`
}

type TaskSessionDTO struct {
	ID                   string                 `json:"id"`
	TaskID               string                 `json:"task_id"`
	AgentInstanceID      string                 `json:"agent_instance_id,omitempty"`
	ContainerID          string                 `json:"container_id,omitempty"`
	AgentProfileID       string                 `json:"agent_profile_id,omitempty"`
	ExecutorID           string                 `json:"executor_id,omitempty"`
	EnvironmentID        string                 `json:"environment_id,omitempty"`
	RepositoryID         string                 `json:"repository_id,omitempty"`
	BaseBranch           string                 `json:"base_branch,omitempty"`
	WorktreeID           string                 `json:"worktree_id,omitempty"`
	WorktreePath         string                 `json:"worktree_path,omitempty"`
	WorktreeBranch       string                 `json:"worktree_branch,omitempty"`
	State                models.TaskSessionState `json:"state"`
	Progress             int                    `json:"progress"`
	ErrorMessage         string                 `json:"error_message,omitempty"`
	Metadata             map[string]interface{} `json:"metadata,omitempty"`
	AgentProfileSnapshot map[string]interface{} `json:"agent_profile_snapshot,omitempty"`
	ExecutorSnapshot     map[string]interface{} `json:"executor_snapshot,omitempty"`
	EnvironmentSnapshot  map[string]interface{} `json:"environment_snapshot,omitempty"`
	RepositorySnapshot   map[string]interface{} `json:"repository_snapshot,omitempty"`
	StartedAt            time.Time              `json:"started_at"`
	CompletedAt          *time.Time             `json:"completed_at,omitempty"`
	UpdatedAt            time.Time              `json:"updated_at"`
}

type ListTaskSessionsResponse struct {
	Sessions []TaskSessionDTO `json:"sessions"`
	Total    int              `json:"total"`
}

type BoardSnapshotDTO struct {
	Board   BoardDTO    `json:"board"`
	Columns []ColumnDTO `json:"columns"`
	Tasks   []TaskDTO   `json:"tasks"`
}

type ListMessagesResponse struct {
	Messages []*v1.Message `json:"messages"`
	Total    int           `json:"total"`
	HasMore  bool          `json:"has_more"`
	Cursor   string        `json:"cursor"`
}

type ListBoardsResponse struct {
	Boards []BoardDTO `json:"boards"`
	Total  int        `json:"total"`
}

type ListWorkspacesResponse struct {
	Workspaces []WorkspaceDTO `json:"workspaces"`
	Total      int            `json:"total"`
}

type ListColumnsResponse struct {
	Columns []ColumnDTO `json:"columns"`
	Total   int         `json:"total"`
}

type ListRepositoriesResponse struct {
	Repositories []RepositoryDTO `json:"repositories"`
	Total        int             `json:"total"`
}

type ListRepositoryScriptsResponse struct {
	Scripts []RepositoryScriptDTO `json:"scripts"`
	Total   int                   `json:"total"`
}

type ListExecutorsResponse struct {
	Executors []ExecutorDTO `json:"executors"`
	Total     int           `json:"total"`
}

type ListEnvironmentsResponse struct {
	Environments []EnvironmentDTO `json:"environments"`
	Total        int              `json:"total"`
}

type BranchDTO struct {
	Name   string `json:"name"`
	Type   string `json:"type"`   // "local" or "remote"
	Remote string `json:"remote"` // remote name (e.g., "origin") for remote branches
}

type RepositoryBranchesResponse struct {
	Branches []BranchDTO `json:"branches"`
	Total    int         `json:"total"`
}

type LocalRepositoryDTO struct {
	Path          string `json:"path"`
	Name          string `json:"name"`
	DefaultBranch string `json:"default_branch,omitempty"`
}

type RepositoryDiscoveryResponse struct {
	Roots        []string             `json:"roots"`
	Repositories []LocalRepositoryDTO `json:"repositories"`
	Total        int                  `json:"total"`
}

type RepositoryPathValidationResponse struct {
	Path          string `json:"path"`
	Exists        bool   `json:"exists"`
	IsGitRepo     bool   `json:"is_git"`
	Allowed       bool   `json:"allowed"`
	DefaultBranch string `json:"default_branch,omitempty"`
	Message       string `json:"message,omitempty"`
}

type ListTasksResponse struct {
	Tasks []TaskDTO `json:"tasks"`
	Total int       `json:"total"`
}

type SuccessResponse struct {
	Success bool `json:"success"`
}

func FromBoard(board *models.Board) BoardDTO {
	var description *string
	if board.Description != "" {
		description = &board.Description
	}

	return BoardDTO{
		ID:          board.ID,
		WorkspaceID: board.WorkspaceID,
		Name:        board.Name,
		Description: description,
		CreatedAt:   board.CreatedAt,
		UpdatedAt:   board.UpdatedAt,
	}
}

func FromWorkspace(workspace *models.Workspace) WorkspaceDTO {
	var description *string
	if workspace.Description != "" {
		description = &workspace.Description
	}

	return WorkspaceDTO{
		ID:                    workspace.ID,
		Name:                  workspace.Name,
		Description:           description,
		OwnerID:               workspace.OwnerID,
		DefaultExecutorID:     workspace.DefaultExecutorID,
		DefaultEnvironmentID:  workspace.DefaultEnvironmentID,
		DefaultAgentProfileID: workspace.DefaultAgentProfileID,
		CreatedAt:             workspace.CreatedAt,
		UpdatedAt:             workspace.UpdatedAt,
	}
}

func FromColumn(column *models.Column) ColumnDTO {
	return ColumnDTO{
		ID:        column.ID,
		BoardID:   column.BoardID,
		Name:      column.Name,
		Position:  column.Position,
		State:     column.State,
		Color:     column.Color,
		CreatedAt: column.CreatedAt,
		UpdatedAt: column.UpdatedAt,
	}
}

func FromRepository(repository *models.Repository) RepositoryDTO {
	return RepositoryDTO{
		ID:             repository.ID,
		WorkspaceID:    repository.WorkspaceID,
		Name:           repository.Name,
		SourceType:     repository.SourceType,
		LocalPath:      repository.LocalPath,
		Provider:       repository.Provider,
		ProviderRepoID: repository.ProviderRepoID,
		ProviderOwner:  repository.ProviderOwner,
		ProviderName:   repository.ProviderName,
		DefaultBranch:  repository.DefaultBranch,
		SetupScript:    repository.SetupScript,
		CleanupScript:  repository.CleanupScript,
		CreatedAt:      repository.CreatedAt,
		UpdatedAt:      repository.UpdatedAt,
	}
}

func FromRepositoryScript(script *models.RepositoryScript) RepositoryScriptDTO {
	return RepositoryScriptDTO{
		ID:           script.ID,
		RepositoryID: script.RepositoryID,
		Name:         script.Name,
		Command:      script.Command,
		Position:     script.Position,
		CreatedAt:    script.CreatedAt,
		UpdatedAt:    script.UpdatedAt,
	}
}

func FromExecutor(executor *models.Executor) ExecutorDTO {
	return ExecutorDTO{
		ID:        executor.ID,
		Name:      executor.Name,
		Type:      executor.Type,
		Status:    executor.Status,
		IsSystem:  executor.IsSystem,
		Config:    executor.Config,
		CreatedAt: executor.CreatedAt,
		UpdatedAt: executor.UpdatedAt,
	}
}

func FromEnvironment(environment *models.Environment) EnvironmentDTO {
	return EnvironmentDTO{
		ID:           environment.ID,
		Name:         environment.Name,
		Kind:         environment.Kind,
		IsSystem:     environment.IsSystem,
		WorktreeRoot: environment.WorktreeRoot,
		ImageTag:     environment.ImageTag,
		Dockerfile:   environment.Dockerfile,
		BuildConfig:  environment.BuildConfig,
		CreatedAt:    environment.CreatedAt,
		UpdatedAt:    environment.UpdatedAt,
	}
}

func FromLocalRepository(repo service.LocalRepository) LocalRepositoryDTO {
	return LocalRepositoryDTO{
		Path:          repo.Path,
		Name:          repo.Name,
		DefaultBranch: repo.DefaultBranch,
	}
}

func FromTask(task *models.Task) TaskDTO {
	var assignedAgentID *string
	if task.AssignedTo != "" {
		assignedAgentID = &task.AssignedTo
	}

	// Convert repositories
	var repositories []TaskRepositoryDTO
	for _, repo := range task.Repositories {
		repositories = append(repositories, TaskRepositoryDTO{
			ID:           repo.ID,
			TaskID:       repo.TaskID,
			RepositoryID: repo.RepositoryID,
			BaseBranch:   repo.BaseBranch,
			Position:     repo.Position,
			Metadata:     repo.Metadata,
			CreatedAt:    repo.CreatedAt,
			UpdatedAt:    repo.UpdatedAt,
		})
	}

	return TaskDTO{
		ID:              task.ID,
		WorkspaceID:     task.WorkspaceID,
		BoardID:         task.BoardID,
		ColumnID:        task.ColumnID,
		Title:           task.Title,
		Description:     task.Description,
		State:           task.State,
		Priority:        task.Priority,
		Repositories:    repositories,
		AssignedAgentID: assignedAgentID,
		Position:        task.Position,
		CreatedAt:       task.CreatedAt,
		UpdatedAt:       task.UpdatedAt,
		Metadata:        task.Metadata,
	}
}

func FromTaskSession(session *models.TaskSession) TaskSessionDTO {
	result := TaskSessionDTO{
		ID:                   session.ID,
		TaskID:               session.TaskID,
		AgentInstanceID:      session.AgentInstanceID,
		ContainerID:          session.ContainerID,
		AgentProfileID:       session.AgentProfileID,
		ExecutorID:           session.ExecutorID,
		EnvironmentID:        session.EnvironmentID,
		RepositoryID:         session.RepositoryID,
		BaseBranch:           session.BaseBranch,
		State:                session.State,
		Progress:             session.Progress,
		ErrorMessage:         session.ErrorMessage,
		Metadata:             session.Metadata,
		AgentProfileSnapshot: session.AgentProfileSnapshot,
		ExecutorSnapshot:     session.ExecutorSnapshot,
		EnvironmentSnapshot:  session.EnvironmentSnapshot,
		RepositorySnapshot:   session.RepositorySnapshot,
		StartedAt:            session.StartedAt,
		CompletedAt:          session.CompletedAt,
		UpdatedAt:            session.UpdatedAt,
	}
	if len(session.Worktrees) > 0 {
		result.WorktreeID = session.Worktrees[0].WorktreeID
		result.WorktreePath = session.Worktrees[0].WorktreePath
		result.WorktreeBranch = session.Worktrees[0].WorktreeBranch
	}
	return result
}