feat(server): ✨ implement check for image mime types

Dan6erbond · May 14, 2023 · 1ba3108 · 1ba3108
1 parent afaf49d
commit 1ba3108
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/server/pkg/tags/commands.go b/server/pkg/tags/commands.go
@@ -32,6 +32,18 @@ func RegisterCommands(app *pocketbase.PocketBase, taggerHost string) {
 					continue
 				}
 
+				var isImageType bool
+				for _, t := range imageMimeTypes {
+					if file.GetString("type") == t {
+						isImageType = true
+						continue
+					}
+				}
+
+				if !isImageType {
+					continue
+				}
+
 				url, err := url.Parse(taggerHost)
 
 				if err != nil {

diff --git a/server/pkg/tags/hooks.go b/server/pkg/tags/hooks.go
@@ -32,6 +32,18 @@ func RegisterHooks(app *pocketbase.PocketBase, taggerHost string) {
 			return nil
 		}
 
+		var isImageType bool
+		for _, t := range imageMimeTypes {
+			if e.Record.GetString("type") == t {
+				isImageType = true
+				continue
+			}
+		}
+
+		if !isImageType {
+			return nil
+		}
+
 		url, err := url.Parse(taggerHost)
 
 		if err != nil {

diff --git a/server/pkg/tags/utils.go b/server/pkg/tags/utils.go
@@ -0,0 +1,3 @@
+package tags
+
+var imageMimeTypes = []string{"image/png", "image/gif", "image/jpeg", "image/svg+xml", "image/webp", "image/avif"}
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		package tags

		var imageMimeTypes = []string{"image/png", "image/gif", "image/jpeg", "image/svg+xml", "image/webp", "image/avif"}