mirror of
https://github.com/opelly27/streetmerchant.git
synced 2026-05-20 04:07:36 +00:00
Jef LeCompte
122 lines
2.9 KiB
TypeScript
122 lines
2.9 KiB
TypeScript
import * as Process from 'process';
|
||
import {config} from './config'; // Needs to be loaded first
|
||
import {startAPIServer, stopAPIServer} from './web';
|
||
import {Browser, launch} from 'puppeteer';
|
||
import {getSleepTime} from './util';
|
||
import {logger} from './logger';
|
||
import {storeList} from './store/model';
|
||
import {tryLookupAndLoop} from './store';
|
||
|
||
let browser: Browser | undefined;
|
||
|
||
async function sleep(ms: number) {
|
||
return new Promise(resolve => setTimeout(resolve, ms));
|
||
}
|
||
|
||
/**
|
||
* Schedules a restart of the bot
|
||
*/
|
||
async function restartMain() {
|
||
if (config.restartTime > 0) {
|
||
await sleep(config.restartTime);
|
||
await stop();
|
||
loopMain();
|
||
}
|
||
}
|
||
|
||
/**
|
||
* Starts the bot.
|
||
*/
|
||
async function main() {
|
||
const args: string[] = [];
|
||
|
||
// Skip Chromium Linux Sandbox
|
||
// https://github.com/puppeteer/puppeteer/blob/main/docs/troubleshooting.md#setting-up-chrome-linux-sandbox
|
||
if (config.browser.isTrusted) {
|
||
args.push('--no-sandbox');
|
||
args.push('--disable-setuid-sandbox');
|
||
}
|
||
|
||
// https://github.com/puppeteer/puppeteer/blob/main/docs/troubleshooting.md#tips
|
||
// https://stackoverflow.com/questions/48230901/docker-alpine-with-node-js-and-chromium-headless-puppeter-failed-to-launch-c
|
||
if (config.docker) {
|
||
args.push('--disable-dev-shm-usage');
|
||
args.push('--no-sandbox');
|
||
args.push('--disable-setuid-sandbox');
|
||
args.push('--headless');
|
||
args.push('--disable-gpu');
|
||
}
|
||
|
||
// Add the address of the proxy server if defined
|
||
if (config.proxy.address) {
|
||
args.push(
|
||
`--proxy-server=${config.proxy.protocol}://${config.proxy.address}:${config.proxy.port}`
|
||
);
|
||
}
|
||
|
||
if (args.length > 0) {
|
||
logger.info('ℹ puppeteer config: ', args);
|
||
}
|
||
|
||
await stop();
|
||
browser = await launch({
|
||
args,
|
||
defaultViewport: {
|
||
height: config.page.height,
|
||
width: config.page.width,
|
||
},
|
||
headless: config.browser.isHeadless,
|
||
});
|
||
|
||
config.browser.userAgent = await browser.userAgent();
|
||
|
||
for (const store of storeList.values()) {
|
||
logger.debug('store links', {meta: {links: store.links}});
|
||
if (store.setupAction !== undefined) {
|
||
store.setupAction(browser);
|
||
}
|
||
|
||
setTimeout(tryLookupAndLoop, getSleepTime(store), browser, store);
|
||
}
|
||
|
||
await startAPIServer();
|
||
}
|
||
|
||
async function stop() {
|
||
await stopAPIServer();
|
||
|
||
if (browser) {
|
||
// Use temporary swap variable to avoid any race condition
|
||
const browserTemporary = browser;
|
||
browser = undefined;
|
||
await browserTemporary.close();
|
||
}
|
||
}
|
||
|
||
async function stopAndExit() {
|
||
await stop();
|
||
Process.exit(0);
|
||
}
|
||
|
||
/**
|
||
* Will continually run until user interferes.
|
||
*/
|
||
async function loopMain() {
|
||
try {
|
||
restartMain();
|
||
await main();
|
||
} catch (error: unknown) {
|
||
logger.error(
|
||
'✖ something bad happened, resetting streetmerchant in 5 seconds',
|
||
error
|
||
);
|
||
setTimeout(loopMain, 5000);
|
||
}
|
||
}
|
||
|
||
void loopMain();
|
||
|
||
process.on('SIGINT', stopAndExit);
|
||
process.on('SIGQUIT', stopAndExit);
|
||
process.on('SIGTERM', stopAndExit);
|