lure/main.go

217 lines
4.8 KiB
Go
Raw Normal View History

2022-09-26 19:28:21 +00:00
/*
* LURE - Linux User REpository
* Copyright (C) 2022 Arsen Musayelyan
*
* 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/>.
*/
2022-09-25 21:00:15 +00:00
package main
import (
"context"
2022-12-02 20:29:55 +00:00
"fmt"
2022-09-25 21:00:15 +00:00
"os"
"os/signal"
"strings"
2022-09-25 21:00:15 +00:00
"syscall"
"time"
"github.com/urfave/cli/v2"
"go.arsenm.dev/logger"
2022-11-30 18:00:50 +00:00
"go.arsenm.dev/logger/log"
"go.arsenm.dev/lure/internal/config"
2022-12-02 20:29:55 +00:00
"go.arsenm.dev/lure/internal/db"
"go.arsenm.dev/lure/manager"
2022-09-25 21:00:15 +00:00
)
//go:generate scripts/gen-version.sh
2022-11-30 18:00:50 +00:00
func init() {
2022-12-20 03:30:35 +00:00
log.Logger = logger.NewCLI(os.Stderr)
2022-11-30 18:00:50 +00:00
}
2022-09-25 21:00:15 +00:00
func main() {
ctx := context.Background()
ctx, cancel := signal.NotifyContext(ctx, syscall.SIGINT, syscall.SIGTERM)
defer cancel()
go func() {
<-ctx.Done()
// Exit the program after a maximum of 200ms
time.Sleep(200 * time.Millisecond)
gdb.Close()
2022-09-25 21:00:15 +00:00
os.Exit(0)
}()
app := &cli.App{
Name: "lure",
Usage: "Linux User REpository",
Flags: []cli.Flag{
&cli.StringFlag{
Name: "pm-args",
Aliases: []string{"P"},
Usage: "Arguments to be passed on to the package manager",
},
},
2022-09-25 21:00:15 +00:00
Commands: []*cli.Command{
{
2022-12-02 20:29:55 +00:00
Name: "install",
Usage: "Install a new package",
Aliases: []string{"in"},
Action: installCmd,
BashComplete: completionInstall,
2022-09-25 21:00:15 +00:00
},
{
Name: "remove",
Usage: "Remove an installed package",
Aliases: []string{"rm"},
Action: removeCmd,
},
{
Name: "upgrade",
Usage: "Upgrade all installed packages",
Aliases: []string{"up"},
Action: upgradeCmd,
},
2022-09-26 21:42:17 +00:00
{
Flags: []cli.Flag{
&cli.BoolFlag{
Name: "all",
Aliases: []string{"a"},
Usage: "Show all information, not just for the current distro",
},
},
2022-09-28 22:23:09 +00:00
Name: "info",
Usage: "Print information about a package",
Action: infoCmd,
2022-09-26 21:42:17 +00:00
},
2022-09-28 08:35:17 +00:00
{
2022-12-01 05:42:04 +00:00
Flags: []cli.Flag{
&cli.BoolFlag{
Name: "installed",
Aliases: []string{"I"},
},
},
2022-09-28 08:35:17 +00:00
Name: "list",
Usage: "List LURE repo packages",
Aliases: []string{"ls"},
Action: listCmd,
},
2022-09-25 21:00:15 +00:00
{
Flags: []cli.Flag{
&cli.StringFlag{
Name: "script",
Aliases: []string{"s"},
Value: "lure.sh",
Usage: "Path to the build script",
},
},
Name: "build",
Usage: "Build a local package",
Action: buildCmd,
},
{
Flags: []cli.Flag{
&cli.StringFlag{
Name: "name",
Aliases: []string{"n"},
Required: true,
Usage: "Name of the new repo",
},
&cli.StringFlag{
Name: "url",
Aliases: []string{"u"},
Required: true,
Usage: "URL of the new repo",
},
},
Name: "addrepo",
Usage: "Add a new repository",
Aliases: []string{"ar"},
Action: addrepoCmd,
},
{
Flags: []cli.Flag{
&cli.StringFlag{
Name: "name",
Aliases: []string{"n"},
Required: true,
Usage: "Name of the repo to be deleted",
},
},
Name: "removerepo",
Usage: "Remove an existing repository",
Aliases: []string{"rr"},
Action: removerepoCmd,
},
{
Name: "refresh",
Usage: "Pull all repositories that have changed",
Aliases: []string{"ref"},
Action: refreshCmd,
},
2022-12-02 20:43:00 +00:00
{
Name: "fix",
Usage: "Attempt to fix problems with LURE",
Action: fixCmd,
},
2022-11-22 20:54:03 +00:00
{
Name: "version",
Usage: "Display the current LURE version and exit",
Action: displayVersion,
},
2022-09-25 21:00:15 +00:00
},
Before: func(c *cli.Context) error {
args := strings.Split(c.String("pm-args"), " ")
if len(args) == 1 && args[0] == "" {
args = nil
}
manager.Args = append(manager.Args, args...)
return nil
},
After: func(ctx *cli.Context) error {
return gdb.Close()
},
2022-12-02 20:29:55 +00:00
EnableBashCompletion: true,
2022-09-25 21:00:15 +00:00
}
err := app.RunContext(ctx, os.Args)
if err != nil {
log.Error("Error while running app").Err(err).Send()
}
}
2022-11-22 20:54:03 +00:00
func displayVersion(c *cli.Context) error {
print(config.Version)
2022-11-22 20:54:03 +00:00
return nil
}
2022-12-02 20:29:55 +00:00
func completionInstall(c *cli.Context) {
result, err := db.GetPkgs(gdb, "true")
if err != nil {
log.Fatal("Error getting packages").Err(err).Send()
}
defer result.Close()
2022-12-24 20:56:02 +00:00
for result.Next() {
2022-12-02 20:29:55 +00:00
var pkg db.Package
2022-12-24 20:56:02 +00:00
err = result.StructScan(&pkg)
2022-12-02 20:29:55 +00:00
if err != nil {
2022-12-24 20:56:02 +00:00
log.Fatal("Error iterating over packages").Err(err).Send()
2022-12-02 20:29:55 +00:00
}
fmt.Println(pkg.Name)
}
}