bzzz/github/integration.go

package github

import (
	"context"
	"fmt"
	"strings"
	"sync"
	"time"

	"chorus.services/bzzz/executor"
	"chorus.services/bzzz/logging"
	"chorus.services/bzzz/pkg/config"
	"chorus.services/bzzz/pkg/types"
	"chorus.services/bzzz/pubsub"
	"github.com/libp2p/go-libp2p/core/peer"
)

// IntegrationConfig holds configuration for GitHub integration
type IntegrationConfig struct {
	AgentID      string
	Capabilities []string
	PollInterval time.Duration
	MaxTasks     int
}

// Conversation represents a meta-discussion conversation
type Conversation struct {
	ID      string
	TaskID  int
	History []string
	Messages []string
}

// Integration handles dynamic repository discovery
type Integration struct {
	githubToken string
	pubsub *pubsub.PubSub
	hlog *logging.HypercoreLog
	ctx context.Context
	config *IntegrationConfig
	agentConfig *config.AgentConfig

	// Repository management
	repositories map[int]*RepositoryClient // projectID -> GitHub client
	repositoryLock sync.RWMutex

	// Conversation tracking
	activeDiscussions map[string]*Conversation // "projectID:taskID" -> conversation
	discussionLock sync.RWMutex
}

// RepositoryClient wraps a GitHub client for a specific repository
type RepositoryClient struct {
	Client     *Client
	Repository types.Repository
	LastSync   time.Time
}

// NewIntegration creates a new GitHub integration
func NewIntegration(ctx context.Context, githubToken string, ps *pubsub.PubSub, hlog *logging.HypercoreLog, config *IntegrationConfig, agentConfig *config.AgentConfig) *Integration {
	if config.PollInterval == 0 {
		config.PollInterval = 30 * time.Second
	}
	if config.MaxTasks == 0 {
		config.MaxTasks = 3
	}

	return &Integration{
		githubToken:       githubToken,
		pubsub:            ps,
		hlog:              hlog,
		ctx:               ctx,
		config:            config,
		agentConfig:       agentConfig,
		repositories:      make(map[int]*RepositoryClient),
		activeDiscussions: make(map[string]*Conversation),
	}
}

// Start begins the GitHub integration
func (hi *Integration) Start() {
	fmt.Printf("🔗 Starting GitHub integration for agent: %s\n", hi.config.AgentID)
	
	// Register the handler for incoming meta-discussion messages
	hi.pubsub.SetAntennaeMessageHandler(hi.handleMetaDiscussion)
	
	// Start task polling
	go hi.taskPollingLoop()
}

// repositoryDiscoveryLoop periodically discovers active repositories
func (hi *Integration) repositoryDiscoveryLoop() {
	// This functionality is now handled by WHOOSH
}

// syncRepositories synchronizes the list of active repositories
func (hi *Integration) syncRepositories() {
	// This functionality is now handled by WHOOSH
}

// taskPollingLoop periodically polls all repositories for available tasks
func (hi *Integration) taskPollingLoop() {
	ticker := time.NewTicker(hi.config.PollInterval)
	defer ticker.Stop()
	
	for {
		select {
		case <-hi.ctx.Done():
			return
		case <-ticker.C:
			hi.pollAllRepositories()
		}
	}
}

// pollAllRepositories checks all active repositories for available tasks
func (hi *Integration) pollAllRepositories() {
	hi.repositoryLock.RLock()
	repositories := make([]*RepositoryClient, 0, len(hi.repositories))
	for _, repo := range hi.repositories {
		repositories = append(repositories, repo)
	}
	hi.repositoryLock.RUnlock()
	
	if len(repositories) == 0 {
		return
	}
	
	fmt.Printf("🔍 Polling %d repositories for available tasks...\n", len(repositories))
	
	var allTasks []*types.EnhancedTask
	
	// Collect tasks from all repositories
	for _, repoClient := range repositories {
		tasks, err := hi.getRepositoryTasks(repoClient)
		if err != nil {
			fmt.Printf("❌ Failed to get tasks from %s/%s: %v\n", 
				repoClient.Repository.Owner, repoClient.Repository.Repository, err)
			continue
		}
		allTasks = append(allTasks, tasks...)
	}
	
	if len(allTasks) == 0 {
		return
	}
	
	fmt.Printf("📋 Found %d total available tasks across all repositories\n", len(allTasks))
	
	// Apply filtering and selection
	suitableTasks := hi.filterSuitableTasks(allTasks)
	if len(suitableTasks) == 0 {
		fmt.Printf("⚠️ No suitable tasks for agent capabilities: %v\n", hi.config.Capabilities)
		return
	}
	
	// Select and claim the highest priority task
	task := suitableTasks[0]
	hi.claimAndExecuteTask(task)
}

