feat: proxy

Author: LouisHaftmann

.env

@@ -1,5 +1,7 @@
 API_BASE_URL=http://localhost:3000
 DEBUG=true
+CA_KEY_PATH=./certs/key.pem
+CA_CERT_PATH=./certs/cert.pem
 
 # filesystem
 STORAGE_DRIVER=filesystem

.gitignore

@@ -10,4 +10,6 @@ data/
 .data
 .idea
 .DS_Store
-tests/temp/
+tests/temp/
+certs/*
+!certs/.gitkeep

certs/.gitkeep

@@ -1,6 +1,7 @@
 import { z } from 'zod'
 
 const booleanSchema = z.string().transform((v) => v.toLowerCase() === 'true')
+const portSchema = z.coerce.number().int().min(1).max(65_535)
 
 const envSchema = z.object({
   ENABLE_DIRECT_DOWNLOADS: booleanSchema.default('false'),
@@ -9,6 +10,10 @@ const envSchema = z.object({
   STORAGE_DRIVER: z.string().toLowerCase().default('filesystem'),
   DB_DRIVER: z.string().toLowerCase().default('sqlite'),
   DEBUG: booleanSchema.default('false'),
+  PROXY_PORT: portSchema.default(8000),
+  NITRO_PORT: portSchema.default(3000),
+  CA_KEY_PATH: z.string(),
+  CA_CERT_PATH: z.string(),
   TEMP_DIR: z.string().default(tmpdir()),
 })
 

lib/env.ts

@@ -0,0 +1,32 @@
+import mockttp from 'mockttp'
+import { ENV } from './env'
+import { logger } from './logger'
+
+export async function initializeProxy() {
+  const port = ENV.PROXY_PORT
+  logger.info(`Starting proxy server on port ${port}...`)
+
+  // generate ca with `openssl req -x509 -newkey rsa:4096 -keyout key.pem -out cert.pem -sha256 -days 365`
+  const server = mockttp.getLocal({
+    https: {
+      keyPath: ENV.CA_KEY_PATH,
+      certPath: ENV.CA_CERT_PATH,
+      tlsPassthrough: [
+        {
+          hostname: 'broker.actions.githubusercontent.com',
+        },
+      ],
+    },
+  })
+
+  await server
+    .forAnyRequest()
+    .withUrlMatching(
+      /(twirp\/github\.actions\.results\.api\.v1\.CacheService)|(_apis\/artifactcache)/,
+    )
+    .thenForwardTo(`http://localhost:${ENV.NITRO_PORT}`)
+
+  await server.forUnmatchedRequest().thenPassThrough()
+
+  await server.start(port)
+}

lib/proxy.ts

@@ -35,6 +35,7 @@
     "execa": "^9.5.2",
     "h3": "^1.14.0",
     "kysely": "^0.27.5",
+    "mockttp": "^3.16.0",
     "mysql2": "^3.12.0",
     "nitropack": "^2.10.4",
     "pg": "^8.13.1",

package.json

@@ -3,12 +3,14 @@ import { H3Error } from 'h3'
 import { initializeDatabase, useDB } from '~/lib/db'
 import { ENV } from '~/lib/env'
 import { logger } from '~/lib/logger'
+import { initializeProxy } from '~/lib/proxy'
 import { initializeStorage, useStorageAdapter } from '~/lib/storage'
 
 export default defineNitroPlugin(async (nitro) => {
   const version = useRuntimeConfig().version
   logger.info(`🚀 Starting GitHub Actions Cache Server (${version})`)
 
+  await initializeProxy()
   await initializeDatabase()
   await initializeStorage()