Toak/Commands/ToggleCommand.cs

using System;
using System.Net.Sockets;
using System.Threading.Tasks;
using Spectre.Console;
using Toak.Core;

namespace Toak.Commands;

public static class ToggleCommand
{
    public static async Task ExecuteAsync(bool pipeToStdout, bool copyToClipboard, bool verbose)
    {
        Logger.Verbose = verbose;
        
        var socketPath = DaemonService.GetSocketPath();
        
        try
        {
            using var socket = new Socket(AddressFamily.Unix, SocketType.Stream, ProtocolType.Unspecified);
            var endPoint = new UnixDomainSocketEndPoint(socketPath);
            await socket.ConnectAsync(endPoint);
            
            // Send TOGGLE (cmd == 4), pipeToStdout, copyToClipboard
            var msg = new byte[] { 4, (byte)(pipeToStdout ? 1 : 0), (byte)(copyToClipboard ? 1 : 0) };
            await socket.SendAsync(msg, SocketFlags.None);
            
            if (verbose)
            {
                Console.WriteLine("Sent TOGGLE command to daemon.");
            }

            // Wait for response text if pipeToStdout or if it takes a while we just wait until the socket closes
            // This is required so the client process stays alive to receive the text through stdout
            var responseBuffer = new byte[4096];
            while (true)
            {
                int received = await socket.ReceiveAsync(responseBuffer, SocketFlags.None);
                if (received == 0) break; // socket closed by daemon
                
                if (pipeToStdout)
                {
                    var text = System.Text.Encoding.UTF8.GetString(responseBuffer, 0, received);
                    Console.Write(text);
                }
            }
        }
        catch (SocketException)
        {
            AnsiConsole.MarkupLine("[red]Failed to connect to Toak daemon.[/]");
            AnsiConsole.MarkupLine("Please ensure the daemon is running in the background:");
            AnsiConsole.MarkupLine("  [dim]toak daemon[/]");
        }
        catch (Exception ex)
        {
            AnsiConsole.MarkupLine($"[red]Error:[/] {ex.Message}");
        }
    }
}