X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/97274864799ac2eea84bf80c9727a810b695db55..c781f79f3c8cb9a7577ee1574d196988da2719b3:/main.go?ds=sidebyside

diff --git a/main.go b/main.go
index 680c58d..400538a 100644
--- a/main.go
+++ b/main.go
@@ -1,3 +1,4 @@
+// Command onechat runs the agent daemon and its terminal client.
 package main
 
 import (
@@ -5,7 +6,6 @@ import (
 	"fmt"
 	"os"
 	"os/signal"
-	"strings"
 	"syscall"
 )
 
@@ -21,36 +21,5 @@ func run() error {
 	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
 	defer cancel()
 
-	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()
+	return parseCommand(ctx, os.Args[1:])
 }