diff --git a/examples/dexa-mfm/src/index.ts b/examples/dexa-mfm/src/index.ts index c756125..f490b4d 100644 --- a/examples/dexa-mfm/src/index.ts +++ b/examples/dexa-mfm/src/index.ts @@ -6,6 +6,7 @@ import pkg from '../package.json' export interface Env { DEXA_API_BASE_URL: string + ENVIRONMENT: string } const router = OpenAPIRouter({ diff --git a/examples/dexa-mfm/src/routes.ts b/examples/dexa-mfm/src/routes.ts index 1f0ed9e..0d554d1 100644 --- a/examples/dexa-mfm/src/routes.ts +++ b/examples/dexa-mfm/src/routes.ts @@ -66,7 +66,7 @@ export class DexaSearch extends OpenAPIRoute { return new Response('invalid source IP', { status: 500 }) } - if (!isValidChatGPTIPAddress(ip)) { + if (env.environment === 'production' && !isValidChatGPTIPAddress(ip)) { // console.warn('search error invalid IP address', ip) return new Response(`Forbidden`, { status: 403 }) } diff --git a/examples/dexa-mfm/wrangler.example.toml b/examples/dexa-mfm/wrangler.example.toml index 67eaf9f..9077438 100644 --- a/examples/dexa-mfm/wrangler.example.toml +++ b/examples/dexa-mfm/wrangler.example.toml @@ -3,4 +3,10 @@ main = "src/index.ts" compatibility_date = "2023-04-04" [vars] -DEXA_API_BASE_URL="TODO" \ No newline at end of file +vars = { DEXA_API_BASE_URL="TODO", ENVIRONMENT="dev" } + +[env.staging] +vars = { DEXA_API_BASE_URL="TODO", ENVIRONMENT = "staging" } + +[env.production] +vars = { DEXA_API_BASE_URL="TODO", ENVIRONMENT = "production" }