package platform

import (
	"fmt"
	"log"
	"strings"
	"time"
)

// WaitForHealthy waits for a Docker container to report healthy status.
func (s *Server) WaitForHealthy(container string, timeoutSecs int) error {
	log.Printf("   Waiting for %s to become healthy...", container)

	for i := 0; i < timeoutSecs; i += 5 {
		status, err := s.SSH("docker", "inspect", "--format={{.State.Health.Status}}", container)
		if err == nil {
			status = strings.TrimSpace(status)
			switch status {
			case "healthy":
				log.Printf("   %s is healthy", container)
				return nil
			case "unhealthy":
				logs, _ := s.SSH("docker", "logs", "--tail=20", container)
				return fmt.Errorf("container unhealthy, logs:\n%s", logs)
			}
		}
		time.Sleep(5 * time.Second)
	}

	return fmt.Errorf("health check timeout after %ds", timeoutSecs)
}

// Backup stops and renames the current container for rollback.
func (s *Server) Backup(name string) {
	backupName := name + "-backup"
	s.SSH("docker", "rm", "-f", backupName)
	s.SSH("docker", "stop", name)
	s.SSH("docker", "rename", name, backupName)
}

// Rollback restores the previous container version from backup.
func (s *Server) Rollback(name string) error {
	backupName := name + "-backup"

	if _, err := s.SSH("docker", "inspect", backupName); err != nil {
		log.Printf("   No backup found for %s, cannot rollback", name)
		return nil
	}

	log.Printf("   Rolling back %s...", name)

	s.SSH("docker", "rm", "-f", name)

	if _, err := s.SSH("docker", "rename", backupName, name); err != nil {
		return fmt.Errorf("rename backup: %w", err)
	}

	if _, err := s.SSH("docker", "start", name); err != nil {
		return fmt.Errorf("start backup: %w", err)
	}

	log.Printf("   Rolled back to previous version of %s", name)
	return nil
}