webhook

deno.json

@@ -13,7 +13,8 @@
     "@std/path": "jsr:@std/path",
     "@std/fs": "jsr:@std/fs",
     "ai": "npm:ai@^4.2.8",
-    "dedent": "npm:dedent@^1.5.3"
+    "dedent": "npm:dedent@^1.5.3",
+    "discord-api-types": "npm:discord-api-types@^0.37.119"
   },
   "lock": false,
   "fmt": {

discord.ts

@@ -0,0 +1,28 @@
+import { RESTPostAPIWebhookWithTokenJSONBody } from 'discord-api-types/v10';
+
+export async function executeWebhook(webhookUrl: string, imageBlob: Blob, filename: string = 'image.jpg') {
+  const formData = new FormData();
+
+  const payload: RESTPostAPIWebhookWithTokenJSONBody = {
+    avatar_url: 'https://upload.wikimedia.org/wikipedia/commons/thumb/f/f0/Facebook_AI_slop%2C_%22Shrimp_Jesus%22_2.jpg/1024px-Facebook_AI_slop%2C_%22Shrimp_Jesus%22_2.jpg',
+    username: '𒐫',
+    attachments: [
+      {
+        id: 0,
+        description: 'ai slop',
+        filename,
+      },
+    ],
+  };
+
+  formData.append('payload_json', JSON.stringify(payload));
+
+  formData.append('files[0]', imageBlob, filename);
+
+  const response = await fetch(webhookUrl, {
+    method: 'post',
+    body: formData,
+  });
+
+  return response;
+}

main.ts

@@ -5,11 +5,13 @@ import { randomIntegerBetween } from '@std/random';
 import * as hub from '@huggingface/hub';
 import { generateImage } from './ai.ts';
 import { getDirectoryStructureString, getFilesInDirectory, randomString } from './utils.ts';
+import { executeWebhook } from './discord.ts';
 
 const HUGGINGFACE_ACCESS_TOKEN = Deno.env.get('HUGGINGFACE_ACCESS_TOKEN')!;
 assertExists(HUGGINGFACE_ACCESS_TOKEN);
 const REPO_ID = Deno.env.get('REPO_ID')!;
 assertExists(REPO_ID);
+const WEBHOOK_URL = Deno.env.get('WEBHOOK_URL');
 
 ensureDirSync('./images');
 
@@ -47,7 +49,13 @@ async function main() {
       num_inference_steps: 4,
     },
   });
-  Deno.writeFileSync(`./images/${Date.now()}.jpg`, new Uint8Array(imgArrayBuffer), { create: true });
+  const now = Date.now();
+  Deno.writeFileSync(`./images/${now}.jpg`, new Uint8Array(imgArrayBuffer), { create: true });
+  try {
+    WEBHOOK_URL && await executeWebhook(WEBHOOK_URL, new Blob([imgArrayBuffer]), `${now}.jpg`);
+  } catch (e) {
+    console.error(e);
+  }
 }
 
 while (true) {