piwigo-cli/internal/piwigocli/categories_list.go
2021-12-29 16:26:11 +01:00

55 lines
990 B
Go

package piwigocli
import (
"os"
"regexp"
"github.com/celogeek/piwigo-cli/internal/piwigo"
"github.com/jedib0t/go-pretty/v6/table"
)
type CategoriesListCommand struct {
Filter string `short:"x" long:"filter" description:"Regexp filter"`
}
func (c *CategoriesListCommand) Execute(args []string) error {
p := piwigo.Piwigo{}
if err := p.LoadConfig(); err != nil {
return err
}
_, err := p.Login()
if err != nil {
return err
}
categories, err := p.Categories()
if err != nil {
return err
}
filter := regexp.MustCompile("")
if c.Filter != "" {
filter = regexp.MustCompile("(?i)" + c.Filter)
}
t := table.NewWriter()
t.AppendHeader(table.Row{"Id", "Name", "Images", "Url"})
for _, category := range categories {
if filter.MatchString(category.Name) {
t.AppendRow(table.Row{
category.Id,
category.Name,
category.ImagesCount,
category.Url,
})
}
}
t.SetOutputMirror(os.Stdout)
t.SetStyle(table.StyleLight)
t.Render()
return nil
}