tretrauit/castorsrm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add change_viewport option

Browse Source
This commit is contained in:
tretrauit
2025-04-13 00:43:55 +07:00
parent 50c5927be7
commit 358fe53be9
3 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/config.ts
View File

@ -1,11 +1,13 @@
class Config {
playwright: {
browser: string;
change_viewport: boolean;
headless: boolean;
cdp: string;
url: string;
} = {
browser: "chromium",
change_viewport: true,
headless: true,
cdp: "ws://127.0.0.1:9222",
url: "https://www.twitch.tv/",

5
src/index.ts
View File

@ -16,9 +16,10 @@ if (fs.existsSync("config.json")) {
config = Config.fromJSON(text);
} else {
logger.info("No configuration file found. Using the default configuration.");
await fs.promises.writeFile("config.json", config.toJSON());
logger.info("Default configuration file 'config.json' created.");
}
// Write the new config file in case we updated something.
await fs.promises.writeFile("config.json", config.toJSON());
logger.level = process.env.LOG_LEVEL || config.logger.level;
logger.info(`Logger level set to '${logger.level}'`);
@ -63,7 +64,7 @@ if (config.proxy.mode === "reflect4") {
const spawnId = `${i + 1}`;
while (true) {
try {
await reflect4.spawn(context, config.playwright.url, spawnId);
await reflect4.spawn(context, config.playwright.url, config.playwright.change_viewport, spawnId);
} catch (e) {
logger.error(`[${spawnId}] Error while running: ${e}`);
}

5
src/proxy/reflect4.ts
View File

@ -3,14 +3,17 @@ import logger from "../logger";
import * as twitch from "../website/twitch";
import * as constants from "../constants";
async function spawn(context: BrowserContext, targetUrl: string, spawnId: string = "unknown") {
async function spawn(context: BrowserContext, targetUrl: string, changeViewport: boolean = false, spawnId: string = "unknown") {
const server = constants.REFLECT4_SERVERS[Math.floor(Math.random() * constants.REFLECT4_SERVERS.length)];
logger.debug(`[${spawnId}] Using reflect4 server: ${server}`);
const page = await context.newPage();
if (changeViewport) {
logger.debug(`[${spawnId}] Changing viewport size...`);
const deviceName = Object.keys(devices)[Math.floor(Math.random() * Object.keys(devices).length)];
const device = devices[deviceName];
logger.debug(`[${spawnId}] Using device: ${deviceName}`);
await page.setViewportSize(device.viewport);
}
try {
await page.goto(server);
} catch (e) {