127 lines
3.0 KiB
Go
127 lines
3.0 KiB
Go
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)
|
|
}
|
|
}
|