220 lines
7.3 KiB
Go
220 lines
7.3 KiB
Go
// 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"
|
||
|
||
"gitea.plemya-x.ru/Plemya-x/ALR/internal/cliutils"
|
||
appbuilder "gitea.plemya-x.ru/Plemya-x/ALR/internal/cliutils/app_builder"
|
||
"gitea.plemya-x.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 as current user, trying with sudo"))
|
||
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)
|
||
})
|
||
}
|