allow chat to use up to 3 different webhooks with --1, --2 and --3

fixes: #170

Author: dmur1

README.md

@@ -159,14 +159,20 @@ flags are stored in the `.ret/flag.json` file
 ### 📢 <u>cha</u>t
 
 ```
-$ ret chat [-] [message1 message2 message3...] 
+$ ret chat [--1] [--2] [--3] [-] [message1 message2 message3...] 
 ```
 
 chat via a discord webhook with ret
 
 use - to read from stdin
 
-requires that `"chatwebhookurl"` from `~/.config/ret` is a valid webhook
+requires a valid webhook this is typically `"chatwebhookurl"` from `~/.config/ret` is a valid webhook
+
+however the command supports up to 3 webhooks using `$ ret --1 chat`, `$ ret --2 chat` and `$ ret --3 chat`
+
+if no numerical override is specified the `"chatwebhookurl"` webhook is used by default
+
+webhooks 2 and 3 are set with `"chatwebhookurl2"` and `"chatwebhookurl3"` respectively
 
 requires that `"username"` from `~/.config/ret` is set to valid string
 
@@ -573,6 +579,8 @@ if a file called `writeup.md` already exists the command will abort
   "wizardpostcommand": "",
   "username": "",
   "chatwebhookurl": "",
+  "chatwebhookurl2": "",
+  "chatwebhookurl3": "",
   "gisttoken": "",
   "openaikey": "",
   "openaimodel": "",

commands/chat.go

