UPLOAD_FROM_SOURCE setting

init-config.js

@@ -16,7 +16,9 @@ const envVariables = [
   { name: 'LIVE_SOURCE_URL', description: 'The live source URL (e.g., https://cdn.example.com/path/)' },
   { name: 'LIVE_PUBLIC_DOMAIN', description: 'The live public domain (e.g., https://cf.example.com)' },
   { name: 'KV_NAMESPACE_ID', description: 'Your KV namespace ID' },
-  { name: 'CACHE_KEY_PREFIX', description: 'Cache key prefix' }
+  { name: 'CACHE_KEY_PREFIX', description: 'Cache key prefix' },
+  { name: 'UPLOAD_FROM_SOURCE', description: 'Upload from source (true/false)' },
+  { name: 'RATELIMIT_ENABLED', description: 'Rate limiting enabled (true/false)' }  
 ];
 
 const envValues = {};
@@ -69,7 +71,8 @@ function updateWranglerToml() {
     '<cloudflare-api-token>': envValues.API_TOKEN,
     '<account_id>': envValues.ACCOUNT_ID,
     '<account_hash>': envValues.ACCOUNT_HASH,
-    '<rate_limit>': process.env.RATELIMIT_ENABLED === 'true' ? 'true' : 'false',
+    '<rate_limit>': envValues.RATELIMIT_ENABLED === 'true' ? 'true' : 'false',
+    '<upload_from_source>': envValues.UPLOAD_FROM_SOURCE === 'true' ? 'true' : 'false',
     'https://cdn.example.com/path/': envValues.LIVE_SOURCE_URL,
     'https://cf.example.com': envValues.LIVE_PUBLIC_DOMAIN
   };

package.json

@@ -6,7 +6,7 @@
   "scripts": {
     "init": "node init-config.js",
     "update-config": "node update-config.js",
-    "build": "npm run update-config && wrangler build",
+    "build": "npm run update-config && wrangler deploy --dry-run --outdir=dist",
     "deploy": "npm run build && wrangler deploy",
     "test": "jest",
     "test:watch": "jest --watch",

src/index.ts

@@ -44,10 +44,9 @@ async function fetchOriginalCloudflareImage(env: Env, id: string): Promise<Respo
     headers: { 'Authorization': `Bearer ${env.API_TOKEN}` },
   });
 
-  // TODO: Handle 404s when original source url is not active
-  // if (!response.ok) {
-  //   throw new CloudflareApiError('Failed to fetch original image', response.status, await response.text());
-  // }
+  if (!response.ok && env.UPLOAD_FROM_SOURCE === false) {
+    throw new CloudflareApiError('Failed to fetch original image', response.status, await response.text());
+  }
 
   return response;
 }
@@ -137,7 +136,7 @@ export async function handleImageRequest(request: Request, env: Env): Promise<Re
       imageResponse = await fetchOriginalCloudflareImage(env, id);
     } 
 
-    if (!imageResponse || !imageResponse.ok) {
+    if (env.UPLOAD_FROM_SOURCE && (!imageResponse || !imageResponse.ok)) {
       const sourceUrl = `${env.LIVE_SOURCE_URL}/${image_path}.${extension}`;
       const uploadResponse = await uploadToCloudflareImages(env, sourceUrl, id);
 

src/types.ts

@@ -9,8 +9,9 @@ export interface Env {
   LIVE_SOURCE_URL: string;
   LIVE_PUBLIC_DOMAIN: string;
   KV_STORE: KVNamespace;
-  RATELIMIT_ENABLED: string;
+  RATELIMIT_ENABLED: boolean;
   CACHE_KEY_PREFIX: string;
+  UPLOAD_FROM_SOURCE: boolean;
 }
 
 /**

update-config.js

@@ -15,7 +15,8 @@ const replacements = {
   '<bucket_name>': process.env.BUCKET,
   '<cachekey_prefix>': process.env.CACHE_KEY_PREFIX,
   '<kv_namespace_id>': process.env.KV_NAMESPACE_ID,
-  '<rate_limit>': process.env.RATELIMIT_ENABLED === 'true' ? 'true' : 'false'
+  '<rate_limit>': process.env.RATELIMIT_ENABLED === 'true' ? 'true' : 'false',
+  '<upload_from_source>': process.env.UPLOAD_FROM_SOURCE === 'true' ? 'true' : 'false'
 };
 
 for (const [placeholder, envValue] of Object.entries(replacements)) {

wrangler.toml.tpl

@@ -13,7 +13,8 @@ ACCOUNT_HASH = "<account_hash>"
 LIVE_SOURCE_URL = "https://cdn.example.com/path/"
 LIVE_PUBLIC_DOMAIN = "https://cf.example.com"
 CACHE_KEY_PREFIX = "<cachekey_prefix>"
-RATELIMIT_ENABLED = "<ratelimit>"
+RATELIMIT_ENABLED = <rate_limit>
+UPLOAD_FROM_SOURCE = <upload_from_source>
 
 [[ r2_buckets ]]
 binding = "R2_IMAGES_BUCKET"