data/slack-queue.db

This is a binary file and will not be displayed.

+159

features/message-queue.ts

··· 1 1 + import { Database } from "bun:sqlite"; 2 2 + import type { Block, SlackAPIClient } from "slack-edge"; 3 3 + 4 4 + export interface SlackMessage { 5 5 + userId?: string; 6 6 + channelId?: string; 7 7 + blocks?: Block[]; 8 8 + text: string; 9 9 + timestamp?: number; 10 10 + status: "pending" | "sent" | "failed"; 11 11 + } 12 12 + 13 13 + export default class SlackMessageQueue { 14 14 + private db: Database; 15 15 + private slack: SlackAPIClient; 16 16 + private isProcessing = false; 17 17 + private batchSize = 50; 18 18 + 19 19 + constructor(slackClient: SlackAPIClient, dbPath = "slack-queue.db") { 20 20 + this.slack = slackClient; 21 21 + this.db = new Database(dbPath); 22 22 + this.initDatabase(); 23 23 + } 24 24 + 25 25 + private initDatabase() { 26 26 + this.db.run(` 27 27 + CREATE TABLE IF NOT EXISTS messages ( 28 28 + id INTEGER PRIMARY KEY AUTOINCREMENT, 29 29 + userId TEXT, 30 30 + channelId TEXT, 31 31 + blocks TEXT, 32 32 + text TEXT NOT NULL, 33 33 + timestamp INTEGER NOT NULL, 34 34 + status TEXT NOT NULL 35 35 + ) 36 36 + `); 37 37 + this.db.run("CREATE INDEX IF NOT EXISTS idx_status ON messages(status)"); 38 38 + } 39 39 + 40 40 + async enqueue(message: SlackMessage): Promise<void> { 41 41 + const stmt = this.db.prepare(` 42 42 + INSERT INTO messages (userId, channelId, blocks, text, timestamp, status) 43 43 + VALUES (?, ?, ?, ?, ?, ?) 44 44 + `); 45 45 + 46 46 + stmt.run( 47 47 + message.userId ?? null, 48 48 + message.channelId ?? null, 49 49 + JSON.stringify(message.blocks) ?? null, 50 50 + message.text, 51 51 + Date.now(), 52 52 + "pending", 53 53 + ); 54 54 + 55 55 + if (!this.isProcessing) { 56 56 + this.processQueue(); 57 57 + } 58 58 + } 59 59 + 60 60 + private async processQueue() { 61 61 + if (this.isProcessing) return; 62 62 + this.isProcessing = true; 63 63 + 64 64 + console.log("Processing queue"); 65 65 + 66 66 + try { 67 67 + while (true) { 68 68 + const messages = this.db 69 69 + .prepare( 70 70 + ` 71 71 + SELECT * FROM messages 72 72 + WHERE status = 'pending' 73 73 + LIMIT ? 74 74 + `, 75 75 + ) 76 76 + .all(this.batchSize) as (SlackMessage & { id: number })[]; 77 77 + 78 78 + console.log(messages); 79 79 + if (messages.length === 0) break; 80 80 + 81 81 + await Promise.all( 82 82 + messages.map(async (message) => { 83 83 + try { 84 84 + if (message.channelId) { 85 85 + await this.slack.chat.postMessage({ 86 86 + channel: message.channelId, 87 87 + blocks: 88 88 + JSON.parse(message.blocks as unknown as string) ?? 89 89 + undefined, 90 90 + text: message.text, 91 91 + }); 92 92 + 93 93 + console.log(res); 94 94 + } else if (message.userId) { 95 95 + await this.slack.chat.postMessage({ 96 96 + channel: message.userId, 97 97 + blocks: 98 98 + JSON.parse(message.blocks as unknown as string) ?? 99 99 + undefined, 100 100 + text: message.text, 101 101 + }); 102 102 + } 103 103 + 104 104 + console.log("Message sent successfully"); 105 105 + 106 106 + this.db 107 107 + .prepare( 108 108 + ` 109 109 + UPDATE messages 110 110 + SET status = 'sent' 111 111 + WHERE id = ? 112 112 + `, 113 113 + ) 114 114 + .run(message.id); 115 115 + } catch (error) { 116 116 + console.error("Error sending message", error); 117 117 + this.db 118 118 + .prepare( 119 119 + ` 120 120 + UPDATE messages 121 121 + SET status = 'failed' 122 122 + WHERE id = ? 123 123 + `, 124 124 + ) 125 125 + .run(message.id); 126 126 + } 127 127 + }), 128 128 + ); 129 129 + } 130 130 + } finally { 131 131 + this.isProcessing = false; 132 132 + } 133 133 + } 134 134 + 135 135 + async cleanup(olderThan: number = 7 * 24 * 60 * 60 * 1000) { 136 136 + const cutoff = Date.now() - olderThan; 137 137 + this.db 138 138 + .prepare( 139 139 + ` 140 140 + DELETE FROM messages 141 141 + WHERE timestamp < ? AND status != 'pending' 142 142 + `, 143 143 + ) 144 144 + .run(cutoff); 145 145 + } 146 146 + 147 147 + async queueLength() { 148 148 + const result = this.db 149 149 + .prepare( 150 150 + ` 151 151 + SELECT COUNT(*) as count 152 152 + FROM messages 153 153 + WHERE status = 'pending' 154 154 + `, 155 155 + ) 156 156 + .get(); 157 157 + return (result as { count: number }).count; 158 158 + } 159 159 + }

+34

index.ts

··· 1 1 import { SlackApp } from "slack-edge"; 2 2 + import SlackMessageQueue, { type SlackMessage } from "./features/message-queue"; 2 3 3 4 import * as features from "./features/index"; 4 5 ··· 33 34 }); 34 35 const slackClient = slackApp.client; 35 36 37 37 + const messageQueue = new SlackMessageQueue(slackClient, "data/slack-queue.db"); 38 38 + console.log(`👔 Message Queue Size: ${await messageQueue.queueLength()}`); 39 39 + 36 40 console.log(`⚒️ Loading ${Object.entries(features).length} features...`); 37 41 for (const [feature, handler] of Object.entries(features)) { 38 42 console.log(`📦 ${feature} loaded`); ··· 54 58 return new Response("OK"); 55 59 case "/slack": 56 60 return slackApp.run(request); 61 61 + case "/slack/message": { 62 62 + if ( 63 63 + request.headers.get("Authorization") !== 64 64 + `Bearer ${process.env.API_KEY}` 65 65 + ) { 66 66 + return new Response("Unauthorized", { status: 401 }); 67 67 + } 68 68 + 69 69 + const message: SlackMessage = await request.json(); 70 70 + const { userId, channelId, text } = message; 71 71 + 72 72 + if ((!userId && !channelId) || (userId && channelId) || !text) { 73 73 + return new Response( 74 74 + `Invalid fields: ${[ 75 75 + !userId && 76 76 + !channelId && 77 77 + "must provide either userId or channelId", 78 78 + userId && channelId && "cannot provide both userId and channelId", 79 79 + !text && "text is required", 80 80 + ] 81 81 + .filter(Boolean) 82 82 + .join(", ")}`, 83 83 + { status: 400 }, 84 84 + ); 85 85 + } 86 86 + 87 87 + await messageQueue.enqueue(message); 88 88 + 89 89 + return new Response(JSON.stringify({ ok: true }), { status: 200 }); 90 90 + } 57 91 default: 58 92 return new Response("404 Not Found", { status: 404 }); 59 93 }