swich to i3ipc library

go.mod

@@ -9,6 +9,7 @@ require (
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
 	github.com/spf13/pflag v1.0.7 // indirect
+	github.com/tlinden/i3ipc v0.0.0-20250815101608-4f7e27528be3 // indirect
 	github.com/tlinden/yadu v0.1.3 // indirect
 	golang.org/x/sys v0.14.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect

go.sum

@@ -11,6 +11,8 @@ github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWE
 github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/spf13/pflag v1.0.7 h1:vN6T9TfwStFPFM5XzjsvmzZkLuaLX+HS+0SeFLRgU6M=
 github.com/spf13/pflag v1.0.7/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/tlinden/i3ipc v0.0.0-20250815101608-4f7e27528be3 h1:/kIZO4852sAVemXtqnsBid0r4Q1h87jDwHa8f7v1h5I=
+github.com/tlinden/i3ipc v0.0.0-20250815101608-4f7e27528be3/go.mod h1:mc0toDHmgqgX6FpE69U5yMPnHuLTdekHRslSLDp8xSE=
 github.com/tlinden/yadu v0.1.3 h1:5cRCUmj+l5yvlM2irtpFBIJwVV2DPEgYSaWvF19FtcY=
 github.com/tlinden/yadu v0.1.3/go.mod h1:l3bRmHKL9zGAR6pnBHY2HRPxBecf7L74BoBgOOpTcUA=
 golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=

main.go

@@ -18,41 +18,21 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
 package main
 
 import (
-	"encoding/binary"
-	"encoding/json"
 	"fmt"
 	"io"
 	"log"
 	"log/slog"
-	"net"
 	"os"
 	"runtime/debug"
 
 	"github.com/lmittmann/tint"
 	"github.com/mattn/go-isatty"
+	"github.com/tlinden/i3ipc"
 	"github.com/tlinden/yadu"
 
 	flag "github.com/spf13/pflag"
 )
 
-type Node struct {
-	Id                int    `json:"id"`
-	Nodetype          string `json:"type"` // output, workspace or container
-	Name              string `json:"name"` // workspace number or app name
-	Nodes             []Node `json:"nodes"`
-	FloatingNodes     []Node `json:"floating_nodes"`
-	Focused           bool   `json:"focused"`
-	Window            int    `json:"window"` // wayland native
-	X11Window         string `json:"app_id"` // x11 compat
-	Current_workspace string `json:"current_workspace"`
-}
-
-type Response struct {
-	Success    bool   `json:"success"`
-	ParseError bool   `json:"parse_error"`
-	Error      string `json:"error"`
-}
-
 const (
 	root = iota + 1
 	output
@@ -62,7 +42,7 @@ const (
 
 	LevelNotice = slog.Level(2)
 
-	VERSION = "v0.2.0"
+	VERSION = "v0.3.0"
 
 	IPC_HEADER_SIZE = 14
 	IPC_MAGIC       = "i3-ipc"
@@ -73,7 +53,7 @@ const (
 )
 
 var (
-	Visibles         = []Node{}
+	Visibles         = []*i3ipc.Node{}
 	CurrentWorkspace = ""
 	Debug            = false
 	Dumptree         = false
@@ -132,19 +112,21 @@ func main() {
 	}
 
 	// connect to sway unix socket
-	unixsock, err := setupIPC()
+	ipc := i3ipc.NewI3ipc()
-	if err != nil {
-		log.Fatalf("Failed to connect to sway unix socket: %s", err)
-	}
 
-	// retrieve the raw json tree
+	err := ipc.Connect()
-	rawjson, err := getTree(unixsock)
 	if err != nil {
-		log.Fatalf("Failed to retrieve raw json tree: %s", err)
+		log.Fatal(err)
+	}
+	defer ipc.Close()
+
+	sway, err := ipc.GetTree()
+	if err != nil {
+		log.Fatal(err)
 	}
 
 	// traverse the tree and find visible windows
-	if err := processJSON(rawjson); err != nil {
+	if err := processJSON(sway); err != nil {
 		log.Fatalf("%s", err)
 	}
 
@@ -156,110 +138,13 @@ func main() {
 	slog.Debug("findNextWindow", "nextid", id)
 
 	if id > 0 && !Notswitch {
-		switchFocus(id, unixsock)
+		switchFocus(id, ipc)
 	}
 }
 
-// connect to unix socket
-func setupIPC() (net.Conn, error) {
-	sockfile := os.Getenv("SWAYSOCK")
-
-	if sockfile == "" {
-		return nil, fmt.Errorf("Environment variable SWAYSOCK does not exist or is empty")
-	}
-
-	conn, err := net.Dial("unix", sockfile)
-	if err != nil {
-		return nil, err
-	}
-
-	return conn, nil
-}
-
-// send a sway message header
-func sendHeaderIPC(sock net.Conn, messageType uint32, len uint32) error {
-	sendPayload := make([]byte, IPC_HEADER_SIZE)
-	copy(sendPayload, []byte(IPC_MAGIC))
-	binary.LittleEndian.PutUint32(sendPayload[6:], len)
-	binary.LittleEndian.PutUint32(sendPayload[10:], messageType)
-
-	_, err := sock.Write(sendPayload)
-
-	if err != nil {
-		return fmt.Errorf("failed to send header to IPC %w", err)
-	}
-
-	return nil
-}
-
-// send a payload, header had to be sent before
-func sendPayloadIPC(sock net.Conn, payload []byte) error {
-	_, err := sock.Write(payload)
-
-	if err != nil {
-		return fmt.Errorf("failed to send payload to IPC %w", err)
-	}
-
-	return nil
-}
-
-// read a response, reads response header and returns payload only
-func readResponseIPC(sock net.Conn) ([]byte, error) {
-	// read header
-	buf := make([]byte, IPC_HEADER_SIZE)
-
-	_, err := sock.Read(buf)
-	if err != nil {
-		return nil, fmt.Errorf("failed to read header from socket: %s", err)
-	}
-
-	// slog.Debug("got IPC header", "header", hex.EncodeToString(buf))
-
-	if string(buf[:6]) != IPC_MAGIC {
-		return nil, fmt.Errorf("got invalid IPC response from sway socket")
-	}
-
-	payloadLen := binary.LittleEndian.Uint32(buf[6:10])
-
-	if payloadLen == 0 {
-		return nil, fmt.Errorf("got empty payload IPC response from sway socket")
-	}
-
-	// read payload
-	payload := make([]byte, payloadLen)
-
-	_, err = sock.Read(payload)
-	if err != nil {
-		return nil, fmt.Errorf("failed to read payload from socket: %s", err)
-	}
-
-	return payload, nil
-}
-
-// get raw JSON tree via sway IPC
-func getTree(sock net.Conn) ([]byte, error) {
-	err := sendHeaderIPC(sock, IPC_GET_TREE, 0)
-	if err != nil {
-		return nil, err
-	}
-
-	payload, err := readResponseIPC(sock)
-	if err != nil {
-		return nil, err
-	}
-
-	return payload, nil
-}
-
 // get into the sway tree, determine current workspace and extract all
 // its visible windows, store them in the global var Visibles
-func processJSON(jsoncode []byte) error {
+func processJSON(sway *i3ipc.Node) error {
-	sway := Node{}
-
-	if err := json.Unmarshal(jsoncode, &sway); err != nil {
-		return fmt.Errorf("Failed to unmarshal json: %w", err)
-	}
-
 	if !istype(sway, root) && len(sway.Nodes) == 0 {
 		return fmt.Errorf("Invalid or empty JSON structure")
 	}
@@ -310,41 +195,11 @@ func findNextWindow() int {
 }
 
 // actually switch focus using a swaymsg command
-func switchFocus(id int, sock net.Conn) error {
+func switchFocus(id int, ipc *i3ipc.I3ipc) error {
-	command := fmt.Sprintf("[con_id=%d] focus", id)
+	responses, err := ipc.RunContainerCommand(id, "focus")
-
-	slog.Debug("sending ipc", "command", command)
-
-	// send switch focus command
-	err := sendHeaderIPC(sock, IPC_RUN_COMMAND, uint32(len(command)))
 	if err != nil {
-		return fmt.Errorf("failed to send run_command to IPC %w", err)
+		log.Fatalf("failed to send focus command to container %d: %w (%s)",
-	}
+			id, responses[0].Error, err)
-
-	err = sendPayloadIPC(sock, []byte(command))
-	if err != nil {
-		return fmt.Errorf("failed to send switch focus command: %w", err)
-	}
-
-	// check response from sway
-	payload, err := readResponseIPC(sock)
-	if err != nil {
-		return err
-	}
-
-	responses := []Response{}
-
-	if err := json.Unmarshal(payload, &responses); err != nil {
-		return fmt.Errorf("Failed to unmarshal json response: %w", err)
-	}
-
-	if len(responses) == 0 {
-		return fmt.Errorf("Got invalid IPC zero response")
-	}
-
-	if !responses[0].Success {
-		slog.Debug("IPC response to switch focus command", "response", responses)
-		return fmt.Errorf("Failed to switch focus: %s", responses[0].Error)
 	}
 
 	slog.Info("switched focus", "con_id", id)
@@ -353,7 +208,7 @@ func switchFocus(id int, sock net.Conn) error {
 }
 
 // iterate recursively over given node list extracting visible windows
-func recurseNodes(nodes []Node) {
+func recurseNodes(nodes []*i3ipc.Node) {
 	for _, node := range nodes {
 		// we handle nodes and floating_nodes identical
 		node.Nodes = append(node.Nodes, node.FloatingNodes...)
@@ -420,8 +275,8 @@ func setupLogging(output io.Writer) {
 }
 
 // little helper to distinguish sway tree node types
-func istype(nd Node, which int) bool {
+func istype(nd *i3ipc.Node, which int) bool {
-	switch nd.Nodetype {
+	switch nd.Type {
 	case "root":
 		return which == root
 	case "output":