90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
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")
|
|
}
|