package updater

import (
	"fmt"
	"os"
	"path/filepath"

	"gitea.unprism.ru/KRBL/FemaInstaller/internal/ssh"
	"gitea.unprism.ru/KRBL/FemaInstaller/pkg/config"
	"gitea.unprism.ru/KRBL/FemaInstaller/pkg/fileutils"
)

// Updater handles the software update process
type Updater struct {
	Config     *config.UpdaterConfig
	BinaryData []byte
}

// NewUpdater creates a new updater with the provided configuration and binary data
func NewUpdater(config *config.UpdaterConfig, binaryData []byte) *Updater {
	return &Updater{
		Config:     config,
		BinaryData: binaryData,
	}
}

// Update performs the software update process
func (u *Updater) Update() error {
	// Save binary to temporary file
	tempFile := "update_binary"
	if err := os.WriteFile(tempFile, u.BinaryData, 0644); err != nil {
		return fmt.Errorf("не удалось сохранить временный файл: %w", err)
	}
	defer os.Remove(tempFile) // Clean up temporary file

	// Create SSH client configuration
	clientConfig := ssh.NewClientConfig(
		u.Config.IP,
		u.Config.Port,
		u.Config.Login,
		u.Config.Password,
	)

	// Connect to SSH server
	sshClient, err := ssh.CreateSSHClient(clientConfig)
	if err != nil {
		return fmt.Errorf("ошибка подключения SSH: %w", err)
	}
	defer sshClient.Close()

	// Create SFTP client
	sftpClient, err := ssh.CreateSFTPClient(clientConfig)
	if err != nil {
		return fmt.Errorf("ошибка подключения SFTP: %w", err)
	}
	defer sftpClient.Close()

	// Upload binary
	remotePath := "/root/fema/build.new"
	if err = fileutils.UploadFile(sftpClient, tempFile, remotePath); err != nil {
		return fmt.Errorf("ошибка загрузки файла: %w", err)
	}

	// Execute update commands
	commands := []string{
		"mv -f /root/fema/build.new /root/fema/build",
		"chmod +x /root/fema/build",
		"systemctl restart fema.service",
	}

	for _, cmd := range commands {
		if err := ssh.ExecuteCommand(sshClient, cmd); err != nil {
			return fmt.Errorf("ошибка выполнения команды '%s': %w", cmd, err)
		}
	}

	return nil
}

// GetConfigFilePath returns the path to the configuration file
func GetConfigFilePath() string {
	// Get executable directory
	execDir, err := filepath.Abs(filepath.Dir(os.Args[0]))
	if err != nil {
		execDir = "."
	}

	return filepath.Join(execDir, "updater_config.json")
}