3 files changed, 16 insertions, 9 deletions

diff --git a/src/api/e621/index.ts b/src/api/e621/index.ts
index 411edae..2fc0d2e 100644
--- a/src/api/e621/index.ts
+++ b/src/api/e621/index.ts
@@ -1,6 +1,11 @@
 import got from "got";
 import config from "../../config";
 
+export interface GetPostQuery {
+    tags: readonly string[];
+    maxPage: number;
+}
+
 export interface Post {
     id: number;
     file: {
@@ -26,7 +31,7 @@ export const client = got.extend({
     },
 });
 
-export async function getPost(id: number) {
+export async function getPostById(id: number) {
     const response = await client
         .get("https://e926.net/posts.json", {
             searchParams: {
@@ -42,9 +47,7 @@ export async function getPost(id: number) {
     return response.posts[0];
 }
 
-export async function getRandomPost() {
-    const queryIndex = Math.floor(Math.random() * config.e621.queries.length);
-    const query = config.e621.queries[queryIndex];
+export async function getRandomPost(query: GetPostQuery) {
     const page = Math.floor(Math.random() * (query.maxPage - 1)) + 1;
 
     const response = await client
@@ -64,5 +67,5 @@ export async function getRandomPost() {
     const postIndex = Math.floor(Math.random() * response.posts.length);
     const post = response.posts[postIndex];
 
-    return { queryIndex, post };
+    return post;
 }
diff --git a/src/config.ts b/src/config.ts
index d854725..6b0816d 100644
--- a/src/config.ts
+++ b/src/config.ts
@@ -1,4 +1,6 @@
-const mainQuery = {
+import { GetPostQuery } from "./api/e621";
+
+const mainQuery: GetPostQuery = {
     tags: [
         "feral",
         "-anthro",
diff --git a/src/index.ts b/src/index.ts
index e2fee01..3b8b0a8 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -19,7 +19,9 @@ const args = cliArgs.parse<{
 async function postRandomPicture() {
     console.log("Fetching random post...");
 
-    const { queryIndex, post } = await e621.getRandomPost();
+    const queryIndex = Math.floor(Math.random() * config.e621.queries.length);
+    const query = config.e621.queries[queryIndex];
+    const post = await e621.getRandomPost(query);
 
     console.log(`Got ${post.id} via query ${queryIndex}`);
     
@@ -27,9 +29,9 @@ async function postRandomPicture() {
 }
 
 async function postSpecificPicture(id: number) {
-    console.log("Fetching post ${id}...");
+    console.log(`Fetching post ${id}...`);
 
-    const post = await e621.getPost(id);
+    const post = await e621.getPostById(id);
 
     console.log(`Got ${post.id}`);