Support posting a specific picture

1 files changed, 17 insertions, 1 deletions

diff --git a/src/api/e621/index.ts b/src/api/e621/index.ts
index c8e0f44..411edae 100644
--- a/src/api/e621/index.ts
+++ b/src/api/e621/index.ts
@@ -26,7 +26,23 @@ export const client = got.extend({
     },
 });
 
-export async function randomPost() {
+export async function getPost(id: number) {
+    const response = await client
+        .get("https://e926.net/posts.json", {
+            searchParams: {
+                tags: `id:${id}`,
+            },
+        })
+        .json<{ posts: readonly Post[] }>();
+
+    if (!response.posts.length) {
+        throw new Error("No posts received");
+    }
+
+    return response.posts[0];
+}
+
+export async function getRandomPost() {
     const queryIndex = Math.floor(Math.random() * config.e621.queries.length);
     const query = config.e621.queries[queryIndex];
     const page = Math.floor(Math.random() * (query.maxPage - 1)) + 1;