birdclaw 0.8.2 → 0.8.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +33 -0
- package/README.md +10 -8
- package/bin/birdclaw.mjs +4 -66
- package/dist/cli/birdclaw.js +28165 -0
- package/dist/client/assets/AvatarChip-DLoqM5xD.js +1 -0
- package/dist/client/assets/ProfilePreview-BejAuYsJ.js +1 -0
- package/dist/client/assets/SavedTimelineView-BIS7HKpp.js +1 -0
- package/dist/client/assets/SmartTimestamp-BR0-9se2.js +1 -0
- package/dist/client/assets/TimelineRouteFrame-CyGPj0eM.js +1 -0
- package/dist/client/assets/blocks-7UDKQp8H.js +1 -0
- package/dist/client/assets/bookmarks-CbBkMvXk.js +1 -0
- package/dist/client/assets/chunk-C0phanGV.js +1 -0
- package/dist/client/assets/circle-check-C_Vq3ElD.js +1 -0
- package/dist/client/assets/client-stream-contracts-ys3XhWG4.js +5 -0
- package/dist/client/assets/createLucideIcon-JV8isFFR.js +1 -0
- package/dist/client/assets/data-sources-Cm2bpNvg.js +1 -0
- package/dist/client/assets/discuss-DTjaOugw.js +1 -0
- package/dist/client/assets/dms-CGd6Eogq.js +1 -0
- package/dist/client/assets/inbox-BHlGuOy2.js +1 -0
- package/dist/client/assets/index-CEIG7S6s.js +31 -0
- package/dist/client/assets/index-bpFrQ5yG.css +1 -0
- package/dist/client/assets/likes-cLrcAQem.js +1 -0
- package/dist/client/assets/links-CM9QO1LC.js +1 -0
- package/dist/client/assets/mapbox-BxTlqXt9.js +2 -0
- package/dist/client/assets/mapbox-gl-dgf_hX3a.js +3215 -0
- package/dist/client/assets/mentions-CxKCgzIf.js +1 -0
- package/dist/client/assets/network-map-B5a3whqj.js +2 -0
- package/dist/client/assets/profile-analyze-CRtNEwlb.js +1 -0
- package/dist/client/assets/profile-hydration-client-YEtJsNTv.js +1 -0
- package/dist/client/assets/profiles._handle-C82_HoKX.js +1 -0
- package/dist/client/assets/query-client-fYu2rzRC.js +17 -0
- package/dist/client/assets/rate-limits-ScaEwdVK.js +1 -0
- package/dist/client/assets/react-dom-DHl6Ug7F.js +1 -0
- package/dist/client/assets/react-nBylq1U5.js +1 -0
- package/dist/client/assets/repeat-2-xCwHM9hD.js +1 -0
- package/dist/client/assets/routes-q7ZaW4En.js +1 -0
- package/dist/client/assets/schemas-I9r6F6ya.js +64 -0
- package/dist/client/assets/search-Dq5kYsxA.js +1 -0
- package/dist/client/assets/sparkles-CQVlvHsZ.js +1 -0
- package/dist/client/assets/styles-B8s3tt3r.css +2 -0
- package/dist/client/assets/today-CeG1O7Py.js +1 -0
- package/dist/client/assets/triangle-alert-CBL7s2pf.js +1 -0
- package/dist/client/assets/useTimelineRouteData-1HsqgvUi.js +1 -0
- package/dist/client/assets/users-CHbKlqUr.js +1 -0
- package/dist/server/assets/AvatarChip-zUDAEmjL.js +105 -0
- package/dist/server/assets/FeedState-ntW0Q2GB.js +214 -0
- package/dist/server/assets/ProfileAnalysisStream-BFvhVdhn.js +198 -0
- package/dist/server/assets/ProfilePreview-C6gAor_j.js +752 -0
- package/dist/server/assets/SavedTimelineView-BG1jnBp1.js +64 -0
- package/dist/server/assets/SmartTimestamp-DfwE1NGf.js +45 -0
- package/dist/server/assets/SyncNowButton-BOn_IBXS.js +76 -0
- package/dist/server/assets/TimelineRouteFrame-CU7ilM7U.js +85 -0
- package/dist/server/assets/TweetRichText-DhCVv1-6.js +74 -0
- package/dist/server/assets/_tanstack-start-manifest_v-C-ZRnG_n.js +148 -0
- package/dist/server/assets/account-selection-CTdFGWRk.js +44 -0
- package/dist/server/assets/blocks-C1mgGTB3.js +299 -0
- package/dist/server/assets/blocks-DSgQghqP.js +17 -0
- package/dist/server/assets/bookmarks-BJN8tykY.js +14 -0
- package/dist/server/assets/client-stream-contracts-C2xbWVjm.js +769 -0
- package/dist/server/assets/data-sources-BnA1Q94p.js +180 -0
- package/dist/server/assets/discuss-CJ1dau1j.js +15 -0
- package/dist/server/assets/discuss-QTo08NCN.js +338 -0
- package/dist/server/assets/dms-BuLj4bp8.js +368 -0
- package/dist/server/assets/dms-DAyQrVM2.js +240 -0
- package/dist/server/assets/empty-plugin-adapters-DzrEkdBF.js +5 -0
- package/dist/server/assets/inbox-CkGOaW1q.js +214 -0
- package/dist/server/assets/inbox-DpWd1EzC.js +146 -0
- package/dist/server/assets/likes-DBOAtsJU.js +14 -0
- package/dist/server/assets/links-DAhSjynu.js +240 -0
- package/dist/server/assets/links-DmH5lxHO.js +468 -0
- package/dist/server/assets/mentions-BH2MzKtB.js +26 -0
- package/dist/server/assets/network-map-D3CzRcC4.js +224 -0
- package/dist/server/assets/network-map-D5DoLOwR.js +582 -0
- package/dist/server/assets/profile-analyze-BP65PED0.js +90 -0
- package/dist/server/assets/profile-analyze-C1NoM0tV.js +9 -0
- package/dist/server/assets/profile-hydration-client-GJ4kwJS8.js +133 -0
- package/dist/server/assets/profiles._handle-DjwlN_ac.js +184 -0
- package/dist/server/assets/profiles._handle-DsLNj2Ca.js +12 -0
- package/dist/server/assets/query-client-BdauHBgd.js +923 -0
- package/dist/server/assets/rate-limits-VAr2qzYD.js +262 -0
- package/dist/server/assets/route-search-BUlQ5lGM.js +117 -0
- package/dist/server/assets/router-DqkwSPiM.js +17026 -0
- package/dist/server/assets/routes-A9c9ZhwN.js +26 -0
- package/dist/server/assets/start-FaDOTmto.js +4 -0
- package/dist/server/assets/today-B5hVM57Z.js +17 -0
- package/dist/server/assets/today-fvGiZ7ud.js +303 -0
- package/dist/server/assets/ui-qbmcMSSX.js +137 -0
- package/{src/components/useDebouncedValue.ts → dist/server/assets/useDebouncedValue-DOnA6Xcy.js} +4 -4
- package/dist/server/assets/useTimelineRouteData-27cxynxa.js +1101 -0
- package/dist/server/server.js +5520 -0
- package/package.json +27 -34
- package/playwright.config.ts +0 -35
- package/scripts/browser-perf.mjs +0 -427
- package/scripts/build-docs-site.mjs +0 -967
- package/scripts/docs-site-assets.mjs +0 -311
- package/scripts/run-vitest.mjs +0 -21
- package/scripts/sanitize-node-options.mjs +0 -23
- package/scripts/start-test-server.mjs +0 -42
- package/src/cli-moderation.ts +0 -210
- package/src/cli.ts +0 -2336
- package/src/components/AccountSwitcher.tsx +0 -186
- package/src/components/AppNav.tsx +0 -118
- package/src/components/AvatarChip.tsx +0 -58
- package/src/components/BrandMark.tsx +0 -67
- package/src/components/ConversationThread.tsx +0 -131
- package/src/components/DmWorkspace.tsx +0 -286
- package/src/components/EmbeddedTweetCard.tsx +0 -55
- package/src/components/FeedState.tsx +0 -147
- package/src/components/FloatingPreview.tsx +0 -382
- package/src/components/InboxCard.tsx +0 -152
- package/src/components/LinkPreviewCard.tsx +0 -292
- package/src/components/MarkdownViewer.tsx +0 -777
- package/src/components/ProfileAnalysisStream.tsx +0 -428
- package/src/components/ProfilePreview.tsx +0 -129
- package/src/components/SavedTimelineView.tsx +0 -155
- package/src/components/SmartTimestamp.tsx +0 -72
- package/src/components/SyncNowButton.tsx +0 -140
- package/src/components/ThemeSlider.tsx +0 -80
- package/src/components/TimelineCard.tsx +0 -533
- package/src/components/TimelineRouteFrame.tsx +0 -186
- package/src/components/TweetArticleCard.tsx +0 -66
- package/src/components/TweetMediaGrid.tsx +0 -185
- package/src/components/TweetRichText.tsx +0 -159
- package/src/components/account-selection.ts +0 -64
- package/src/components/useTimelineRouteData.ts +0 -323
- package/src/lib/account-sync-job.ts +0 -666
- package/src/lib/actions-transport.ts +0 -300
- package/src/lib/api-client.ts +0 -419
- package/src/lib/archive-finder.ts +0 -185
- package/src/lib/archive-import.ts +0 -2764
- package/src/lib/authored-live.ts +0 -1072
- package/src/lib/avatar-cache.ts +0 -349
- package/src/lib/backup.ts +0 -2741
- package/src/lib/bird-actions.ts +0 -294
- package/src/lib/bird-command.ts +0 -130
- package/src/lib/bird.ts +0 -1140
- package/src/lib/blocklist.ts +0 -136
- package/src/lib/blocks-write.ts +0 -167
- package/src/lib/blocks.ts +0 -389
- package/src/lib/bookmark-sync-job.ts +0 -536
- package/src/lib/client-cache.ts +0 -109
- package/src/lib/config.ts +0 -224
- package/src/lib/conversation-surface.ts +0 -205
- package/src/lib/data-sources.ts +0 -219
- package/src/lib/db.ts +0 -1135
- package/src/lib/dms-live.ts +0 -933
- package/src/lib/effect-runtime.ts +0 -45
- package/src/lib/follow-graph.ts +0 -1097
- package/src/lib/geocoding.ts +0 -296
- package/src/lib/http-effect.ts +0 -222
- package/src/lib/identity-search-index.ts +0 -369
- package/src/lib/inbox.ts +0 -241
- package/src/lib/link-index.ts +0 -641
- package/src/lib/link-insights.ts +0 -860
- package/src/lib/link-preview-metadata.ts +0 -754
- package/src/lib/location.ts +0 -137
- package/src/lib/media-fetch.ts +0 -860
- package/src/lib/media-includes.ts +0 -165
- package/src/lib/mention-threads-live.ts +0 -917
- package/src/lib/mentions-export.ts +0 -147
- package/src/lib/mentions-live.ts +0 -1238
- package/src/lib/moderation-target.ts +0 -216
- package/src/lib/moderation-write.ts +0 -131
- package/src/lib/mutes-write.ts +0 -167
- package/src/lib/mutes.ts +0 -84
- package/src/lib/ndjson-stream.ts +0 -106
- package/src/lib/network-map.ts +0 -382
- package/src/lib/openai.ts +0 -117
- package/src/lib/period-digest.ts +0 -1552
- package/src/lib/present.ts +0 -118
- package/src/lib/profile-affiliation-hydration.ts +0 -228
- package/src/lib/profile-affiliations.ts +0 -262
- package/src/lib/profile-analysis.ts +0 -1272
- package/src/lib/profile-bio-entities.ts +0 -318
- package/src/lib/profile-history.ts +0 -164
- package/src/lib/profile-hydration.ts +0 -180
- package/src/lib/profile-replies.ts +0 -77
- package/src/lib/profile-resolver.ts +0 -650
- package/src/lib/queries.ts +0 -2381
- package/src/lib/research.ts +0 -652
- package/src/lib/search-discussion.ts +0 -1016
- package/src/lib/seed.ts +0 -613
- package/src/lib/sqlite.ts +0 -147
- package/src/lib/sync-cache.ts +0 -65
- package/src/lib/theme-transition.ts +0 -113
- package/src/lib/theme.tsx +0 -165
- package/src/lib/timeline-collections-live.ts +0 -581
- package/src/lib/timeline-live.ts +0 -452
- package/src/lib/tweet-account-edges.ts +0 -48
- package/src/lib/tweet-lookup.ts +0 -46
- package/src/lib/tweet-render.ts +0 -430
- package/src/lib/tweet-search-live.ts +0 -566
- package/src/lib/types.ts +0 -723
- package/src/lib/ui.ts +0 -529
- package/src/lib/url-expansion-store.ts +0 -110
- package/src/lib/url-expansion.ts +0 -343
- package/src/lib/url-safety.ts +0 -220
- package/src/lib/web-sync.ts +0 -516
- package/src/lib/whois.ts +0 -1079
- package/src/lib/x-profile.ts +0 -365
- package/src/lib/x-web.ts +0 -199
- package/src/lib/xurl-rate-limits.ts +0 -267
- package/src/lib/xurl.ts +0 -1677
- package/src/routeTree.gen.ts +0 -723
- package/src/router.tsx +0 -20
- package/src/routes/__root.tsx +0 -79
- package/src/routes/api/action.tsx +0 -147
- package/src/routes/api/avatar.tsx +0 -50
- package/src/routes/api/blocks.tsx +0 -35
- package/src/routes/api/conversation.tsx +0 -45
- package/src/routes/api/data-sources.tsx +0 -24
- package/src/routes/api/inbox.tsx +0 -43
- package/src/routes/api/link-insights.tsx +0 -85
- package/src/routes/api/link-preview.tsx +0 -47
- package/src/routes/api/network-map.tsx +0 -55
- package/src/routes/api/period-digest.tsx +0 -101
- package/src/routes/api/profile-analysis.tsx +0 -97
- package/src/routes/api/profile-hydrate.tsx +0 -54
- package/src/routes/api/query.tsx +0 -115
- package/src/routes/api/search-discussion.tsx +0 -113
- package/src/routes/api/status.tsx +0 -28
- package/src/routes/api/sync.tsx +0 -105
- package/src/routes/api/xurl-rate-limits.tsx +0 -24
- package/src/routes/blocks.tsx +0 -377
- package/src/routes/bookmarks.tsx +0 -18
- package/src/routes/data-sources.tsx +0 -257
- package/src/routes/discuss.tsx +0 -419
- package/src/routes/dms.tsx +0 -432
- package/src/routes/inbox.tsx +0 -226
- package/src/routes/index.tsx +0 -32
- package/src/routes/likes.tsx +0 -18
- package/src/routes/links.tsx +0 -994
- package/src/routes/mentions.tsx +0 -32
- package/src/routes/network-map.tsx +0 -1035
- package/src/routes/profile-analyze.tsx +0 -112
- package/src/routes/profiles.$handle.tsx +0 -228
- package/src/routes/rate-limits.tsx +0 -309
- package/src/routes/today.tsx +0 -511
- package/src/styles.css +0 -257
- package/tsconfig.json +0 -29
- package/vite.config.ts +0 -29
- /package/{public → dist/client}/birdclaw-mark.png +0 -0
- /package/{public → dist/client}/favicon.ico +0 -0
- /package/{public → dist/client}/logo192.png +0 -0
- /package/{public → dist/client}/logo512.png +0 -0
- /package/{public → dist/client}/manifest.json +0 -0
- /package/{public → dist/client}/robots.txt +0 -0
package/src/lib/media-fetch.ts
DELETED
|
@@ -1,860 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Respectful media caching for tweet records already present in birdclaw.
|
|
3
|
-
*
|
|
4
|
-
* This is not a scraper: it never crawls, enumerates, or derives Twitter/X CDN
|
|
5
|
-
* URLs. It only downloads media URLs already stored in `tweets.media_json`,
|
|
6
|
-
* skips files present on disk, paces requests, and backs off on 429.
|
|
7
|
-
*/
|
|
8
|
-
import {
|
|
9
|
-
existsSync,
|
|
10
|
-
mkdirSync,
|
|
11
|
-
readFileSync,
|
|
12
|
-
readdirSync,
|
|
13
|
-
statSync,
|
|
14
|
-
} from "node:fs";
|
|
15
|
-
import { appendFile, copyFile, rename, rm, writeFile } from "node:fs/promises";
|
|
16
|
-
import path from "node:path";
|
|
17
|
-
import { Effect } from "effect";
|
|
18
|
-
import { getBirdclawPaths } from "./config";
|
|
19
|
-
import { getNativeDb } from "./db";
|
|
20
|
-
import { runEffectPromise } from "./effect-runtime";
|
|
21
|
-
|
|
22
|
-
type FetchLike = (input: string, init?: RequestInit) => Promise<Response>;
|
|
23
|
-
type Row = { id: string; media_json: string };
|
|
24
|
-
type MediaKind = "image" | "video" | "gif";
|
|
25
|
-
type Candidate = {
|
|
26
|
-
kind: MediaKind;
|
|
27
|
-
mediaKey: string;
|
|
28
|
-
tweetId: string;
|
|
29
|
-
url: string;
|
|
30
|
-
path: string;
|
|
31
|
-
tmpPath: string;
|
|
32
|
-
archivePath?: string;
|
|
33
|
-
};
|
|
34
|
-
type FetchOneResult = {
|
|
35
|
-
fetched: number;
|
|
36
|
-
bytes: number;
|
|
37
|
-
rateLimited: boolean;
|
|
38
|
-
kind?: MediaKind;
|
|
39
|
-
reusedFromArchive?: boolean;
|
|
40
|
-
failure?: MediaFetchResult["failures"][number];
|
|
41
|
-
};
|
|
42
|
-
|
|
43
|
-
export type MediaFetchResult = {
|
|
44
|
-
ok: true;
|
|
45
|
-
fetched: number;
|
|
46
|
-
images_fetched: number;
|
|
47
|
-
videos_fetched: number;
|
|
48
|
-
gifs_fetched: number;
|
|
49
|
-
reused_from_archive: number;
|
|
50
|
-
skipped_cached: number;
|
|
51
|
-
failed: number;
|
|
52
|
-
rate_limited: number;
|
|
53
|
-
bytes: number;
|
|
54
|
-
image_bytes: number;
|
|
55
|
-
video_bytes: number;
|
|
56
|
-
gif_bytes: number;
|
|
57
|
-
duration_ms: number;
|
|
58
|
-
failures: Array<{ media_key: string; url: string; reason: string }>;
|
|
59
|
-
dry_run?: true;
|
|
60
|
-
would_fetch?: Array<{
|
|
61
|
-
media_key: string;
|
|
62
|
-
tweet_id: string;
|
|
63
|
-
kind: MediaKind;
|
|
64
|
-
url: string;
|
|
65
|
-
path: string;
|
|
66
|
-
}>;
|
|
67
|
-
};
|
|
68
|
-
|
|
69
|
-
export type MediaFetchOptions = {
|
|
70
|
-
account?: string;
|
|
71
|
-
limit?: number;
|
|
72
|
-
kind?: string;
|
|
73
|
-
since?: string;
|
|
74
|
-
parallel?: number;
|
|
75
|
-
pacingMs?: number;
|
|
76
|
-
videoPacingMs?: number;
|
|
77
|
-
retryMax?: number;
|
|
78
|
-
dryRun?: boolean;
|
|
79
|
-
includeVideo?: boolean;
|
|
80
|
-
maxBytes?: number;
|
|
81
|
-
fetchImpl?: FetchLike;
|
|
82
|
-
sleep?: (ms: number) => Promise<void>;
|
|
83
|
-
now?: () => number;
|
|
84
|
-
userAgent?: string;
|
|
85
|
-
};
|
|
86
|
-
|
|
87
|
-
const DEFAULT_MAX_BYTES = 100 * 1024 * 1024;
|
|
88
|
-
const PBS_PREFIXES = [
|
|
89
|
-
"/media/",
|
|
90
|
-
"/ext_tw_video_thumb/",
|
|
91
|
-
"/amplify_video_thumb/",
|
|
92
|
-
"/tweet_video_thumb/",
|
|
93
|
-
"/profile_images/",
|
|
94
|
-
] as const;
|
|
95
|
-
const packageVersion = (
|
|
96
|
-
JSON.parse(
|
|
97
|
-
readFileSync(new URL("../../package.json", import.meta.url), "utf8"),
|
|
98
|
-
) as { version?: string }
|
|
99
|
-
).version;
|
|
100
|
-
|
|
101
|
-
function defaultSleep(ms: number) {
|
|
102
|
-
return new Promise<void>((resolve) => setTimeout(resolve, ms));
|
|
103
|
-
}
|
|
104
|
-
|
|
105
|
-
function toMediaError(error: unknown) {
|
|
106
|
-
return error instanceof Error ? error : new Error(String(error));
|
|
107
|
-
}
|
|
108
|
-
|
|
109
|
-
function tryMediaSync<T>(try_: () => T) {
|
|
110
|
-
return Effect.try({
|
|
111
|
-
try: try_,
|
|
112
|
-
catch: toMediaError,
|
|
113
|
-
});
|
|
114
|
-
}
|
|
115
|
-
|
|
116
|
-
function tryMediaPromise<T>(try_: () => Promise<T>) {
|
|
117
|
-
return Effect.tryPromise({
|
|
118
|
-
try: try_,
|
|
119
|
-
catch: toMediaError,
|
|
120
|
-
});
|
|
121
|
-
}
|
|
122
|
-
|
|
123
|
-
function fileSize(filePath: string) {
|
|
124
|
-
try {
|
|
125
|
-
return statSync(filePath).size;
|
|
126
|
-
} catch {
|
|
127
|
-
return 0;
|
|
128
|
-
}
|
|
129
|
-
}
|
|
130
|
-
|
|
131
|
-
function basenameKey(url: URL) {
|
|
132
|
-
return path.posix.parse(path.posix.basename(url.pathname)).name;
|
|
133
|
-
}
|
|
134
|
-
|
|
135
|
-
function imageExtension(url: URL) {
|
|
136
|
-
const ext = path.posix.extname(url.pathname).toLowerCase();
|
|
137
|
-
if (ext === ".jpeg" || ext === ".jpg") return ".jpg";
|
|
138
|
-
if (ext === ".png" || ext === ".webp" || ext === ".gif" || ext === ".svg")
|
|
139
|
-
return ext;
|
|
140
|
-
const format = url.searchParams.get("format")?.toLowerCase();
|
|
141
|
-
return format === "png" || format === "webp" || format === "gif"
|
|
142
|
-
? `.${format}`
|
|
143
|
-
: ".jpg";
|
|
144
|
-
}
|
|
145
|
-
|
|
146
|
-
function imageCandidate(
|
|
147
|
-
urlValue: string,
|
|
148
|
-
dir: string,
|
|
149
|
-
tweetId: string,
|
|
150
|
-
): Candidate | null {
|
|
151
|
-
let url: URL;
|
|
152
|
-
try {
|
|
153
|
-
url = new URL(urlValue);
|
|
154
|
-
} catch {
|
|
155
|
-
return null;
|
|
156
|
-
}
|
|
157
|
-
if (
|
|
158
|
-
url.protocol !== "https:" ||
|
|
159
|
-
url.hostname !== "pbs.twimg.com" ||
|
|
160
|
-
!PBS_PREFIXES.some((prefix) => url.pathname.startsWith(prefix))
|
|
161
|
-
) {
|
|
162
|
-
return null;
|
|
163
|
-
}
|
|
164
|
-
const mediaKey = basenameKey(url);
|
|
165
|
-
const ext = imageExtension(url);
|
|
166
|
-
return {
|
|
167
|
-
kind: "image",
|
|
168
|
-
mediaKey,
|
|
169
|
-
tweetId,
|
|
170
|
-
url: url.toString(),
|
|
171
|
-
path: path.join(dir, `${mediaKey}${ext}`),
|
|
172
|
-
tmpPath: path.join(dir, `${mediaKey}${ext}.tmp`),
|
|
173
|
-
};
|
|
174
|
-
}
|
|
175
|
-
|
|
176
|
-
function record(value: unknown) {
|
|
177
|
-
return value && typeof value === "object" && !Array.isArray(value)
|
|
178
|
-
? (value as Record<string, unknown>)
|
|
179
|
-
: null;
|
|
180
|
-
}
|
|
181
|
-
|
|
182
|
-
function variantUrl(value: unknown) {
|
|
183
|
-
const item = record(value);
|
|
184
|
-
if (!item) return null;
|
|
185
|
-
const contentType = String(item.content_type ?? item.contentType ?? "");
|
|
186
|
-
if (contentType !== "video/mp4" || typeof item.url !== "string") return null;
|
|
187
|
-
const bitrate = item.bitRate ?? item.bit_rate ?? item.bitrate;
|
|
188
|
-
return {
|
|
189
|
-
url: item.url,
|
|
190
|
-
bitrate: Number.isFinite(Number(bitrate)) ? Number(bitrate) : 0,
|
|
191
|
-
};
|
|
192
|
-
}
|
|
193
|
-
|
|
194
|
-
function videoCandidate(
|
|
195
|
-
item: Record<string, unknown>,
|
|
196
|
-
dir: string,
|
|
197
|
-
tweetId: string,
|
|
198
|
-
): Candidate | null {
|
|
199
|
-
const rawType = String(item.type ?? "");
|
|
200
|
-
const kind: MediaKind | null =
|
|
201
|
-
rawType === "video"
|
|
202
|
-
? "video"
|
|
203
|
-
: rawType === "animated_gif" || rawType === "gif"
|
|
204
|
-
? "gif"
|
|
205
|
-
: null;
|
|
206
|
-
if (!kind) return null;
|
|
207
|
-
const variants = Array.isArray(item.variants)
|
|
208
|
-
? item.variants
|
|
209
|
-
: Array.isArray(record(item.video_info)?.variants)
|
|
210
|
-
? (record(item.video_info)?.variants as unknown[])
|
|
211
|
-
: [];
|
|
212
|
-
const best = variants
|
|
213
|
-
.map(variantUrl)
|
|
214
|
-
.filter(
|
|
215
|
-
(variant): variant is { url: string; bitrate: number } =>
|
|
216
|
-
variant !== null,
|
|
217
|
-
)
|
|
218
|
-
.sort((left, right) => right.bitrate - left.bitrate)[0];
|
|
219
|
-
if (!best) return null;
|
|
220
|
-
|
|
221
|
-
let url: URL;
|
|
222
|
-
try {
|
|
223
|
-
url = new URL(best.url);
|
|
224
|
-
} catch {
|
|
225
|
-
return null;
|
|
226
|
-
}
|
|
227
|
-
if (url.protocol !== "https:" || url.hostname !== "video.twimg.com")
|
|
228
|
-
return null;
|
|
229
|
-
const mediaKey = basenameKey(url);
|
|
230
|
-
return {
|
|
231
|
-
kind,
|
|
232
|
-
mediaKey,
|
|
233
|
-
tweetId,
|
|
234
|
-
url: url.toString(),
|
|
235
|
-
path: path.join(dir, `${mediaKey}.mp4`),
|
|
236
|
-
tmpPath: path.join(dir, `${mediaKey}.mp4.tmp`),
|
|
237
|
-
};
|
|
238
|
-
}
|
|
239
|
-
|
|
240
|
-
function archiveTweetDirs(dir: string, tweetId: string) {
|
|
241
|
-
const archiveRoot = path.join(dir, "archive");
|
|
242
|
-
try {
|
|
243
|
-
return readdirSync(archiveRoot, { withFileTypes: true })
|
|
244
|
-
.filter((entry) => entry.isDirectory())
|
|
245
|
-
.map((entry) => path.join(archiveRoot, entry.name, tweetId))
|
|
246
|
-
.filter((tweetDir) => existsSync(tweetDir));
|
|
247
|
-
} catch {
|
|
248
|
-
return [];
|
|
249
|
-
}
|
|
250
|
-
}
|
|
251
|
-
|
|
252
|
-
function archiveVideoCandidates(
|
|
253
|
-
item: Record<string, unknown>,
|
|
254
|
-
dir: string,
|
|
255
|
-
tweetId: string,
|
|
256
|
-
) {
|
|
257
|
-
const rawType = String(item.type ?? "");
|
|
258
|
-
const kind: MediaKind | null =
|
|
259
|
-
rawType === "video"
|
|
260
|
-
? "video"
|
|
261
|
-
: rawType === "animated_gif" || rawType === "gif"
|
|
262
|
-
? "gif"
|
|
263
|
-
: null;
|
|
264
|
-
if (!kind) return [];
|
|
265
|
-
|
|
266
|
-
const candidates: Candidate[] = [];
|
|
267
|
-
for (const tweetDir of archiveTweetDirs(dir, tweetId)) {
|
|
268
|
-
let entries;
|
|
269
|
-
try {
|
|
270
|
-
entries = readdirSync(tweetDir, { withFileTypes: true });
|
|
271
|
-
} catch {
|
|
272
|
-
continue;
|
|
273
|
-
}
|
|
274
|
-
for (const entry of entries) {
|
|
275
|
-
const ext = path.extname(entry.name).toLowerCase();
|
|
276
|
-
if (!entry.isFile() || ext !== ".mp4") continue;
|
|
277
|
-
const archivePath = path.join(tweetDir, entry.name);
|
|
278
|
-
const prefix = `${tweetId}-`;
|
|
279
|
-
const rawKey = path.basename(entry.name, ext);
|
|
280
|
-
const mediaKey = rawKey.startsWith(prefix)
|
|
281
|
-
? rawKey.slice(prefix.length)
|
|
282
|
-
: rawKey;
|
|
283
|
-
if (!mediaKey) continue;
|
|
284
|
-
candidates.push({
|
|
285
|
-
kind,
|
|
286
|
-
mediaKey,
|
|
287
|
-
tweetId,
|
|
288
|
-
url: `archive:${archivePath}`,
|
|
289
|
-
path: path.join(dir, `${mediaKey}${ext}`),
|
|
290
|
-
tmpPath: path.join(dir, `${mediaKey}${ext}.tmp`),
|
|
291
|
-
archivePath,
|
|
292
|
-
});
|
|
293
|
-
}
|
|
294
|
-
}
|
|
295
|
-
return candidates;
|
|
296
|
-
}
|
|
297
|
-
|
|
298
|
-
function rowCandidates(row: Row, dir: string, includeVideo: boolean) {
|
|
299
|
-
let items: unknown;
|
|
300
|
-
try {
|
|
301
|
-
items = JSON.parse(row.media_json);
|
|
302
|
-
} catch {
|
|
303
|
-
return [];
|
|
304
|
-
}
|
|
305
|
-
if (!Array.isArray(items)) return [];
|
|
306
|
-
|
|
307
|
-
const candidates: Candidate[] = [];
|
|
308
|
-
for (const value of items) {
|
|
309
|
-
const item = record(value);
|
|
310
|
-
if (!item) continue;
|
|
311
|
-
if (typeof item.url === "string") {
|
|
312
|
-
const image = imageCandidate(item.url, dir, row.id);
|
|
313
|
-
if (image) candidates.push(image);
|
|
314
|
-
}
|
|
315
|
-
if (includeVideo) {
|
|
316
|
-
const archiveVideos = archiveVideoCandidates(item, dir, row.id);
|
|
317
|
-
if (archiveVideos.length > 0) {
|
|
318
|
-
candidates.push(...archiveVideos);
|
|
319
|
-
continue;
|
|
320
|
-
}
|
|
321
|
-
const video = videoCandidate(item, dir, row.id);
|
|
322
|
-
if (video) candidates.push(video);
|
|
323
|
-
}
|
|
324
|
-
}
|
|
325
|
-
return candidates;
|
|
326
|
-
}
|
|
327
|
-
|
|
328
|
-
function queryRows(options: MediaFetchOptions) {
|
|
329
|
-
const params: Array<string | number> = [];
|
|
330
|
-
const account =
|
|
331
|
-
options.account && options.account !== "all" ? options.account : undefined;
|
|
332
|
-
const kind = normalizeKind(options.kind);
|
|
333
|
-
let sql = `
|
|
334
|
-
select t.id, t.media_json
|
|
335
|
-
from tweets t
|
|
336
|
-
where t.media_json not in ('', '[]', 'null')
|
|
337
|
-
`;
|
|
338
|
-
const scopeClause = buildScopeClause(params, account, kind);
|
|
339
|
-
if (scopeClause) sql += ` and (${scopeClause})`;
|
|
340
|
-
if (options.since) {
|
|
341
|
-
params.push(options.since);
|
|
342
|
-
sql += " and t.created_at >= ?";
|
|
343
|
-
}
|
|
344
|
-
sql += " order by t.created_at desc, t.id desc";
|
|
345
|
-
if (options.limit !== undefined) {
|
|
346
|
-
params.push(Math.max(0, Math.floor(options.limit)));
|
|
347
|
-
sql += " limit ?";
|
|
348
|
-
}
|
|
349
|
-
return getNativeDb().prepare(sql).all(params) as Row[];
|
|
350
|
-
}
|
|
351
|
-
|
|
352
|
-
function normalizeKind(kind?: string) {
|
|
353
|
-
const value = kind?.trim().toLowerCase();
|
|
354
|
-
if (!value || value === "all") return undefined;
|
|
355
|
-
if (value === "likes") return "like";
|
|
356
|
-
if (value === "bookmarks") return "bookmark";
|
|
357
|
-
return value;
|
|
358
|
-
}
|
|
359
|
-
|
|
360
|
-
function collectionKind(kind?: string) {
|
|
361
|
-
if (kind === "like") return "likes";
|
|
362
|
-
if (kind === "bookmark") return "bookmarks";
|
|
363
|
-
return undefined;
|
|
364
|
-
}
|
|
365
|
-
|
|
366
|
-
function buildScopeClause(
|
|
367
|
-
params: Array<string | number>,
|
|
368
|
-
account?: string,
|
|
369
|
-
kind?: string,
|
|
370
|
-
) {
|
|
371
|
-
const clauses: string[] = [];
|
|
372
|
-
const accountClause = (alias: string) =>
|
|
373
|
-
account ? ` and ${alias}.account_id = ?` : "";
|
|
374
|
-
const pushAccount = () => {
|
|
375
|
-
if (account) params.push(account);
|
|
376
|
-
};
|
|
377
|
-
if (kind) {
|
|
378
|
-
params.push(kind);
|
|
379
|
-
pushAccount();
|
|
380
|
-
clauses.push(
|
|
381
|
-
`exists (select 1 from tweet_account_edges edge where edge.tweet_id = t.id and edge.kind = ?${accountClause("edge")})`,
|
|
382
|
-
);
|
|
383
|
-
const savedKind = collectionKind(kind);
|
|
384
|
-
if (savedKind) {
|
|
385
|
-
params.push(savedKind);
|
|
386
|
-
pushAccount();
|
|
387
|
-
clauses.push(
|
|
388
|
-
`exists (select 1 from tweet_collections collection where collection.tweet_id = t.id and collection.kind = ?${accountClause("collection")})`,
|
|
389
|
-
);
|
|
390
|
-
const legacyColumn = savedKind === "likes" ? "liked" : "bookmarked";
|
|
391
|
-
pushAccount();
|
|
392
|
-
clauses.push(
|
|
393
|
-
`t.${legacyColumn} = 1${account ? " and t.account_id = ?" : ""}`,
|
|
394
|
-
);
|
|
395
|
-
}
|
|
396
|
-
params.push(kind);
|
|
397
|
-
pushAccount();
|
|
398
|
-
clauses.push(`t.kind = ?${account ? " and t.account_id = ?" : ""}`);
|
|
399
|
-
return clauses.join(" or ");
|
|
400
|
-
}
|
|
401
|
-
if (account) {
|
|
402
|
-
params.push(account, account, account);
|
|
403
|
-
clauses.push(
|
|
404
|
-
"exists (select 1 from tweet_account_edges edge where edge.tweet_id = t.id and edge.account_id = ?)",
|
|
405
|
-
);
|
|
406
|
-
clauses.push(
|
|
407
|
-
"exists (select 1 from tweet_collections collection where collection.tweet_id = t.id and collection.account_id = ?)",
|
|
408
|
-
);
|
|
409
|
-
clauses.push("t.account_id = ?");
|
|
410
|
-
}
|
|
411
|
-
return clauses.join(" or ");
|
|
412
|
-
}
|
|
413
|
-
|
|
414
|
-
function collect(options: MediaFetchOptions, dir: string) {
|
|
415
|
-
const seen = new Set<string>();
|
|
416
|
-
const candidates: Candidate[] = [];
|
|
417
|
-
const would_fetch: NonNullable<MediaFetchResult["would_fetch"]> = [];
|
|
418
|
-
let skipped_cached = 0;
|
|
419
|
-
|
|
420
|
-
for (const row of queryRows(options)) {
|
|
421
|
-
for (const item of rowCandidates(row, dir, options.includeVideo ?? true)) {
|
|
422
|
-
const identity = `${item.kind}:${item.mediaKey}`;
|
|
423
|
-
if (seen.has(identity)) continue;
|
|
424
|
-
seen.add(identity);
|
|
425
|
-
if (existsSync(item.path)) {
|
|
426
|
-
skipped_cached += 1;
|
|
427
|
-
} else if (options.dryRun) {
|
|
428
|
-
would_fetch.push({
|
|
429
|
-
media_key: item.mediaKey,
|
|
430
|
-
tweet_id: item.tweetId,
|
|
431
|
-
kind: item.kind,
|
|
432
|
-
url: item.url,
|
|
433
|
-
path: item.path,
|
|
434
|
-
});
|
|
435
|
-
} else {
|
|
436
|
-
candidates.push(item);
|
|
437
|
-
}
|
|
438
|
-
}
|
|
439
|
-
}
|
|
440
|
-
return { candidates, skipped_cached, would_fetch };
|
|
441
|
-
}
|
|
442
|
-
|
|
443
|
-
function fail(
|
|
444
|
-
item: Candidate,
|
|
445
|
-
reason: string,
|
|
446
|
-
rateLimited = false,
|
|
447
|
-
): FetchOneResult {
|
|
448
|
-
return {
|
|
449
|
-
fetched: 0,
|
|
450
|
-
bytes: 0,
|
|
451
|
-
rateLimited,
|
|
452
|
-
failure: { media_key: item.mediaKey, url: item.url, reason },
|
|
453
|
-
};
|
|
454
|
-
}
|
|
455
|
-
|
|
456
|
-
/**
|
|
457
|
-
* Archive reuse consumes files extracted by the sibling
|
|
458
|
-
* `feat/import-archive-followers-following` branch into
|
|
459
|
-
* `media/originals/archive/<kind>/<tweet_id>/...`.
|
|
460
|
-
*
|
|
461
|
-
* This fetcher intentionally does not extract archive ZIP media itself; it only
|
|
462
|
-
* reuses that layout when it is already present.
|
|
463
|
-
*/
|
|
464
|
-
function archivePathFor(item: Candidate, mediaOriginalsDir: string) {
|
|
465
|
-
if (item.archivePath) return item.archivePath;
|
|
466
|
-
if (!item.tweetId || !item.mediaKey) return null;
|
|
467
|
-
const ext = path.extname(item.path);
|
|
468
|
-
if (!ext) return null;
|
|
469
|
-
const fileName = `${item.tweetId}-${item.mediaKey}${ext}`;
|
|
470
|
-
return (
|
|
471
|
-
archiveTweetDirs(mediaOriginalsDir, item.tweetId)
|
|
472
|
-
.map((tweetDir) => path.join(tweetDir, fileName))
|
|
473
|
-
.find((archivePath) => existsSync(archivePath)) ?? null
|
|
474
|
-
);
|
|
475
|
-
}
|
|
476
|
-
|
|
477
|
-
function reuseFromArchiveEffect(
|
|
478
|
-
item: Candidate,
|
|
479
|
-
mediaOriginalsDir: string,
|
|
480
|
-
maxBytes: number,
|
|
481
|
-
): Effect.Effect<FetchOneResult | null, Error> {
|
|
482
|
-
return Effect.gen(function* () {
|
|
483
|
-
const archivePath = archivePathFor(item, mediaOriginalsDir);
|
|
484
|
-
if (!archivePath || !existsSync(archivePath)) return null;
|
|
485
|
-
const bytes = fileSize(archivePath);
|
|
486
|
-
if (bytes > maxBytes) return fail(item, "max-bytes");
|
|
487
|
-
yield* tryMediaPromise(() => copyFile(archivePath, item.tmpPath));
|
|
488
|
-
yield* tryMediaPromise(() => rename(item.tmpPath, item.path));
|
|
489
|
-
return {
|
|
490
|
-
fetched: 1,
|
|
491
|
-
bytes,
|
|
492
|
-
rateLimited: false,
|
|
493
|
-
kind: item.kind,
|
|
494
|
-
reusedFromArchive: true,
|
|
495
|
-
} satisfies FetchOneResult;
|
|
496
|
-
});
|
|
497
|
-
}
|
|
498
|
-
|
|
499
|
-
function contentLength(response: Response) {
|
|
500
|
-
const value = Number(response.headers.get("content-length"));
|
|
501
|
-
return Number.isFinite(value) && value >= 0 ? value : null;
|
|
502
|
-
}
|
|
503
|
-
|
|
504
|
-
function contentRangeTotal(response: Response) {
|
|
505
|
-
const total = /\/(\d+)\s*$/.exec(
|
|
506
|
-
response.headers.get("content-range") ?? "",
|
|
507
|
-
)?.[1];
|
|
508
|
-
return total ? Number(total) : null;
|
|
509
|
-
}
|
|
510
|
-
|
|
511
|
-
function writeResponseBodyEffect(
|
|
512
|
-
response: Response,
|
|
513
|
-
tmpPath: string,
|
|
514
|
-
append: boolean,
|
|
515
|
-
maxBytes: number,
|
|
516
|
-
initialBytes: number,
|
|
517
|
-
): Effect.Effect<number, Error> {
|
|
518
|
-
return Effect.gen(function* () {
|
|
519
|
-
if (!response.body)
|
|
520
|
-
return yield* Effect.fail(new Error("missing response body"));
|
|
521
|
-
let bytes = 0;
|
|
522
|
-
if (!append) {
|
|
523
|
-
yield* tryMediaPromise(() => writeFile(tmpPath, Buffer.alloc(0)));
|
|
524
|
-
}
|
|
525
|
-
const reader = response.body.getReader();
|
|
526
|
-
const result = yield* Effect.gen(function* () {
|
|
527
|
-
for (;;) {
|
|
528
|
-
const { done, value } = yield* tryMediaPromise(() => reader.read());
|
|
529
|
-
if (done) {
|
|
530
|
-
break;
|
|
531
|
-
}
|
|
532
|
-
const chunk = Buffer.from(value);
|
|
533
|
-
bytes += chunk.length;
|
|
534
|
-
if (initialBytes + bytes > maxBytes) {
|
|
535
|
-
yield* tryMediaPromise(() => rm(tmpPath, { force: true })).pipe(
|
|
536
|
-
Effect.catchAll(() => Effect.void),
|
|
537
|
-
);
|
|
538
|
-
return yield* Effect.fail(new Error("max-bytes"));
|
|
539
|
-
}
|
|
540
|
-
yield* tryMediaPromise(() => appendFile(tmpPath, chunk));
|
|
541
|
-
}
|
|
542
|
-
return bytes;
|
|
543
|
-
}).pipe(
|
|
544
|
-
Effect.map((writtenBytes) => ({ ok: true as const, writtenBytes })),
|
|
545
|
-
Effect.catchAll((error) =>
|
|
546
|
-
Effect.gen(function* () {
|
|
547
|
-
if (error.message === "max-bytes") {
|
|
548
|
-
yield* tryMediaPromise(() => rm(tmpPath, { force: true })).pipe(
|
|
549
|
-
Effect.catchAll(() => Effect.void),
|
|
550
|
-
);
|
|
551
|
-
}
|
|
552
|
-
yield* tryMediaPromise(() => reader.cancel(error)).pipe(
|
|
553
|
-
Effect.catchAll(() => Effect.void),
|
|
554
|
-
);
|
|
555
|
-
return { error, ok: false as const };
|
|
556
|
-
}),
|
|
557
|
-
),
|
|
558
|
-
Effect.ensuring(Effect.sync(() => reader.releaseLock())),
|
|
559
|
-
);
|
|
560
|
-
if (!result.ok) return yield* Effect.fail(result.error);
|
|
561
|
-
return result.writtenBytes;
|
|
562
|
-
});
|
|
563
|
-
}
|
|
564
|
-
|
|
565
|
-
function fetchOneEffect({
|
|
566
|
-
item,
|
|
567
|
-
fetchImpl,
|
|
568
|
-
sleep,
|
|
569
|
-
retryMax,
|
|
570
|
-
userAgent,
|
|
571
|
-
maxBytes,
|
|
572
|
-
}: {
|
|
573
|
-
item: Candidate;
|
|
574
|
-
fetchImpl: FetchLike;
|
|
575
|
-
sleep: (ms: number) => Promise<void>;
|
|
576
|
-
retryMax: number;
|
|
577
|
-
userAgent: string;
|
|
578
|
-
maxBytes: number;
|
|
579
|
-
}): Effect.Effect<FetchOneResult, Error> {
|
|
580
|
-
return Effect.gen(function* () {
|
|
581
|
-
let rateLimited = false;
|
|
582
|
-
for (let attempt = 0; attempt <= retryMax; attempt += 1) {
|
|
583
|
-
const partialBytes = item.kind === "image" ? 0 : fileSize(item.tmpPath);
|
|
584
|
-
if (partialBytes > maxBytes) {
|
|
585
|
-
yield* tryMediaPromise(() => rm(item.tmpPath, { force: true })).pipe(
|
|
586
|
-
Effect.catchAll(() => Effect.void),
|
|
587
|
-
);
|
|
588
|
-
return fail(item, "max-bytes");
|
|
589
|
-
}
|
|
590
|
-
const responseResult = yield* tryMediaPromise(() =>
|
|
591
|
-
fetchImpl(item.url, {
|
|
592
|
-
headers: {
|
|
593
|
-
"user-agent": userAgent,
|
|
594
|
-
...(partialBytes > 0 ? { range: `bytes=${partialBytes}-` } : {}),
|
|
595
|
-
},
|
|
596
|
-
}),
|
|
597
|
-
).pipe(
|
|
598
|
-
Effect.map((response) => ({ ok: true as const, response })),
|
|
599
|
-
Effect.catchAll((error) =>
|
|
600
|
-
Effect.succeed({ error, ok: false as const }),
|
|
601
|
-
),
|
|
602
|
-
);
|
|
603
|
-
if (!responseResult.ok) {
|
|
604
|
-
return fail(item, responseResult.error.message, rateLimited);
|
|
605
|
-
}
|
|
606
|
-
const { response } = responseResult;
|
|
607
|
-
if (response.status === 429) {
|
|
608
|
-
rateLimited = true;
|
|
609
|
-
if (attempt < retryMax) {
|
|
610
|
-
yield* tryMediaPromise(() => sleep(1000 * 2 ** attempt)).pipe(
|
|
611
|
-
Effect.catchAll(() => Effect.void),
|
|
612
|
-
);
|
|
613
|
-
continue;
|
|
614
|
-
}
|
|
615
|
-
return fail(item, "429", true);
|
|
616
|
-
}
|
|
617
|
-
if (!response.ok && response.status !== 206) {
|
|
618
|
-
return fail(item, String(response.status), rateLimited);
|
|
619
|
-
}
|
|
620
|
-
|
|
621
|
-
const expectedTotal =
|
|
622
|
-
contentRangeTotal(response) ??
|
|
623
|
-
(contentLength(response) ?? 0) +
|
|
624
|
-
(response.status === 206 ? partialBytes : 0);
|
|
625
|
-
if (expectedTotal > maxBytes) {
|
|
626
|
-
yield* tryMediaPromise(() => rm(item.tmpPath, { force: true })).pipe(
|
|
627
|
-
Effect.catchAll(() => Effect.void),
|
|
628
|
-
);
|
|
629
|
-
return fail(item, "max-bytes");
|
|
630
|
-
}
|
|
631
|
-
|
|
632
|
-
const append = partialBytes > 0 && response.status === 206;
|
|
633
|
-
const bytesResult = yield* writeResponseBodyEffect(
|
|
634
|
-
response,
|
|
635
|
-
item.tmpPath,
|
|
636
|
-
append,
|
|
637
|
-
maxBytes,
|
|
638
|
-
append ? partialBytes : 0,
|
|
639
|
-
).pipe(
|
|
640
|
-
Effect.map((bytes) => ({ bytes, ok: true as const })),
|
|
641
|
-
Effect.catchAll((error) =>
|
|
642
|
-
Effect.succeed({ error, ok: false as const }),
|
|
643
|
-
),
|
|
644
|
-
);
|
|
645
|
-
if (!bytesResult.ok) {
|
|
646
|
-
return fail(item, bytesResult.error.message, rateLimited);
|
|
647
|
-
}
|
|
648
|
-
yield* tryMediaPromise(() => rename(item.tmpPath, item.path));
|
|
649
|
-
return {
|
|
650
|
-
fetched: 1,
|
|
651
|
-
bytes: bytesResult.bytes,
|
|
652
|
-
rateLimited,
|
|
653
|
-
kind: item.kind,
|
|
654
|
-
};
|
|
655
|
-
}
|
|
656
|
-
return fail(item, "retry exhausted", rateLimited);
|
|
657
|
-
});
|
|
658
|
-
}
|
|
659
|
-
|
|
660
|
-
function applyFetched(result: MediaFetchResult, fetched: FetchOneResult) {
|
|
661
|
-
result.fetched += fetched.fetched;
|
|
662
|
-
result.bytes += fetched.bytes;
|
|
663
|
-
if (fetched.kind === "image") {
|
|
664
|
-
result.images_fetched += 1;
|
|
665
|
-
result.image_bytes += fetched.bytes;
|
|
666
|
-
}
|
|
667
|
-
if (fetched.kind === "video") {
|
|
668
|
-
result.videos_fetched += 1;
|
|
669
|
-
result.video_bytes += fetched.bytes;
|
|
670
|
-
}
|
|
671
|
-
if (fetched.kind === "gif") {
|
|
672
|
-
result.gifs_fetched += 1;
|
|
673
|
-
result.gif_bytes += fetched.bytes;
|
|
674
|
-
}
|
|
675
|
-
if (fetched.reusedFromArchive) result.reused_from_archive += 1;
|
|
676
|
-
if (fetched.rateLimited) result.rate_limited += 1;
|
|
677
|
-
if (fetched.failure) result.failures.push(fetched.failure);
|
|
678
|
-
}
|
|
679
|
-
|
|
680
|
-
function runGroupEffect(
|
|
681
|
-
items: Candidate[],
|
|
682
|
-
parallel: number,
|
|
683
|
-
pacingMs: number,
|
|
684
|
-
now: () => number,
|
|
685
|
-
sleep: (ms: number) => Promise<void>,
|
|
686
|
-
worker: (item: Candidate) => Promise<void>,
|
|
687
|
-
) {
|
|
688
|
-
let lastStart: number | null = null;
|
|
689
|
-
let pace = Promise.resolve();
|
|
690
|
-
const pacedItems = items.map((item) => {
|
|
691
|
-
const previous = pace;
|
|
692
|
-
let release = () => {};
|
|
693
|
-
pace = new Promise<void>((resolve) => {
|
|
694
|
-
release = resolve;
|
|
695
|
-
});
|
|
696
|
-
return { item, previous, release };
|
|
697
|
-
});
|
|
698
|
-
const runPaced = ({
|
|
699
|
-
item,
|
|
700
|
-
previous,
|
|
701
|
-
release,
|
|
702
|
-
}: {
|
|
703
|
-
item: Candidate;
|
|
704
|
-
previous: Promise<void>;
|
|
705
|
-
release: () => void;
|
|
706
|
-
}) =>
|
|
707
|
-
tryMediaPromise(() =>
|
|
708
|
-
previous.then(() => {
|
|
709
|
-
const waitMs =
|
|
710
|
-
lastStart !== null ? Math.max(0, lastStart + pacingMs - now()) : 0;
|
|
711
|
-
const wait = waitMs > 0 ? sleep(waitMs) : Promise.resolve();
|
|
712
|
-
return wait.then(
|
|
713
|
-
() => {
|
|
714
|
-
let work: Promise<void>;
|
|
715
|
-
try {
|
|
716
|
-
lastStart = now();
|
|
717
|
-
work = worker(item);
|
|
718
|
-
} finally {
|
|
719
|
-
release();
|
|
720
|
-
}
|
|
721
|
-
return work;
|
|
722
|
-
},
|
|
723
|
-
(error: unknown) => {
|
|
724
|
-
release();
|
|
725
|
-
throw error;
|
|
726
|
-
},
|
|
727
|
-
);
|
|
728
|
-
}),
|
|
729
|
-
);
|
|
730
|
-
return Effect.forEach(pacedItems, runPaced, {
|
|
731
|
-
concurrency: Math.min(parallel, items.length),
|
|
732
|
-
discard: true,
|
|
733
|
-
});
|
|
734
|
-
}
|
|
735
|
-
|
|
736
|
-
export function fetchTweetMedia(options: MediaFetchOptions = {}) {
|
|
737
|
-
return runEffectPromise(fetchTweetMediaEffect(options));
|
|
738
|
-
}
|
|
739
|
-
|
|
740
|
-
export function fetchTweetMediaEffect(options: MediaFetchOptions = {}) {
|
|
741
|
-
return Effect.gen(function* () {
|
|
742
|
-
const now = options.now ?? Date.now;
|
|
743
|
-
const startedAt = now();
|
|
744
|
-
const sleep = options.sleep ?? defaultSleep;
|
|
745
|
-
const fetchImpl = options.fetchImpl ?? fetch;
|
|
746
|
-
const retryMax = Math.max(0, Math.floor(options.retryMax ?? 3));
|
|
747
|
-
const parallel = Math.min(
|
|
748
|
-
5,
|
|
749
|
-
Math.max(1, Math.floor(options.parallel ?? 1)),
|
|
750
|
-
);
|
|
751
|
-
const pacingMs = Math.max(0, Math.floor(options.pacingMs ?? 250));
|
|
752
|
-
const videoPacingMs = Math.max(
|
|
753
|
-
0,
|
|
754
|
-
Math.floor(options.videoPacingMs ?? pacingMs),
|
|
755
|
-
);
|
|
756
|
-
const maxBytes = Math.max(
|
|
757
|
-
0,
|
|
758
|
-
Math.floor(options.maxBytes ?? DEFAULT_MAX_BYTES),
|
|
759
|
-
);
|
|
760
|
-
const userAgent =
|
|
761
|
-
options.userAgent ??
|
|
762
|
-
`birdclaw/${packageVersion ?? "0.0.0"} (https://github.com/steipete/birdclaw)`;
|
|
763
|
-
const { mediaOriginalsDir } = getBirdclawPaths();
|
|
764
|
-
yield* tryMediaSync(() =>
|
|
765
|
-
mkdirSync(mediaOriginalsDir, { recursive: true }),
|
|
766
|
-
);
|
|
767
|
-
|
|
768
|
-
const { candidates, skipped_cached, would_fetch } = yield* tryMediaSync(
|
|
769
|
-
() => collect(options, mediaOriginalsDir),
|
|
770
|
-
);
|
|
771
|
-
const result: MediaFetchResult = {
|
|
772
|
-
ok: true,
|
|
773
|
-
fetched: 0,
|
|
774
|
-
images_fetched: 0,
|
|
775
|
-
videos_fetched: 0,
|
|
776
|
-
gifs_fetched: 0,
|
|
777
|
-
reused_from_archive: 0,
|
|
778
|
-
skipped_cached,
|
|
779
|
-
failed: 0,
|
|
780
|
-
rate_limited: 0,
|
|
781
|
-
bytes: 0,
|
|
782
|
-
image_bytes: 0,
|
|
783
|
-
video_bytes: 0,
|
|
784
|
-
gif_bytes: 0,
|
|
785
|
-
duration_ms: 0,
|
|
786
|
-
failures: [],
|
|
787
|
-
...(options.dryRun ? { dry_run: true as const, would_fetch } : {}),
|
|
788
|
-
};
|
|
789
|
-
|
|
790
|
-
if (!options.dryRun) {
|
|
791
|
-
const httpCandidates: Candidate[] = [];
|
|
792
|
-
for (const item of candidates) {
|
|
793
|
-
const reused = yield* reuseFromArchiveEffect(
|
|
794
|
-
item,
|
|
795
|
-
mediaOriginalsDir,
|
|
796
|
-
maxBytes,
|
|
797
|
-
);
|
|
798
|
-
if (reused) {
|
|
799
|
-
applyFetched(result, reused);
|
|
800
|
-
} else {
|
|
801
|
-
httpCandidates.push(item);
|
|
802
|
-
}
|
|
803
|
-
}
|
|
804
|
-
const fetchCandidate = (item: Candidate) =>
|
|
805
|
-
runEffectPromise(
|
|
806
|
-
fetchOneEffect({
|
|
807
|
-
item,
|
|
808
|
-
fetchImpl,
|
|
809
|
-
sleep,
|
|
810
|
-
retryMax,
|
|
811
|
-
userAgent,
|
|
812
|
-
maxBytes,
|
|
813
|
-
}).pipe(Effect.map((fetched) => applyFetched(result, fetched))),
|
|
814
|
-
);
|
|
815
|
-
yield* runGroupEffect(
|
|
816
|
-
httpCandidates.filter((item) => item.kind === "image"),
|
|
817
|
-
parallel,
|
|
818
|
-
pacingMs,
|
|
819
|
-
now,
|
|
820
|
-
sleep,
|
|
821
|
-
fetchCandidate,
|
|
822
|
-
);
|
|
823
|
-
yield* runGroupEffect(
|
|
824
|
-
httpCandidates.filter((item) => item.kind !== "image"),
|
|
825
|
-
1,
|
|
826
|
-
videoPacingMs,
|
|
827
|
-
now,
|
|
828
|
-
sleep,
|
|
829
|
-
fetchCandidate,
|
|
830
|
-
);
|
|
831
|
-
}
|
|
832
|
-
|
|
833
|
-
result.failed = result.failures.length;
|
|
834
|
-
result.duration_ms = Math.max(0, Math.round(now() - startedAt));
|
|
835
|
-
return result;
|
|
836
|
-
});
|
|
837
|
-
}
|
|
838
|
-
|
|
839
|
-
export function formatMediaFetchResult(result: MediaFetchResult) {
|
|
840
|
-
if (result.dry_run) {
|
|
841
|
-
return [
|
|
842
|
-
...(result.would_fetch ?? []).map(
|
|
843
|
-
(item) => `${item.kind}\t${item.media_key}\t${item.url}\t${item.path}`,
|
|
844
|
-
),
|
|
845
|
-
`would_fetch=${result.would_fetch?.length ?? 0} skipped_cached=${result.skipped_cached}`,
|
|
846
|
-
].join("\n");
|
|
847
|
-
}
|
|
848
|
-
return [
|
|
849
|
-
`fetched=${result.fetched}`,
|
|
850
|
-
`images=${result.images_fetched}`,
|
|
851
|
-
`videos=${result.videos_fetched}`,
|
|
852
|
-
`gifs=${result.gifs_fetched}`,
|
|
853
|
-
`reused_from_archive=${result.reused_from_archive}`,
|
|
854
|
-
`skipped_cached=${result.skipped_cached}`,
|
|
855
|
-
`failed=${result.failed}`,
|
|
856
|
-
`rate_limited=${result.rate_limited}`,
|
|
857
|
-
`bytes=${result.bytes}`,
|
|
858
|
-
`duration_ms=${result.duration_ms}`,
|
|
859
|
-
].join(" ");
|
|
860
|
-
}
|