Merge pull request #191 from dmur1/add-share-command-#185

Add share command #185

Author: dmur1

commands/gist.go

@@ -1,15 +1,12 @@
 package commands
 
 import (
-	"bytes"
-	"encoding/json"
 	"fmt"
-	"io"
 	"log"
-	"net/http"
 	"os"
 	"ret/config"
 	"ret/theme"
+	"ret/util"
 	"strings"
 )
 
@@ -36,7 +33,7 @@ func GistHelp() string {
 
 func Gist(args []string) {
 	if len(args) == 0 {
-		log.Fatalf("💥 " + theme.ColorRed + "error" + theme.ColorReset + ": exepcted 1 or more arguments\n")
+		log.Fatalf("💥 " + theme.ColorRed + "error" + theme.ColorReset + ": expected 1 or more arguments\n")
 	}
 
 	if len(config.GistToken) == 0 {
@@ -59,42 +56,5 @@ func Gist(args []string) {
 		}
 	}
 
-	gist := map[string]interface{}{
-		"description": "🐙 made with https://github.com/rerrorctf/ret",
-		"public":      false,
-		"files":       files,
-	}
-
-	body, err := json.Marshal(gist)
-	if err != nil {
-		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
-	}
-
-	req, err := http.NewRequest("POST", "https://api.github.com/gists", bytes.NewBuffer(body))
-	if err != nil {
-		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
-	}
-	req.Header.Set("Accept", "application/vnd.github+json")
-	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", config.GistToken))
-	req.Header.Set("X-GitHub-Api-Version", "2022-11-28")
-
-	client := &http.Client{}
-	resp, err := client.Do(req)
-	if err != nil {
-		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
-	}
-
-	body, err = io.ReadAll(resp.Body)
-	if err != nil {
-		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
-	}
-
-	var result map[string]interface{}
-	if err := json.Unmarshal(body, &result); err != nil {
-		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
-	}
-
-	fmt.Printf("%s\n", result["html_url"].(string))
-
-	resp.Body.Close()
+	fmt.Printf("%s\n", util.Gist(files))
 }

commands/share.go

@@ -0,0 +1,64 @@
+package commands
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"ret/config"
+	"ret/theme"
+	"ret/util"
+)
+
+func init() {
+	Commands = append(Commands, Command{
+		Name:      "share",
+		Emoji:     "🌐",
+		Func:      Share,
+		Help:      ShareHelp,
+		Url:       "https://github.com/rerrorctf/ret/blob/main/commands/share.go",
+		Arguments: nil,
+		SeeAlso:   []string{"notes", "capture", "chat", "gist", "pwn"}})
+}
+
+func ShareHelp() string {
+	return "share task progress with ret\n\n"
+}
+
+func Share(args []string) {
+	if len(config.GistToken) == 0 {
+		log.Fatalf("💥 " + theme.ColorRed + "error" + theme.ColorReset + ": no gist token in ~/.config/ret\n")
+	}
+
+	files := map[string]interface{}{}
+
+	buffer, err := os.ReadFile(config.PwnScriptName)
+	if err == nil {
+		files[config.PwnScriptName] = map[string]interface{}{
+			"content": string(buffer),
+		}
+	}
+
+	buffer, err = os.ReadFile(config.NotesFileName)
+	if err == nil {
+		// does not like .ret/notes.json
+		files["notes.json"] = map[string]interface{}{
+			"content": string(buffer),
+		}
+	}
+
+	flag, err := util.GetCurrentFlag()
+	if err != nil {
+		flag = config.FlagFormat
+	} else {
+		files["flag.txt"] = map[string]interface{}{
+			"content": string(flag),
+		}
+	}
+
+	gistUrl := ""
+	if len(files) > 0 {
+		gistUrl = util.Gist(files)
+	}
+
+	Chat([]string{fmt.Sprintf("🏁 %s\n**%s**", flag, gistUrl)})
+}

util/gist.go

@@ -0,0 +1,55 @@
+package util
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"ret/config"
+	"ret/theme"
+)
+
+func Gist(files map[string]interface{}) string {
+	gist := map[string]interface{}{
+		"description": "🐙 made with https://github.com/rerrorctf/ret",
+		"public":      false,
+		"files":       files,
+	}
+
+	body, err := json.Marshal(gist)
+	if err != nil {
+		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
+	}
+
+	req, err := http.NewRequest("POST", "https://api.github.com/gists", bytes.NewBuffer(body))
+	if err != nil {
+		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
+	}
+	req.Header.Set("Accept", "application/vnd.github+json")
+	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", config.GistToken))
+	req.Header.Set("X-GitHub-Api-Version", "2022-11-28")
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
+	}
+
+	body, err = io.ReadAll(resp.Body)
+	if err != nil {
+		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
+	}
+
+	var result map[string]interface{}
+	if err := json.Unmarshal(body, &result); err != nil {
+		log.Fatalf("💥 "+theme.ColorRed+" error"+theme.ColorReset+": %v\n", err)
+	}
+
+	url := result["html_url"].(string)
+
+	resp.Body.Close()
+
+	return url
+}