fmt

compose.yml

@@ -10,7 +10,7 @@ services:
       MONGO_INITDB_DATABASE: bot
     volumes:
       - mongodb_data:/data/db
-  
+
   bot-api:
     image: git.fai.st/fedi-image-bot/bot-api:latest
     container_name: bot-api
@@ -38,8 +38,8 @@ services:
     environment:
       PORT: 8081
       BOT_API_URI: "http://bot-api:8080"
-      GELBOORU_IMAGES_PER_REQUEST: 100  # Number of images per request, maximum 100
-      GELBOORU_TAGS: "2girls sleeping"  # Tags of the images. The images will have all of these tags
+      GELBOORU_IMAGES_PER_REQUEST: 100 # Number of images per request, maximum 100
+      GELBOORU_TAGS: "2girls sleeping" # Tags of the images. The images will have all of these tags
     volumes:
       - ./:/usr/src/app:ro
 

package.json

@@ -24,4 +24,4 @@
     "dev": "docker compose down -v && docker compose up",
     "test": "docker compose down -v && docker compose run fe-middleware 'sleep 5 && bun test'"
   }
-}
+}

src/app.ts

@@ -9,10 +9,10 @@ app.use(express.json());
 app.use(compression());
 
 app.get("/", (_, res) => {
-    const endpoints = listEndpoints(app);
-    res.json({ endpoints });
+  const endpoints = listEndpoints(app);
+  res.json({ endpoints });
 });
 
 app.get("/image", ImageController.get);
 
-export default app;
+export default app;

src/controllers/ImageController.ts

@@ -3,15 +3,15 @@ import logger from "src/logger";
 import ImageService from "src/services/ImageService";
 
 class ImageController {
-    async get(_: Request, res: Response) {
-        try {
-            const image = await ImageService.get();
-            res.json(image);
-        } catch (error: any) {
-            logger.error(error)
-            res.status(500).json({ "error": `Internal server error: ${error}` });
-        }
+  async get(_: Request, res: Response) {
+    try {
+      const image = await ImageService.get();
+      res.json(image);
+    } catch (error: any) {
+      logger.error(error);
+      res.status(500).json({ error: `Internal server error: ${error}` });
     }
+  }
 }
 
-export default new ImageController();
+export default new ImageController();

src/index.ts

@@ -5,12 +5,12 @@ import logger from "src/logger";
 const PORT = env.PORT;
 
 const server = app.listen(PORT, () =>
-    logger.info(`Express server listening on port ${PORT}`)
+  logger.info(`Express server listening on port ${PORT}`)
 );
 
 process.on("SIGTERM", () => {
-    server.close(() => {
-        logger.info('Server closed.');
-        process.exit(0);
-    });
-});
+  server.close(() => {
+    logger.info("Server closed.");
+    process.exit(0);
+  });
+});

src/services/BotApiService.ts

@@ -3,22 +3,20 @@ import logger from "src/logger";
 import { BotApiResponse } from "src/types/BotApiResponse";
 
 class BotApiService {
-    readonly BOT_API_URI = env.BOT_API_URI;
-
-    async getAll(): Promise<BotApiResponse> {
-        const get_url = `${this.BOT_API_URI}/images`;
-        const response: BotApiResponse = await fetch(get_url)
-            .then(async res => {
-                if (!res.ok) {
-                    logger.error(`${res.status}: ${res.statusText}, ${await res.text()}`)
-                    throw new Error("Error fetching images");
-                } else {
-                    return res.json();
-                }
-            }) as BotApiResponse;
-        return response;
-    }
+  readonly BOT_API_URI = env.BOT_API_URI;
 
+  async getAll(): Promise<BotApiResponse> {
+    const get_url = `${this.BOT_API_URI}/images`;
+    const response: BotApiResponse = (await fetch(get_url).then(async (res) => {
+      if (!res.ok) {
+        logger.error(`${res.status}: ${res.statusText}, ${await res.text()}`);
+        throw new Error("Error fetching images");
+      } else {
+        return res.json();
+      }
+    })) as BotApiResponse;
+    return response;
+  }
 }
 
-export default new BotApiService();
+export default new BotApiService();

src/services/GelbooruApiService.ts

