appease the linter

Author: SasSwart

agent/agentsocket/server.go

@@ -18,6 +18,25 @@ import (
 	"github.com/coder/coder/v2/codersdk/drpcsdk"
 )
 
+// Client wraps a DRPC client with connection management.
+// This type is defined here so it's available on all platforms.
+type Client struct {
+	proto.DRPCAgentSocketClient
+	conn    net.Conn
+	session *yamux.Session
+}
+
+// Close closes the client connection.
+func (c *Client) Close() error {
+	if c.session != nil {
+		_ = c.session.Close()
+	}
+	if c.conn != nil {
+		return c.conn.Close()
+	}
+	return nil
+}
+
 // Server provides access to the DRPCAgentSocketService via a Unix domain socket.
 // Do not invoke Server{} directly. Use NewServer() instead.
 type Server struct {

agent/agentsocket/socket_unix.go

@@ -3,14 +3,20 @@
 package agentsocket
 
 import (
+	"context"
 	"crypto/rand"
 	"encoding/hex"
 	"net"
 	"os"
 	"path/filepath"
 	"time"
 
+	"github.com/hashicorp/yamux"
 	"golang.org/x/xerrors"
+
+	"cdr.dev/slog"
+	"github.com/coder/coder/v2/agent/agentsocket/proto"
+	"github.com/coder/coder/v2/codersdk/drpcsdk"
 )
 
 // createSocket creates a Unix domain socket listener
@@ -81,3 +87,25 @@ func isSocketAvailable(path string) bool {
 	// Socket is in use
 	return false
 }
+
+// NewClient creates a DRPC client for the agent socket at the given path.
+func NewClient(path string, logger slog.Logger) (*Client, error) {
+	conn, err := net.Dial("unix", path)
+	if err != nil {
+		return nil, xerrors.Errorf("dial unix socket: %w", err)
+	}
+
+	config := yamux.DefaultConfig()
+	config.LogOutput = nil
+	config.Logger = slog.Stdlib(context.Background(), logger, slog.LevelInfo)
+	session, err := yamux.Client(conn, config)
+	if err != nil {
+		_ = conn.Close()
+		return nil, xerrors.Errorf("multiplex client: %w", err)
+	}
+	return &Client{
+		DRPCAgentSocketClient: proto.NewDRPCAgentSocketClient(drpcsdk.MultiplexedConn(session)),
+		conn:                  conn,
+		session:               session,
+	}, nil
+}

cli/testdata/coder_agent_--help.golden

@@ -67,6 +67,9 @@ OPTIONS:
       --script-data-dir string, $CODER_AGENT_SCRIPT_DATA_DIR (default: /tmp)
           Specify the location for storing script data.
 
+      --socket-path string, $CODER_AGENT_SOCKET_PATH
+          Specify the path for the agent socket.
+
       --ssh-max-timeout duration, $CODER_AGENT_SSH_MAX_TIMEOUT (default: 72h)
           Specify the max timeout for a SSH connection, it is advisable to set
           it to a minimum of 60s, but no more than 72h.