feat: update main entry point to src/worker.ts and add UPSTREAM_ORIGIN variable

Author: pphatdev

src/worker.ts

@@ -0,0 +1,47 @@
+type WorkerEnv = {
+    UPSTREAM_ORIGIN?: string;
+};
+
+function jsonResponse(body: unknown, status: number): Response {
+    return new Response(JSON.stringify(body, null, 2), {
+        status,
+        headers: {
+            'content-type': 'application/json; charset=utf-8',
+            'cache-control': 'no-store',
+        },
+    });
+}
+
+export default {
+    async fetch(request: Request, env: WorkerEnv): Promise<Response> {
+        const configuredOrigin = (env.UPSTREAM_ORIGIN || '').trim();
+
+        if (!configuredOrigin) {
+            return jsonResponse(
+                {
+                    error: 'UPSTREAM_ORIGIN is not configured',
+                    message:
+                        'Set UPSTREAM_ORIGIN in your Wrangler env vars to the Node API origin, for example https://api.example.com',
+                },
+                500,
+            );
+        }
+
+        const incomingUrl = new URL(request.url);
+        const upstreamBase = new URL(configuredOrigin.endsWith('/') ? configuredOrigin : `${configuredOrigin}/`);
+        const upstreamUrl = new URL(`${incomingUrl.pathname}${incomingUrl.search}`, upstreamBase);
+
+        const headers = new Headers(request.headers);
+        headers.set('x-forwarded-host', incomingUrl.host);
+        headers.delete('host');
+
+        const init: RequestInit = {
+            method: request.method,
+            headers,
+            redirect: 'manual',
+            body: request.method === 'GET' || request.method === 'HEAD' ? undefined : request.body,
+        };
+
+        return fetch(upstreamUrl.toString(), init);
+    },
+};

wrangler.toml

@@ -1,5 +1,5 @@
 name = "github-stats"
-main = "dist/index.js"
+main = "src/worker.ts"
 compatibility_date = "2024-12-16"
 compatibility_flags = ["nodejs_compat"]
 
@@ -12,6 +12,7 @@ cwd = "."
 [vars]
 ENVIRONMENT = "cloudflare"
 PORT = "3000"
+UPSTREAM_ORIGIN = ""
 
 # D1 Database Binding - SQLite for Cloudflare
 [[d1_databases]]
@@ -32,12 +33,13 @@ preview_id = "YOUR_BADGE_CACHE_PREVIEW_NAMESPACE_ID"
 # Development environment
 [env.development]
 name = "github-stats-dev"
-main = "dist/index.js"
+main = "src/worker.ts"
 compatibility_date = "2024-12-16"
 
 [env.development.vars]
 ENVIRONMENT = "cloudflare-dev"
 DEBUG = "true"
+UPSTREAM_ORIGIN = ""
 
 [[env.development.d1_databases]]
 binding = "DB"
@@ -57,12 +59,13 @@ preview_id = "YOUR_DEV_BADGE_CACHE_PREVIEW_NAMESPACE_ID"
 # Production environment
 [env.production]
 name = "github-stats-prod"
-main = "dist/index.js"
+main = "src/worker.ts"
 compatibility_date = "2024-12-16"
 
 [env.production.vars]
 ENVIRONMENT = "cloudflare-production"
 DEBUG = "false"
+UPSTREAM_ORIGIN = ""
 
 [[env.production.d1_databases]]
 binding = "DB"