@@ -22,6 +22,24 @@ func init() {
 		Help:  ChatHelp,
 		Url:   "https://github.com/rerrorctf/ret/blob/main/commands/chat.go",
 		Arguments: []Argument{
+			{
+				Name:     "--1",
+				Optional: true,
+				List:     false,
+				Override: true,
+			},
+			{
+				Name:     "--2",
+				Optional: true,
+				List:     false,
+				Override: true,
+			},
+			{
+				Name:     "--3",
+				Optional: true,
+				List:     false,
+				Override: true,
+			},
 			{
 				Name:     "-",
 				Optional: true,
@@ -38,20 +56,23 @@ func init() {
 func ChatHelp() string {
 	return "chat via a discord webhook with ret\n\n" +
 		"use - to read from stdin\n\n" +
-		"requires that " + theme.ColorYellow + "`\"chatwebhookurl\"`" + theme.ColorReset + " from " + theme.ColorCyan + "`~/.config/ret`" + theme.ColorReset + " is a valid webhook\n\n" +
+		"requires a valid webhook this is typically " + theme.ColorYellow + "`\"chatwebhookurl\"`" + theme.ColorReset + " from " + theme.ColorCyan + "`~/.config/ret`" + theme.ColorReset + " is a valid webhook\n\n" +
+		"however the command supports up to 3 webhooks using " + theme.ColorGray + "`$ ret --1 chat`" + theme.ColorReset + ", " + theme.ColorGray + "`$ ret --2 chat` " + theme.ColorReset + "and " + theme.ColorGray + "`$ ret --3 chat`" + theme.ColorReset + "\n\n" +
+		"if no numerical override is specified the " + theme.ColorYellow + "`\"chatwebhookurl\"`" + theme.ColorReset + " webhook is used by default\n\n" +
+		"webhooks 2 and 3 are set with " + theme.ColorYellow + "`\"chatwebhookurl2\"`" + theme.ColorReset + " and " + theme.ColorYellow + "`\"chatwebhookurl3\"`" + theme.ColorReset + " respectively\n\n" +
 		"requires that " + theme.ColorYellow + "`\"username\"`" + theme.ColorReset + " from " + theme.ColorCyan + "`~/.config/ret`" + theme.ColorReset + " is set to valid string\n\n" +
 		"when data is read from stdin, due to the use of the - argument, it will be sent as an embed with an accurate timestamp and a random color\n\n" +
 		"color codes, such as the ones used by this tool, are stripped by this code prior to sending\n\n" +
 		"for more information please see " + theme.ColorPurple + "https://support.discord.com/hc/en-us/articles/228383668-Intro-to-Webhooks\n" + theme.ColorReset
 }
 
-func sendMessage(message map[string]interface{}) {
+func sendMessage(message map[string]interface{}, webhook string) {
 	body, err := json.Marshal(message)
 	if err != nil {
 		log.Fatalf("💥 "+theme.ColorRed+"error"+theme.ColorReset+": %v\n", err)
 	}
 
-	req, err := http.NewRequest("POST", config.ChatWebhookUrl, bytes.NewBuffer(body))
+	req, err := http.NewRequest("POST", webhook, bytes.NewBuffer(body))
 	if err != nil {
 		log.Fatalf("💥 "+theme.ColorRed+"error"+theme.ColorReset+": %v\n", err)
 	}
@@ -66,18 +87,18 @@ func sendMessage(message map[string]interface{}) {
 	resp.Body.Close()
 }
 
-func sendChat(message string) {
+func sendChat(message string, webhook string) {
 	message = theme.RemoveColors(message)
 
 	body := map[string]interface{}{
 		"username": config.Username,
 		"content":  message,
 	}
 
-	sendMessage(body)
+	sendMessage(body, webhook)
 }
 
-func sendEmbed(message string) {
+func sendEmbed(message string, webhook string) {
 	message = theme.RemoveColors(message)
 
 	embed := map[string]interface{}{
@@ -97,38 +118,46 @@ func sendEmbed(message string) {
 		"embeds":   []interface{}{embed},
 	}
 
-	sendMessage(body)
+	sendMessage(body, webhook)
 }
 
 func Chat(args []string) {
 	if len(args) == 0 {
 		log.Fatalf("💥 " + theme.ColorRed + " error" + theme.ColorReset + ": expected 1 or more arguments\n")
 	}
 
-	if config.ChatWebhookUrl == "" {
-		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": no chat webhook url found in %s\n", config.UserConfig)
-	}
-
 	if config.Username == "" {
 		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": no username found in %s\n", config.UserConfig)
 	}
 
+	webhook := config.ChatWebhookUrl
+
+	if CommandsChoice2 {
+		webhook = config.ChatWebhookUrl2
+	} else if CommandsChoice3 {
+		webhook = config.ChatWebhookUrl3
+	}
+
+	if webhook == "" {
+		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": no chat webhook url found in %s\n", config.UserConfig)
+	}
+
 	if len(args) > 0 {
 		if strings.Compare("-", args[0]) == 0 {
 			var buffer bytes.Buffer
 			_, err := io.Copy(&buffer, os.Stdin)
 			if err != nil {
 				log.Fatalf("💥 "+theme.ColorRed+"error"+theme.ColorReset+": %v\n", err)
 			}
-			sendEmbed(buffer.String())
+			sendEmbed(buffer.String(), webhook)
 
 			if len(args) > 1 {
-				sendChat(strings.Join(args[1:], " "))
+				sendChat(strings.Join(args[1:], " "), webhook)
 			}
 
 			return
 		}
 	}
 
-	sendChat(strings.Join(args, " "))
+	sendChat(strings.Join(args, " "), webhook)
 }

config/config.go

@@ -32,6 +32,8 @@ var (
 	WizardPostCommand        = ""
 	Username                 = ""
 	ChatWebhookUrl           = ""
+	ChatWebhookUrl2          = ""
+	ChatWebhookUrl3          = ""
 	GistToken                = ""
 	OpenAIKey                = ""
 	OpenAIModel              = "gpt-4o"
@@ -54,6 +56,8 @@ type Config struct {
 	WizardPostCommand        string `json:"wizardpostcommand"`
 	Username                 string `json:"username"`
 	ChatWebhookUrl           string `json:"chatwebhookurl"`
+	ChatWebhookUrl2          string `json:"chatwebhookurl2"`
+	ChatWebhookUrl3          string `json:"chatwebhookurl3"`
 	GistToken                string `json:"gisttoken"`
 	OpenAIKey                string `json:"openaikey"`
 	OpenAIModel              string `json:"openaimodel"`
@@ -129,6 +133,14 @@ func ParseUserConfig() {
 		ChatWebhookUrl = userConfig.ChatWebhookUrl
 	}
 
+	if len(userConfig.ChatWebhookUrl2) > 0 {
+		ChatWebhookUrl2 = userConfig.ChatWebhookUrl2
+	}
+
+	if len(userConfig.ChatWebhookUrl3) > 0 {
+		ChatWebhookUrl3 = userConfig.ChatWebhookUrl3
+	}
+
 	if len(userConfig.GistToken) > 0 {
 		GistToken = userConfig.GistToken
 	}
@@ -191,6 +203,8 @@ func WriteUserConfig() {
 	userConfig.WizardPostCommand = WizardPostCommand
 	userConfig.Username = Username
 	userConfig.ChatWebhookUrl = ChatWebhookUrl
+	userConfig.ChatWebhookUrl2 = ChatWebhookUrl2
+	userConfig.ChatWebhookUrl3 = ChatWebhookUrl3
 	userConfig.GistToken = GistToken
 	userConfig.OpenAIKey = OpenAIKey
 	userConfig.OpenAIModel = OpenAIModel