1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
import fs from "fs";
import puppeteer, { TimeoutError } from "puppeteer";
import { wait } from "./utils";
import { getBrowser } from "./browser";
export interface TikTokTask {
type: "download_tiktok";
url: string;
respondWithFile: (filePath: string) => Promise<void>;
respondTaskFailed: (err: unknown) => Promise<void>;
}
export const runDownloadTiktokTask = async (task: TikTokTask) => {
const browser = await getBrowser();
const page = await browser.newPage();
const client = await page.createCDPSession();
await client.send("Page.setDownloadBehavior", {
behavior: "allow",
downloadPath: "./downloads",
});
try {
console.log("Starting task:", task.url);
const downloadDirFiles = fs.readdirSync("./downloads");
console.log("Emptying downloads folder");
downloadDirFiles.forEach((file) => {
fs.unlinkSync(`./downloads/${file}`);
});
await page.goto("https://snaptik.app/");
try {
const consent = await page.waitForSelector("button.fc-cta-consent", {
timeout: 3000,
});
if (consent) {
console.log("Looks like a cookie consent dialog, clicking agree");
await consent.click();
}
} catch (e) {
if (e instanceof TimeoutError) {
console.log("No cookie consent dialog found");
} else {
throw e;
}
}
await page.type("input#url", task.url);
await page.click("button[type=submit]");
console.log("Waiting for a bit");
await wait(1000);
console.log("Done waiting");
// try {
// const adModalClose = await page.waitForSelector("#adOverlay button", {
// timeout: 3000,
// });
// console.log("Looks like an ad modal, clicking close");
// await adModalClose?.click();
// } catch (e) {
// if (e instanceof TimeoutError) {
// console.log("No ad modal");
// } else {
// throw e;
// }
// }
// try {
// const secondAdModalClose = await page.waitForSelector("#dismiss-button", {
// timeout: 3000,
// });
// console.log("Looks like a second ad modal, clicking close");
// await secondAdModalClose?.click();
// } catch (e) {
// if (e instanceof TimeoutError) {
// console.log("No second ad modal");
// } else {
// throw e;
// }
// }
const downloadButton = await page.waitForSelector(
"a[data-event=server01_file]",
{ timeout: 10_000 }
);
if (!downloadButton) {
throw new Error("Could not find download button");
}
console.log("Found download button, pressing enter");
await downloadButton?.scrollIntoView();
await downloadButton?.focus();
await downloadButton?.press("Enter");
console.log("Waiting to see if one more ad pops up");
await wait(1000);
// Click outside shadow-dommed ad modal to close it and start the download
// console.log(
// "Clicking outside the possible ad modal to close it and start the download"
// );
// await page.mouse.click(10, 10);
console.log("Waiting for download to start");
await wait(3000);
let downloadedFiles = fs.readdirSync("./downloads");
while (downloadedFiles.some((file) => file.endsWith(".crdownload"))) {
console.log("Download still in progress, waiting");
await wait(1000);
downloadedFiles = fs.readdirSync("./downloads");
}
console.log("Downloaded files", downloadedFiles);
if (downloadedFiles.length === 0) {
throw new Error("No files downloaded");
} else if (downloadedFiles.length > 1) {
throw new Error("More than one file downloaded, not sure what to do");
}
const downloadedFile = downloadedFiles[0];
await task.respondWithFile(`./downloads/${downloadedFile}`);
} finally {
await page.close();
}
};
|
