openlan/cmd/api/v5/lease.go

package v5

import (
	"github.com/luscis/openlan/cmd/api"
	"github.com/luscis/openlan/pkg/schema"
	"github.com/urfave/cli/v2"
)

type Lease struct {
	Cmd
}

func (u Lease) Url(prefix, name string) string {
	if name == "" {
		return prefix + "/api/lease"
	} else {
		return prefix + "/api/lease/" + name
	}
}

func (u Lease) Tmpl() string {
	return `# total {{ len . }}
{{ps -16 "alias"}} {{ ps -16 "address" }} {{ps -22 "client"}} {{ps -8 "network"}} {{ ps -6 "type"}}
{{- range . }}
{{ps -16 .Alias}} {{ ps -16 .Address}} {{ps -22 .Client}} {{ps -8 .Network}} {{ ps -6 .Type}}
{{- end }}
`
}

func (u Lease) List(c *cli.Context) error {
	url := u.Url(c.String("url"), "")
	clt := u.NewHttp(c.String("token"))
	var items []schema.Lease
	if err := clt.GetJSON(url, &items); err != nil {
		return err
	}
	name := c.String("network")
	if len(name) > 0 {
		tmp := items[:0]
		for _, obj := range items {
			if obj.Network == name {
				tmp = append(tmp, obj)
			}
		}
		items = tmp
	}
	return u.Out(items, c.String("format"), u.Tmpl())
}

func (u Lease) Commands(app *api.App) {
	app.Command(&cli.Command{
		Name:  "lease",
		Usage: "Address lease",
		Subcommands: []*cli.Command{
			{
				Name:    "list",
				Usage:   "Display all lease",
				Aliases: []string{"ls"},
				Flags: []cli.Flag{
					&cli.StringFlag{Name: "network"},
				},
				Action: u.List,
			},
		},
	})
}