// This file was originally part of the project "LURE - Linux User REpository", created by Elara Musayelyan.

// It has been modified as part of "ALR - Any Linux Repository" by the ALR Authors.

//

// ALR - Any Linux Repository

// Copyright (C) 2025 The ALR Authors

//

// This program is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

//

// This program is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

// GNU General Public License for more details.

//

// You should have received a copy of the GNU General Public License

// along with this program. If not, see <http://www.gnu.org/licenses/>.

package main

import (

"io/fs"

"log/slog"

"os"

"os/exec"

"path/filepath"

"github.com/leonelquinteros/gotext"

"github.com/urfave/cli/v2"

"git.alr-pkg.ru/Plemya-x/ALR/internal/cliutils"

appbuilder "git.alr-pkg.ru/Plemya-x/ALR/internal/cliutils/app_builder"

"git.alr-pkg.ru/Plemya-x/ALR/internal/utils"

)

// execWithPrivileges выполняет команду напрямую если root или CI, иначе через sudo

func execWithPrivileges(name string, args ...string) *exec.Cmd {

isRoot := os.Geteuid() == 0

isCI := os.Getenv("CI") == "true"

if !isRoot && !isCI {

// Если не root и не в CI, используем sudo

allArgs := append([]string{name}, args...)

return exec.Command("sudo", allArgs...)

} else {

// Если root или в CI, запускаем напрямую

return exec.Command(name, args...)

}

}

func FixCmd() *cli.Command {

return &cli.Command{

Name: "fix",

Usage: gotext.Get("Attempt to fix problems with ALR"),

Action: func(c *cli.Context) error {

// Команда выполняется от текущего пользователя

// При необходимости будет запрошен sudo для удаления файлов root

ctx := c.Context

deps, err := appbuilder.

New(ctx).

WithConfig().

Build()

if err != nil {

return cli.Exit(err, 1)

}

defer deps.Defer()

cfg := deps.Cfg

paths := cfg.GetPaths()

slog.Info(gotext.Get("Clearing cache and temporary directories"))

// Проверяем, существует ли директория кэша

dir, err := os.Open(paths.CacheDir)

if err != nil {

if os.IsNotExist(err) {

// Директория не существует, просто создадим её позже

slog.Info(gotext.Get("Cache directory does not exist, will create it"))

} else {

return cliutils.FormatCliExit(gotext.Get("Unable to open cache directory"), err)

}

} else {

defer dir.Close()

entries, err := dir.Readdirnames(-1)

if err != nil {

return cliutils.FormatCliExit(gotext.Get("Unable to read cache directory contents"), err)

}

for _, entry := range entries {

fullPath := filepath.Join(paths.CacheDir, entry)

// Пробуем сделать файлы доступными для записи

if err := makeWritableRecursive(fullPath); err != nil {

slog.Debug("Failed to make path writable", "path", fullPath, "error", err)

}

// Пробуем удалить

err = os.RemoveAll(fullPath)

if err != nil {

// Если не получилось удалить, пробуем через sudo

slog.Warn(gotext.Get("Unable to remove cache item (%s) as current user, trying with sudo", entry))

sudoCmd := execWithPrivileges("rm", "-rf", fullPath)

if sudoErr := sudoCmd.Run(); sudoErr != nil {

// Если и через sudo не получилось, пропускаем с предупреждением

slog.Error(gotext.Get("Unable to remove cache item (%s)", entry), "error", err)

continue

}

}

}

}

// Очищаем временные директории

slog.Info(gotext.Get("Clearing temporary directory"))

tmpDir := "/tmp/alr"

if _, err := os.Stat(tmpDir); err == nil {

// Директория существует, пробуем очистить

err = os.RemoveAll(tmpDir)

if err != nil {

// Если не получилось удалить, пробуем через sudo

slog.Warn(gotext.Get("Unable to remove temporary directory (%s) as current user, trying with sudo", tmpDir))

sudoCmd := execWithPrivileges("rm", "-rf", tmpDir)

if sudoErr := sudoCmd.Run(); sudoErr != nil {

slog.Error(gotext.Get("Unable to remove temporary directory"), "error", err)

}

}

}

// Создаем базовый каталог /tmp/alr с владельцем root:wheel и правами 2775

err = utils.EnsureTempDirWithRootOwner(tmpDir, 0o2775)

if err != nil {

slog.Warn(gotext.Get("Unable to create temporary directory"), "error", err)

}

// Создаем каталог dl с правами для группы wheel

dlDir := filepath.Join(tmpDir, "dl")

err = utils.EnsureTempDirWithRootOwner(dlDir, 0o2775)

if err != nil {

slog.Warn(gotext.Get("Unable to create download directory"), "error", err)

}

// Создаем каталог pkgs с правами для группы wheel

pkgsDir := filepath.Join(tmpDir, "pkgs")

err = utils.EnsureTempDirWithRootOwner(pkgsDir, 0o2775)

if err != nil {

slog.Warn(gotext.Get("Unable to create packages directory"), "error", err)

}

// Исправляем права на все существующие файлы в /tmp/alr, если там что-то есть

if _, err := os.Stat(tmpDir); err == nil {

slog.Info(gotext.Get("Fixing permissions on temporary files"))

// Проверяем, есть ли файлы в директории

entries, err := os.ReadDir(tmpDir)

if err == nil && len(entries) > 0 {

group := utils.GetPrivilegedGroup()

fixCmd := execWithPrivileges("chown", "-R", "root:"+group, tmpDir)

if fixErr := fixCmd.Run(); fixErr != nil {

slog.Warn(gotext.Get("Unable to fix file ownership"), "error", fixErr)

}

fixCmd = execWithPrivileges("chmod", "-R", "2775", tmpDir)

if fixErr := fixCmd.Run(); fixErr != nil {

slog.Warn(gotext.Get("Unable to fix file permissions"), "error", fixErr)

}

}

}

slog.Info(gotext.Get("Rebuilding cache"))

// Создаем директорию кэша с правильными правами

slog.Info(gotext.Get("Creating cache directory"))

err = utils.EnsureTempDirWithRootOwner(paths.CacheDir, 0o2775)

if err != nil {

return cliutils.FormatCliExit(gotext.Get("Unable to create new cache directory"), err)

}

deps, err = appbuilder.

New(ctx).

WithConfig().

WithDB().

WithReposForcePull().

Build()

if err != nil {

return cli.Exit(err, 1)

}

defer deps.Defer()

slog.Info(gotext.Get("Done"))

return nil

},

}

}

func makeWritableRecursive(path string) error {

return filepath.WalkDir(path, func(path string, d fs.DirEntry, err error) error {

if err != nil {

return err

}

info, err := d.Info()

if err != nil {

return err

}

newMode := info.Mode() | 0o200

if d.IsDir() {

newMode |= 0o100

}

return os.Chmod(path, newMode)

})

}