// getRepositoryTasks fetches available tasks from a specific repository
func (hi *Integration) getRepositoryTasks(repoClient *RepositoryClient) ([]*types.EnhancedTask, error) {
	// Get tasks from GitHub
	githubTasks, err := repoClient.Client.ListAvailableTasks()
	if err != nil {
		return nil, err
	}
	
	// Convert to enhanced tasks with project context
	var enhancedTasks []*types.EnhancedTask
	for _, task := range githubTasks {
		enhancedTask := &types.EnhancedTask{
			ID:          task.ID,
			Number:      task.Number,
			Title:       task.Title,
			Description: task.Description,
			State:       task.State,
			Labels:      task.Labels,
			Assignee:    task.Assignee,
			CreatedAt:   task.CreatedAt,
			UpdatedAt:   task.UpdatedAt,
			TaskType:    task.TaskType,
			Priority:    task.Priority,
			Requirements: task.Requirements,
			Deliverables: task.Deliverables,
			Context:     task.Context,
			ProjectID:  repoClient.Repository.ProjectID,
			GitURL:     repoClient.Repository.GitURL,
			Repository: repoClient.Repository,
		}
		enhancedTasks = append(enhancedTasks, enhancedTask)
	}
	
		return enhancedTasks, nil
}

// filterSuitableTasks filters tasks based on agent capabilities
func (hi *Integration) filterSuitableTasks(tasks []*types.EnhancedTask) []*types.EnhancedTask {
	var suitable []*types.EnhancedTask
	
	for _, task := range tasks {
		if hi.canHandleTaskType(task.TaskType) {
			suitable = append(suitable, task)
		}
	}
	
	return suitable
}

// canHandleTaskType checks if this agent can handle the given task type
func (hi *Integration) canHandleTaskType(taskType string) bool {
	for _, capability := range hi.config.Capabilities {
		if capability == taskType || capability == "general" || capability == "task-coordination" {
			return true
		}
	}
	return false
}

// claimAndExecuteTask claims a task and begins execution
func (hi *Integration) claimAndExecuteTask(task *types.EnhancedTask) {
	hi.repositoryLock.RLock()
	repoClient, exists := hi.repositories[task.ProjectID]
	hi.repositoryLock.RUnlock()
	
	if !exists {
		fmt.Printf("❌ Repository client not found for project %d\n", task.ProjectID)
		return
	}
	
	// Claim the task in GitHub
	_, err := repoClient.Client.ClaimTask(task.Number, hi.config.AgentID)
	if err != nil {
		fmt.Printf("❌ Failed to claim task %d in %s/%s: %v\n", 
			task.Number, task.Repository.Owner, task.Repository.Repository, err)
		return
	}
	
	fmt.Printf("✋ Claimed task #%d from %s/%s: %s\n", 
		task.Number, task.Repository.Owner, task.Repository.Repository, task.Title)
	
	// Log the claim
	hi.hlog.Append(logging.TaskClaimed, map[string]interface{}{
		"task_id":    task.Number,
		"repository": fmt.Sprintf("%s/%s", task.Repository.Owner, task.Repository.Repository),
		"title":      task.Title,
	})

	// Start task execution
	go hi.executeTask(task, repoClient)
}

// executeTask executes a claimed task with reasoning and coordination
func (hi *Integration) executeTask(task *types.EnhancedTask, repoClient *RepositoryClient) {
	// Define the dynamic topic for this task
	taskTopic := fmt.Sprintf("bzzz/meta/issue/%d", task.Number)
	hi.pubsub.JoinDynamicTopic(taskTopic)
	defer hi.pubsub.LeaveDynamicTopic(taskTopic)

	fmt.Printf("🚀 Starting execution of task #%d in sandbox...\n", task.Number)

	// The executor now handles the entire iterative process.
	result, err := executor.ExecuteTask(hi.ctx, task, hi.hlog, hi.agentConfig)
	if err != nil {
		fmt.Printf("❌ Failed to execute task #%d: %v\n", task.Number, err)
		hi.hlog.Append(logging.TaskFailed, map[string]interface{}{"task_id": task.Number, "reason": "task execution failed in sandbox"})
		return
	}

	// Ensure sandbox cleanup happens regardless of PR creation success/failure
	defer result.Sandbox.DestroySandbox()

	// Create a pull request
	pr, err := repoClient.Client.CreatePullRequest(task.Number, result.BranchName, hi.config.AgentID)
	if err != nil {
		fmt.Printf("❌ Failed to create pull request for task #%d: %v\n", task.Number, err)
		fmt.Printf("📝 Note: Branch '%s' has been pushed to repository and work is preserved\n", result.BranchName)
		
		// Escalate PR creation failure to humans via N8N webhook
		escalationReason := fmt.Sprintf("Failed to create pull request: %v. Task execution completed successfully and work is preserved in branch '%s', but PR creation failed.", err, result.BranchName)
		hi.requestAssistance(task, escalationReason, fmt.Sprintf("bzzz/meta/issue/%d", task.Number))
		
		hi.hlog.Append(logging.TaskFailed, map[string]interface{}{
			"task_id": task.Number, 
			"reason": "failed to create pull request",
			"branch_name": result.BranchName,
			"work_preserved": true,
			"escalated": true,
		})
		return
	}

	fmt.Printf("✅ Successfully created pull request for task #%d: %s\n", task.Number, pr.GetHTMLURL())
	hi.hlog.Append(logging.TaskCompleted, map[string]interface{}{
		"task_id":   task.Number,
		"pr_url":    pr.GetHTMLURL(),
		"pr_number": pr.GetNumber(),
	})
}

