Code improvements

2 files changed, 119 insertions, 0 deletions

diff --git a/src/services/dedupe.ts b/src/services/dedupe.ts
new file mode 100644
index 0000000..2eeb5ee
--- /dev/null
+++ b/src/services/dedupe.ts
@@ -0,0 +1,65 @@
+import fs from "fs/promises";
+import path from "path";
+import * as f from "fp-ts";
+import * as t from "io-ts";
+
+export const E926DedupeEntryC = t.type({
+    provider: t.literal("e926"),
+    id: t.number,
+});
+
+export const DedupeEntryC = E926DedupeEntryC;
+
+export type E926DedupeEntry = t.TypeOf<typeof E926DedupeEntryC>;
+
+export type DedupeEntry = t.TypeOf<typeof DedupeEntryC>;
+
+export class Dedupe {
+    private entries: DedupeEntry[] = [];
+
+    private readonly filePath: string;
+
+    private isLoaded = false;
+
+    constructor(private max: number, filename: string) {
+        this.filePath = path.join(process.cwd(), filename);
+    }
+
+    private async load() {
+        if (this.isLoaded) {
+            return;
+        }
+
+        try {
+            await fs.stat(this.filePath);
+        } catch {
+            await this.save();
+        }
+
+        const fileContent = await fs.readFile(this.filePath, "utf8");
+        const entries = t.array(DedupeEntryC).decode(fileContent);
+
+        if (f.either.isRight(entries)) {
+            this.entries = entries.right;
+        }
+    }
+
+    private async save() {
+        await fs.writeFile(this.filePath, JSON.stringify(this.entries ?? []), "utf8");
+    }
+
+    async check(entry: DedupeEntry) {
+        await this.load();
+
+        const has = !!this.entries.find((e) => e.provider === entry.provider && e.id === entry.id);
+
+        if (!has) {
+            this.entries.push(entry);
+            await this.save();
+        }
+
+        return has;
+    }
+}
+
+export default new Dedupe(50, "dedupe.json");
diff --git a/src/services/jobs.ts b/src/services/jobs.ts
new file mode 100644
index 0000000..cf3b894
--- /dev/null
+++ b/src/services/jobs.ts
@@ -0,0 +1,54 @@
+import * as e621 from "../api/e621";
+import * as mastodon from "../api/mastodon";
+import config from "../config";
+import Sharp from "sharp";
+
+export async function postRandomPicture() {
+  console.log("Fetching random post...");
+
+  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}`);
+
+  await handlePost(post);
+}
+
+export async function postSpecificPicture(id: number) {
+    console.log(`Fetching post ${id}...`);
+
+    const post = await e621.getPostById(id);
+
+    console.log(`Got ${post.id}`);
+
+    await handlePost(post);
+}
+
+async function handlePost(post: e621.Post) {
+    const source = post.sources.length ? post.sources[0] : undefined;
+    const cws = config.cw.filter((w) => post.tags.general.includes(w));
+
+    console.log(`Downloading image...`);
+
+    let file = await e621.client.get(post.file.url).buffer();
+
+    if (Buffer.byteLength(file) > 1024 * 1024 * 9) {
+        console.log(`Compressing...`);
+
+        file = await Sharp(file)
+            .resize(1800, 1800, { fit: "inside", withoutEnlargement: true })
+            .jpeg({ quality: 80, mozjpeg: true })
+            .toBuffer();
+    }
+
+    console.log(`Uploading...`);
+
+    const attachment = await mastodon.upload(file, post.id.toString(10));
+
+    console.log(`Posting status...`);
+
+    const status = await mastodon.createStatus(`https://e926.net/posts/${post.id}`, source, cws, attachment.id);
+
+    console.log(`Done! ${status.url}`);
+}