Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit eefdd6b

Browse files
tamndtamnd
committed
feat: initial uci CLI for UCI ML Repository datasets
1 parent f23a0cd commit eefdd6b

11 files changed

Lines changed: 855 additions & 78 deletions

File tree

‎cli/cmd_list.go‎

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
package cli
2+
3+
import (
4+
"github.com/spf13/cobra"
5+
)
6+
7+
func (a *App) listCmd() *cobra.Command {
8+
cmd := &cobra.Command{
9+
Use: "list",
10+
Short: "List all UCI ML Repository datasets",
11+
RunE: func(cmd *cobra.Command, _ []string) error {
12+
limit := a.effectiveLimit(0)
13+
datasets, err := a.client.List(cmd.Context(), limit)
14+
if err != nil {
15+
return mapFetchErr(err)
16+
}
17+
return a.renderOrEmpty(datasets, len(datasets))
18+
},
19+
}
20+
return cmd
21+
}

‎cli/cmd_search.go‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
package cli
2+
3+
import (
4+
"fmt"
5+
6+
"github.com/spf13/cobra"
7+
)
8+
9+
func (a *App) searchCmd() *cobra.Command {
10+
cmd := &cobra.Command{
11+
Use: "search <query>",
12+
Short: "Search UCI ML Repository datasets by name",
13+
Args: cobra.ExactArgs(1),
14+
RunE: func(cmd *cobra.Command, args []string) error {
15+
query := args[0]
16+
if query == "" {
17+
return codeError(exitUsage, fmt.Errorf("query cannot be empty"))
18+
}
19+
limit := a.effectiveLimit(0)
20+
datasets, err := a.client.Search(cmd.Context(), query, limit)
21+
if err != nil {
22+
return mapFetchErr(err)
23+
}
24+
return a.renderOrEmpty(datasets, len(datasets))
25+
},
26+
}
27+
return cmd
28+
}

‎cli/errors.go‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
package cli
2+
3+
func isNotFound(_ error) bool { return false }

‎cli/output.go‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
package cli
2+
3+
import (
4+
"io"
5+
6+
"github.com/tamnd/uci-cli/pkg/render"
7+
)
8+
9+
type Format = render.Format
10+
11+
const (
12+
FormatTable = render.FormatTable
13+
FormatJSON = render.FormatJSON
14+
FormatJSONL = render.FormatJSONL
15+
FormatCSV = render.FormatCSV
16+
FormatTSV = render.FormatTSV
17+
FormatURL = render.FormatURL
18+
FormatRaw = render.FormatRaw
19+
)
20+
21+
func NewRenderer(w io.Writer, format Format, fields []string, noHeader bool, tmpl string) *render.Renderer {
22+
return render.New(w, format, fields, noHeader, tmpl)
23+
}

‎cli/root.go‎

Lines changed: 117 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,30 +2,137 @@
22
package cli
33

44
import (
5+
"fmt"
6+
"os"
7+
8+
"github.com/mattn/go-isatty"
59
"github.com/spf13/cobra"
10+
"github.com/tamnd/uci-cli/uci"
611
)
712

8-
// Build metadata, set via -ldflags at release time.
913
var (
1014
Version = "dev"
1115
Commit = "none"
1216
Date = "unknown"
1317
)
1418

