This commit is contained in:
Simon Vieille 2023-08-24 18:41:12 +02:00
commit 6c524e2c41
5 changed files with 345 additions and 0 deletions

23
actions.go Normal file
View file

@ -0,0 +1,23 @@
package main
import (
"golang.org/x/net/websocket"
)
type Actions struct {
Functions map[string]func(ws *websocket.Conn, msg string) error
}
func (actions Actions) add(name string, callback func(ws *websocket.Conn, msg string) error) {
actions.Functions[name] = callback
}
func (actions Actions) has(name string) bool {
_, exists := actions.Functions[name]
return exists
}
func (actions Actions) exec(name string, ws *websocket.Conn, msg string) error {
return actions.Functions[name](ws, msg)
}

18
go.mod Normal file
View file

@ -0,0 +1,18 @@
module gitnet.fr/deblan/remote-i3wm-go
go 1.21.0
require (
github.com/golang-jwt/jwt v3.2.2+incompatible // indirect
github.com/labstack/echo/v4 v4.11.1 // indirect
github.com/labstack/gommon v0.4.0 // indirect
github.com/mattn/go-colorable v0.1.13 // indirect
github.com/mattn/go-isatty v0.0.19 // indirect
github.com/valyala/bytebufferpool v1.0.0 // indirect
github.com/valyala/fasttemplate v1.2.2 // indirect
golang.org/x/crypto v0.12.0 // indirect
golang.org/x/net v0.14.0 // indirect
golang.org/x/sys v0.11.0 // indirect
golang.org/x/text v0.12.0 // indirect
golang.org/x/time v0.3.0 // indirect
)

49
go.sum Normal file
View file

@ -0,0 +1,49 @@
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
github.com/labstack/echo/v4 v4.11.1 h1:dEpLU2FLg4UVmvCGPuk/APjlH6GDpbEPti61srUUUs4=
github.com/labstack/echo/v4 v4.11.1/go.mod h1:YuYRTSM3CHs2ybfrL8Px48bO6BAnYIN4l8wSTMP6BDQ=
github.com/labstack/gommon v0.4.0 h1:y7cvthEAEbU0yHOf4axH8ZG2NH8knB9iNSoTO8dyIk8=
github.com/labstack/gommon v0.4.0/go.mod h1:uW6kP17uPlLJsD3ijUYn3/M5bAxtlZhMI6m3MFxTMTM=
github.com/mattn/go-colorable v0.1.11/go.mod h1:u5H1YNBxpqRaxsYJYSkiCWKzEfiAb1Gb520KVy5xxl4=
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
github.com/valyala/fasttemplate v1.2.1/go.mod h1:KHLXt3tVN2HBp8eijSv/kGJopbvo7S+qRAEEKiv+SiQ=
github.com/valyala/fasttemplate v1.2.2 h1:lxLXG0uE3Qnshl9QyaK6XJxMXlQZELvChBOCmQD0Loo=
github.com/valyala/fasttemplate v1.2.2/go.mod h1:KHLXt3tVN2HBp8eijSv/kGJopbvo7S+qRAEEKiv+SiQ=
golang.org/x/crypto v0.11.0 h1:6Ewdq3tDic1mg5xRO4milcWCfMVQhI4NkqWWvqejpuA=
golang.org/x/crypto v0.11.0/go.mod h1:xgJhtzW8F9jGdVFWZESrid1U1bjeNy4zgy5cRr/CIio=
golang.org/x/crypto v0.12.0 h1:tFM/ta59kqch6LlvYnPa0yx5a83cL2nHflFhYKvv9Yk=
golang.org/x/crypto v0.12.0/go.mod h1:NF0Gs7EO5K4qLn+Ylc+fih8BSTeIjAP05siRnAh98yw=
golang.org/x/net v0.12.0 h1:cfawfvKITfUsFCeJIHJrbSxpeu/E81khclypR0GVT50=
golang.org/x/net v0.12.0/go.mod h1:zEVYFnQC7m/vmpQFELhcD1EWkZlX69l4oqgmer6hfKA=
golang.org/x/net v0.14.0 h1:BONx9s002vGdD9umnlX1Po8vOZmrgH34qlHcD1MfK14=
golang.org/x/net v0.14.0/go.mod h1:PpSgVXXLK0OxS0F31C1/tv6XNguvCrnXIDrFMspZIUI=
golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20211103235746-7861aae1554b/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.10.0 h1:SqMFp9UcQJZa+pmYuAKjd9xq1f0j5rLcDIk0mj4qAsA=
golang.org/x/sys v0.10.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM=
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/text v0.11.0 h1:LAntKIrcmeSKERyiOh0XMV39LXS8IE9UL2yP7+f5ij4=
golang.org/x/text v0.11.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
golang.org/x/text v0.12.0 h1:k+n5B8goJNdU7hSvEtMUz3d1Q6D/XW4COJSJR6fN0mc=
golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
golang.org/x/time v0.3.0 h1:rg5rLMjNzMS1RkNLzCG38eapWhnYLFYXDXj2gOlr8j4=
golang.org/x/time v0.3.0/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

255
main.go Normal file
View file

