⚡ Better frontend asset serving

Signed-off-by: Muhammed Hussain Karimi <[email protected]>

Author: mhkarimi1383

cmd/root.go

@@ -212,27 +212,36 @@ func start(_ *cobra.Command, _ []string) {
 
 	uiGroup := rootGroup.Group("/ui")
 	uiGroup.Any("", nil, addTrailingSlashMiddleware)
+	uiCache := make(map[string]string)
 	uiGroup.GET("/*", func(c echo.Context) error {
 		prefix := filepath.Join("/"+strings.TrimPrefix(cfg.BaseURI, "/"), "/ui/")
 		filePath := strings.TrimPrefix(strings.TrimPrefix(c.Request().URL.Path, prefix), "/")
-		file, err := ui.MainFS.Open(filePath)
-		if err != nil {
-			filePath = "index.html"
-			file, _ = ui.MainFS.Open(filePath)
-		}
-		defer file.Close()
-		buf := new(bytes.Buffer)
-		_, err = io.Copy(buf, file)
-		if err != nil {
-			return err
-		}
-		newURI, err := neturl.JoinPath("/BASE_URI/", "../"+cfg.BaseURI+"/")
-		if err != nil {
-			return err
+		modifiedContent, ok := uiCache[filePath]
+		if !ok {
+			file, err := ui.MainFS.Open(filePath)
+			if err != nil {
+				filePath = "index.html"
+				file, _ = ui.MainFS.Open(filePath)
+			}
+			defer file.Close()
+			buf := new(bytes.Buffer)
+			_, err = io.Copy(buf, file)
+			if err != nil {
+				return err
+			}
+			newURI, err := neturl.JoinPath("/BASE_URI/", "../"+cfg.BaseURI+"/")
+			if err != nil {
+				return err
+			}
+			modifiedContent = strings.ReplaceAll(buf.String(), "/BASE_URI/", newURI)
+			uiCache[filePath] = modifiedContent
 		}
-		modifiedContent := strings.ReplaceAll(buf.String(), "/BASE_URI/", newURI)
 		fileParts := strings.Split(filePath, ".")
-		return c.Blob(200, mime.TypeByExtension("."+fileParts[len(fileParts)-1]), []byte(modifiedContent))
+		mimeType := mime.TypeByExtension("." + fileParts[len(fileParts)-1])
+		if mimeType == "" {
+			mimeType = "text/html; charset=utf-8"
+		}
+		return c.Blob(200, mimeType, []byte(modifiedContent))
 	})
 
 	if configuration.CurrentConfig.RunServer {
@@ -243,7 +252,6 @@ func start(_ *cobra.Command, _ []string) {
 		)
 	}
 
-	commandGroup := apiGroup.Group("/command", checkUserAdmin)
-	commandGroup.DELETE("/remove-old-links", url.RemoveOldIds)
-
+	commandGroup := apiGroup.Group("/command")
+	commandGroup.DELETE("/remove-old-links", url.RemoveUnusedUrls)
 }

constrains/params.go

@@ -6,4 +6,5 @@ const (
 	ShortCodeParamName   = "ShortCode"
 	IdParamName          = "Id"
 	RefererQueryParam    = "referer"
+	CutoffQueryParamName = "Cutoff"
 )

internal/endpoint/url/url.go

@@ -80,22 +80,25 @@ func Delete(c echo.Context) error {
 	return c.NoContent(http.StatusNoContent)
 }
 
-func RemoveOldIds(c echo.Context) error {
+func RemoveUnusedUrls(c echo.Context) error {
 	user := c.Get(constrains.UserInfoContextVar).(databasemodels.User)
+	cutoffStr := c.QueryParam(constrains.CutoffQueryParamName) // Example: 2006-01-02T15:04:05Z07:00
+	if cutoffStr == "" {
+		cutoffStr = time.Now().AddDate(0, -1, 0).Format(time.RFC3339)
+	}
 
-	// Define cutoff date (6 months ago)
-	cutoff := time.Now().AddDate(0, -1, 0)
+	cutoff, err := time.Parse(time.RFC3339, cutoffStr)
+	if err != nil {
+		return err
+	}
 
-	// Prepare query conditions
 	session := database.Engine.Where("last_visited_at < ?", cutoff)
 
 	if !user.Admin {
-		// Non-admins can only delete their own URLs
 		session = session.And("creator_id = ?", user.Id)
 	}
 
-	// Delete matching records
-	_, err := session.Delete(&databasemodels.Url{})
+	_, err = session.Delete(&databasemodels.Url{})
 	if err != nil {
 		return err
 	}