initial commit

README.md

@@ -0,0 +1,3 @@
+Requires `rofi-blocks-git` from the AUR. 
+
+See [Rofi-blocks](https://github.com/OmarCastro/rofi-blocks) for doc.

main.go

@@ -0,0 +1,126 @@
+package main
+
+import (
+	"bufio"
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"os"
+	"os/exec"
+)
+
+type Output struct {
+	Message     string `json:"message,omitempty"`      // Sets rofi message, hides it if empty or null
+	Overlay     string `json:"overlay,omitempty"`      // Shows overlay with text, hides it if empty or null
+	Prompt      string `json:"prompt,omitempty"`       // sets prompt text
+	Input       string `json:"input,omitempty"`        // sets input text, to clear use empty string
+	InputAction string `json:"input action,omitempty"` // "send" or "filter"
+	EventFormat string `json:"event format,omitempty"` // event format used to send to input
+	Lines       []Line `json:"lines"`                  // a list of sting or json object to set rofi list content
+}
+
+type Line struct {
+	Text      string `json:"text,required"`
+	Urgent    bool   `json:"urgent,omitempty"`
+	Highlight bool   `json:"highlight,omitempty"`
+	Markup    bool   `json:"markup,omitempty"`
+	Icon      string `json:"icon,omitempty"`
+}
+
+type Input struct {
+	Name  string `json:"name"`
+	Value string `json:"value"`
+}
+
+func handleSelect(input Input) {
+	param := url.Values{}
+	param.Add("q", input.Value)
+	exec.Command("xdg-open", "https://duckduckgo.com/?"+param.Encode()).Start()
+	os.Exit(0)
+}
+
+func getSuggestions(query string) []string {
+	param := url.Values{}
+	param.Add("q", query)
+	param.Add("type", "list")
+	param.Add("kl", "en-US")
+
+	resp, _ := http.Get("https://ac.duckduckgo.com/ac?" + param.Encode())
+	body, _ := ioutil.ReadAll(resp.Body)
+	resp.Body.Close()
+
+	var result []interface{}
+	json.Unmarshal(body, &result)
+
+	if len(result) < 2 {
+		return []string{}
+	}
+
+	rtn := make([]string, len(result[1].([]interface{})))
+	for i, v := range result[1].([]interface{}) {
+		rtn[i] = v.(string)
+	}
+
+	return rtn
+}
+
+func convertStringsToLine(strings []string) []Line {
+	lines := make([]Line, len(strings))
+	for i, v := range strings {
+		lines[i] = Line{v, false, false, false, ""}
+	}
+	return lines
+}
+
+func handleInput(input Input) {
+	switch input.Name {
+	case "input change":
+		output := newOutput()
+		output.Lines = convertStringsToLine(getSuggestions(input.Value))
+		sendOutput(output)
+	case "custom key":
+	case "active entry":
+	case "select entry":
+		handleSelect(input)
+	case "delete entry":
+	case "exec custom input":
+	}
+}
+
+func sendOutput(output Output) {
+	res, _ := json.Marshal(output)
+	fmt.Printf("%s\n", res)
+}
+
+func simpleLines(lines []string) []Line {
+	outputLines := make([]Line, len(lines))
+
+	for i := 0; i < len(lines); i++ {
+		outputLines[i] = Line{lines[i], false, false, false, ""}
+	}
+
+	return outputLines
+}
+
+func newOutput() Output {
+	return Output{"", "", "", "", "", "", []Line{}}
+}
+
+func main() {
+	// Setup rofi
+	initialOutput := newOutput()
+	initialOutput.Prompt = "Search"
+	initialOutput.InputAction = "send"
+	sendOutput(initialOutput)
+
+	// Input loop
+	scanner := bufio.NewScanner(os.Stdin)
+	for scanner.Scan() {
+		var input Input
+		bytes := scanner.Bytes()
+		json.Unmarshal(bytes, &input)
+		handleInput(input)
+	}
+}