feat: Hash images

Author: Axorax

core.py

@@ -1,8 +1,11 @@
 import os
+import hashlib
 import requests
 import threading
 from utils import rgb_to_hex, get_foreground_color
 
+HASH_IMAGES = True
+
 def get_file(file, token):
     response = requests.get(f"https://api.figma.com/v1/files/{file}", headers={'X-FIGMA-TOKEN': token})
 
@@ -11,6 +14,11 @@ def get_file(file, token):
     else:
         return response.status_code, response.text
 
+downloaded_images = {}
+
+def get_image_hash(image_content):
+    return hashlib.sha256(image_content).hexdigest()
+
 def download_image(file, id, name, token, out=None, frame=None):
     response = requests.get(f"https://api.figma.com/v1/images/{file}", headers={'X-FIGMA-TOKEN': token}, params={'ids': id})
 
@@ -32,13 +40,20 @@ def download_image(file, id, name, token, out=None, frame=None):
             image_response = requests.get(image_url)
 
             if image_response.status_code == 200:
+                image_content = image_response.content
+                if HASH_IMAGES:
+                    image_hash = get_image_hash(image_content)
+
+                    if image_hash in downloaded_images:
+                        return downloaded_images[image_hash]
+                
                 with open(file_path, 'wb') as f:
-                    f.write(image_response.content)
+                    f.write(image_content)
 
-                if frame is not None:
-                    return os.path.join(f'frame_{frame}', file_name).replace('\\', '/')
-                else:
-                    return file_name
+                if HASH_IMAGES:
+                    downloaded_images[image_hash] = os.path.join(f'frame_{frame}', file_name).replace('\\', '/') if frame else file_name
+
+                return os.path.join(f'frame_{frame}', file_name).replace('\\', '/') if frame else file_name
             else:
                 print("Failed to download the image.")
         else:
@@ -58,8 +73,6 @@ def parse_file(file, token, download_images=True, out=None):
     try:
         frames = result['document']['children'][0]['children']
         frame_count = 1 if len(frames) > 1 else 0
-        # import json
-        # print(json.dumps(frames, indent=4, sort_keys=True))
 
         def parse_frame(frame, frame_count):
             nonlocal output