@@ -4,23 +4,29 @@ import GelbooruApiResponse from "src/types/GelbooruApiResponse";
 import GelbooruServiceResponse from "src/types/GelbooruServiceResponse";
 
 class GelbooruApiService {
-    async get(): Promise<GelbooruServiceResponse> {
-        const LIMIT = env.GELBOORU_IMAGES_PER_REQUEST || 100;
-        const TAGS = encodeURIComponent(env.GELBOORU_TAGS || "");
-        const url: string = `https://gelbooru.com/index.php?page=dapi&s=post&q=index&limit=${LIMIT}&json=1&tags=${TAGS}`;
+  async get(): Promise<GelbooruServiceResponse> {
+    const LIMIT = env.GELBOORU_IMAGES_PER_REQUEST || 100;
+    const TAGS = encodeURIComponent(env.GELBOORU_TAGS || "");
+    const url: string = `https://gelbooru.com/index.php?page=dapi&s=post&q=index&limit=${LIMIT}&json=1&tags=${TAGS}`;
 
-        const response: GelbooruApiResponse = await fetch(url)
-            .then(async res => {
-                if (!res.ok) {
-                    logger.error(`${res.status}: ${res.statusText}, ${await res.text()}`)
-                    throw new Error("Error fetching images");
-                } else {
-                    return res.json();
-                }
-            }) as GelbooruApiResponse;
+    const response: GelbooruApiResponse = (await fetch(url).then(
+      async (res) => {
+        if (!res.ok) {
+          logger.error(`${res.status}: ${res.statusText}, ${await res.text()}`);
+          throw new Error("Error fetching images");
+        } else {
+          return res.json();
+        }
+      }
+    )) as GelbooruApiResponse;
 
-        return { posts: response.post.map(post => ({ url: post.file_url, tags: post.tags.split(" ") })) };
-    }
+    return {
+      posts: response.post.map((post) => ({
+        url: post.file_url,
+        tags: post.tags.split(" "),
+      })),
+    };
+  }
 }
 
-export default new GelbooruApiService();
+export default new GelbooruApiService();

src/services/ImageService.ts

@@ -5,32 +5,35 @@ import GelbooruApiService from "src/services/GelbooruApiService";
 import logger from "src/logger";
 
 class ImageService {
-    postsQueue: Image[] = [];
+  postsQueue: Image[] = [];
 
-    async get(): Promise<Image> {
-        while (this.postsQueue.length === 0) {
-            const validPosts = await this.getNewValidImages();
-            this.postsQueue = validPosts;
-            logger.info(`Got ${validPosts.length} images from remote`)
-        }
-        return this.postsQueue.pop() as Image;
+  async get(): Promise<Image> {
+    while (this.postsQueue.length === 0) {
+      const validPosts = await this.getNewValidImages();
+      this.postsQueue = validPosts;
+      logger.info(`Got ${validPosts.length} images from remote`);
     }
+    return this.postsQueue.pop() as Image;
+  }
 
-    private async getNewValidImages(): Promise<Image[]> {
-        const gelbooruResponse: GelbooruServiceResponse = await GelbooruApiService.get();
-        const posts = gelbooruResponse.posts;
+  private async getNewValidImages(): Promise<Image[]> {
+    const gelbooruResponse: GelbooruServiceResponse =
+      await GelbooruApiService.get();
+    const posts = gelbooruResponse.posts;
 
-        const botResponse = await BotApiService.getAll();
-        const imagesUrls = botResponse.images.map(image => image.url);
+    const botResponse = await BotApiService.getAll();
+    const imagesUrls = botResponse.images.map((image) => image.url);
 
-        const validPosts = Promise.all(posts
-            .filter(post => !imagesUrls.some(url => url === post.url))
-            .map((post): Image => {
-                return { url: post.url, tags: post.tags };
-            }));
-        
-        return validPosts;
-    }
+    const validPosts = Promise.all(
+      posts
+        .filter((post) => !imagesUrls.some((url) => url === post.url))
+        .map((post): Image => {
+          return { url: post.url, tags: post.tags };
+        })
+    );
+
+    return validPosts;
+  }
 }
 
-export default new ImageService();
+export default new ImageService();

src/types/BotApiResponse.ts

@@ -1,5 +1,5 @@
 import BotImage from "src/types/BotImage";
 
 export interface BotApiResponse {
-    images: BotImage[]
-}
+  images: BotImage[];
+}

