mirror of
https://github.com/AvengeMedia/DankMaterialShell.git
synced 2026-01-24 13:32:50 -05:00
115 lines
2.9 KiB
Go
115 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/AvengeMedia/DankMaterialShell/core/internal/server"
|
|
"github.com/AvengeMedia/DankMaterialShell/core/internal/server/models"
|
|
)
|
|
|
|
func sendServerRequest(req models.Request) (*models.Response[any], error) {
|
|
socketPath := getServerSocketPath()
|
|
|
|
conn, err := net.Dial("unix", socketPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to connect to server (is it running?): %w", err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
scanner := bufio.NewScanner(conn)
|
|
scanner.Scan() // discard initial capabilities message
|
|
|
|
reqData, err := json.Marshal(req)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to marshal request: %w", err)
|
|
}
|
|
|
|
if _, err := conn.Write(reqData); err != nil {
|
|
return nil, fmt.Errorf("failed to write request: %w", err)
|
|
}
|
|
|
|
if _, err := conn.Write([]byte("\n")); err != nil {
|
|
return nil, fmt.Errorf("failed to write newline: %w", err)
|
|
}
|
|
|
|
if !scanner.Scan() {
|
|
return nil, fmt.Errorf("failed to read response")
|
|
}
|
|
|
|
var resp models.Response[any]
|
|
if err := json.Unmarshal(scanner.Bytes(), &resp); err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal response: %w", err)
|
|
}
|
|
|
|
return &resp, nil
|
|
}
|
|
|
|
// sendServerRequestFireAndForget sends a request without waiting for a response.
|
|
// Useful for commands that trigger UI or async operations.
|
|
func sendServerRequestFireAndForget(req models.Request) error {
|
|
socketPath := getServerSocketPath()
|
|
|
|
conn, err := net.Dial("unix", socketPath)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to connect to server (is it running?): %w", err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
scanner := bufio.NewScanner(conn)
|
|
scanner.Scan() // discard initial capabilities message
|
|
|
|
reqData, err := json.Marshal(req)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal request: %w", err)
|
|
}
|
|
|
|
if _, err := conn.Write(reqData); err != nil {
|
|
return fmt.Errorf("failed to write request: %w", err)
|
|
}
|
|
|
|
if _, err := conn.Write([]byte("\n")); err != nil {
|
|
return fmt.Errorf("failed to write newline: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// tryServerRequest attempts to send a request but returns false if server unavailable.
|
|
// Does not log errors - caller can decide what to do on failure.
|
|
func tryServerRequest(req models.Request) (*models.Response[any], bool) {
|
|
resp, err := sendServerRequest(req)
|
|
if err != nil {
|
|
return nil, false
|
|
}
|
|
return resp, true
|
|
}
|
|
|
|
func getServerSocketPath() string {
|
|
runtimeDir := os.Getenv("XDG_RUNTIME_DIR")
|
|
if runtimeDir == "" {
|
|
runtimeDir = os.TempDir()
|
|
}
|
|
|
|
entries, err := os.ReadDir(runtimeDir)
|
|
if err != nil {
|
|
return filepath.Join(runtimeDir, "danklinux.sock")
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
name := entry.Name()
|
|
if name == "danklinux.sock" {
|
|
return filepath.Join(runtimeDir, name)
|
|
}
|
|
if len(name) > 10 && name[:10] == "danklinux-" && filepath.Ext(name) == ".sock" {
|
|
return filepath.Join(runtimeDir, name)
|
|
}
|
|
}
|
|
|
|
return server.GetSocketPath()
|
|
}
|