15-
// Root builds the root command and its subtree.
19+
const (
20+
exitError = 1
21+
exitUsage = 2
22+
exitNoData = 3
23+
)
24+
25+
type ExitError struct {
26+
Code int
27+
Err error
28+
}
29+
30+
func (e *ExitError) Error() string {
31+
if e.Err != nil {
32+
return e.Err.Error()
33+
}
34+
return fmt.Sprintf("exit %d", e.Code)
35+
}
36+
37+
func (e *ExitError) Unwrap() error { return e.Err }
38+
39+
func codeError(code int, err error) error { return &ExitError{Code: code, Err: err} }
40+
41+
type App struct {
42+
client *uci.Client
43+
cfg uci.Config
44+
45+
output string
46+
fields []string
47+
noHeader bool
48+
template string
49+
limit int
50+
quiet bool
51+
}
52+
1653
func Root() *cobra.Command {
17-
root := &cobra.Command{
18-
Use: "uci",
19-
Short: "A command line for uci.",
20-
Long: `A command line for uci.
54+
app := &App{cfg: uci.DefaultConfig()}
2155

22-
This is a fresh scaffold. Add your commands here on top of the uci
23-
library package, then wire them into Root with root.AddCommand.`,
56+
root := &cobra.Command{
57+
Use: "uci",
58+
Short: "Browse UCI Machine Learning Repository datasets",
2459
SilenceUsage: true,
2560
SilenceErrors: true,
61+
PersistentPreRunE: func(cmd *cobra.Command, _ []string) error {
62+
return app.setup()
63+
},
2664
}
2765

28-
root.AddCommand(newVersionCmd())
29-
// TODO: root.AddCommand(newGetCmd()), etc.
66+
pf := root.PersistentFlags()
67+
pf.StringVarP(&app.output, "output", "o", "auto", "output: table|json|jsonl|csv|tsv|url|raw")
68+
pf.StringSliceVar(&app.fields, "fields", nil, "comma-separated columns to include")
69+
pf.BoolVar(&app.noHeader, "no-header", false, "omit header row in table/csv/tsv")
70+
pf.StringVar(&app.template, "template", "", "Go text/template per record")
71+
pf.IntVarP(&app.limit, "limit", "n", 0, "limit number of records (0 = all)")
72+
pf.BoolVarP(&app.quiet, "quiet", "q", false, "suppress progress on stderr")
73+
pf.DurationVar(&app.cfg.Rate, "delay", app.cfg.Rate, "minimum spacing between requests")
74+
pf.DurationVar(&app.cfg.Timeout, "timeout", app.cfg.Timeout, "per-request timeout")
75+
pf.IntVar(&app.cfg.Retries, "retries", app.cfg.Retries, "retry attempts on 429/5xx")
76+
pf.StringVar(&app.cfg.UserAgent, "user-agent", app.cfg.UserAgent, "User-Agent header")
77+
78+
root.AddCommand(
79+
app.listCmd(),
80+
app.searchCmd(),
81+
newVersionCmd(),
82+
)
3083
return root
3184
}
85+
86+
func (a *App) setup() error {
87+
if a.output == "" || a.output == "auto" {
88+
if isatty.IsTerminal(os.Stdout.Fd()) {
89+
a.output = string(FormatTable)
90+
} else {
91+
a.output = string(FormatJSONL)
92+
}
93+
}
94+
if !Format(a.output).Valid() {
95+
return codeError(exitUsage, fmt.Errorf("unknown output format %q", a.output))
96+
}
97+
a.client = uci.NewClient(a.cfg)
98+
return nil
99+
}
100+
101+
func (a *App) render(records any) error {
102+
r := NewRenderer(os.Stdout, Format(a.output), a.fields, a.noHeader, a.template)
103+
return r.Render(records)
104+
}
105+
106+
func (a *App) renderOrEmpty(records any, n int) error {
107+
if err := a.render(records); err != nil {
108+
return err
109+
}
110+
if n == 0 {
111+
return codeError(exitNoData, nil)
112+
}
113+
return nil
114+
}
115+
116+
func (a *App) progressf(format string, args ...any) {
117+
if a.quiet {
118+
return
119+
}
120+
_, _ = fmt.Fprintf(os.Stderr, format+"\n", args...)
121+
}
122+
123+
func mapFetchErr(err error) error {
124+
if err == nil {
125+
return nil
126+
}
127+
if isNotFound(err) {
128+
return codeError(exitNoData, err)
129+
}
130+
return codeError(exitError, err)
131+
}
132+
133+
func (a *App) effectiveLimit(def int) int {
134+
if a.limit > 0 {
135+
return a.limit
136+
}
137+
return def
138+
}

‎go.mod‎

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,5 +4,34 @@ go 1.26
44

55
require (
66
github.com/charmbracelet/fang v1.0.0
7+
github.com/mattn/go-isatty v0.0.22
78
github.com/spf13/cobra v1.10.2
89
)
10+
11+
require (
12+
charm.land/lipgloss/v2 v2.0.0-beta.3.0.20251106193318-19329a3e8410 // indirect
13+
github.com/charmbracelet/colorprofile v0.3.3 // indirect
14+
github.com/charmbracelet/ultraviolet v0.0.0-20251106190538-99ea45596692 // indirect
15+
github.com/charmbracelet/x/ansi v0.11.0 // indirect
16+
github.com/charmbracelet/x/exp/charmtone v0.0.0-20250603201427-c31516f43444 // indirect
17+
github.com/charmbracelet/x/term v0.2.2 // indirect
18+
github.com/charmbracelet/x/termios v0.1.1 // indirect
19+
github.com/charmbracelet/x/windows v0.2.2 // indirect
20+
github.com/clipperhouse/displaywidth v0.4.1 // indirect
21+
github.com/clipperhouse/stringish v0.1.1 // indirect
22+
github.com/clipperhouse/uax29/v2 v2.3.0 // indirect
23+
github.com/inconshreveable/mousetrap v1.1.0 // indirect
24+
github.com/lucasb-eyer/go-colorful v1.3.0 // indirect
25+
github.com/mattn/go-runewidth v0.0.19 // indirect
26+
github.com/muesli/cancelreader v0.2.2 // indirect
27+
github.com/muesli/mango v0.1.0 // indirect
28+
github.com/muesli/mango-cobra v1.2.0 // indirect
29+
github.com/muesli/mango-pflag v0.1.0 // indirect
30+
github.com/muesli/roff v0.1.0 // indirect
31+
github.com/rivo/uniseg v0.4.7 // indirect
32+
github.com/spf13/pflag v1.0.9 // indirect
33+
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
34+
golang.org/x/sync v0.17.0 // indirect
35+
golang.org/x/sys v0.37.0 // indirect
36+
golang.org/x/text v0.24.0 // indirect
37+
)

‎go.sum‎

Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,74 @@
1+
charm.land/lipgloss/v2 v2.0.0-beta.3.0.20251106193318-19329a3e8410 h1:D9PbaszZYpB4nj+d6HTWr1onlmlyuGVNfL9gAi8iB3k=
2+
charm.land/lipgloss/v2 v2.0.0-beta.3.0.20251106193318-19329a3e8410/go.mod h1:1qZyvvVCenJO2M1ac2mX0yyiIZJoZmDM4DG4s0udJkU=
3+
github.com/aymanbagabas/go-udiff v0.3.1 h1:LV+qyBQ2pqe0u42ZsUEtPiCaUoqgA9gYRDs3vj1nolY=
4+
github.com/aymanbagabas/go-udiff v0.3.1/go.mod h1:G0fsKmG+P6ylD0r6N/KgQD/nWzgfnl8ZBcNLgcbrw8E=
5+
github.com/charmbracelet/colorprofile v0.3.3 h1:DjJzJtLP6/NZ8p7Cgjno0CKGr7wwRJGxWUwh2IyhfAI=
6+
github.com/charmbracelet/colorprofile v0.3.3/go.mod h1:nB1FugsAbzq284eJcjfah2nhdSLppN2NqvfotkfRYP4=
7+
github.com/charmbracelet/fang v1.0.0 h1:jESBY40agJOlLYnnv9jE0mLqDGTxEk0hkOnx7YGyRlQ=
8+
github.com/charmbracelet/fang v1.0.0/go.mod h1:P5/DNb9DddQ0Z0dbc0P3ol4/ix5Po7Ofr2KMBfAqoCo=
9+
github.com/charmbracelet/ultraviolet v0.0.0-20251106190538-99ea45596692 h1:r/3jQZ1LjWW6ybp8HHfhrKrwHIWiJhUuY7wwYIWZulQ=
10+
github.com/charmbracelet/ultraviolet v0.0.0-20251106190538-99ea45596692/go.mod h1:Y8B4DzWeTb0ama8l3+KyopZtkE8fZjwRQ3aEAPEXHE0=
11+
github.com/charmbracelet/x/ansi v0.11.0 h1:uuIVK7GIplwX6UBIz8S2TF8nkr7xRlygSsBRjSJqIvA=
12+
github.com/charmbracelet/x/ansi v0.11.0/go.mod h1:uQt8bOrq/xgXjlGcFMc8U2WYbnxyjrKhnvTQluvfCaE=
13+
github.com/charmbracelet/x/exp/charmtone v0.0.0-20250603201427-c31516f43444 h1:IJDiTgVE56gkAGfq0lBEloWgkXMk4hl/bmuPoicI4R0=
14+
github.com/charmbracelet/x/exp/charmtone v0.0.0-20250603201427-c31516f43444/go.mod h1:T9jr8CzFpjhFVHjNjKwbAD7KwBNyFnj2pntAO7F2zw0=
15+
github.com/charmbracelet/x/exp/golden v0.0.0-20250806222409-83e3a29d542f h1:pk6gmGpCE7F3FcjaOEKYriCvpmIN4+6OS/RD0vm4uIA=
16+
github.com/charmbracelet/x/exp/golden v0.0.0-20250806222409-83e3a29d542f/go.mod h1:IfZAMTHB6XkZSeXUqriemErjAWCCzT0LwjKFYCZyw0I=
17+
github.com/charmbracelet/x/term v0.2.2 h1:xVRT/S2ZcKdhhOuSP4t5cLi5o+JxklsoEObBSgfgZRk=
18+
github.com/charmbracelet/x/term v0.2.2/go.mod h1:kF8CY5RddLWrsgVwpw4kAa6TESp6EB5y3uxGLeCqzAI=
19+
github.com/charmbracelet/x/termios v0.1.1 h1:o3Q2bT8eqzGnGPOYheoYS8eEleT5ZVNYNy8JawjaNZY=
20+
github.com/charmbracelet/x/termios v0.1.1/go.mod h1:rB7fnv1TgOPOyyKRJ9o+AsTU/vK5WHJ2ivHeut/Pcwo=
21+
github.com/charmbracelet/x/windows v0.2.2 h1:IofanmuvaxnKHuV04sC0eBy/smG6kIKrWG2/jYn2GuM=
22+
github.com/charmbracelet/x/windows v0.2.2/go.mod h1:/8XtdKZzedat74NQFn0NGlGL4soHB0YQZrETF96h75k=
23+
github.com/clipperhouse/displaywidth v0.4.1 h1:uVw9V8UDfnggg3K2U84VWY1YLQ/x2aKSCtkRyYozfoU=
24+
github.com/clipperhouse/displaywidth v0.4.1/go.mod h1:R+kHuzaYWFkTm7xoMmK1lFydbci4X2CicfbGstSGg0o=
25+
github.com/clipperhouse/stringish v0.1.1 h1:+NSqMOr3GR6k1FdRhhnXrLfztGzuG+VuFDfatpWHKCs=
26+
github.com/clipperhouse/stringish v0.1.1/go.mod h1:v/WhFtE1q0ovMta2+m+UbpZ+2/HEXNWYXQgCt4hdOzA=
27+
github.com/clipperhouse/uax29/v2 v2.3.0 h1:SNdx9DVUqMoBuBoW3iLOj4FQv3dN5mDtuqwuhIGpJy4=
28+
github.com/clipperhouse/uax29/v2 v2.3.0/go.mod h1:Wn1g7MK6OoeDT0vL+Q0SQLDz/KpfsVRgg6W7ihQeh4g=
29+
github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
30+
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
31+
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
32+
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
33+
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
34+
github.com/lucasb-eyer/go-colorful v1.3.0 h1:2/yBRLdWBZKrf7gB40FoiKfAWYQ0lqNcbuQwVHXptag=
35+
github.com/lucasb-eyer/go-colorful v1.3.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
36+
github.com/mattn/go-isatty v0.0.22 h1:j8l17JJ9i6VGPUFUYoTUKPSgKe/83EYU2zBC7YNKMw4=
37+
github.com/mattn/go-isatty v0.0.22/go.mod h1:ZXfXG4SQHsB/w3ZeOYbR0PrPwLy+n6xiMrJlRFqopa4=
38+
github.com/mattn/go-runewidth v0.0.19 h1:v++JhqYnZuu5jSKrk9RbgF5v4CGUjqRfBm05byFGLdw=
39+
github.com/mattn/go-runewidth v0.0.19/go.mod h1:XBkDxAl56ILZc9knddidhrOlY5R/pDhgLpndooCuJAs=
40+
github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
41+
github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo=
42+
github.com/muesli/mango v0.1.0 h1:DZQK45d2gGbql1arsYA4vfg4d7I9Hfx5rX/GCmzsAvI=
43+
github.com/muesli/mango v0.1.0/go.mod h1:5XFpbC8jY5UUv89YQciiXNlbi+iJgt29VDC5xbzrLL4=
44+
github.com/muesli/mango-cobra v1.2.0 h1:DQvjzAM0PMZr85Iv9LIMaYISpTOliMEg+uMFtNbYvWg=
45+
github.com/muesli/mango-cobra v1.2.0/go.mod h1:vMJL54QytZAJhCT13LPVDfkvCUJ5/4jNUKF/8NC2UjA=
46+
github.com/muesli/mango-pflag v0.1.0 h1:UADqbYgpUyRoBja3g6LUL+3LErjpsOwaC9ywvBWe7Sg=
47+
github.com/muesli/mango-pflag v0.1.0/go.mod h1:YEQomTxaCUp8PrbhFh10UfbhbQrM/xJ4i2PB8VTLLW0=
48+
github.com/muesli/roff v0.1.0 h1:YD0lalCotmYuF5HhZliKWlIx7IEhiXeSfq7hNjFqGF8=
49+
github.com/muesli/roff v0.1.0/go.mod h1:pjAHQM9hdUUwm/krAfrLGgJkXJ+YuhtsfZ42kieB2Ig=
50+
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
51+
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
52+
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
53+
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
54+
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
55+
github.com/spf13/cobra v1.10.2 h1:DMTTonx5m65Ic0GOoRY2c16WCbHxOOw6xxezuLaBpcU=
56+
github.com/spf13/cobra v1.10.2/go.mod h1:7C1pvHqHw5A4vrJfjNwvOdzYu0Gml16OCs2GRiTUUS4=
57+
github.com/spf13/pflag v1.0.9 h1:9exaQaMOCwffKiiiYk6/BndUBv+iRViNW+4lEMi0PvY=
58+
github.com/spf13/pflag v1.0.9/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
59+
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
60+
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
61+
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
62+
github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
63+
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
64+
golang.org/x/exp v0.0.0-20231006140011-7918f672742d h1:jtJma62tbqLibJ5sFQz8bKtEM8rJBtfilJ2qTU199MI=
65+
golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQzzkH36rKxoZW1tw7ZJpeKx+hdo=
66+
golang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug=
67+
golang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
68+
golang.org/x/sys v0.37.0 h1:fdNQudmxPjkdUTPnLn5mdQv7Zwvbvpaxqs831goi9kQ=
69+
golang.org/x/sys v0.37.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
70+
golang.org/x/text v0.24.0 h1:dd5Bzh4yt5KYA8f9CJHCP4FB4D51c2c6JvN37xJJkJ0=
71+
golang.org/x/text v0.24.0/go.mod h1:L8rBsPeo2pSS+xqN0d5u2ikmjtmoJbDBT1b7nHvFCdU=
72+
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
73+
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
74+
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

0 commit comments

Comments
 (0)