comfychan/internal/util/posts.go

package util

import (
	"fmt"
	"html/template"
	"image"
	"io"
	"log"
	"mime/multipart"
	"net/http"
	"os"
	"os/exec"
	"path"
	"path/filepath"
	"strings"

	"github.com/disintegration/imaging"
)

// 10 MB memory limit
const FILE_MEM_LIMIT int64 = 10 << 20
const POST_MEDIA_FULL_PATH = "web/static/media/posts/full"
const POST_MEDIA_THUMB_PATH = "web/static/media/posts/thumb"
const MAX_THREAD_COUNT = 50

var SUPPORTED_VID_FORMATS = []string{".mp4", ".webm", ".ogg"}

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, string, string) {
	// read file type
	buffer := make([]byte, 512)
	file.Read(buffer)
	file.Seek(0, 0)

	fileType := http.DetectContentType(buffer)
	isFileVideo := strings.HasPrefix(fileType, "video/")
	fileExt := strings.ToLower(filepath.Ext(fileName))

	// FULL
	if err := os.MkdirAll(POST_MEDIA_FULL_PATH, 0755); err != nil {
		log.Printf("MkdirAll (full): %v", err)
		return err, "", ""
	}

	dstPathFull := filepath.Join(POST_MEDIA_FULL_PATH, fileName)
	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
	dstPathThumb := filepath.Join(POST_MEDIA_THUMB_PATH, fileName)
	if err := os.MkdirAll(POST_MEDIA_THUMB_PATH, 0755); err != nil {
		log.Printf("MkdirAll (thumb): %v", err)
		return err, "", ""
	}

	var thumbFileName string
	if isFileVideo {
		fileNameNoExt := strings.TrimSuffix(fileName, fileExt)
		thumbFileName = fileNameNoExt + ".jpg"

		inputPath := filepath.Join(POST_MEDIA_FULL_PATH, fileName)
		outputPath := filepath.Join(POST_MEDIA_THUMB_PATH, thumbFileName)

		cmd := exec.Command(
			"ffmpeg",
			"-i", inputPath,
			"-ss", "00:00:01.000",
			"-vframes", "1",
			"-vf", "scale=300:-1",
			outputPath,
		)

		if err := cmd.Run(); err != nil {
			log.Printf("ffmpeg error: %v", err)
			return err, "", ""
		}

	} else {
		thumbFileName = fileName

		img, _, err := image.Decode(file)
		if err != nil {
			log.Printf("image.Decode: %v", err)
			return err, "", ""
		}

		var thumb image.Image
		if img.Bounds().Dx() > 300 {
			thumb = imaging.Resize(img, 300, 0, imaging.Lanczos)
		} else {
			thumb = img
		}

		if err = imaging.Save(thumb, dstPathThumb); err != nil {
			log.Printf("imaging.Save: %v", err)
			return err, "", ""
		}
	}

	return nil, fileName, thumbFileName
}

type PostFileInfo struct {
	Size    int64
	Height  int
	Width   int
	IsVideo bool
}

func GetPostFileInfo(mediaPath string) PostFileInfo {
	var result PostFileInfo

	filePath := path.Join(POST_MEDIA_FULL_PATH, mediaPath)

	fileInfo, err := os.Stat(filePath)
	if err != nil {
		log.Printf("Failed to get file info at %s: %v", mediaPath, err)
		return PostFileInfo{}
	}

	file, err := os.Open(filePath)
	if err != nil {
		log.Printf("Failed to open file at %s: %v", mediaPath, err)
		return PostFileInfo{}
	}
	defer file.Close()

	// read file type
	buffer := make([]byte, 512)
	file.Read(buffer)
	file.Seek(0, 0)

	fileType := http.DetectContentType(buffer)
	isFileVideo := strings.HasPrefix(fileType, "video/")

	result.Size = fileInfo.Size()

	if !isFileVideo {
		cfg, _, err := image.DecodeConfig(file)
		if err != nil {
			log.Printf("Failed to decode image at %s: %v", mediaPath, err)
			return PostFileInfo{}
		}

		result.Width = cfg.Width
		result.Height = cfg.Height
		result.IsVideo = false

	} else {
		result.IsVideo = true
	}

	return result
}

func FormatBytes(bytes int64) string {
	const (
		KB = 1 << 10 // 1024
		MB = 1 << 20
		GB = 1 << 30
		TB = 1 << 40
	)

	switch {
	case bytes >= TB:
		return fmt.Sprintf("%.2f TB", float64(bytes)/float64(TB))
	case bytes >= GB:
		return fmt.Sprintf("%.2f GB", float64(bytes)/float64(GB))
	case bytes >= MB:
		return fmt.Sprintf("%.2f MB", float64(bytes)/float64(MB))
	case bytes >= KB:
		return fmt.Sprintf("%.2f KB", float64(bytes)/float64(KB))
	default:
		return fmt.Sprintf("%d B", bytes)
	}
}

func FormatFileInfo(fileInfo PostFileInfo) string {
	humanSize := FormatBytes(fileInfo.Size)
	if fileInfo.IsVideo {
		return fmt.Sprintf("(%s)", humanSize)
	}
	return fmt.Sprintf("(%s, %dx%d)", humanSize, fileInfo.Width, fileInfo.Height)
}