102 lines
2.0 KiB
Go
102 lines
2.0 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"text/tabwriter"
|
|
|
|
"go.jolheiser.com/tmpl/cmd/flags"
|
|
"go.jolheiser.com/tmpl/registry"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
"go.jolheiser.com/beaver"
|
|
)
|
|
|
|
var (
|
|
Source = &cli.Command{
|
|
Name: "source",
|
|
Usage: "Commands for working with sources",
|
|
Description: "Commands for working with sources, short-hand flags for easier downloads",
|
|
Action: SourceList.Action,
|
|
Subcommands: []*cli.Command{
|
|
SourceList,
|
|
SourceAdd,
|
|
SourceRemove,
|
|
},
|
|
}
|
|
|
|
SourceList = &cli.Command{
|
|
Name: "list",
|
|
Usage: "List available sources",
|
|
Description: "List all available sources in the registry",
|
|
Action: runSourceList,
|
|
}
|
|
|
|
SourceAdd = &cli.Command{
|
|
Name: "add",
|
|
Usage: "Add a source",
|
|
Description: "Add a new source to the registry",
|
|
Action: runSourceAdd,
|
|
}
|
|
|
|
SourceRemove = &cli.Command{
|
|
Name: "remove",
|
|
Usage: "Remove a source",
|
|
Description: "Remove a source from the registry",
|
|
Action: runSourceRemove,
|
|
}
|
|
)
|
|
|
|
func runSourceList(_ *cli.Context) error {
|
|
reg, err := registry.Open(flags.Registry)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
wr := tabwriter.NewWriter(os.Stdout, 0, 8, 0, '\t', 0)
|
|
for _, s := range reg.Sources {
|
|
if _, err := fmt.Fprintf(wr, "%s\t%s\n", s.Name, s.URL); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return wr.Flush()
|
|
}
|
|
|
|
func runSourceAdd(ctx *cli.Context) error {
|
|
if ctx.NArg() < 2 {
|
|
return errors.New("<repo> <name>")
|
|
}
|
|
|
|
reg, err := registry.Open(flags.Registry)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
s, err := reg.AddSource(ctx.Args().First(), ctx.Args().Get(1))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
beaver.Infof("Added new source %s", s.Name)
|
|
return nil
|
|
}
|
|
|
|
func runSourceRemove(ctx *cli.Context) error {
|
|
if ctx.NArg() < 1 {
|
|
return errors.New("<name>")
|
|
}
|
|
|
|
reg, err := registry.Open(flags.Registry)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := reg.RemoveSource(ctx.Args().First()); err != nil {
|
|
return err
|
|
}
|
|
|
|
beaver.Infof("Successfully removed source for %s", ctx.Args().First())
|
|
return nil
|
|
}
|