From 220976ca377865124898daffc12568b220fc7f68 Mon Sep 17 00:00:00 2001
From: Oliver Gugger <gugger@gmail.com>
Date: Fri, 17 Jul 2020 10:00:26 +0200
Subject: [PATCH] cmd/llm: fix unnecessary help print

---
 cmd/llm/main.go | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/cmd/llm/main.go b/cmd/llm/main.go
index fcceba8..a977cdb 100644
--- a/cmd/llm/main.go
+++ b/cmd/llm/main.go
@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"encoding/hex"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"os"
 	"strconv"
@@ -17,6 +18,15 @@ import (
 	"google.golang.org/grpc"
 )
 
+type invalidUsageError struct {
+	ctx     *cli.Context
+	command string
+}
+
+func (e *invalidUsageError) Error() string {
+	return fmt.Sprintf("invalid usage of command %s", e.command)
+}
+
 func printJSON(resp interface{}) {
 	b, err := json.Marshal(resp)
 	if err != nil {
@@ -46,7 +56,12 @@ func printRespJSON(resp proto.Message) { // nolint
 }
 
 func fatal(err error) {
-	_, _ = fmt.Fprintf(os.Stderr, "[llm] %v\n", err)
+	var e *invalidUsageError
+	if errors.As(err, &e) {
+		_ = cli.ShowCommandHelp(e.ctx, e.command)
+	} else {
+		_, _ = fmt.Fprintf(os.Stderr, "[llm] %v\n", err)
+	}
 	os.Exit(1)
 }
 
@@ -117,7 +132,7 @@ func parseStr(ctx *cli.Context, argIdx int, flag, cmd string) (string, error) {
 	case ctx.Args().Get(argIdx) != "":
 		str = ctx.Args().Get(argIdx)
 	default:
-		return "", cli.ShowCommandHelp(ctx, cmd)
+		return "", &invalidUsageError{ctx, cmd}
 	}
 	return str, nil
 }