Implement cooldowns

internal/util/cooldowns.go

@@ -0,0 +1,45 @@
+package util
+
+import (
+	"net"
+	"net/http"
+	"strings"
+	"sync"
+	"time"
+)
+
+var (
+	PostCooldowns   = make(map[string]time.Time)
+	ThreadCooldowns = make(map[string]time.Time)
+)
+
+var cooldownMutex sync.Mutex
+
+const POST_COOLDOWN = 15 * time.Second
+const THREAD_COOLDOWN = 2 * time.Minute
+
+func IsOnCooldown(ip string, m map[string]time.Time, duration time.Duration) bool {
+	cooldownMutex.Lock()
+	defer cooldownMutex.Unlock()
+
+	last, exists := m[ip]
+	if !exists || time.Since(last) >= duration {
+		m[ip] = time.Now()
+		return false
+	}
+	return true
+}
+
+func GetIP(r *http.Request) string {
+	if xff := r.Header.Get("X-Forwarded-For"); xff != "" {
+		parts := strings.Split(xff, ",")
+		return strings.TrimSpace(parts[0])
+	}
+
+	// Fallback to RemoteAddr
+	ip, _, err := net.SplitHostPort(r.RemoteAddr)
+	if err != nil {
+		return r.RemoteAddr
+	}
+	return ip
+}

internal/util/posts.go

@@ -0,0 +1,104 @@
+package util
+
+import (
+	"fmt"
+	"html/template"
+	"image"
+	"io"
+	"log"
+	"mime/multipart"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/disintegration/imaging"
+)
+
+// 10 MB memory limit
+const FILE_MEM_LIMIT int64 = 10 << 20
+const POST_IMG_FULL_PATH = "web/static/img/posts/full"
+const POST_IMG_THUMB_PATH = "web/static/img/posts/thumb"
+
+func EnrichPost(body string) string {
+	var b strings.Builder
+	for _, rawLine := range strings.Split(body, "\n") {
+		var outLine string
+
+		for i, rawWord := range strings.Split(rawLine, " ") {
+			var outWord string
+			if strings.HasPrefix(rawWord, ">>") {
+				postId := strings.TrimPrefix(rawWord, ">>")
+				esc := template.HTMLEscapeString(rawWord)
+				outWord = fmt.Sprintf(
+					`<a onclick="onReplyLinkClick(event)" 
+						 onmouseover="highlightPost(%[1]s, event)" `+
+						`onmouseleave="highlightPost(%[1]s, event, false)" `+
+						`href="#post-%[1]s" class="reply-link">%s</a>`,
+					postId, esc,
+				)
+			} else {
+				outWord = template.HTMLEscapeString(rawWord)
+			}
+
+			if i != 0 {
+				outLine += " "
+			}
+			outLine += outWord
+		}
+
+		if strings.HasPrefix(rawLine, ">") && !strings.HasPrefix(rawLine, ">>") {
+			outLine = `<span class="greentext">` + outLine + `</span>`
+		}
+
+		b.WriteString(outLine)
+		b.WriteString("<br/>")
+	}
+	return b.String()
+}
+
+func SavePostFile(file *multipart.File, filename string) error {
+	dstPathFull := filepath.Join(POST_IMG_FULL_PATH, filename)
+	dstPathThumb := filepath.Join(POST_IMG_THUMB_PATH, filename)
+
+	// FULL
+	if err := os.MkdirAll(POST_IMG_FULL_PATH, 0755); err != nil {
+		log.Printf("MkdirAll (full): %v", err)
+		return err
+	}
+
+	dstFull, err := os.Create(dstPathFull)
+	if err != nil {
+		log.Printf("os.Create (full): %v", err)
+		return err
+	}
+	defer dstFull.Close()
+	if _, err := io.Copy(dstFull, *file); err != nil {
+		log.Printf("io.Copy (full): %v", err)
+		return err
+	}
+
+	if _, err := (*file).Seek(0, 0); err != nil { // rewind file
+		log.Printf("seek file (thumb): %v", err)
+		return err
+	}
+
+	// THUMBNAIL
+	if err := os.MkdirAll(POST_IMG_THUMB_PATH, 0755); err != nil {
+		log.Printf("MkdirAll (thumb): %v", err)
+		return err
+	}
+
+	img, _, err := image.Decode(*file)
+	if err != nil {
+		log.Printf("image.Decode: %v", err)
+		return err
+	}
+
+	thumb := imaging.Resize(img, 300, 0, imaging.Lanczos)
+	if err = imaging.Save(thumb, dstPathThumb); err != nil {
+		log.Printf("imaging.Save: %v", err)
+		return err
+	}
+
+	return nil
+}

internal/util/util.go

@@ -1,44 +0,0 @@
-package util
-
-import (
-	"fmt"
-	"html/template"
-	"strings"
-)
-
-func EnrichPost(body string) string {
-	var b strings.Builder
-	for _, rawLine := range strings.Split(body, "\n") {
-		var outLine string
-
-		for i, rawWord := range strings.Split(rawLine, " ") {
-			var outWord string
-			if strings.HasPrefix(rawWord, ">>") {
-				postId := strings.TrimPrefix(rawWord, ">>")
-				esc := template.HTMLEscapeString(rawWord)
-				outWord = fmt.Sprintf(
-					`<a onclick="onReplyLinkClick(event)" 
-						 onmouseover="highlightPost(%[1]s, event)" `+
-						`onmouseleave="highlightPost(%[1]s, event, false)" `+
-						`href="#post-%[1]s" class="reply-link">%s</a>`,
-					postId, esc,
-				)
-			} else {
-				outWord = template.HTMLEscapeString(rawWord)
-			}
-
-			if i != 0 {
-				outLine += " "
-			}
-			outLine += outWord
-		}
-
-		if strings.HasPrefix(rawLine, ">") && !strings.HasPrefix(rawLine, ">>") {
-			outLine = `<span class="greentext">` + outLine + `</span>`
-		}
-
-		b.WriteString(outLine)
-		b.WriteString("<br/>")
-	}
-	return b.String()
-}