add hm-search (home-manager)

3 files changed, 212 insertions, 0 deletions

diff --git a/cmd/hm-search/main.go b/cmd/hm-search/main.go
new file mode 100644
index 0000000..e2a43dc
--- /dev/null
+++ b/cmd/hm-search/main.go
@@ -0,0 +1,51 @@
+package main
+
+import (
+        "flag"
+        "fmt"
+        "log"
+        "strings"
+
+        colors "git.sixfoisneuf.fr/nix-tools"
+)
+
+var (
+        source   = flag.String("source", "<home-manager>", "Path to the home-manager folder")
+        examples = flag.Bool("examples", false, "Display examples")
+        desc     = flag.Bool("desc", false, "Also search in module description")
+)
+
+func main() {
+        flag.Parse()
+
+        query := flag.Arg(0)
+
+        options, err := LoadOptions(*source)
+        if err != nil {
+                log.Fatalf("error loading options: %v", err)
+        }
+
+        for _, o := range options {
+                if !strings.Contains(strings.ToLower(o.Name), strings.ToLower(query)) {
+                        if !*desc {
+                                continue
+                        }
+
+                        if !strings.Contains(strings.ToLower(o.Description), strings.ToLower(query)) {
+                                continue
+                        }
+                }
+
+                colors.Red.Printf("%s", o.Name)
+                if o.Default.Type != "" {
+                        fmt.Printf(" %s", o.Default)
+                }
+                fmt.Printf("\n%s", o.RenderDescription())
+
+                if o.Example.Type != "" && *examples {
+                        fmt.Println(o.Example)
+                }
+
+                fmt.Println()
+        }
+}
diff --git a/cmd/hm-search/options.go b/cmd/hm-search/options.go
new file mode 100644
index 0000000..accd2af
--- /dev/null
+++ b/cmd/hm-search/options.go
@@ -0,0 +1,137 @@
+package main
+
+import (
+        "encoding/json"
+        "fmt"
+        "os"
+        "strings"
+
+        nix "git.sixfoisneuf.fr/nix-tools/nix"
+)
+
+type Option struct {
+        Name        string
+        Declaration []string
+        Description string
+        ReadOnly    bool
+        Type        string
+        Default     DefaultValue
+        Example     Example
+}
+
+type Example struct {
+        Type string
+        Text string
+}
+
+type DefaultValue struct {
+        Type string
+        Text string
+}
+
+var optionsExpr = `
+let
+  hm = import <home-manager> {};  
+in
+hm.docs.json
+`
+
+func LoadOptions(nixpkgs string) ([]Option, error) {
+        pkgs, err := nix.LoadNixpkgs(nixpkgs)
+        if err != nil {
+                return nil, fmt.Errorf("loading '%s': %v", nixpkgs, err)
+        }
+
+        drv, err := pkgs.EvalString(optionsExpr)
+        if err != nil {
+                return nil, fmt.Errorf("running Nix query: %v", err)
+        }
+
+        path, err := nix.BuildDerivation(drv)
+        if err != nil {
+                return nil, fmt.Errorf("building options.json: %v", err)
+        }
+
+        path = strings.TrimSpace(path)
+        path = path + "/share/doc/home-manager/options.json"
+
+        var res map[string]map[string]interface{}
+        data, err := os.ReadFile(path)
+        if err != nil {
+                return nil, fmt.Errorf("reading %s: %v", path, err)
+        }
+
+        err = json.Unmarshal(data, &res)
+        if err != nil {
+                return nil, fmt.Errorf("parsing JSON in %s: %v", path, err)
+        }
+
+        var options []Option
+        for option, params := range res {
+                o := Option{
+                        Name: option,
+                }
+
+                if desc, ok := params["description"]; ok {
+                        if desc, ok := desc.(string); ok {
+                                o.Description = desc
+                        }
+                }
+
+                if expl, ok := params["example"]; ok {
+                        expl := expl.(map[string]interface{})
+                        o.Example.Type = expl["_type"].(string)
+                        o.Example.Text = expl["text"].(string)
+                }
+
+                if def, ok := params["default"]; ok {
+                        def := def.(map[string]interface{})
+                        o.Default.Type = def["_type"].(string)
+                        o.Default.Text = def["text"].(string)
+                }
+
+                options = append(options, o)
+        }
+
+        return options, nil
+}
+
+func (e Example) String() string {
+        sb := strings.Builder{}
+
+        sb.WriteString(" | [Example]\n")
+        sb.WriteString(" | Type: ")
+        sb.WriteString(e.Type)
+        sb.WriteString("\n")
+        sb.WriteString(" | \n")
+
+        parts := strings.Split(e.Text, "\n")
+        for _, p := range parts {
+                sb.WriteString(" | ")
+                sb.WriteString(p)
+                sb.WriteString("\n")
+        }
+
+        return sb.String()
+}
+
+func (d DefaultValue) String() string {
+        return fmt.Sprintf("(default: %s)", d.Text)
+}
+
+func (o Option) RenderDescription() string {
+        sb := strings.Builder{}
+
+        o.Description = strings.Trim(o.Description, "\n")
+
+        parts := strings.Split(o.Description, "\n")
+        for _, p := range parts {
+                p = strings.TrimSpace(p)
+                sb.WriteString("  ")
+                sb.WriteString(p)
+
+                sb.WriteString("\n")
+        }
+
+        return sb.String()
+}
diff --git a/nix/drv.go b/nix/drv.go
new file mode 100644
index 0000000..3d92dab
--- /dev/null
+++ b/nix/drv.go
@@ -0,0 +1,24 @@
+package nix
+
+import (
+        "bytes"
+        "fmt"
+        "os/exec"
+)
+
+func BuildDerivation(path string) (string, error) {
+        cmd := exec.Command("nix-store", "-r", path)
+
+        stdout := bytes.Buffer{}
+        stderr := bytes.Buffer{}
+
+        cmd.Stdout = &stdout
+        cmd.Stderr = &stderr
+
+        err := cmd.Run()
+        if err != nil {
+                return "", fmt.Errorf("realising '%s': %v\n\"\"\"\n%s\n\"\"\"", path, err, string(stderr.Bytes()))
+        }
+
+        return string(stdout.Bytes()), nil
+}