src/types/BotImage.ts

@@ -1,7 +1,7 @@
 export default interface BotImage {
-    _id: string
-    url: string
-    status: string
-    tags: string[]
-    __v: number
+  _id: string;
+  url: string;
+  status: string;
+  tags: string[];
+  __v: number;
 }

src/types/GelbooruApiResponse.ts

@@ -1,42 +1,42 @@
 export default interface GelbooruApiResponse {
-    "@attributes": Attributes
-    post: Post[]
+  "@attributes": Attributes;
+  post: Post[];
 }
-  
+
 export interface Attributes {
-    limit: number
-    offset: number
-    count: number
+  limit: number;
+  offset: number;
+  count: number;
 }
 
 export interface Post {
-    id: number
-    created_at: string
-    score: number
-    width: number
-    height: number
-    md5: string
-    directory: string
-    image: string
-    rating: string
-    source: string
-    change: number
-    owner: string
-    creator_id: number
-    parent_id: number
-    sample: number
-    preview_height: number
-    preview_width: number
-    tags: string
-    title: string
-    has_notes: string
-    has_comments: string
-    file_url: string
-    preview_url: string
-    sample_url: string
-    sample_height: number
-    sample_width: number
-    status: string
-    post_locked: number
-    has_children: string
+  id: number;
+  created_at: string;
+  score: number;
+  width: number;
+  height: number;
+  md5: string;
+  directory: string;
+  image: string;
+  rating: string;
+  source: string;
+  change: number;
+  owner: string;
+  creator_id: number;
+  parent_id: number;
+  sample: number;
+  preview_height: number;
+  preview_width: number;
+  tags: string;
+  title: string;
+  has_notes: string;
+  has_comments: string;
+  file_url: string;
+  preview_url: string;
+  sample_url: string;
+  sample_height: number;
+  sample_width: number;
+  status: string;
+  post_locked: number;
+  has_children: string;
 }

src/types/GelbooruPost.ts

@@ -1,4 +1,4 @@
 export default interface GelbooruPost {
-    url: string;
-    tags: string[];
-}
+  url: string;
+  tags: string[];
+}

src/types/GelbooruServiceResponse.ts

@@ -1,5 +1,5 @@
 import GelbooruPost from "src/types/GelbooruPost";
 
 export default interface GelbooruServiceResponse {
-    posts: GelbooruPost[];
-}
+  posts: GelbooruPost[];
+}

src/types/Image.ts

@@ -1,4 +1,4 @@
 export default interface Image {
-    url: string;
-    tags: string[];
-}
+  url: string;
+  tags: string[];
+}

test/ImageController/ImageController.test.ts

@@ -6,31 +6,31 @@ import request from "supertest";
 const imageServiceOriginal = ImageService;
 
 afterEach(() => {
-    mock.restore();
-    mock.module("src/services/ImageService", () => ({
-        default: imageServiceOriginal,
-    }));
-})
+  mock.restore();
+  mock.module("src/services/ImageService", () => ({
+    default: imageServiceOriginal,
+  }));
+});
 
 describe("endpoint returns the correct status codes", () => {
-    it("should return 200 if successful", async () => {
-        // Can't mock ImageService partially because of bun incomplete implementation of testing :(
-        // It goes to the network directly to test
-        const res = await request(app).get("/image");
-        expect(res.status).toBe(200);
-    });
+  it("should return 200 if successful", async () => {
+    // Can't mock ImageService partially because of bun incomplete implementation of testing :(
+    // It goes to the network directly to test
+    const res = await request(app).get("/image");
+    expect(res.status).toBe(200);
+  });
 
-    it("should return 500 if any error happens", async () => {
-        mock.module("src/services/ImageService", () => {
-            return {
-                default: {
-                    get: () => {
-                        throw new Error("Controlled error");
-                    }
-                }
-            }
-        });
-        const res = await request(app).get("/image");
-        expect(res.status).toBe(500);
+  it("should return 500 if any error happens", async () => {
+    mock.module("src/services/ImageService", () => {
+      return {
+        default: {
+          get: () => {
+            throw new Error("Controlled error");
+          },
+        },
+      };
     });
-})
+    const res = await request(app).get("/image");
+    expect(res.status).toBe(500);
+  });
+});

