Check the socket path before the daemon writes

[onechat] / main.go

diff --git a/main.go b/main.go
index 271a91a56d749d67880213d921a949ecfa6b8b15..680c58d9a1c3884b5f8bab7bc8ef04d3874ee12c 100644 (file)
--- a/main.go
+++ b/main.go
@@ -5,6 +5,7 @@ import (
        "fmt"
        "os"
        "os/signal"
+       "strings"
        "syscall"
 )
 
@@ -20,9 +21,35 @@ func run() error {
        ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
        defer cancel()
 
-       _, err := fmt.Println("onechat")
-       if err != nil {
-               return err
+       args := os.Args[1:]
+       switch {
+       case len(args) == 0:
+               _, err := fmt.Println("onechat")
+               if err != nil {
+                       return err
+               }
+       case len(args) == 1 && args[0] == "login":
+               err := login(ctx, os.Stdout)
+               if err != nil {
+                       return err
+               }
+       case len(args) == 1 && args[0] == "daemon":
+               err := daemon(ctx, os.Stdout)
+               if err != nil {
+                       return err
+               }
+       case len(args) == 1 && args[0] == "chat":
+               err := chat(ctx, os.Stdin, os.Stdout)
+               if err != nil {
+                       return err
+               }
+       case len(args) > 0 && args[0] == "probe":
+               err := probe(ctx, args[1:], os.Stdout)
+               if err != nil {
+                       return err
+               }
+       default:
+               return fmt.Errorf("unknown command %q", strings.Join(args, " "))
        }
 
        return ctx.Err()