package v5

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

type Qos struct {
	Cmd
}

func (q Qos) Commands() *cli.Command {
	return &cli.Command{
		Name:  "qos",
		Usage: "QoS for client",
		Subcommands: []*cli.Command{
			QosRule{}.Commands(),
		},
	}
}

type QosRule struct {
	Cmd
}

func (qr QosRule) Url(prefix, name string) string {
	return prefix + "/api/network/" + name + "/qos"
}

func (qr QosRule) Add(c *cli.Context) error {
	name := c.String("name")
	url := qr.Url(c.String("url"), name)

	rule := &schema.Qos{
		Name:    c.String("client"),
		InSpeed: c.Float64("inspeed"),
	}

	clt := qr.NewHttp(c.String("token"))
	if err := clt.PostJSON(url, rule, nil); err != nil {
		return err
	}

	return nil
}

func (qr QosRule) Remove(c *cli.Context) error {
	name := c.String("name")
	url := qr.Url(c.String("url"), name)

	rule := &schema.Qos{
		Name: c.String("client"),
	}

	clt := qr.NewHttp(c.String("token"))
	if err := clt.DeleteJSON(url, rule, nil); err != nil {
		return err
	}

	return nil
}

func (qr QosRule) Tmpl() string {
	return `# total {{ len . }}
{{ps -28 "Name"}} {{ps -10 "Device"}} {{ps -15 "Ip"}} {{ps -8 "InSpeed"}}
{{- range . }}
{{ps -28 .Name}} {{ps -10 .Device}} {{ps -15 .Ip}} {{pf -8 2 .InSpeed}}
{{- end }}
`
}

func (qr QosRule) List(c *cli.Context) error {
	name := c.String("name")

	url := qr.Url(c.String("url"), name)
	clt := qr.NewHttp(c.String("token"))

	var items []schema.Qos
	if err := clt.GetJSON(url, &items); err != nil {
		return err
	}

	return qr.Out(items, c.String("format"), qr.Tmpl())
}

func (qr QosRule) Save(c *cli.Context) error {
	name := c.String("name")
	url := qr.Url(c.String("url"), name)

	clt := qr.NewHttp(c.String("token"))
	if err := clt.PutJSON(url, nil, nil); err != nil {
		return err
	}

	return nil
}

func (qr QosRule) Commands() *cli.Command {
	return &cli.Command{
		Name:  "rule",
		Usage: "Access Control Qos Rule",
		Subcommands: []*cli.Command{
			{
				Name:  "add",
				Usage: "Add a new qos rule for client",
				Flags: []cli.Flag{
					&cli.StringFlag{Name: "client", Aliases: []string{"c"}},
					&cli.Float64Flag{Name: "inspeed", Aliases: []string{"is"}},
				},
				Action: qr.Add,
			},
			{
				Name:    "remove",
				Usage:   "remove a qos rule",
				Aliases: []string{"rm"},
				Flags: []cli.Flag{
					&cli.StringFlag{Name: "client", Aliases: []string{"c"}},
				},
				Action: qr.Remove,
			},
			{
				Name:    "list",
				Usage:   "Display all qos rules",
				Aliases: []string{"ls"},
				Action:  qr.List,
			},
			{
				Name:    "save",
				Usage:   "Save all qos rules",
				Aliases: []string{"sa"},
				Action:  qr.Save,
			},
		},
	}
}