Code improvements, support a manual post queue

1 files changed, 5 insertions, 3 deletions

diff --git a/src/api/e621/index.ts b/src/api/e926/index.ts
index a8abbcf..43a3179 100644
--- a/src/api/e621/index.ts
+++ b/src/api/e926/index.ts
@@ -1,7 +1,9 @@
 import got from "got";
 import config from "../../config";
 import delay from "../../util/delay";
-import dedupe from "../../services/dedupe";
+import PostDatabase from "../../services/postDatabase";
+
+export const dedupeDb = new PostDatabase("e926dedupe.json", 50);
 
 export interface GetPostQuery {
     tags: readonly string[];
@@ -49,7 +51,7 @@ export async function getPostById(id: number) {
     return response.posts[0];
 }
 
-export async function getRandomPost(query: GetPostQuery) {
+export async function getRandomPost(query: GetPostQuery): Promise<Post> {
     const page = Math.floor(Math.random() * (query.maxPage - 1)) + 1;
 
     const response = await client
@@ -69,7 +71,7 @@ export async function getRandomPost(query: GetPostQuery) {
     const postIndex = Math.floor(Math.random() * response.posts.length);
     const post = response.posts[postIndex];
 
-    const isDupe = await dedupe.check({ provider: "e926", id: post.id });
+    const isDupe = await dedupeDb.insertIfNotExists({ provider: "e926", id: post.id });
 
     if (isDupe) {
         await delay(1000);