chore: add ARXIGNIS_API_URL to configuration, remove test-buffer.js, and enhance error handling in logging and metrics functions

Author: pigri

original_wrangler.jsonc

@@ -44,6 +44,7 @@
     }
   ],
 	"vars": {
+		"ARXIGNIS_API_URL": "https://api.arxignis.com/v1",
 		"ARXIGNIS_API_KEY": "<ARXIGNIS_API_KEY>",
 		"TURNSTILE_SITE_KEY": "<TURNSTILE_SITE_KEY>",
 		"TURNSTILE_SECRET_KEY": "<TURNSTILE_SECRET_KEY>",

src/index.ts

@@ -56,20 +56,38 @@ const handler = {
 
 			switch (decision) {
 				case 'block':
+					try {
+					metrics(request, env, {
+						decision,
+						cached: decisionCached || false,
+						ruleId: remediationResult.ruleId || '',
+					});
 					if (env.MODE != 'block') {
 						return fetch(request);
 					}
-					metrics(request, env, { decision, cached: decisionCached || false, score: remediationResult.score || 0 });
 					console.log(JSON.stringify({ ipAddress: clientIP, remediation: decision, remediationCached: decisionCached || false }));
-					const assetResponse = await env.ASSETS.fetch(new URL('block.html', request.url));
-					return assetResponse;
+						const assetResponse = await env.ASSETS.fetch(new URL('block.html', request.url));
+						return assetResponse;
+					} catch (error) {
+						console.warn('Something went wrong with the block:', error);
+						return fetch(request);
+					}
 				case 'captcha':
+					try {
+					metrics(request, env, {
+						decision,
+						cached: decisionCached || false,
+						ruleId: remediationResult.ruleId || '',
+					});
 					if (env.MODE != 'block') {
 						return fetch(request);
 					}
-					metrics(request, env, { decision, cached: decisionCached || false, score: remediationResult.score || 0 });
 					console.log(JSON.stringify({ ipAddress: clientIP, remediation: decision, remediationCached: decisionCached || false }));
 					return captcha(request, env);
+					} catch (error) {
+						console.warn('Something went wrong with the captcha:', error);
+						return fetch(request);
+					}
 				default:
 					log(request, env);
 					console.log(JSON.stringify({ ipAddress: clientIP, remediation: decision, remediationCached: decisionCached || false }));
@@ -89,7 +107,7 @@ export const config: ResolveConfigFn = (env: Env, _trigger) => {
 
 	try {
 		// More explicit check for enabled metrics
-		if (env.PERFORMANCE_METRICS === 'true' && env.PROMETHEUS_URL) {
+		if (env.PERFORMANCE_METRICS && env.PROMETHEUS_URL) {
 			console.log('Enabling OpenTelemetry export to:', env.PROMETHEUS_URL);
 			return {
         exporter: {

src/lib/log.ts

@@ -3,6 +3,7 @@ import { LogTemplate } from "./types";
 import version from "./version";
 
 export function log(request: Request, env: Env) {
+	try {
 	// Extract request body if present
 	let requestBody: object | string | null = null;
 	if (request.body) {
@@ -46,6 +47,9 @@ export function log(request: Request, env: Env) {
 
 	// Add to Durable Object buffer instead of immediate API call
 	addToLogBuffer(env, logTemplate);
+	} catch (error) {
+		console.warn('Failed to send log:', error);
+	}
 }
 
 

src/lib/metrics.ts

@@ -2,14 +2,18 @@ import { addToMetricsBuffer } from "./durable-buffer";
 import { RemediationResult, MetricsTemplate } from "./types";
 
 export function metrics(request: Request, env: Env, result: RemediationResult): void {
+	try {
 	const metricsTemplate: MetricsTemplate = {
 		timestamp: new Date().toISOString(),
 		clientIp: request.headers.get('CF-Connecting-IP') || 'unknown',
 		hostName: new URL(request.url).hostname,
 		remediation: result.decision,
 		cached: result.cached,
-		score: result.score,
+		ruleId: result.ruleId,
 	};
 
 	addToMetricsBuffer(env, metricsTemplate);
+	} catch (error) {
+		console.warn('Failed to send metrics:', error);
+	}
 }