// requestAssistance publishes a help request to the task-specific topic.
func (hi *Integration) requestAssistance(task *types.EnhancedTask, reason, topic string) {
	fmt.Printf("🆘 Agent %s is requesting assistance for task #%d: %s\n", hi.config.AgentID, task.Number, reason)
	hi.hlog.Append(logging.TaskHelpRequested, map[string]interface{}{
		"task_id": task.Number,
		"reason":  reason,
	})

	helpRequest := map[string]interface{}{
		"issue_id":   task.Number,
		"repository": fmt.Sprintf("%s/%s", task.Repository.Owner, task.Repository.Repository),
		"reason":     reason,
	}

	hi.pubsub.PublishToDynamicTopic(topic, pubsub.TaskHelpRequest, helpRequest)
}

// handleMetaDiscussion handles all incoming messages from dynamic and static topics.
func (hi *Integration) handleMetaDiscussion(msg pubsub.Message, from peer.ID) {
	switch msg.Type {
	case pubsub.TaskHelpRequest:
		hi.handleHelpRequest(msg, from)
	case pubsub.TaskHelpResponse:
		hi.handleHelpResponse(msg, from)
	default:
		// Handle other meta-discussion messages (e.g., peer feedback)
	}
}

// handleHelpRequest is called when another agent requests assistance.
func (hi *Integration) handleHelpRequest(msg pubsub.Message, from peer.ID) {
	issueID, _ := msg.Data["issue_id"].(float64)
	reason, _ := msg.Data["reason"].(string)
	fmt.Printf("🙋 Received help request for task #%d from %s: %s\n", int(issueID), from.ShortString(), reason)

	// Simple logic: if we are not busy, we can help.
	// TODO: A more advanced agent would check its capabilities against the reason.
	canHelp := true // Placeholder for more complex logic

	if canHelp {
		fmt.Printf("✅ Agent %s can help with task #%d\n", hi.config.AgentID, int(issueID))
		hi.hlog.Append(logging.TaskHelpOffered, map[string]interface{}{
			"task_id":      int(issueID),
			"requester_id": from.ShortString(),
		})

		response := map[string]interface{}{
			"issue_id":     issueID,
			"can_help":     true,
			"capabilities": hi.config.Capabilities,
		}
		taskTopic := fmt.Sprintf("bzzz/meta/issue/%d", int(issueID))
		hi.pubsub.PublishToDynamicTopic(taskTopic, pubsub.TaskHelpResponse, response)
	}
}

// handleHelpResponse is called when an agent receives an offer for help.
func (hi *Integration) handleHelpResponse(msg pubsub.Message, from peer.ID) {
	issueID, _ := msg.Data["issue_id"].(float64)
	canHelp, _ := msg.Data["can_help"].(bool)

	if canHelp {
		fmt.Printf("🤝 Received help offer for task #%d from %s\n", int(issueID), from.ShortString())
		hi.hlog.Append(logging.TaskHelpReceived, map[string]interface{}{
			"task_id":   int(issueID),
			"helper_id": from.ShortString(),
		})
		// In a full implementation, the agent would now delegate a sub-task
		// or use the helper's capabilities. For now, we just log it.
	}
}

// shouldEscalate determines if a task needs human intervention
func (hi *Integration) shouldEscalate(response string, history []string) bool {
	// Check for escalation keywords
	lowerResponse := strings.ToLower(response)
	keywords := []string{"stuck", "help", "human", "escalate", "clarification needed", "manual intervention"}
	
	for _, keyword := range keywords {
		if strings.Contains(lowerResponse, keyword) {
			return true
		}
	}
	
	// Check conversation length
	if len(history) >= 10 {
		return true
	}
	
	return false
}

// triggerHumanEscalation sends escalation to N8N
func (hi *Integration) triggerHumanEscalation(projectID int, convo *Conversation, reason string) {
	hi.hlog.Append(logging.Escalation, map[string]interface{}{
		"task_id": convo.TaskID,
		"reason":  reason,
	})
	
	fmt.Printf("✅ Task #%d in project %d escalated for human intervention\n", convo.TaskID, projectID)
}