62 lines
2.3 KiB
TypeScript
62 lines
2.3 KiB
TypeScript
import { Route } from '@/types';
|
|
import ofetch from '@/utils/ofetch';
|
|
import * as cheerio from 'cheerio';
|
|
import type { Context } from 'hono';
|
|
import { ChartTitleSearchConnection } from './types';
|
|
import path from 'path';
|
|
import { getCurrentPath } from '@/utils/helpers';
|
|
import { art } from '@/utils/render';
|
|
|
|
const __dirname = getCurrentPath(import.meta.url);
|
|
const render = (data) => art(path.join(__dirname, 'templates', 'chart.art'), data);
|
|
|
|
export const route: Route = {
|
|
path: '/chart/:chart?',
|
|
categories: ['multimedia'],
|
|
parameters: { chart: 'The chart to display, `top` by default' },
|
|
example: '/imdb/chart',
|
|
radar: [
|
|
{
|
|
source: ['www.imdb.com/chart/:chart/'],
|
|
},
|
|
],
|
|
name: 'Charts',
|
|
maintainers: ['TonyRL'],
|
|
handler,
|
|
url: 'www.imdb.com/chart/top/',
|
|
description: `| Top 250 Movies | Most Popular Movies | Top 250 TV Shows | Most Popular TV Shows |
|
|
| -------------- | ------------------- | ---------------- | --------------------- |
|
|
| top | moviemeter | toptv | tvmeter |`,
|
|
};
|
|
|
|
async function handler(ctx: Context) {
|
|
const { chart = 'top' } = ctx.req.param();
|
|
const baseUrl = 'https://www.imdb.com';
|
|
const link = `${baseUrl}/chart/${chart}/`;
|
|
|
|
const response = await ofetch(link);
|
|
const $ = cheerio.load(response);
|
|
const nextData = JSON.parse($('script#__NEXT_DATA__').text());
|
|
const chartTitles = nextData.props.pageProps.pageData.chartTitles as ChartTitleSearchConnection;
|
|
|
|
const items = chartTitles.edges.map(({ currentRank, node }) => ({
|
|
title: `${currentRank}. ${node.titleText.text} (${node.releaseYear.year}${node.releaseYear.endYear ? `-${node.releaseYear.endYear}` : ''})`,
|
|
description: render({
|
|
primaryImage: node.primaryImage,
|
|
originalTitleText: node.originalTitleText,
|
|
certificate: node.certificate,
|
|
ratingsSummary: node.ratingsSummary,
|
|
plot: node.plot,
|
|
}),
|
|
link: `${baseUrl}/title/${node.id}`,
|
|
category: node.titleGenres.genres.map((g) => chartTitles.genres.find((genre) => genre.filterId === g.genre.text)?.text),
|
|
}));
|
|
|
|
return {
|
|
title: $('head title').text(),
|
|
description: $('head meta[name="description"]').attr('content'),
|
|
link,
|
|
item: items,
|
|
};
|
|
}
|