mirror of
https://github.com/opelly27/streetmerchant.git
synced 2026-05-20 02:57:34 +00:00
Vincent
96 lines
2.5 KiB
TypeScript
96 lines
2.5 KiB
TypeScript
import {Config} from './config';
|
||
import {Store, Stores} from './store';
|
||
import puppeteer from 'puppeteer';
|
||
import open from 'open';
|
||
import sendNotification from './notification';
|
||
import {Logger} from './logger';
|
||
|
||
/**
|
||
* Starts the bot.
|
||
*/
|
||
async function main() {
|
||
const results = [];
|
||
for (const store of Stores) {
|
||
Logger.debug(store.links);
|
||
results.push(lookup(store));
|
||
}
|
||
|
||
await Promise.all(results);
|
||
|
||
Logger.info('↗ trying stores again');
|
||
setTimeout(main, Config.rateLimitTimeout);
|
||
}
|
||
|
||
/**
|
||
* Responsible for looking up information about a each product within
|
||
* a `Store`. It's important that we ignore `no-await-in-loop` here
|
||
* because we don't want to get rate limited within the same store.
|
||
*
|
||
* @param store Vendor of graphics cards.
|
||
*/
|
||
async function lookup(store: Store) {
|
||
/* eslint-disable no-await-in-loop */
|
||
for (const link of store.links) {
|
||
const browser = await puppeteer.launch();
|
||
const page = await browser.newPage();
|
||
await page.setUserAgent(Config.page.userAgent);
|
||
await page.setViewport({
|
||
height: Config.page.height,
|
||
width: Config.page.width
|
||
});
|
||
|
||
const graphicsCard = `${link.brand} ${link.model}`;
|
||
|
||
try {
|
||
await page.goto(link.url, {waitUntil: 'networkidle0'});
|
||
} catch {
|
||
Logger.error(`✖ [${store.name}] ${graphicsCard} skipping; timed out`);
|
||
return;
|
||
}
|
||
|
||
const bodyHandle = await page.$('body');
|
||
const textContent = await page.evaluate(body => body.textContent, bodyHandle);
|
||
|
||
Logger.debug(textContent);
|
||
|
||
if (isOutOfStock(textContent, link.oosLabels)) {
|
||
Logger.info(`✖ [${store.name}] ${graphicsCard} is still out of stock`);
|
||
} else {
|
||
Logger.info(`🚀🚀🚀 [${store.name}] ${graphicsCard} IN STOCK 🚀🚀🚀`);
|
||
Logger.info(link.url);
|
||
|
||
Logger.debug('ℹ saving screenshot');
|
||
await page.screenshot({path: `success-${Date.now()}.png`});
|
||
|
||
const givenUrl = store.cartUrl ? store.cartUrl : link.url;
|
||
await open(givenUrl);
|
||
sendNotification(givenUrl);
|
||
}
|
||
|
||
await browser.close();
|
||
}
|
||
/* eslint-enable no-await-in-loop */
|
||
}
|
||
|
||
/**
|
||
* Checks if DOM has any out-of-stock related text.
|
||
*
|
||
* @param domText Complete DOM of website.
|
||
* @param oosLabels Out-of-stock labels.
|
||
*/
|
||
function isOutOfStock(domText: string, oosLabels: string[]) {
|
||
const domTextLowerCase = domText.toLowerCase();
|
||
return oosLabels.some(label => domTextLowerCase.includes(label));
|
||
}
|
||
|
||
/**
|
||
* Will continually run until user interferes.
|
||
*/
|
||
try {
|
||
void main();
|
||
} catch (error) {
|
||
// Ignoring errors; more than likely due to rate limits
|
||
Logger.error(error);
|
||
void main();
|
||
}
|