Feat: Live thumbnails for office files

backend/handlers/onlyoffice.go

@@ -10,13 +10,19 @@ import (
 	"strings"
 
 	"git.elijahkuntz.com/Elijah/drive/config"
+	"git.elijahkuntz.com/Elijah/drive/database"
 	"git.elijahkuntz.com/Elijah/drive/middleware"
+	"git.elijahkuntz.com/Elijah/drive/workers"
 	"github.com/gofiber/fiber/v2"
 	"github.com/golang-jwt/jwt/v5"
+	"sync"
+	"time"
+	"mime"
 )
 
 type OnlyOfficeHandler struct {
 	Config *config.Config
+	DB     *database.DB
 }
 
 type CallbackRequest struct {
@@ -73,12 +79,51 @@ func (h *OnlyOfficeHandler) Callback(c *fiber.Ctx) error {
 		if _, err := io.Copy(out, resp.Body); err != nil {
 			return c.Status(500).JSON(fiber.Map{"error": "failed to write file"})
 		}
+		
+		h.scheduleThumbnailUpdate(targetPath, path)
 	}
 
 	// Send successful response according to OnlyOffice API
 	return c.JSON(fiber.Map{"error": 0})
 }
 
+var (
+	saveDebounceMutex sync.Mutex
+	saveTimers        = make(map[string]*time.Timer)
+)
+
+func (h *OnlyOfficeHandler) scheduleThumbnailUpdate(targetPath, relativePath string) {
+	saveDebounceMutex.Lock()
+	defer saveDebounceMutex.Unlock()
+
+	if timer, exists := saveTimers[targetPath]; exists {
+		timer.Stop()
+	}
+
+	saveTimers[targetPath] = time.AfterFunc(10*time.Second, func() {
+		saveDebounceMutex.Lock()
+		delete(saveTimers, targetPath)
+		saveDebounceMutex.Unlock()
+
+		checksum, err := generateChecksum(targetPath)
+		if err != nil {
+			return
+		}
+		info, err := os.Stat(targetPath)
+		if err != nil {
+			return
+		}
+
+		h.DB.Exec("UPDATE files SET checksum = ?, size = ?, updated_at = CURRENT_TIMESTAMP WHERE path = ?", checksum, info.Size(), filepath.ToSlash(relativePath))
+
+		workers.Instance.Enqueue(workers.ThumbnailJob{
+			FilePath: relativePath,
+			Checksum: checksum,
+			MimeType: mime.TypeByExtension(filepath.Ext(relativePath)),
+		})
+	})
+}
+
 // DownloadForEditor serves a file to the OnlyOffice Document Server.
 // This is a public endpoint that validates the download token manually,
 // bypassing the auth middleware (which can fail when requests are proxied