From 3458f18e16962acf27d2b917c9c1da72f6ba4325 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ivan=20=C5=A0ari=C4=87?= <ivan@path-variable.com>
Date: Sun, 24 Dec 2023 14:14:04 +0100
Subject: [PATCH] filter on mimetype - image only

---
 docker-compose.yml | 4 +++-
 main.py            | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 0ad69aa..a487551 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,7 +1,9 @@
 version: '3.0'
 services:
   api:
-    build: .
+    build:
+      context: .
+      dockerfile: dockerfile-amd64
     ports:
       - "5000:5000"
       - "5678:5678"
diff --git a/main.py b/main.py
index 5affc2d..13d89eb 100644
--- a/main.py
+++ b/main.py
@@ -20,7 +20,9 @@ def process(self, content):
             return
         if "event" in content and "files" in content["event"]:
             app.logger.info("Received event contains files")
-            # TODO: switch to manageable list of channel ids
+            if "image" not in content["event"]["files"][0]["mimetype"]:
+                app.logger.info("Received file is not an image")
+                return
             if content["event"]["channel"] in self.read_channels:
                 detector.detect(content["event"]["files"][0]["url_private_download"])
             else: