package service

import (
	"log/slog"

	"codespace/internal/process"
	"codespace/internal/shell"
	"codespace/internal/workspace"
)

// WorkspaceService orchestrates workspace lifecycle operations.
type WorkspaceService struct {
	workspaces workspace.Manager
	processes  process.Manager
	shells     shell.Manager
	logger     *slog.Logger
}

// NewWorkspaceService creates a WorkspaceService.
// If lg is nil, slog.Default() is used.
func NewWorkspaceService(workspaces workspace.Manager, processes process.Manager, shells shell.Manager, lg *slog.Logger) *WorkspaceService {
	if lg == nil {
		lg = slog.Default()
	}
	return &WorkspaceService{workspaces: workspaces, processes: processes, shells: shells, logger: lg}
}

// Create creates a new workspace.
func (s *WorkspaceService) Create(id string) (*workspace.Workspace, error) {
	ws, err := s.workspaces.Create(id)
	if err != nil {
		return nil, err
	}
	s.logger.Info("workspace created", "workspace_id", id)

	if err := s.shells.Start(ws.ID, ws.Root); err != nil {
		s.logger.Warn("failed to auto-start shell for workspace", "workspace_id", id, "error", err)
	}

	return ws, nil
}

// Get retrieves a workspace by id.
func (s *WorkspaceService) Get(id string) (*workspace.Workspace, error) {
	return s.workspaces.Get(id)
}

// List returns all workspaces sorted by ID.
func (s *WorkspaceService) List() ([]workspace.Workspace, error) {
	return s.workspaces.List()
}

// Delete stops any running process for the workspace, then removes it.
func (s *WorkspaceService) Delete(id string) error {
	if err := s.shells.Stop(id); err != nil {
		s.logger.Warn("failed to stop workspace shell before delete", "workspace_id", id, "error", err)
	}

	status := s.processes.Status(id)
	if status.Running {
		if err := s.processes.Stop(id); err != nil {
			s.logger.Error("failed to stop workspace process before delete", "workspace_id", id, "error", err)
		}
	}
	if err := s.workspaces.Delete(id); err != nil {
		return err
	}
	s.logger.Info("workspace deleted", "workspace_id", id)
	return nil
}