feat: add address remapping

ethersphere · Nov 20, 2023 · 95b34a5 · 95b34a5
1 parent 128f0a7
commit 95b34a5
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/src/bzz-link.ts b/src/bzz-link.ts
@@ -3,7 +3,15 @@ import { logger } from './logger'
 
 export class NotEnabledError extends Error {}
 
-export function subdomainToBzz(subdomain: string, isCidEnabled: boolean, isEnsEnabled: boolean): string {
+export function subdomainToBzz(
+  subdomain: string,
+  isCidEnabled: boolean,
+  isEnsEnabled: boolean,
+  remap: Record<string, string>,
+): string {
+  if (subdomain in remap) {
+    return remap[subdomain]
+  }
   try {
     const result = swarmCid.decodeCid(subdomain)
 

diff --git a/src/proxy.ts b/src/proxy.ts
@@ -21,6 +21,7 @@ interface Options {
   hostname?: string
   cidSubdomains?: boolean
   ensSubdomains?: boolean
+  remap: Record<string, string>
 }
 
 export function createProxyEndpoints(app: Application, options: Options) {
@@ -32,11 +33,13 @@ export function createProxyEndpoints(app: Application, options: Options) {
 
       return
     }
+
     try {
       const newUrl = subdomainToBzz(
         subdomain.slice(0, -1), // remove trailing dot
         options.cidSubdomains ?? false,
         options.ensSubdomains ?? false,
+        options.remap,
       )
       await fetchAndRespond(
         'GET',

diff --git a/src/server.ts b/src/server.ts
@@ -1,5 +1,6 @@
 import { Bee, BeeDebug } from '@ethersphere/bee-js'
 import bodyParser from 'body-parser'
+import { Arrays } from 'cafe-utility'
 import express, { Application } from 'express'
 import { AppConfig, DEFAULT_HOSTNAME } from './config'
 import { HASHED_IDENTITY_HEADER, fetchBeeIdentity, getHashedIdentity } from './identity'
@@ -113,6 +114,9 @@ export const createApp = (
     cidSubdomains,
     ensSubdomains,
     hostname,
+    remap: Object.fromEntries(
+      (Arrays.getArgument(process.argv, 'remap', process.env as any, 'REMAP') || '').split(';').map(x => x.split('=')),
+    ),
   })
 
   app.use(express.static('public'))