93 lines
2.4 KiB
TypeScript
93 lines
2.4 KiB
TypeScript
import Axios from "axios"
|
|
import { _tiktokGetCollection } from "../../constants/api"
|
|
import { _getCollectionParams } from "../../constants/params"
|
|
import { HttpsProxyAgent } from "https-proxy-agent"
|
|
import { SocksProxyAgent } from "socks-proxy-agent"
|
|
import { TiktokCollectionResponse } from "../../types/get/getCollection"
|
|
import { ERROR_MESSAGES } from "../../constants"
|
|
import retry from "async-retry"
|
|
|
|
/** Types */
|
|
interface ProxyConfig {
|
|
httpsAgent?: HttpsProxyAgent<string> | SocksProxyAgent
|
|
}
|
|
|
|
const createProxyAgent = (proxy?: string): ProxyConfig => {
|
|
if (!proxy) return {}
|
|
|
|
if (proxy.startsWith("socks")) {
|
|
return {
|
|
httpsAgent: new SocksProxyAgent(proxy)
|
|
}
|
|
}
|
|
|
|
return {
|
|
httpsAgent: new HttpsProxyAgent(proxy)
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get TikTok Collection
|
|
* @param {string} collectionId - Collection ID
|
|
* @param {string} proxy - Your Proxy (optional)
|
|
* @param {string} page - Page for pagination (optional)
|
|
* @param {number} count - Number of items to fetch (optional)
|
|
* @returns {Promise<TiktokCollectionResponse>}
|
|
*/
|
|
export const getCollection = async (
|
|
collectionId: string,
|
|
proxy?: string,
|
|
page: number = 1,
|
|
count: number = 5
|
|
): Promise<TiktokCollectionResponse> => {
|
|
try {
|
|
const response = await retry(
|
|
async () => {
|
|
const res = await Axios(
|
|
_tiktokGetCollection(
|
|
_getCollectionParams(collectionId, page, count)
|
|
),
|
|
{
|
|
method: "GET",
|
|
headers: {
|
|
"User-Agent":
|
|
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36",
|
|
Accept: "*/*",
|
|
"Accept-Language": "en-US,en;q=0.7",
|
|
Referer: "https://www.tiktok.com/",
|
|
Origin: "https://www.tiktok.com"
|
|
},
|
|
...createProxyAgent(proxy)
|
|
}
|
|
)
|
|
|
|
if (res.data && res.data.statusCode === 0) {
|
|
return res.data
|
|
}
|
|
|
|
throw new Error(ERROR_MESSAGES.NETWORK_ERROR)
|
|
},
|
|
{
|
|
retries: 20,
|
|
minTimeout: 200,
|
|
maxTimeout: 1000
|
|
}
|
|
)
|
|
|
|
return {
|
|
status: "success",
|
|
result: {
|
|
hasMore: response.hasMore,
|
|
itemList: response.itemList || [],
|
|
extra: response.extra
|
|
}
|
|
}
|
|
} catch (error) {
|
|
return {
|
|
status: "error",
|
|
message:
|
|
error instanceof Error ? error.message : ERROR_MESSAGES.NETWORK_ERROR
|
|
}
|
|
}
|
|
}
|