/*
 * ALR - Any Linux Repository
 * ALR - Любой Linux Репозиторий
 * Copyright (C) 2024 Евгений Храмов
 *
 * This program является свободным: вы можете распространять его и/или изменять
 * на условиях GNU General Public License, опубликованной Free Software Foundation,
 * либо версии 3 лицензии, либо (по вашему выбору) любой более поздней версии.
 *
 * Это программное обеспечение распространяется в надежде, что оно будет полезным,
 * но БЕЗ КАКИХ-ЛИБО ГАРАНТИЙ; без подразумеваемой гарантии
 * КОММЕРЧЕСКОЙ ПРИГОДНОСТИ или ПРИГОДНОСТИ ДЛЯ ОПРЕДЕЛЕННОЙ ЦЕЛИ.
 * Подробности см. в GNU General Public License.
 *
 * Вы должны были получить копию GNU General Public License
 * вместе с этой программой. Если нет, см. <http://www.gnu.org/licenses/>.
 */

package manager

import (
    "bufio"
    "fmt"
    "os/exec"
    "strings"
)

// DNF представляет менеджер пакетов DNF
type DNF struct {
    rootCmd string  // rootCmd хранит команду, используемую для выполнения команд с правами root
}

// Exists проверяет, доступен ли DNF в системе, возвращает true если да
func (*DNF) Exists() bool {
    _, err := exec.LookPath("dnf")
    return err == nil
}

// Name возвращает имя менеджера пакетов, в данном случае "dnf"
func (*DNF) Name() string {
    return "dnf"
}

// Format возвращает формат пакетов "rpm", используемый DNF
func (*DNF) Format() string {
    return "rpm"
}

// SetRootCmd устанавливает команду, используемую для выполнения операций с правами root
func (d *DNF) SetRootCmd(s string) {
    d.rootCmd = s
}

// Sync выполняет upgrade всех установленных пакетов, обновляя их до более новых версий
func (d *DNF) Sync(opts *Opts) error {
    opts = ensureOpts(opts)  // Гарантирует, что opts не равен nil и содержит допустимые значения
    cmd := d.getCmd(opts, "dnf", "upgrade")
    setCmdEnv(cmd)  // Устанавливает переменные окружения для команды
    err := cmd.Run()  // Выполняет команду
    if err != nil {
        return fmt.Errorf("dnf: sync: %w", err)
    }
    return nil
}

// Install устанавливает указанные пакеты с помощью DNF
func (d *DNF) Install(opts *Opts, pkgs ...string) error {
    opts = ensureOpts(opts)
    cmd := d.getCmd(opts, "dnf", "install", "--allowerasing")
    cmd.Args = append(cmd.Args, pkgs...)  // Добавляем названия пакетов к команде
    setCmdEnv(cmd)
    err := cmd.Run()
    if err != nil {
        return fmt.Errorf("dnf: install: %w", err)
    }
    return nil
}

// InstallLocal расширяет метод Install для установки пакетов, расположенных локально
func (d *DNF) InstallLocal(opts *Opts, pkgs ...string) error {
    opts = ensureOpts(opts)
    return d.Install(opts, pkgs...)
}

// Remove удаляет указанные пакеты с помощью DNF
func (d *DNF) Remove(opts *Opts, pkgs ...string) error {
    opts = ensureOpts(opts)
    cmd := d.getCmd(opts, "dnf", "remove")
    cmd.Args = append(cmd.Args, pkgs...)
    setCmdEnv(cmd)
    err := cmd.Run()
    if err != nil {
        return fmt.Errorf("dnf: remove: %w", err)
    }
    return nil
}

// Upgrade обновляет указанные пакеты до более новых версий
func (d *DNF) Upgrade(opts *Opts, pkgs ...string) error {
    opts = ensureOpts(opts)
    cmd := d.getCmd(opts, "dnf", "upgrade")
    cmd.Args = append(cmd.Args, pkgs...)
    setCmdEnv(cmd)
    err := cmd.Run()
    if err != nil {
        return fmt.Errorf("dnf: upgrade: %w", err)
    }
    return nil
}

// UpgradeAll обновляет все установленные пакеты
func (d *DNF) UpgradeAll(opts *Opts) error {
    opts = ensureOpts(opts)
    cmd := d.getCmd(opts, "dnf", "upgrade")
    setCmdEnv(cmd)
    err := cmd.Run()
    if err != nil {
        return fmt.Errorf("dnf: upgradeall: %w", err)
    }
    return nil
}

// ListInstalled возвращает список установленных пакетов и их версий
func (d *DNF) ListInstalled(opts *Opts) (map[string]string, error) {
    out := map[string]string{}
    cmd := exec.Command("rpm", "-qa", "--queryformat", "%{NAME}\u200b%|EPOCH?{%{EPOCH}:}:{}|%{VERSION}-%{RELEASE}\\n")

    stdout, err := cmd.StdoutPipe()
    if err != nil {
        return nil, err
    }

    err = cmd.Start()
    if err != nil {
        return nil, err
    }

    scanner := bufio.NewScanner(stdout)
    for scanner.Scan() {
        name, version, ok := strings.Cut(scanner.Text(), "\u200b")
        if !ok {
            continue
        }
        version = strings.TrimPrefix(version, "0:")
        out[name] = version
    }

    err = scanner.Err()
    if err != nil {
        return nil, err
    }

    return out, nil
}

// getCmd создает и возвращает команду exec.Cmd для менеджера пакетов DNF
func (d *DNF) getCmd(opts *Opts, mgrCmd string, args ...string) *exec.Cmd {
    var cmd *exec.Cmd
    if opts.AsRoot {
        cmd = exec.Command(getRootCmd(d.rootCmd), mgrCmd)
        cmd.Args = append(cmd.Args, opts.Args...)
        cmd.Args = append(cmd.Args, args...)
    } else {
        cmd = exec.Command(mgrCmd, args...)
    }

    if opts.NoConfirm {
        cmd.Args = append(cmd.Args, "-y")  // Добавляет параметр автоматического подтверждения (-y)
    }

    return cmd
}