feat: add ImageFind application and release pipelines

2026-08-11 18:02:40 +08:00
commit 16239d7525
270 changed files with 59163 additions and 0 deletions
@@ -0,0 +1,73 @@
+from pathlib import Path
+
+import httpx
+from imagefind.sources import LocalConnector, WebDavConnector
+
+
+def test_local_connector_ignores_symlinks_and_non_video(tmp_path: Path):
+    (tmp_path / "a.mp4").write_bytes(b"video")
+    (tmp_path / "note.txt").write_text("no")
+    outside = tmp_path.parent / "outside.mkv"
+    outside.write_bytes(b"outside")
+    (tmp_path / "link.mkv").symlink_to(outside)
+    items = list(LocalConnector(str(tmp_path)).items())
+    assert [item.key for item in items] == ["a.mp4"]
+
+
+def test_webdav_depth_one_and_external_href_rejection():
+    xml = b"""<?xml version="1.0"?>
+    <d:multistatus xmlns:d="DAV:">
+      <d:response>
+        <d:href>/videos/</d:href><d:propstat><d:prop>
+          <d:resourcetype><d:collection/></d:resourcetype>
+        </d:prop><d:status>HTTP/1.1 200 OK</d:status></d:propstat>
+      </d:response>
+      <d:response>
+        <d:href>/videos/movie.mp4</d:href><d:propstat><d:prop>
+          <d:resourcetype/><d:getcontentlength>123</d:getcontentlength><d:getetag>abc</d:getetag>
+        </d:prop><d:status>HTTP/1.1 200 OK</d:status></d:propstat>
+      </d:response>
+      <d:response>
+        <d:href>https://evil.example/leak.mp4</d:href><d:propstat><d:prop>
+          <d:resourcetype/>
+        </d:prop><d:status>HTTP/1.1 200 OK</d:status></d:propstat>
+      </d:response>
+    </d:multistatus>"""
+
+    def handler(request: httpx.Request):
+        assert request.method == "PROPFIND"
+        assert request.headers["Depth"] == "1"
+        return httpx.Response(207, content=xml)
+
+    connector = WebDavConnector("https://dav.example/videos/", "user", "pass")
+    connector.client.close()
+    connector.client = httpx.Client(transport=httpx.MockTransport(handler))
+    items = list(connector.items())
+    connector.close()
+    assert len(items) == 1
+    assert items[0].key == "movie.mp4"
+    assert items[0].etag == "abc"
+
+
+def test_webdav_tolerates_alist_xml_with_extra_payload():
+    xml = b"""alist notice
+    <D:multistatus xmlns:D="DAV:">
+      <D:response>
+        <D:href>/dav/movie.mp4</D:href><D:propstat><D:prop>
+          <D:resourcetype/><D:getcontentlength>42</D:getcontentlength>
+        </D:prop><D:status>HTTP/1.1 200 OK</D:status></D:propstat>
+      </D:response>
+    </D:multistatus><ignored/>"""
+
+    def handler(_request: httpx.Request):
+        return httpx.Response(207, content=xml)
+
+    connector = WebDavConnector("https://dav.example/dav/", "user", "pass")
+    connector.client.close()
+    connector.client = httpx.Client(transport=httpx.MockTransport(handler))
+    items = list(connector.items())
+    connector.close()
+
+    assert len(items) == 1
+    assert items[0].key == "movie.mp4"
+    assert items[0].size_bytes == 42