88 lines
2 KiB
Go
88 lines
2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os/exec"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func selector(history []string, max int, tool string) error {
|
|
// reverse the history
|
|
for i, j := 0, len(history)-1; i < j; i, j = i+1, j-1 {
|
|
history[i], history[j] = history[j], history[i]
|
|
}
|
|
|
|
selected, err := dmenu(history, max, tool)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// serve selection to the OS
|
|
err = exec.Command("wl-copy", []string{"--", selected}...).Run()
|
|
|
|
return err
|
|
}
|
|
|
|
func dmenu(list []string, max int, tool string) (string, error) {
|
|
if len(list) == 0 {
|
|
return "", nil
|
|
}
|
|
|
|
bin, err := exec.LookPath(tool)
|
|
if err != nil {
|
|
return "", fmt.Errorf("%s is not installed", tool)
|
|
}
|
|
|
|
var args []string
|
|
if tool == "dmenu" {
|
|
args = []string{"dmenu", "-b",
|
|
"-fn",
|
|
"-misc-dejavu sans mono-medium-r-normal--17-120-100-100-m-0-iso8859-16",
|
|
"-l",
|
|
strconv.Itoa(max)}
|
|
} else {
|
|
args = []string{"rofi", "-dmenu",
|
|
"-lines",
|
|
strconv.Itoa(max)}
|
|
}
|
|
|
|
// dmenu will break if items contain newlines, so we must pass them as literals.
|
|
// however, when it sends them back, we need a way to restore them
|
|
var escaped []string
|
|
guide := make(map[string]string)
|
|
for _, original := range list {
|
|
repr := fmt.Sprintf("%#v", original)
|
|
|
|
// dmenu will split lines longer than 1200 something; we cut at 400 to spare memory
|
|
max := len(repr) - 1 // drop right quote
|
|
maxChars := 400
|
|
if max > maxChars {
|
|
max = maxChars
|
|
}
|
|
repr = repr[1:max] // drop left quote
|
|
|
|
guide[repr] = original
|
|
escaped = append(escaped, repr)
|
|
}
|
|
|
|
input := strings.NewReader(strings.Join(escaped, "\n"))
|
|
|
|
cmd := exec.Cmd{Path: bin, Args: args, Stdin: input}
|
|
selected, err := cmd.Output()
|
|
if err != nil {
|
|
if err.Error() == "exit status 1" {
|
|
// dmenu exits with this error when no selection done
|
|
return "", nil
|
|
}
|
|
return "", err
|
|
}
|
|
trimmed := selected[:len(selected)-1] // drop newline added by dmenu
|
|
|
|
sel, ok := guide[string(trimmed)]
|
|
if !ok {
|
|
return "", fmt.Errorf("couldn't recover original string; please report this bug along with a copy of your clipman.json")
|
|
}
|
|
|
|
return sel, nil
|
|
}
|