// This approach injects a notification indicator with a counter into the top toolbar // of the Hacker News proxy for logged-in users. It modifies the HTML response // to include the notification count in an orange box with rounded corners. import { extractValInfo } from "https://esm.town/v/pomdtr/extractValInfo?v=29"; import { sqlite } from "https://esm.town/v/stevekrouse/sqlite"; const { httpEndpoint, name } = extractValInfo(import.meta.url); // Initialize the SQLite table await sqlite.execute(` CREATE TABLE IF NOT EXISTS user_cookie_${name} ( id INTEGER PRIMARY KEY CHECK (id = 1), cookie_value TEXT ) `); function createErrorPage(title: string, message: string): string { return ` ${title}

HN Proxy

${title}

${message}

`; } const checkUser = async (req: Request): Promise => { // Extract the "user" cookie value const userCookie = req.headers.get("cookie")?.split(";") .find(c => c.trim().startsWith("user="))?.split("=")[1]; // Check if a user cookie is already stored const storedCookie = await sqlite.execute( `SELECT cookie_value FROM user_cookie_${name} WHERE id = 1`, ); if (!userCookie) { const errorPage = createErrorPage( "Authentication Error", "No user cookie provided. Please log in to access this page.", ); return new Response(errorPage, { status: 401, headers: { "Content-Type": "text/html" } }); } if (storedCookie.rows.length === 0) { // No stored cookie, save the current one await sqlite.execute( `INSERT INTO user_cookie_${name} (id, cookie_value) VALUES (1, ?)`, [userCookie], ); } else if (userCookie !== storedCookie.rows[0].cookie_value) { if (storedCookie.rows[0].cookie_value.split("&")[0] === userCookie.split("&")[0]) { // Same user, different cookie value, update it await sqlite.execute( `UPDATE user_cookie_${name} SET cookie_value = ? WHERE id = 1`, [userCookie], ); } else { // Different user, return an error const errorPage = createErrorPage( "Access Denied", "You are not authorized to access this page. Only one user is allowed to use this proxy.", ); return new Response(errorPage, { status: 403, headers: { "Content-Type": "text/html" } }); } } return userCookie; }; function injectNotificationIndicator(html: string): string { const notificationCount = 21; // Hardcoded for now const notificationIndicator = ` ${notificationCount} `; return html.replace( /( $.*?<\/span>$) \|/, `$1 |${notificationIndicator}|` ); } export default async function(req: Request): Promise { const url = new URL(req.url); let userCookie: string | undefined; if (url.pathname !== "/login" && !url.pathname.endsWith(".svg")) { let uc = await checkUser(req); if (typeof uc !== "string") return uc; userCookie = uc; } // Proceed with the proxy request let resp = await fetch("https://news.ycombinator.com" + url.pathname + url.search, { headers: req.headers, method: req.method, body: req.body, redirect: "manual", }); let text = await resp.text(); text = text.replace("https://news.ycombinator.com", httpEndpoint); // Inject notification indicator for logged-in users if (userCookie && text.includes('