Skip to content

Commit

Permalink
feat(notification): add price to links (#1209)
Browse files Browse the repository at this point in the history
fix(store): label selection ordering and pricing
fix(nvidia): deprecation notice removed outside of usa
fix(amazon): maxPrice selector

Resolves #1188
Resolves #673
Resolves #1187
  • Loading branch information
jef authored Dec 6, 2020
1 parent f7b32e8 commit 15ec12b
Show file tree
Hide file tree
Showing 17 changed files with 157 additions and 112 deletions.
67 changes: 50 additions & 17 deletions package-lock.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion package.json
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
"@slack/web-api": "^5.14.0",
"chalk": "^4.1.0",
"cheerio": "^1.0.0-rc.3",
"discord-webhook-node": "^1.1.8",
"discord.js": "^12.5.1",
"dotenv": "^8.2.0",
"messaging-api-telegram": "^1.0.1",
"mqtt": "^4.2.6",
Expand Down
1 change: 1 addition & 0 deletions src/__test__/notification-test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ const link: Link = {
brand: 'test:brand',
cartUrl: 'https://www.example.com/cartUrl',
model: 'test:model',
price: 100,
series: 'test:series',
url: 'https://www.example.com/url'
};
Expand Down
2 changes: 1 addition & 1 deletion src/config.ts
Original file line number Diff line number Diff line change
Expand Up @@ -180,7 +180,7 @@ const notifications = {
desktop: process.env.DESKTOP_NOTIFICATIONS === 'true',
discord: {
notifyGroup: envOrArray(process.env.DISCORD_NOTIFY_GROUP),
webHookUrl: envOrArray(process.env.DISCORD_WEB_HOOK)
webhooks: envOrArray(process.env.DISCORD_WEB_HOOK)
},
email: {
password: envOrString(process.env.EMAIL_PASSWORD),
Expand Down
11 changes: 4 additions & 7 deletions src/logger.ts
Original file line number Diff line number Diff line change
Expand Up @@ -118,11 +118,9 @@ export const Print = {

return `ℹ ${buildProductString(link, store)} :: IN STOCK, WAITING`;
},
// eslint-disable-next-line max-params
maxPrice(
link: Link,
store: Store,
price: number,
maxPrice: number,
color?: boolean
): string {
Expand All @@ -131,14 +129,13 @@ export const Print = {
'✖ ' +
buildProductString(link, store, true) +
' :: ' +
chalk.yellow(`PRICE ${price} EXCEEDS LIMIT ${maxPrice}`)
chalk.yellow(`PRICE ${link.price ?? ''} EXCEEDS LIMIT ${maxPrice}`)
);
}

return `✖ ${buildProductString(
link,
store
)} :: PRICE ${price} EXCEEDS LIMIT ${maxPrice}`;
return `✖ ${buildProductString(link, store)} :: PRICE ${
link.price ?? ''
} EXCEEDS LIMIT ${maxPrice}`;
},
message(
message: string,
Expand Down
62 changes: 42 additions & 20 deletions src/notification/discord.ts
Original file line number Diff line number Diff line change
@@ -1,41 +1,63 @@
import {Link, Store} from '../store/model';
import {MessageBuilder, Webhook} from 'discord-webhook-node';
import Discord from 'discord.js';
import {config} from '../config';
import {logger} from '../logger';

const discord = config.notifications.discord;
const hooks = discord.webHookUrl;
const notifyGroup = discord.notifyGroup;
const {notifyGroup, webhooks} = discord;

function getIdAndToken(webhook: string) {
const match = /.*\/webhooks\/(\d+)\/(.+)/.exec(webhook);

if (!match) {
throw new Error('could not get discord webhook');
}

return {
id: match[1],
token: match[2]
};
}

export function sendDiscordMessage(link: Link, store: Store) {
if (discord.webHookUrl.length > 0) {
if (webhooks.length > 0) {
logger.debug('↗ sending discord message');

(async () => {
try {
const embed = new MessageBuilder();
embed.setTitle('Stock Notification');
if (link.cartUrl)
embed.addField('Add To Cart Link', link.cartUrl, true);
embed.addField('Product Page', link.url, true);
const embed = new Discord.MessageEmbed()
.setTitle('_**Stock alert!**_')
.setDescription(
'> provided by [streetmerchant](https://github.com/jef/streetmerchant) with :heart:'
)
.setThumbnail(
'https://raw.githubusercontent.com/jef/streetmerchant/main/media/streetmerchant-square.png'
)
.setColor('#52b788')
.setTimestamp();

embed.addField('Store', store.name, true);
if (link.price) embed.addField('Price', `$${link.price}`, true);
embed.addField('Product Page', link.url);
if (link.cartUrl) embed.addField('Add to Cart', link.cartUrl);
embed.addField('Brand', link.brand, true);
embed.addField('Series', link.series, true);
embed.addField('Model', link.model, true);

if (notifyGroup) {
embed.setText(notifyGroup.join(' '));
}

embed.setColor(0x76b900);
embed.setTimestamp();
embed.addField('Series', link.series, true);

const promises = [];
for (const hook of hooks) {
promises.push(new Webhook(hook).send(embed));
for (const webhook of webhooks) {
const {id, token} = getIdAndToken(webhook);
const client = new Discord.WebhookClient(id, token);
promises.push({
client,
message: client.send(notifyGroup.join(' '), {
embeds: [embed],
username: 'streetmerchant'
})
});
}

await Promise.all(promises);
(await Promise.all(promises)).forEach(({client}) => client.destroy());

logger.info('✔ discord message sent');
} catch (error: unknown) {
Expand Down
21 changes: 8 additions & 13 deletions src/store/includes-labels.ts
Original file line number Diff line number Diff line change
Expand Up @@ -116,27 +116,22 @@ export function includesLabels(
);
}

export async function cardPrice(
export async function getPrice(
page: Page,
query: Pricing,
max: number,
options: Selector
): Promise<number | null> {
if (!max || max === -1) {
return null;
}

const selector = {...options, selector: query.container};
const cardPrice = await extractPageContents(page, selector);
const priceString = await extractPageContents(page, selector);

if (cardPrice) {
const priceSeperator = query.euroFormat ? /\./g : /,/g;
const cardpriceNumber = Number.parseFloat(
cardPrice.replace(priceSeperator, '').match(/\d+/g)!.join('.')
if (priceString) {
const priceSeparator = query.euroFormat ? /\./g : /,/g;
const price = Number.parseFloat(
priceString.replace(priceSeparator, '').match(/\d+/g)!.join('.')
);

logger.debug(`Raw card price: ${cardPrice} | Limit: ${max}`);
return cardpriceNumber > max ? cardpriceNumber : null;
logger.debug('received price', price);
return price;
}

return null;
Expand Down
79 changes: 38 additions & 41 deletions src/store/lookup.ts
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import {Browser, Page, PageEventObj, Request, Response} from 'puppeteer';
import {Link, Store, getStores} from './model';
import {Print, logger} from '../logger';
import {Selector, cardPrice, pageIncludesLabels} from './includes-labels';
import {Selector, getPrice, pageIncludesLabels} from './includes-labels';
import {
closePage,
delay,
Expand Down Expand Up @@ -303,6 +303,43 @@ async function lookupCardInStock(store: Store, page: Page, link: Link) {
type: 'textContent'
};

if (store.labels.captcha) {
if (await pageIncludesLabels(page, store.labels.captcha, baseOptions)) {
logger.warn(Print.captcha(link, store, true));
await delay(getSleepTime(store));
return false;
}
}

if (store.labels.bannedSeller) {
if (
await pageIncludesLabels(page, store.labels.bannedSeller, baseOptions)
) {
logger.warn(Print.bannedSeller(link, store, true));
return false;
}
}

if (store.labels.maxPrice) {
const maxPrice = config.store.maxPrice.series[link.series];

link.price = await getPrice(page, store.labels.maxPrice, baseOptions);

if (link.price && link.price > maxPrice && maxPrice > 0) {
logger.info(Print.maxPrice(link, store, maxPrice, true));
return false;
}
}

// Fixme: currently causing issues
// Do API inventory validation in realtime (no cache) if available
// if (
// store.realTimeInventoryLookup !== undefined &&
// link.itemNumber !== undefined
// ) {
// return store.realTimeInventoryLookup(link.itemNumber);
// }

if (store.labels.inStock) {
const options = {
...baseOptions,
Expand Down Expand Up @@ -336,46 +373,6 @@ async function lookupCardInStock(store: Store, page: Page, link: Link) {
}
}

if (store.labels.bannedSeller) {
if (
await pageIncludesLabels(page, store.labels.bannedSeller, baseOptions)
) {
logger.warn(Print.bannedSeller(link, store, true));
return false;
}
}

if (store.labels.maxPrice) {
const price = await cardPrice(
page,
store.labels.maxPrice,
config.store.maxPrice.series[link.series],
baseOptions
);
const maxPrice = config.store.maxPrice.series[link.series];
if (price) {
logger.info(Print.maxPrice(link, store, price, maxPrice, true));
return false;
}
}

if (store.labels.captcha) {
if (await pageIncludesLabels(page, store.labels.captcha, baseOptions)) {
logger.warn(Print.captcha(link, store, true));
await delay(getSleepTime(store));
return false;
}
}

// Fixme: currently causing issues
// Do API inventory validation in realtime (no cache) if available
// if (
// store.realTimeInventoryLookup !== undefined &&
// link.itemNumber !== undefined
// ) {
// return store.realTimeInventoryLookup(link.itemNumber);
// }

return true;
}

Expand Down
3 changes: 1 addition & 2 deletions src/store/model/amazon-ca.ts
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,7 @@ export const AmazonCa: Store = {
text: ['add to cart']
},
maxPrice: {
container: 'span[class*="PriceString"]',
euroFormat: false
container: '#priceblock_ourprice'
}
},
links: [
Expand Down
Loading

0 comments on commit 15ec12b

Please sign in to comment.