@ -0,0 +1,255 @@
package main
import (
"crypto/subtle"
"encoding/json"
"errors"
"fmt"
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
"golang.org/x/net/websocket"
"net/http"
"os/exec"
"strconv"
"strings"
)
type Message struct {
Type string `json:type`
}
type SimpleMessageData struct {
Value string `json:type`
}
type PointerMessageData struct {
X string `json:x`
Y string `json:y`
Click string `json:click`
}
func getSimpleMessageValue(msg string) string {
data := SimpleMessageData{}
json.Unmarshal([]byte(msg), &data)
return data.Value
}
func ws(c echo.Context) error {
var actions = Actions{
Functions: make(map[string]func(ws *websocket.Conn, msg string) error),
}
actions.add("pointer", func(ws *websocket.Conn, msg string) error {
data := PointerMessageData{}
json.Unmarshal([]byte(msg), &data)
if data.Click != "" {
keys := make(map[string]string)
keys["left"] = "1"
keys["middle"] = "2"
keys["right"] = "3"
key, exists := keys[data.Click]
if !exists {
return errors.New("Invalid value")
}
cmd := exec.Command("xdotool", "click", key)
return cmd.Run()
}
location := exec.Command("xdotool", "getmouselocation")
output, _ := location.Output()
position := string(output)
currentX := 0.0
currentY := 0.0
for key, value := range strings.Split(position, " ") {
if key == 0 {
currentX, _ = strconv.ParseFloat(strings.Replace(value, "x:", "", 1), 32)
} else if key == 1 {
currentY, _ = strconv.ParseFloat(strings.Replace(value, "y:", "", 1), 32)
}
}
fmt.Printf("%+v\n", currentX)
fmt.Printf("%+v\n", currentY)
newX, _ := strconv.ParseFloat(data.X, 32)
newY, _ := strconv.ParseFloat(data.Y, 32)
x := currentX + newX*2.5
y := currentY + newY*2.5
cmd := exec.Command("xdotool", "mousemove", fmt.Sprintf("%.0f", x), fmt.Sprintf("%.0f", y))
return cmd.Run()
})
actions.add("scroll", func(ws *websocket.Conn, msg string) error {
value := getSimpleMessageValue(msg)
key := ""
if value == "down" {
key = "5"
} else if value == "up" {
key = "4"
} else {
return errors.New("Invalid value")
}
for i := 0; i < 2; i++ {
cmd := exec.Command("xdotool", "click", key)
cmd.Run()
}
return nil
})
actions.add("workspace", func(ws *websocket.Conn, msg string) error {
value := getSimpleMessageValue(msg)
if value == "" {
return errors.New("Invalid value")
}
cmd := exec.Command("i3-msg", fmt.Sprintf("workspace \"%s\"", value))
return cmd.Run()
})
actions.add("volume", func(ws *websocket.Conn, msg string) error {
return nil
})
actions.add("media", func(ws *websocket.Conn, msg string) error {
return nil
})
actions.add("keys", func(ws *websocket.Conn, msg string) error {
value := strings.TrimSpace(getSimpleMessageValue(msg))
if value == "" {
return errors.New("Invalid value")
}
keys := []string{}
for _, key := range strings.Split(value, ",") {
if key == "win" {
key = "super"
} else if key == "ctrl" {
key = "Control_L"
} else if key == "alt" {
key = "Alt_L"
}
if key != "" {
keys = append(keys, key)
}
}
if len(keys) == 0 {
return errors.New("Invalid value")
}
cmd := exec.Command("xdotool", "key", strings.Join(keys, "+"))
return cmd.Run()
})
actions.add("key", func(ws *websocket.Conn, msg string) error {
value := strings.TrimSpace(getSimpleMessageValue(msg))
keys := make(map[string]string)
keys["up"] = "Up"
keys["down"] = "Down"
keys["left"] = "Left"
keys["right"] = "Right"
keys["tab"] = "Tab"
keys["backspace"] = "BackSpace"
keys["enter"] = "Return"
keys["space"] = "space"
keys["escape"] = "Escape"
key, exists := keys[value]
if !exists {
return errors.New("Invalid value")
}
cmd := exec.Command("xdotool", "key", key)
return cmd.Run()
})
actions.add("text", func(ws *websocket.Conn, msg string) error {
value := strings.TrimSpace(getSimpleMessageValue(msg))
if value == "" {
return errors.New("Invalid value")
}
cmd := exec.Command("xdotool", "type", value)
return cmd.Run()
})
actions.add("screenshot", func(ws *websocket.Conn, msg string) error {
return nil
})
actions.add("messages", func(ws *websocket.Conn, msg string) error {
return nil
})
websocket.Handler(func(ws *websocket.Conn) {
defer ws.Close()
for {
// Write
// err := websocket.Message.Send(ws, "Hello, Client!")
// if err != nil {
// c.Logger().Error(err)
// }
msg := ""
websocket.Message.Receive(ws, &msg)
message := Message{}
json.Unmarshal([]byte(msg), &message)
if message.Type != "" && actions.has(message.Type) {
fmt.Printf("%+v\n", msg)
err := actions.exec(message.Type, ws, msg)
fmt.Printf("%+v\n", err)
}
}
}).ServeHTTP(c.Response(), c.Request())
return nil
}
func main() {
e := echo.New()
e.Use(middleware.BasicAuth(func(username, password string, c echo.Context) (bool, error) {
if subtle.ConstantTimeCompare([]byte(username), []byte("admin")) == 1 &&
subtle.ConstantTimeCompare([]byte(password), []byte("admin")) == 1 {
return true, nil
}
return false, nil
}))
e.GET("/", func(c echo.Context) error {
return c.String(http.StatusOK, "Hello, World!")
})
e.GET("/ws", ws)
e.Logger.Fatal(e.Start(":4000"))
}

BIN
remote-i3wm-go Executable file

Binary file not shown.