SalexAI commited on
Commit
ef08f0b
·
verified ·
1 Parent(s): 45a431f

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +140 -0
app.py ADDED
@@ -0,0 +1,140 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from fastapi.middleware.cors import CORSMiddleware
3
+ import httpx
4
+ import json
5
+ import logging
6
+
7
+ app = FastAPI()
8
+ logging.basicConfig(level=logging.INFO)
9
+
10
+ # Allow all CORS so browser apps can call this API
11
+ def configure_cors(app: FastAPI):
12
+ app.add_middleware(
13
+ CORSMiddleware,
14
+ allow_origins=["*"],
15
+ allow_methods=["*"],
16
+ allow_headers=["*"],
17
+ )
18
+
19
+ configure_cors(app)
20
+
21
+ # Precompute Base62 index mapping for O(1) lookup
22
+ BASE_62_MAP = {c: i for i, c in enumerate(
23
+ "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
24
+ )}
25
+
26
+ async def get_client() -> httpx.AsyncClient:
27
+ if not hasattr(app.state, "client"):
28
+ app.state.client = httpx.AsyncClient(timeout=10.0)
29
+ return app.state.client
30
+
31
+ # Convert base62 string to integer
32
+ def base62_to_int(token: str) -> int:
33
+ result = 0
34
+ for ch in token:
35
+ result = result * 62 + BASE_62_MAP[ch]
36
+ return result
37
+
38
+ async def get_base_url(token: str) -> str:
39
+ first = token[0]
40
+ if first == "A":
41
+ n = base62_to_int(token[1])
42
+ else:
43
+ n = base62_to_int(token[1:3])
44
+ return f"https://p{n:02d}-sharedstreams.icloud.com/{token}/sharedstreams/"
45
+
46
+ # Common headers and payload for iCloud requests
47
+ ICLOUD_HEADERS = {"Origin": "https://www.icloud.com", "Content-Type": "text/plain"}
48
+ ICLOUD_PAYLOAD = '{"streamCtag":null}'
49
+
50
+ async def get_redirected_base_url(base_url: str, token: str) -> str:
51
+ client = await get_client()
52
+ resp = await client.post(
53
+ f"{base_url}webstream", headers=ICLOUD_HEADERS, data=ICLOUD_PAYLOAD, follow_redirects=False
54
+ )
55
+ if resp.status_code == 330:
56
+ body = resp.json()
57
+ host = body.get("X-Apple-MMe-Host")
58
+ return f"https://{host}/{token}/sharedstreams/"
59
+ return base_url
60
+
61
+ async def post_json(path: str, base_url: str, payload: str) -> dict:
62
+ client = await get_client()
63
+ resp = await client.post(
64
+ f"{base_url}{path}", headers=ICLOUD_HEADERS, data=payload
65
+ )
66
+ resp.raise_for_status()
67
+ return resp.json()
68
+
69
+ async def get_metadata(base_url: str) -> list:
70
+ data = await post_json("webstream", base_url, ICLOUD_PAYLOAD)
71
+ return data.get("photos", [])
72
+
73
+ async def get_asset_urls(base_url: str, guids: list) -> dict:
74
+ payload = json.dumps({"photoGuids": guids})
75
+ data = await post_json("webasseturls", base_url, payload)
76
+ return data.get("items", {})
77
+
78
+ @app.get("/album/{token}")
79
+ async def get_album(token: str):
80
+ try:
81
+ base_url = await get_base_url(token)
82
+ base_url = await get_redirected_base_url(base_url, token)
83
+
84
+ metadata = await get_metadata(base_url)
85
+ guids = [photo["photoGuid"] for photo in metadata]
86
+ asset_map = await get_asset_urls(base_url, guids)
87
+
88
+ videos = []
89
+ for photo in metadata:
90
+ if photo.get("mediaAssetType", "").lower() != "video":
91
+ logging.info(f"Photo {photo.get('photoGuid')} is not a video.")
92
+ continue
93
+
94
+ derivatives = photo.get("derivatives", {})
95
+ best = max(
96
+ (d for k, d in derivatives.items() if k.lower() != "posterframe"),
97
+ key=lambda d: int(d.get("fileSize") or 0),
98
+ default=None
99
+ )
100
+ if not best:
101
+ logging.info(f"No video derivative for photo {photo.get('photoGuid')}")
102
+ continue
103
+
104
+ checksum = best.get("checksum")
105
+ info = asset_map.get(checksum)
106
+ if not info:
107
+ logging.info(f"Missing asset for checksum {checksum}")
108
+ continue
109
+ video_url = f"https://{info['url_location']}{info['url_path']}"
110
+
111
+ poster = None
112
+ pf = derivatives.get("PosterFrame")
113
+ if pf:
114
+ pf_info = asset_map.get(pf.get("checksum"))
115
+ if pf_info:
116
+ poster = f"https://{pf_info['url_location']}{pf_info['url_path']}"
117
+
118
+ videos.append({
119
+ "caption": photo.get("caption", ""),
120
+ "url": video_url,
121
+ "poster": poster
122
+ })
123
+
124
+ return {"videos": videos}
125
+ except Exception as e:
126
+ logging.error(f"Error in get_album: {e}")
127
+ return {"error": str(e)}
128
+
129
+ @app.get("/album/{token}/raw")
130
+ async def get_album_raw(token: str):
131
+ try:
132
+ base_url = await get_base_url(token)
133
+ base_url = await get_redirected_base_url(base_url, token)
134
+ metadata = await get_metadata(base_url)
135
+ guids = [photo["photoGuid"] for photo in metadata]
136
+ asset_map = await get_asset_urls(base_url, guids)
137
+ return {"metadata": metadata, "asset_urls": asset_map}
138
+ except Exception as e:
139
+ logging.error(f"Error in get_album_raw: {e}")
140
+ return {"error": str(e)}