yay/cmd.go

354 lines
7.2 KiB
Go
Raw Normal View History

2017-04-29 17:12:12 +00:00
package main
import (
"bufio"
"encoding/json"
2017-04-29 17:12:12 +00:00
"fmt"
"io"
2017-04-29 17:12:12 +00:00
"os"
"strconv"
"strings"
"time"
2017-04-29 17:12:12 +00:00
)
func usage() {
fmt.Println(`usage: yay <operation> [...]
operations:
yay {-h --help}
yay {-V --version}
yay {-D --database} <options> <package(s)>
yay {-F --files} [options] [package(s)]
yay {-Q --query} [options] [package(s)]
yay {-R --remove} [options] <package(s)>
yay {-S --sync} [options] [package(s)]
yay {-T --deptest} [options] [package(s)]
yay {-U --upgrade} [options] <file(s)>
New operations:
yay -Qstats displays system information
yay -Cd remove unneeded dependencies
yay -G [package(s)] get pkgbuild from ABS or AUR
New options:
--topdown shows repository's packages first and then aur's
--bottomup shows aur's packages first and then repository's
--noconfirm skip user input on package install
2017-06-14 07:47:17 +00:00
--devel Check -git/-svn/-hg development version
--nodevel Disable development version checking
2017-04-29 17:12:12 +00:00
`)
}
func init() {
defaultSettings(&config)
var err error
configfile := os.Getenv("HOME") + "/.config/yay/config.json"
if _, err = os.Stat(configfile); os.IsNotExist(err) {
_ = os.MkdirAll(os.Getenv("HOME")+"/.config/yay", 0755)
// Save the default config if nothing is found
config.saveConfig()
} else {
file, err := os.Open(configfile)
if err != nil {
fmt.Println("Error reading config:", err)
} else {
decoder := json.NewDecoder(file)
err = decoder.Decode(&config)
if err != nil {
fmt.Println("Loading default Settings\nError reading config:", err)
defaultSettings(&config)
}
}
}
AlpmConf, err = readAlpmConfig(config.PacmanConf)
if err != nil {
fmt.Println("Unable to read Pacman conf", err)
os.Exit(1)
}
AlpmHandle, err = AlpmConf.CreateHandle()
if err != nil {
fmt.Println("Unable to CreateHandle", err)
os.Exit(1)
}
updated = false
configfile = os.Getenv("HOME") + "/.config/yay/yay_vcs.json"
if _, err := os.Stat(configfile); os.IsNotExist(err) {
_ = os.MkdirAll(os.Getenv("HOME")+"/.config/yay", 0755)
return
}
file, err := os.Open(configfile)
if err != nil {
fmt.Println("error:", err)
return
}
decoder := json.NewDecoder(file)
err = decoder.Decode(&savedInfo)
if err != nil {
fmt.Println("error:", err)
}
}
2017-04-29 17:12:12 +00:00
2017-05-07 01:43:49 +00:00
func parser() (op string, options []string, packages []string, changedConfig bool, err error) {
2017-04-29 17:12:12 +00:00
if len(os.Args) < 2 {
err = fmt.Errorf("no operation specified")
return
}
2017-05-07 01:43:49 +00:00
changedConfig = false
2017-04-29 17:12:12 +00:00
op = "yogurt"
for _, arg := range os.Args[1:] {
if arg[0] == '-' && arg[1] != '-' {
switch arg {
default:
op = arg
}
continue
}
if arg[0] == '-' && arg[1] == '-' {
changedConfig = true
2017-04-29 17:12:12 +00:00
switch arg {
case "--printconfig":
fmt.Printf("%+v", config)
os.Exit(0)
case "--gendb":
err = createDevelDB()
if err != nil {
fmt.Println(err)
}
err = saveVCSInfo()
if err != nil {
fmt.Println(err)
}
os.Exit(0)
case "--devel":
config.Devel = true
case "--nodevel":
config.Devel = false
2017-06-23 08:19:18 +00:00
case "--timeupdate":
config.TimeUpdate = true
2017-06-23 08:19:18 +00:00
case "--notimeupdate":
config.TimeUpdate = false
2017-04-29 17:12:12 +00:00
case "--topdown":
config.SortMode = TopDown
2017-04-29 17:12:12 +00:00
case "--complete":
config.Shell = "sh"
_ = complete()
2017-04-29 17:12:12 +00:00
os.Exit(0)
case "--fcomplete":
config.Shell = "fish"
_ = complete()
2017-04-29 17:12:12 +00:00
os.Exit(0)
case "--help":
usage()
os.Exit(0)
case "--noconfirm":
config.NoConfirm = true
2017-04-29 17:12:12 +00:00
fallthrough
default:
options = append(options, arg)
}
continue
}
packages = append(packages, arg)
}
return
}
func main() {
2017-05-07 01:43:49 +00:00
op, options, pkgs, changedConfig, err := parser()
2017-04-29 17:12:12 +00:00
if err != nil {
fmt.Println(err)
os.Exit(1)
}
switch op {
case "-Cd":
err = cleanDependencies(pkgs)
2017-04-29 17:12:12 +00:00
case "-G":
for _, pkg := range pkgs {
err = getPkgbuild(pkg)
2017-04-29 17:12:12 +00:00
if err != nil {
fmt.Println(pkg+":", err)
}
}
case "-Qstats":
err = localStatistics(version)
2017-04-29 17:12:12 +00:00
case "-Ss", "-Ssq", "-Sqs":
if op == "-Ss" {
config.SearchMode = Detailed
2017-04-29 17:12:12 +00:00
} else {
config.SearchMode = Minimal
2017-04-29 17:12:12 +00:00
}
if pkgs != nil {
err = syncSearch(pkgs)
2017-04-29 17:12:12 +00:00
}
case "-S":
err = install(pkgs, options)
2017-07-24 09:32:11 +00:00
case "-Sy":
err = passToPacman("-Sy", nil, nil)
2017-07-24 09:32:11 +00:00
if err != nil {
break
}
err = install(pkgs, options)
2017-07-19 09:32:32 +00:00
case "-Syu", "-Suy", "-Su":
if strings.Contains(op, "y") {
err = passToPacman("-Sy", nil, nil)
2017-07-19 09:32:32 +00:00
if err != nil {
break
}
}
2017-07-14 17:03:54 +00:00
err = upgradePkgs(options)
2017-04-29 17:12:12 +00:00
case "-Si":
err = syncInfo(pkgs, options)
2017-04-29 17:12:12 +00:00
case "yogurt":
config.SearchMode = NumberMenu
2017-04-29 17:12:12 +00:00
if pkgs != nil {
err = numberMenu(pkgs, options)
}
default:
if op[0] == 'R' {
removeVCSPackage(pkgs)
}
err = passToPacman(op, pkgs, options)
2017-05-07 01:43:49 +00:00
}
var erra error
if updated {
erra = saveVCSInfo()
if erra != nil {
fmt.Println(err)
}
}
2017-05-07 01:43:49 +00:00
if changedConfig {
erra = config.saveConfig()
if erra != nil {
fmt.Println(err)
}
}
erra = AlpmHandle.Release()
if erra != nil {
fmt.Println(err)
2017-04-29 17:12:12 +00:00
}
if err != nil {
fmt.Println(err)
os.Exit(1)
}
}
// NumberMenu presents a CLI for selecting packages to install.
func numberMenu(pkgS []string, flags []string) (err error) {
var num int
aq, err := narrowSearch(pkgS, true)
2017-04-29 17:12:12 +00:00
if err != nil {
fmt.Println("Error during AUR search:", err)
}
numaq := len(aq)
2017-08-04 09:26:53 +00:00
pq, numpq, err := queryRepo(pkgS)
2017-04-29 17:12:12 +00:00
if err != nil {
return
}
if numpq == 0 && numaq == 0 {
return fmt.Errorf("no packages match search")
}
if config.SortMode == BottomUp {
aq.printSearch(numpq)
2017-08-04 09:26:53 +00:00
pq.printSearch()
2017-04-29 17:12:12 +00:00
} else {
pq.printSearch()
aq.printSearch(numpq)
2017-04-29 17:12:12 +00:00
}
2017-05-21 13:49:29 +00:00
fmt.Printf("\x1b[32m%s\x1b[0m\nNumbers: ", "Type numbers to install. Separate each number with a space.")
2017-04-29 17:12:12 +00:00
reader := bufio.NewReader(os.Stdin)
numberBuf, overflow, err := reader.ReadLine()
if err != nil || overflow {
fmt.Println(err)
return
}
numberString := string(numberBuf)
2017-08-04 09:26:53 +00:00
var aurI []string
var repoI []string
2017-04-29 17:12:12 +00:00
result := strings.Fields(numberString)
for _, numS := range result {
num, err = strconv.Atoi(numS)
if err != nil {
continue
}
// Install package
if num > numaq+numpq-1 || num < 0 {
continue
} else if num > numpq-1 {
if config.SortMode == BottomUp {
2017-08-04 09:26:53 +00:00
aurI = append(aurI, aq[numaq+numpq-num-1].Name)
2017-04-29 17:12:12 +00:00
} else {
2017-08-04 09:26:53 +00:00
aurI = append(aurI, aq[num-numpq].Name)
2017-04-29 17:12:12 +00:00
}
} else {
if config.SortMode == BottomUp {
2017-08-04 09:26:53 +00:00
repoI = append(repoI, pq[numpq-num-1].Name())
2017-04-29 17:12:12 +00:00
} else {
2017-08-04 09:26:53 +00:00
repoI = append(repoI, pq[num].Name())
2017-04-29 17:12:12 +00:00
}
}
}
2017-08-04 09:26:53 +00:00
if len(repoI) != 0 {
err = passToPacman("-S", repoI, flags)
2017-04-29 17:12:12 +00:00
}
2017-08-04 09:26:53 +00:00
if len(aurI) != 0 {
err = aurInstall(aurI, flags)
2017-04-29 17:12:12 +00:00
}
return err
2017-04-29 17:12:12 +00:00
}
// Complete provides completion info for shells
func complete() (err error) {
path := os.Getenv("HOME") + "/.cache/yay/aur_" + config.Shell + ".cache"
if info, err := os.Stat(path); os.IsNotExist(err) || time.Since(info.ModTime()).Hours() > 48 {
os.MkdirAll(os.Getenv("HOME")+"/.cache/yay/", 0755)
out, err := os.Create(path)
if err != nil {
return err
}
if createAURList(out) != nil {
defer os.Remove(path)
}
2017-08-04 09:26:53 +00:00
err = createRepoList(out)
out.Close()
return err
}
in, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE, 0755)
if err != nil {
return err
}
defer in.Close()
_, err = io.Copy(os.Stdout, in)
return err
}