implemented more stuff along with samples

i3ipc.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"net"
+	"strings"
 )
 
 const (
@@ -53,15 +54,24 @@ type Response struct {
 	Error      string `json:"error"`
 }
 
-func NewI3ipc(file string) *I3ipc {
-	if file == "" {
-		file = "SWAYSOCK"
-	}
-	return &I3ipc{SocketFile: file}
+type Config struct {
+	Config string `json:"config"`
 }
 
-func (ipc *I3ipc) GetTree() (*Node, error) {
-	err := ipc.sendHeader(GET_TREE, 0)
+func NewI3ipc(file ...string) *I3ipc {
+	ipc := &I3ipc{}
+
+	if len(file) == 0 {
+		ipc.SocketFile = "SWAYSOCK"
+	} else {
+		ipc.SocketFile = file[0]
+	}
+
+	return ipc
+}
+
+func (ipc *I3ipc) get(command uint32) ([]byte, error) {
+	err := ipc.sendHeader(command, 0)
 	if err != nil {
 		return nil, err
 	}
@@ -71,10 +81,110 @@ func (ipc *I3ipc) GetTree() (*Node, error) {
 		return nil, err
 	}
 
-	node := &Node{}
-	if err := json.Unmarshal(payload, &node); err != nil {
+	return payload, nil
+}
+
+func (ipc *I3ipc) RunGlobalCommand(command ...string) ([]Response, error) {
+	return ipc.RunCommand(0, command...)
+}
+
+func (ipc *I3ipc) RunCommand(id int, command ...string) ([]Response, error) {
+	if len(command) == 0 {
+		return nil, fmt.Errorf("empty command arg")
+	}
+
+	commands := strings.Join(command, ",")
+
+	if id > 0 {
+		commands = fmt.Sprintf("[con_id=%d] %s", id, commands)
+	}
+
+	err := ipc.sendHeader(RUN_COMMAND, uint32(len(commands)))
+	if err != nil {
+		return nil, fmt.Errorf("failed to send run_command to IPC %w", err)
+	}
+
+	err = ipc.sendPayload([]byte(commands))
+	if err != nil {
+		return nil, fmt.Errorf("failed to send switch focus command: %w", err)
+	}
+
+	payload, err := ipc.readResponse()
+	if err != nil {
+		return nil, err
+	}
+
+	responses := []Response{}
+
+	if err := json.Unmarshal(payload, &responses); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal json response: %w", err)
+	}
+
+	if len(responses) == 0 {
+		return nil, fmt.Errorf("got zero IPC response")
+	}
+
+	for _, response := range responses {
+		if !response.Success {
+			return responses, fmt.Errorf("one or more commands failed")
+		}
+	}
+
+	return responses, nil
+}
+
+func (ipc *I3ipc) GetWorkspaces() ([]*Node, error) {
+	payload, err := ipc.get(GET_WORKSPACES)
+	if err != nil {
+		return nil, err
+	}
+
+	nodes := []*Node{}
+	if err := json.Unmarshal(payload, &nodes); err != nil {
 		return nil, fmt.Errorf("failed to unmarshal json: %w", err)
 	}
 
-	return node, nil
+	return nodes, nil
+}
+
+func (ipc *I3ipc) GetMarks() ([]string, error) {
+	payload, err := ipc.get(GET_MARKS)
+	if err != nil {
+		return nil, err
+	}
+
+	marks := []string{}
+	if err := json.Unmarshal(payload, &marks); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal json: %w", err)
+	}
+
+	return marks, nil
+}
+
+func (ipc *I3ipc) GetBindingModes() ([]string, error) {
+	payload, err := ipc.get(GET_BINDING_MODES)
+	if err != nil {
+		return nil, err
+	}
+
+	modes := []string{}
+	if err := json.Unmarshal(payload, &modes); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal json: %w", err)
+	}
+
+	return modes, nil
+}
+
+func (ipc *I3ipc) GetConfig() (string, error) {
+	payload, err := ipc.get(GET_CONFIG)
+	if err != nil {
+		return "", err
+	}
+
+	config := &Config{}
+	if err := json.Unmarshal(payload, &config); err != nil {
+		return "", fmt.Errorf("failed to unmarshal json: %w", err)
+	}
+
+	return config.Config, nil
 }