simple escape markdown

rostrovsky · Mar 23, 2024 · b266a3b · b266a3b
1 parent e2fe7f3
commit b266a3b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -144,7 +144,7 @@ func run(cmd *cobra.Command, args []string) {
 	slog.Debug("Processing done")
 
 	if oFlag != "" {
-		slog.Debug("Saving", "output file", oFlag)
+		slog.Debug("Saving output", "file", oFlag)
 		err := writeToFile(oFlag, []byte(sb.String()))
 		if err != nil {
 			logErrAndExit(err)

diff --git a/cmd/utils.go b/cmd/utils.go
@@ -45,6 +45,10 @@ func isBinary(filename string) (bool, error) {
 	return !utf8.ValidString(string(buf)), nil
 }
 
+func isMarkdown(filename string) bool {
+	return strings.HasSuffix(strings.ToLower(filename), ".md")
+}
+
 func processPath(path string, prefixToRemove string, include *regexp.Regexp, exclude *regexp.Regexp, stringBuilder *strings.Builder) error {
 	return filepath.Walk(path, func(filePath string, info os.FileInfo, err error) error {
 		if err != nil {
@@ -92,10 +96,16 @@ func processPath(path string, prefixToRemove string, include *regexp.Regexp, exc
 			filePath = trimmedPath
 		}
 
+		fences := "```"
+
+		if isMarkdown(filePath) {
+			fences = "````"
+		}
+
 		stringBuilder.WriteString("`" + filePath + "`\n\n")
-		stringBuilder.WriteString("```" + "\n")
+		stringBuilder.WriteString(fences + "\n")
 		stringBuilder.Write(content)
-		stringBuilder.WriteString("```" + "\n\n")
+		stringBuilder.WriteString(fences + "\n\n")
 
 		return nil
 	})