add logging

Author: qnton

index.ts

@@ -1,78 +1,95 @@
-import { Router, IRequest } from 'itty-router';
+import { Router, IRequest } from "itty-router";
 
 // Constants
-const API_ENDPOINT = 'https://api.mailchannels.net/tx/v1/send';
-const CONTENT_TYPE_JSON = 'application/json';
+const API_ENDPOINT = "https://api.mailchannels.net/tx/v1/send";
+const CONTENT_TYPE_JSON = "application/json";
 
 // Helper function to create a response
 const createResponse = (body: any, status: number) =>
-new Response(JSON.stringify(body), {
-headers: { 'content-type': CONTENT_TYPE_JSON },
-status,
-});
+ new Response(JSON.stringify(body), {
+ headers: { "content-type": CONTENT_TYPE_JSON },
+ status,
+ });
 
 // Curry function for handling errors
 const handleError = (status: number) => (message: string, request: IRequest) =>
-createResponse({ error: message }, status);
+ createResponse({ error: message }, status);
 
 // Middleware for authentication
 const authMiddleware = (token: string | null) => (request: IRequest) => {
-const requestToken = request.headers.get('Authorization');
+ const requestToken = request.headers.get("Authorization");
 
-const unauthorizedError = handleError(401)('Unauthorized', request);
-const missingTokenError = handleError(401)('You must set the TOKEN environment variable.', request);
+ const unauthorizedError = handleError(401)("Unauthorized", request);
+ const missingTokenError = handleError(401)(
+ "You must set the TOKEN environment variable.",
+ request
+ );
 
-return token
-? requestToken === token
-? null
-: unauthorizedError
-: missingTokenError;
+ return token
+ ? requestToken === token
+ ? null
+ : unauthorizedError
+ : missingTokenError;
 };
 
 // Service function to send email
 const sendEmail = async (email: any) => {
-const resp = await fetch(API_ENDPOINT, {
-method: 'POST',
-headers: {
-'content-type': CONTENT_TYPE_JSON,
-},
-body: JSON.stringify(email),
-});
-
-if (!resp.ok) {
-const errorMessage = `Error sending email: ${resp.status} ${resp.statusText}`;
-throw new Error(`${errorMessage}\n${await resp.text()}`);
-}
+ const resp = await fetch(API_ENDPOINT, {
+ method: "POST",
+ headers: {
+ "content-type": CONTENT_TYPE_JSON,
+ },
+ body: JSON.stringify(email),
+ });
+
+ if (!resp.ok) {
+ const errorMessage = `Error sending email: ${resp.status} ${resp.statusText}`;
+ throw new Error(`${errorMessage}\n${await resp.text()}`);
+ }
+};
+
+// Log batch of emails
+const logBatch = async (emails: any[], env: Env) => {
+ if (emails.length > 0) {
+ await Promise.all(
+ emails.map((email) => env.log.put(`${Date.now()}`, JSON.stringify(email)))
+ );
+ }
 };
 
 // Route handler for sending email
 const sendEmailHandler = async (request: IRequest, env: Env) => {
-try {
-const authError = authMiddleware(env.TOKEN)(request);
-if (authError) return authError;
-
-const email = await request.json();
-await sendEmail(email);
-
-return createResponse({
-status: 'SUCCESS',
-statusCode: 1000,
-message: 'NA',
-}, 200);
-} catch (error) {
-console.error(`Error processing request: ${error}`);
-return handleError(500)('Internal Server Error', request);
-}
+ try {
+ const authError = authMiddleware(env.TOKEN)(request);
+ if (authError) return authError;
+
+ const email = await request.json();
+ await sendEmail(email);
+
+ logBatch([email], env);
+
+ return createResponse(
+ {
+ status: "SUCCESS",
+ statusCode: 1000,
+ message: "NA",
+ },
+ 200
+ );
+ } catch (error) {
+ console.error(`Error processing request: ${error}`);
+ return handleError(500)("Internal Server Error", request);
+ }
 };
 
 // Create router
 const router = Router();
 
 // Define routes
-router.post('/api/email', sendEmailHandler);
-router.all('*', (request: IRequest) => handleError(404)('Not Found', request));
+router.post("/api/email", sendEmailHandler);
+router.all("*", (request: IRequest) => handleError(404)("Not Found", request));
 
 // Export the router handler
 export default {
-fetch: router.handle,
+ fetch: router.handle,
 };

package.json

@@ -1,10 +1,10 @@
 {
- "name": "email",
+ "name": "email-proxy",
  "version": "0.0.0",
  "private": true,
  "scripts": {
  "deploy": "wrangler publish",
- "start": "wrangler dev"
+ "dev": "wrangler dev"
  },
  "devDependencies": {
  "@cloudflare/workers-types": "^4.20230419.0",
@@ -13,7 +13,6 @@
  "typescript": "^5.0.4"
  },
  "dependencies": {
- "wrangler": "^3.1.1",
- "zod": "^3.22.4"
+ "wrangler": "^3.1.1"
  }
 }

worker-configuration.d.ts

@@ -1,3 +1,4 @@
 interface Env {
-TOKEN: string;
+ TOKEN: string;
+ log: KVNamespace;
 }

wrangler.toml

@@ -1,4 +1,7 @@
-name = "email"
+name = "email-proxy"
 main = "index.ts"
 compatibility_date = "2023-05-18"
 
+kv_namespaces = [
+ { binding = "log", id = "cfa4e6ed8c054bf5bcb3cb4d0ab107fb", preview_id = "cec98816e55748fd958c0fec7db67131" },
+]