fixed func

FlameInTheDark · May 22, 2019 · 7c6f03e · 7c6f03e
1 parent 9f2199e
commit 7c6f03e
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/cmd/funcommand.go b/cmd/funcommand.go
@@ -12,12 +12,14 @@ func SlapCommand(ctx bot.Context) {
 		ctx.MetricsCommand("fun", "slap")
 		url, err := fun.GetImageURL("slap")
 		if err == nil {
-			user := ctx.GetGuildUser(ctx.Args[0][2 : len(ctx.Args[0])-1])
-			var mention = ctx.Args[0]
-			if user != nil {
-				mention = user.Username
+			if len(ctx.Args) > 0 {
+				user := ctx.GetGuildUser(ctx.Args[0][2 : len(ctx.Args[0])-1])
+				var mention= ctx.Args[0]
+				if user != nil {
+					mention = user.Username
+				}
+				ctx.ReplyEmbedAttachmentImageURL("", fmt.Sprintf(ctx.Loc("slapping"), ctx.User.Username, mention), url)
 			}
-			ctx.ReplyEmbedAttachmentImageURL("", fmt.Sprintf(ctx.Loc("slapping"), ctx.User.Username, mention), url)
 		} else {
 			fmt.Printf(err.Error())
 		}
@@ -30,12 +32,14 @@ func FUCommand(ctx bot.Context) {
 		ctx.MetricsCommand("fun", "fu")
 		url, err := fun.GetImageURL("fu")
 		if err == nil {
-			user := ctx.GetGuildUser(ctx.Args[0][2 : len(ctx.Args[0])-1])
-			var mention = ctx.Args[0]
-			if user != nil {
-				mention = user.Username
+			if len(ctx.Args) > 0 {
+				user := ctx.GetGuildUser(ctx.Args[0][2 : len(ctx.Args[0])-1])
+				var mention = ctx.Args[0]
+				if user != nil {
+					mention = user.Username
+				}
+				ctx.ReplyEmbedAttachmentImageURL("", fmt.Sprintf(ctx.Loc("send_fu"), ctx.User.Username, mention), url)
 			}
-			ctx.ReplyEmbedAttachmentImageURL("", fmt.Sprintf(ctx.Loc("send_fu"), ctx.User.Username, mention), url)
 		} else {
 			fmt.Printf(err.Error())
 		}