-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
90 lines (80 loc) · 2.58 KB
/
index.js
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
const cheerio = require('cheerio')
const axios = require('axios')
const express = require('express')
const { PrismaClient } = require('@prisma/client')
const prisma = new PrismaClient()
const scrapMoviseInfo = async (url) => {
const html = await axios.get(url)
const $ = cheerio.load(html.data)
const list = $(".lister-list > tr").map((i, el) => {
const title = $(el).find(".titleColumn > a").text()
const link = `https://www.imdb.com${$(el).find(".titleColumn > a").attr("href").split('?')[0]}`
const posterUrl = $(el).find("td.posterColumn > a > img").attr("src").split("._")[0] + "._V1_FMjpg_UX1000_.jpg"
const rating = $(el).find(".imdbRating").text().trim().replace(/^$/, "0.0")
return ({ title, rating, posterUrl, link })
}).get()
return (list)
}
const app = express()
const port = 5556
app.get('/', (req, res) => {
const api = [
{
top: '/api/top',
origin: 'https://www.imdb.com/chart/top/'
},
{
moviemeter: '/api/moviemeter',
origin: 'https://www.imdb.com/chart/moviemeter/'
},
{
toptv: '/api/toptv',
origin: 'https://www.imdb.com/chart/toptv/'
},
{
tvmeter: '/api/tvmeter',
origin: 'https://www.imdb.com/chart/tvmeter/'
}
]
res.send(api)
})
const listToDb = async (list) => {
try {
const createdMovies = await prisma.movie.createMany({
data: list.map((movie) => ({
title: movie.title,
rating: parseFloat(movie.rating),
posterUrl: movie.posterUrl,
link: movie.link,
})),
skipDuplicates: false,
});
return createdMovies;
} catch (error) {
console.error('Error creating movies:', error);
throw error;
}
};
app.get('/api/top', async (req, res) => {
const list = await scrapMoviseInfo('https://www.imdb.com/chart/top/')
res.send(list)
// await listToDb(list)
})
app.get('/api/moviemeter', async (req, res) => {
const list = await scrapMoviseInfo('https://www.imdb.com/chart/moviemeter/')
res.send(list)
// await listToDb(list)
})
app.get('/api/toptv', async (req, res) => {
const list = await scrapMoviseInfo('https://www.imdb.com/chart/toptv/')
res.send(list)
// await listToDb(list)
})
app.get('/api/tvmeter', async (req, res) => {
const list = await scrapMoviseInfo('https://www.imdb.com/chart/tvmeter/')
res.send(list)
// await listToDb(list)
})
app.listen(port, async () => {
console.log(`Example app listening on port ${port}`)
})