From 9132c2d0533f49917a81f19a858a6a63a3221bff Mon Sep 17 00:00:00 2001
From: Yiheng <yihengmen@gmail.com>
Date: Thu, 9 May 2024 16:17:18 +0800
Subject: [PATCH] first commit

---
 __init__.py              |  3 ++
 get_filename_from_url.py | 64 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 67 insertions(+)
 create mode 100644 __init__.py
 create mode 100644 get_filename_from_url.py

diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..f4ddb81
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,3 @@
+from .get_filename_from_url import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
\ No newline at end of file
diff --git a/get_filename_from_url.py b/get_filename_from_url.py
new file mode 100644
index 0000000..2439353
--- /dev/null
+++ b/get_filename_from_url.py
@@ -0,0 +1,64 @@
+import requests
+from urllib.parse import unquote
+import cgi
+
+class GetFileNameFromURL:
+    def __init__(self):
+        pass
+        
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "url": ("STRING", {"default": ""}),
+            },
+        }
+
+    RETURN_TYPES = ("STRING", "STRING")
+    RETURN_NAMES = ("filename", "url")
+    FUNCTION = "get_filename"
+    CATEGORY = "Tools"
+
+    def get_filename(self, url):
+        response = requests.get(url, stream=True)
+        if response.status_code != 200:
+            raise Exception(f"Error downloading file: {response.status_code}")
+
+        # 处理重定向
+        if response.status_code == 303:
+            redirect_url = response.headers.get('Location')
+            response = requests.get(redirect_url, stream=True)
+            if response.status_code != 200:
+                raise Exception(f"Error downloading file: {response.status_code}")
+
+        # 尝试从 Content-Disposition 头中获取文件名
+        content_disposition = response.headers.get('Content-Disposition')
+        if content_disposition:
+            _, params = cgi.parse_header(content_disposition)
+            filename = params.get("filename*")
+            if filename:
+                filename = unquote(filename.replace("UTF-8''", ""))
+            else:
+                filename = params.get("filename")
+        else:
+            # 尝试从 X-Goog-Filename 头中获取文件名
+            filename = response.headers.get('X-Goog-Filename')
+
+        if not filename:
+            raise Exception("Unable to determine filename")
+
+        return (filename, url)
+
+# Set the web directory, any .js file in that directory will be loaded by the frontend as a frontend extension
+# WEB_DIRECTORY = "./somejs"
+
+# A dictionary that contains all nodes you want to export with their names
+# NOTE: names should be globally unique
+NODE_CLASS_MAPPINGS = {
+    "GetFileNameFromURL": GetFileNameFromURL
+}
+
+# A dictionary that contains the friendly/humanly readable titles for the nodes
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "GetFileNameFromURL": "Get FileName From URL"
+}