From 16dd6434ad4d9569fdbfdf39f1f3dc372280b735 Mon Sep 17 00:00:00 2001
From: NotAShelf <raf@notashelf.dev>
Date: Sun, 17 Dec 2023 03:08:02 +0300
Subject: [PATCH] feat: log paste contents to a logs directory if private mode
 is not enabled

---
 internal/paste/paste.go | 38 +++++++++++++++++++++++++++++++++-----
 1 file changed, 33 insertions(+), 5 deletions(-)

diff --git a/internal/paste/paste.go b/internal/paste/paste.go
index e9327f7..6d2a1ab 100644
--- a/internal/paste/paste.go
+++ b/internal/paste/paste.go
@@ -58,6 +58,8 @@ func InitializeTemplates() error {
 }
 
 func CreatePasteHandler(w http.ResponseWriter, r *http.Request) {
+	private := viper.GetBool("Private")
+
 	// Parse the request body
 	content, err := io.ReadAll(r.Body)
 	if err != nil {
@@ -66,15 +68,41 @@ func CreatePasteHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	// Check if the private flag is set
-	private := viper.GetBool("Private")
-
-	// Log the event
+	// log the event's occurance - not the content
 	log.Infof("Received request to create a paste (Private mode: %v)", private)
 
-	// Log the content if private mode is not enabled
 	if !private {
+		// log the content if private mode is not enabled
 		log.Infof("Received content: %s", content)
+		// also log the content into a file with the timestamp in the log directory
+
+		// check if the log directory exists and create it if it doesn't
+		if _, err := os.Stat(viper.GetString("LogDir")); os.IsNotExist(err) {
+			err := os.MkdirAll(viper.GetString("LogDir"), 0755)
+			if err != nil {
+				logger.Errorf("Failed to create log directory: %v", err)
+				http.Error(w, "Internal Server Error", http.StatusInternalServerError)
+			}
+		}
+
+		logFile := filepath.Join(viper.GetString("LogDir"), fmt.Sprintf("%d.log", time.Now().UnixNano()))
+		f, err := os.Create(logFile)
+
+		if err != nil {
+			logger.Errorf("Failed to create log file: %v", err)
+			http.Error(w, "Internal Server Error", http.StatusInternalServerError)
+			metrics.IncrementPasteCreatedCounter("failure")
+			return
+		}
+		defer f.Close()
+
+		if _, err := f.Write(content); err != nil {
+			logger.Errorf("Failed to write to log file: %v", err)
+			http.Error(w, "Internal Server Error", http.StatusInternalServerError)
+			metrics.IncrementPasteCreatedCounter("failure")
+			return
+		}
+
 	}
 
 	// Generate a unique ID for the paste