test/ImageService/ImageService.test.ts

@@ -11,50 +11,58 @@ const gelbooruApiServiceOriginal = GelbooruApiService;
 const botApiServiceOriginal = BotApiService;
 
 afterEach(() => {
-    mock.restore();
-    mock.module("src/services/ImageService", () => ({
-        default: imageServiceOriginal,
-    }));
-    mock.module("src/services/GelbooruApiService", () => ({
-        default: gelbooruApiServiceOriginal,
-    }));
-    mock.module("src/services/BotApiService", () => ({
-        default: botApiServiceOriginal
-    }));
-})
+  mock.restore();
+  mock.module("src/services/ImageService", () => ({
+    default: imageServiceOriginal,
+  }));
+  mock.module("src/services/GelbooruApiService", () => ({
+    default: gelbooruApiServiceOriginal,
+  }));
+  mock.module("src/services/BotApiService", () => ({
+    default: botApiServiceOriginal,
+  }));
+});
 
 describe("endpoint gets a non repeated image", () => {
-    it("should return an image that is not in the response of the /images endpoint of the bot API", async () => {
-        const REPEATED_URL = "https://fastly.picsum.photos/id/1/10/20.jpg?hmac=gY6PvUXFacKfYpBpTTVcNLxumpyMmoCamM-J5DOPwNc";
-        const UNIQUE_URL = "https://fastly.picsum.photos/id/2/10/20.jpg?hmac=zy6lz21CuRIstr9ETx9h5AuoH50s_L2uIEct3dROpY8";
+  it("should return an image that is not in the response of the /images endpoint of the bot API", async () => {
+    const REPEATED_URL =
+      "https://fastly.picsum.photos/id/1/10/20.jpg?hmac=gY6PvUXFacKfYpBpTTVcNLxumpyMmoCamM-J5DOPwNc";
+    const UNIQUE_URL =
+      "https://fastly.picsum.photos/id/2/10/20.jpg?hmac=zy6lz21CuRIstr9ETx9h5AuoH50s_L2uIEct3dROpY8";
 
-        mock.module("src/services/GelbooruApiService", () => {
-            let alreadyCalled = false;
-            return {
-                default: {
-                    get: (): GelbooruApiResponse => {
-                        if (alreadyCalled) {
-                            return { posts: [{ url: UNIQUE_URL, tags: [] }] };
-                        } else {
-                            alreadyCalled = true;
-                            return { posts: [{ url: REPEATED_URL, tags: [] }] };
-                        }
-                    }
-                }
+    mock.module("src/services/GelbooruApiService", () => {
+      let alreadyCalled = false;
+      return {
+        default: {
+          get: (): GelbooruApiResponse => {
+            if (alreadyCalled) {
+              return { posts: [{ url: UNIQUE_URL, tags: [] }] };
+            } else {
+              alreadyCalled = true;
+              return { posts: [{ url: REPEATED_URL, tags: [] }] };
             }
-        });
-
-        mock.module("src/services/BotApiService", () => ({
-            default: {
-                getAll: (): BotApiResponse => ({
-                    images: [
-                        { _id: "0", url: REPEATED_URL, status: "consumed", tags: ["pokemon", "computer"], "__v": 0 }
-                    ]
-                })
-            }
-        }));
-
-        const image: Image = await ImageService.get();
-        expect(image.url).not.toBe(REPEATED_URL);
+          },
+        },
+      };
     });
-});
+
+    mock.module("src/services/BotApiService", () => ({
+      default: {
+        getAll: (): BotApiResponse => ({
+          images: [
+            {
+              _id: "0",
+              url: REPEATED_URL,
+              status: "consumed",
+              tags: ["pokemon", "computer"],
+              __v: 0,
+            },
+          ],
+        }),
+      },
+    }));
+
+    const image: Image = await ImageService.get();
+    expect(image.url).not.toBe(REPEATED_URL);
+  });
+});

tsconfig.json

@@ -18,6 +18,6 @@
     "types": [
       "bun-types" // add Bun global
     ],
-    "baseUrl": "./",
+    "baseUrl": "./"
   }
 }