1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
const puppeteer = require('puppeteer');
const MongoClient = require('mongodb').MongoClient;
require('dotenv').config();
// Specifically for scraping
// Stores in database once finished scraping
// query is the movie or show name
// type is the media type ("tv" for tv show or "movie")
exports.performSearch = async function (id, query, type) {
try {
const uri = "mongodb+srv://hahaUthought:HhKrH8NbhgrcRScq@cluster0.kfyrm.mongodb.net/StreamFinder?retryWrites=true&w=majority";
const client = new MongoClient(uri, { useNewUrlParser: true, useUnifiedTopology: true });
await client.connect();
const database = client.db("db");
const tv = database.collection("tv");
const movie = database.collection('movie')
const browser = await puppeteer.launch({
ignoreDefaultArgs: ["--disable-extensions"],
args: [`--no-sandbox`, `--no-zygote`, `--single-process`, `--args`, `--disable-dev-shm-usage`],
headless: true
})
const pages = await browser.pages();
const page = pages[0];
await page.goto('https://google.com');
const searchBox = await page.$x("/html/body/div[1]/div[3]/form/div[1]/div[1]/div[1]/div/div[2]/input");
await searchBox[0].type(`${query} ${type} streaming`);
await page.keyboard.press('Enter');
await page.waitForNavigation();
const resultsContainer = await page.$$('.r0VsPb')
var streamingPlatforms;
var streamingPrices;
for (let i = 0; i < resultsContainer.length; i++) {
streamingPlatforms = await resultsContainer[i].$$eval('.i3LlFf', nodes => nodes.map(n => n.innerText));
streamingPrices = await resultsContainer[i].$$eval('.V8xno', nodes => nodes.map(n => n.innerText))
}
if (type == "movie") {
await movie.insertOne({
title: query,
id: id,
service: streamingPlatforms,
price: streamingPrices
})
} else if (type == "tv") {
await tv.insertOne({
title: query,
id: id,
service: streamingPlatforms,
price: streamingPrices
})
}
await client.close()
await browser.close();
} catch (error) {
console.log(error)
}
}
|
