From e8d271edc66237612bf6ba61fd02440f79b6ff6b Mon Sep 17 00:00:00 2001
From: Nick Leeman <nick@aterve.nl>
Date: Fri, 26 Mar 2021 15:24:00 +0100
Subject: [PATCH] added wifimedia support

---
 src/core.ts             |  6 +++++-
 src/modules/websites.ts | 18 ++++++++++++++++++
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/core.ts b/src/core.ts
index f3de40a..6c8bdd1 100644
--- a/src/core.ts
+++ b/src/core.ts
@@ -31,7 +31,8 @@ import {
     Conrad,
     HuisdierExpress,
     HondenBed,
-    Coolblue
+    Coolblue,
+    WifiMedia
 } from "./modules/websites";
 
 // Globals
@@ -280,6 +281,9 @@ async function crawlProductStock(url: string) {
             case "hondenbed.nl":
                 return [domain, page.url(), await HondenBed.check(html)];
 
+            case "wifimedia.eu":
+                return [domain, page.url(), await WifiMedia.check(html)];
+
             default:
                 console.error(`-- ${domain} is not an available website module! Can't check stock!`);
                 return [domain, page.url(), false];
diff --git a/src/modules/websites.ts b/src/modules/websites.ts
index 92984c2..533b6ad 100644
--- a/src/modules/websites.ts
+++ b/src/modules/websites.ts
@@ -509,3 +509,21 @@ export namespace Coolblue {
     }
 }
 
+export namespace WifiMedia {
+    export async function check(html: string) {
+        try {
+            const $ = cheerio.load(html);
+            
+            if ($("[itemprop=availability]").first().attr("content") == "http://schema.org/InStock") {
+                return true;
+            }
+            
+            return false;
+        } catch (error) {
+            console.log(error);
+            console.error(`Error occured during stock check!`);
+            return false;
+        }
+    }
+}
+