tmpl/cmd/source.go

116 lines
2.3 KiB
Go
Raw Normal View History

package cmd
import (
"encoding/json"
"fmt"
"os"
"text/tabwriter"
"go.jolheiser.com/tmpl/registry"
"github.com/rs/zerolog/log"
"github.com/urfave/cli/v2"
)
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",
Flags: []cli.Flag{
&cli.BoolFlag{
Name: "json",
Usage: "JSON format",
},
},
Action: runSourceList,
}
SourceAdd = &cli.Command{
Name: "add",
Usage: "Add a source",
Description: "Add a new source to the registry",
ArgsUsage: "[base URL] [name]",
Action: runSourceAdd,
}
SourceRemove = &cli.Command{
Name: "remove",
Usage: "Remove a source",
Description: "Remove a source from the registry",
ArgsUsage: "[name]",
Action: runSourceRemove,
}
)
func runSourceList(ctx *cli.Context) error {
reg, err := registry.Open(registryFlag)
if err != nil {
return err
}
if ctx.Bool("json") {
return json.NewEncoder(os.Stdout).Encode(reg.Sources)
}
wr := tabwriter.NewWriter(os.Stdout, 0, 8, 1, '\t', 0)
if _, err := fmt.Fprintf(wr, "NAME\tURL\n"); err != nil {
return err
}
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 cli.ShowCommandHelp(ctx, ctx.Command.Name)
}
reg, err := registry.Open(registryFlag)
if err != nil {
return err
}
s, err := reg.AddSource(ctx.Args().First(), ctx.Args().Get(1))
if err != nil {
return err
}
log.Info().Msgf("Added new source %q", s.Name)
return nil
}
func runSourceRemove(ctx *cli.Context) error {
if ctx.NArg() < 1 {
return cli.ShowCommandHelp(ctx, ctx.Command.Name)
}
reg, err := registry.Open(registryFlag)
if err != nil {
return err
}
if err := reg.RemoveSource(ctx.Args().First()); err != nil {
return err
}
log.Info().Msgf("Successfully removed source for %q", ctx.Args().First())
return nil
}