hypr/utils.go

package hypr

import (
	"context"
	"encoding/json"
	"io"
	"net"
	"time"
)

func readSocketRaw(socket string, body io.Reader) (io.ReadCloser, error) {
	conn, err := net.Dial("unix", socket)
	if err != nil {
		return nil, err
	}

	if _, err := io.Copy(conn, body); err != nil {
		conn.Close()
		return nil, err
	}

	return conn, nil
}

func readSocketString(socket string, body io.Reader) (string, error) {
	r, err := readSocketRaw(socket, body)
	if err != nil {
		return "", err
	}
	defer r.Close()

	data, err := io.ReadAll(r)
	if err != nil {
		return "", err
	}

	return string(data), nil
}

func readSocket[T any](socket string, body io.Reader) (T, error) {
	r, err := readSocketRaw(socket, body)
	if err != nil {
		return *new(T), err
	}
	defer r.Close()

	value := new(T)
	if err := json.NewDecoder(r).Decode(value); err != nil {
		return *new(T), err
	}

	return *value, nil
}

func waitFor(ctx context.Context, condition func() bool) bool {
	if condition() {
		return true
	}

	ticker := time.NewTicker(100 * time.Millisecond)
	defer ticker.Stop()

	for {
		select {
		case <-ticker.C:
			if condition() {
				return true
			}
		case <-ctx.Done():
			return false
		}
	}
}