mirror of
https://github.com/Jguer/yay
synced 2024-09-06 16:43:03 +00:00
![J Guerreiro](/assets/img/avatar_default.png)
* fix(alpm): fix callback text * feat(yay): Add mixed search result * remove old result structs * add option for controlling query builder * only set query builder after parsing args * add parser args * update manpage * write test for results * write test for results * mixed source test * only sort 1 time with every mode
190 lines
4.1 KiB
Go
190 lines
4.1 KiB
Go
package main // import "github.com/Jguer/yay"
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
|
|
pacmanconf "github.com/Morganamilo/go-pacmanconf"
|
|
"github.com/leonelquinteros/gotext"
|
|
"golang.org/x/term"
|
|
|
|
"github.com/Jguer/yay/v11/pkg/db"
|
|
"github.com/Jguer/yay/v11/pkg/db/ialpm"
|
|
"github.com/Jguer/yay/v11/pkg/query"
|
|
"github.com/Jguer/yay/v11/pkg/settings"
|
|
"github.com/Jguer/yay/v11/pkg/settings/parser"
|
|
"github.com/Jguer/yay/v11/pkg/text"
|
|
)
|
|
|
|
func initGotext() {
|
|
if envLocalePath := os.Getenv("LOCALE_PATH"); envLocalePath != "" {
|
|
localePath = envLocalePath
|
|
}
|
|
|
|
if lc := os.Getenv("LANGUAGE"); lc != "" {
|
|
gotext.Configure(localePath, lc, "yay")
|
|
} else if lc := os.Getenv("LC_ALL"); lc != "" {
|
|
gotext.Configure(localePath, lc, "yay")
|
|
} else if lc := os.Getenv("LC_MESSAGES"); lc != "" {
|
|
gotext.Configure(localePath, lc, "yay")
|
|
} else {
|
|
gotext.Configure(localePath, os.Getenv("LANG"), "yay")
|
|
}
|
|
}
|
|
|
|
func initAlpm(cmdArgs *parser.Arguments, pacmanConfigPath string) (*pacmanconf.Config, bool, error) {
|
|
root := "/"
|
|
if value, _, exists := cmdArgs.GetArg("root", "r"); exists {
|
|
root = value
|
|
}
|
|
|
|
pacmanConf, stderr, err := pacmanconf.PacmanConf("--config", pacmanConfigPath, "--root", root)
|
|
if err != nil {
|
|
cmdErr := err
|
|
if stderr != "" {
|
|
cmdErr = fmt.Errorf("%s\n%s", err, stderr)
|
|
}
|
|
|
|
return nil, false, cmdErr
|
|
}
|
|
|
|
if dbPath, _, exists := cmdArgs.GetArg("dbpath", "b"); exists {
|
|
pacmanConf.DBPath = dbPath
|
|
}
|
|
|
|
if arch := cmdArgs.GetArgs("arch"); arch != nil {
|
|
pacmanConf.Architecture = append(pacmanConf.Architecture, arch...)
|
|
}
|
|
|
|
if ignoreArray := cmdArgs.GetArgs("ignore"); ignoreArray != nil {
|
|
pacmanConf.IgnorePkg = append(pacmanConf.IgnorePkg, ignoreArray...)
|
|
}
|
|
|
|
if ignoreGroupsArray := cmdArgs.GetArgs("ignoregroup"); ignoreGroupsArray != nil {
|
|
pacmanConf.IgnoreGroup = append(pacmanConf.IgnoreGroup, ignoreGroupsArray...)
|
|
}
|
|
|
|
if cacheArray := cmdArgs.GetArgs("cachedir"); cacheArray != nil {
|
|
pacmanConf.CacheDir = cacheArray
|
|
}
|
|
|
|
if gpgDir, _, exists := cmdArgs.GetArg("gpgdir"); exists {
|
|
pacmanConf.GPGDir = gpgDir
|
|
}
|
|
|
|
useColor := pacmanConf.Color && term.IsTerminal(int(os.Stdout.Fd()))
|
|
|
|
switch value, _, _ := cmdArgs.GetArg("color"); value {
|
|
case "always":
|
|
useColor = true
|
|
case "auto":
|
|
useColor = term.IsTerminal(int(os.Stdout.Fd()))
|
|
case "never":
|
|
useColor = false
|
|
}
|
|
|
|
return pacmanConf, useColor, nil
|
|
}
|
|
|
|
func main() {
|
|
var (
|
|
err error
|
|
ctx = context.Background()
|
|
ret = 0
|
|
)
|
|
|
|
defer func() { os.Exit(ret) }()
|
|
|
|
initGotext()
|
|
|
|
if os.Geteuid() == 0 {
|
|
text.Warnln(gotext.Get("Avoid running yay as root/sudo."))
|
|
}
|
|
|
|
config, err = settings.NewConfig(yayVersion)
|
|
if err != nil {
|
|
if str := err.Error(); str != "" {
|
|
text.Errorln(str)
|
|
}
|
|
|
|
ret = 1
|
|
|
|
return
|
|
}
|
|
|
|
cmdArgs := parser.MakeArguments()
|
|
|
|
if err = config.ParseCommandLine(cmdArgs); err != nil {
|
|
if str := err.Error(); str != "" {
|
|
text.Errorln(str)
|
|
}
|
|
|
|
ret = 1
|
|
|
|
return
|
|
}
|
|
|
|
if config.Runtime.SaveConfig {
|
|
if errS := config.Save(config.Runtime.ConfigPath); errS != nil {
|
|
text.Errorln(errS)
|
|
}
|
|
}
|
|
|
|
if config.SeparateSources {
|
|
config.Runtime.QueryBuilder = query.NewSourceQueryBuilder(config.SortBy,
|
|
config.Runtime.Mode, config.SearchBy, config.BottomUp, config.SingleLineResults)
|
|
} else {
|
|
config.Runtime.QueryBuilder = query.NewMixedSourceQueryBuilder(config.SortBy,
|
|
config.Runtime.Mode, config.SearchBy, config.BottomUp, config.SingleLineResults)
|
|
}
|
|
|
|
var useColor bool
|
|
|
|
config.Runtime.PacmanConf, useColor, err = initAlpm(cmdArgs, config.PacmanConf)
|
|
if err != nil {
|
|
if str := err.Error(); str != "" {
|
|
text.Errorln(str)
|
|
}
|
|
|
|
ret = 1
|
|
|
|
return
|
|
}
|
|
|
|
config.Runtime.CmdBuilder.SetPacmanDBPath(config.Runtime.PacmanConf.DBPath)
|
|
|
|
text.UseColor = useColor
|
|
|
|
dbExecutor, err := ialpm.NewExecutor(config.Runtime.PacmanConf)
|
|
if err != nil {
|
|
if str := err.Error(); str != "" {
|
|
text.Errorln(str)
|
|
}
|
|
|
|
ret = 1
|
|
|
|
return
|
|
}
|
|
|
|
defer dbExecutor.Cleanup()
|
|
|
|
if err = handleCmd(ctx, cmdArgs, db.Executor(dbExecutor)); err != nil {
|
|
if str := err.Error(); str != "" {
|
|
text.Errorln(str)
|
|
}
|
|
|
|
if exitError, ok := err.(*exec.ExitError); ok {
|
|
// mirror pacman exit code when applicable
|
|
ret = exitError.ExitCode()
|
|
return
|
|
}
|
|
|
|
// fallback
|
|
ret = 1
|
|
|
|
return
|
|
}
|
|
}
|