VERSE
ffa336196a
This pull request modernizes the codebase by adopting Go 1.21+ features, including built-in functions like min and max, standard library iterators such as strings.SplitSeq and strings.FieldsSeq, utility functions like maps.Copy and slices.Contains, and string manipulation helpers like strings.Cut and strings.CutPrefix. Feedback is provided on a test file where the usage of strings.Builder can be further optimized to avoid unnecessary string allocations during concatenation. `go fix -diff ./...`
82 lines
2.5 KiB
Go
82 lines
2.5 KiB
Go
package python
|
|
|
|
import (
|
|
"context"
|
|
"os/exec"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/versenilvis/iris/commands/core"
|
|
)
|
|
|
|
func pipPackageGenerator(tokens []string, _ string, _ string) []core.Suggestion {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
cmd := exec.CommandContext(ctx, "pip", "list", "--format=freeze")
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
// try pip3
|
|
cmd = exec.CommandContext(ctx, "pip3", "list", "--format=freeze")
|
|
out, err = cmd.Output()
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
var results []core.Suggestion
|
|
for line := range strings.SplitSeq(string(out), "\n") {
|
|
line = strings.TrimSpace(line)
|
|
if line == "" {
|
|
continue
|
|
}
|
|
// format is "package==version"
|
|
parts := strings.SplitN(line, "==", 2)
|
|
name := parts[0]
|
|
desc := ""
|
|
if len(parts) == 2 {
|
|
desc = "v" + parts[1]
|
|
}
|
|
results = append(results, core.Suggestion{Cmd: name, Desc: desc})
|
|
}
|
|
return results
|
|
}
|
|
|
|
func makePipSpec(name string) *core.Spec {
|
|
return &core.Spec{
|
|
Name: name,
|
|
Description: "python packages",
|
|
Subcommands: []core.Subcommand{
|
|
{Name: "install", Description: "install package", Options: []core.Option{
|
|
{Name: "-r", Description: "from requirements file"},
|
|
{Name: "-U", Description: "upgrade"},
|
|
{Name: "--user", Description: "install for user only"},
|
|
{Name: "--editable", Description: "editable install"},
|
|
{Name: "--index-url", Description: "custom package index"},
|
|
{Name: "--no-deps", Description: "skip dependencies"},
|
|
}},
|
|
{Name: "uninstall", Description: "remove package", Generator: pipPackageGenerator, Options: []core.Option{
|
|
{Name: "-y", Description: "yes to all prompts"},
|
|
}},
|
|
{Name: "freeze", Description: "list installed packages"},
|
|
{Name: "list", Description: "list installed", Options: []core.Option{
|
|
{Name: "--outdated", Description: "show outdated"},
|
|
{Name: "--format", Description: "output format"},
|
|
}},
|
|
{Name: "show", Description: "show package info", Generator: pipPackageGenerator},
|
|
{Name: "download", Description: "download packages"},
|
|
{Name: "check", Description: "verify compatibility"},
|
|
{Name: "config", Description: "manage config"},
|
|
{Name: "search", Description: "search pypi"},
|
|
{Name: "wheel", Description: "build wheel archive"},
|
|
{Name: "hash", Description: "compute hashes"},
|
|
{Name: "completion", Description: "generate completion"},
|
|
},
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
core.Register(makePipSpec("pip"))
|
|
core.Register(makePipSpec("pip3"))
|
|
}
|