-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
138 lines (120 loc) · 3.22 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
package main
import (
"encoding/json"
"flag"
"fmt"
"log"
"math/rand"
"net"
"net/http"
"net/url"
"strings"
"time"
"github.com/gorilla/websocket"
)
// generate uniq id for each call
func getID(strlen int) string {
const chars = "abcdefghijklmnopqrstuvwxyz0123456789"
r := rand.New(rand.NewSource(time.Now().UnixNano()))
result := make([]byte, strlen)
for i := range result {
result[i] = chars[r.Intn(len(chars))]
}
return string(result)
}
func command(addr net.IP, cmd *request, data *LGPair) []byte {
client := connect(addr, data)
defer client.Close()
dict, err := json.Marshal(cmd)
if err != nil {
log.Fatal(err.Error())
}
log.Println("Payload: ", string(dict[:]))
err = client.WriteMessage(websocket.TextMessage, dict)
if err != nil {
log.Println("Try again...", err)
}
_, message, err := client.ReadMessage()
if err != nil {
log.Println("Error while reading the message from server:", err)
}
log.Printf("Response from WebOS: %s", message)
return message
}
// Initialize websocket on each command call
func connect(addr net.IP, data *LGPair) *websocket.Conn {
dst := fmt.Sprintf("%s:3000", addr)
u := url.URL{Scheme: "ws", Host: dst}
log.Printf("connecting to %s", u.String())
client, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
if err != nil {
log.Println("dial:", err)
}
register(client, data)
return client
}
func main() {
username := flag.String("username", "username", "REST API username")
password := flag.String("password", "password", "REST API password")
flag.Parse()
// Prepare payload for initial
data := LGPair{
ForcePairing: false,
PairingType: "PROMPT",
}
data.WebosHandshake()
// Addr of webos device
addr := webosDiscovery()
http.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
user, pass, _ := req.BasicAuth()
if user != *username || pass != *password {
http.Error(w, "Unauthorized.", 401)
return
}
/// Parse JSON's POST request
decoder := json.NewDecoder(req.Body)
var post map[string]interface{}
err := decoder.Decode(&post)
if err != nil {
log.Println("Can not decode json, ", err)
return
}
ssap := fmt.Sprintf("ssap://%v", req.URL.Path[1:])
cmd := &request{Type: "request", ID: getID(32), URI: ssap}
for key := range post {
switch key {
case "volume":
if volume, ok := post[key].(float64); ok {
cmd.Payload.Volume = int(volume)
}
case "id":
if id, ok := post[key].(string); ok {
cmd.URI = "ssap://com.webos.applicationManager/listLaunchPoints"
apps := &Receiver{}
json.Unmarshal(command(addr.IP, cmd, &data), apps)
for _, app := range apps.Payload.LaunchPoints {
if strings.Contains(strings.ToLower(app.Title), strings.ToLower(id)) {
fmt.Println(app.Title, app.ID)
cmd = &request{Type: "request", ID: getID(32), URI: ssap}
cmd.Payload.ID = app.ID
break
}
}
}
case "mute":
if mute, ok := post[key].(bool); ok {
cmd.Payload.Mute = mute
}
case "message":
if message, ok := post[key].(string); ok {
cmd.Payload.Message = message
}
}
}
command(addr.IP, cmd, &data)
})
err := http.ListenAndServeTLS(":8888", "server.crt", "server.key", nil)
if err != nil {
log.Fatal("ListenAndServe: ", err)
}
}