70 lines
1.7 KiB
TypeScript
70 lines
1.7 KiB
TypeScript
import * as cache from "@lib/cache/cache.ts";
|
|
import * as openai from "@lib/openai.ts";
|
|
import { GenericResource } from "@lib/types.ts";
|
|
import { parseRating } from "@lib/helpers.ts";
|
|
|
|
type RecommendationResource = {
|
|
id: string;
|
|
type: string;
|
|
rating: number;
|
|
tags?: string[];
|
|
description?: string;
|
|
keywords?: string[];
|
|
author?: string;
|
|
year?: number;
|
|
};
|
|
|
|
export async function createRecommendationResource(
|
|
res: GenericResource,
|
|
description?: string,
|
|
) {
|
|
const cacheId = `recommendations:${res.type}:${res.id.replaceAll(":", "")}`;
|
|
const resource: RecommendationResource = await cache.get(cacheId) || {
|
|
id: res.id,
|
|
type: res.type,
|
|
rating: -1,
|
|
};
|
|
if (description && !resource.keywords) {
|
|
const keywords = await openai.createKeywords(res.type, description, res.id);
|
|
if (keywords?.length) {
|
|
resource.keywords = keywords;
|
|
}
|
|
}
|
|
|
|
const { author, date, rating } = res.meta || {};
|
|
|
|
if (res?.tags) {
|
|
resource.tags = res.tags;
|
|
}
|
|
|
|
if (typeof rating !== "undefined") {
|
|
resource.rating = parseRating(rating);
|
|
}
|
|
|
|
if (author) {
|
|
resource.author = author;
|
|
}
|
|
|
|
if (description) {
|
|
resource.description = description;
|
|
}
|
|
|
|
if (date) {
|
|
const d = typeof date === "string" ? new Date(date) : date;
|
|
resource.year = d.getFullYear();
|
|
}
|
|
|
|
cache.set(cacheId, JSON.stringify(resource));
|
|
}
|
|
|
|
export function getRecommendation(id: string, type: string) {
|
|
return cache.get(`recommendations:${type}:${id}`);
|
|
}
|
|
|
|
export async function getAllRecommendations() {
|
|
const keys = await cache.keys("recommendations:movie:*");
|
|
return Promise.all(keys.map((k) => cache.get(k))).then((res) =>
|
|
res.map((r) => JSON.parse(r))
|
|
);
|
|
}
|