formidable: Add support for site-specific filesize

.gitignore

@@ -60,9 +60,6 @@ typings/
 # Local development config file
 config/*.json
 
-# Public folder should be ignored
-public/*
-
 # lol
 package-lock.json
 

api/media/formidable.mjs

@@ -39,6 +39,12 @@ export function uploadFile(ctx, siteName, noprefix = false) {
     form.uploadDir = `${config.get('uploadFolder')}/${siteName}`
     form.maxFileSize = config.get('fileSize')
 
+    let siteSize = config.get(`sites:${siteName}:fileSize`)
+
+    if (siteSize && typeof(siteSize) === 'number') {
+      form.maxFileSize = siteSize
+    }
+
     form.parse(ctx.req, function(err, fields, files) {
       if (err) return rej(err)
       if (!files || !files.file) return rej(new HttpError(422, 'File in body was missing'))

api/media/routes.mjs

@@ -2,7 +2,7 @@ import path from 'path'
 import sharp from 'sharp'
 import fs from 'fs/promises'
 import config from '../config.mjs'
-import { HttpError } from 'flaska'
+import { HttpError, CorsHandler } from 'flaska'
 import * as security from './security.mjs'
 import * as formidable from './formidable.mjs'
 
@@ -16,6 +16,7 @@ export default class MediaRoutes {
     })
     this.siteCache = new Map()
     this.collator = new Intl.Collator('is-IS', { numeric: false, sensitivity: 'accent' })
+    this.cors = CorsHandler({ allowedOrigins: ['*'] })
   }
 
   register(server) {
@@ -30,6 +31,7 @@ export default class MediaRoutes {
     server.flaska.post('/media/resize', [server.queryHandler()], this.resize.bind(this))
     server.flaska.post('/media/resize/:filename', [server.queryHandler(), server.jsonHandler()], this.resizeExisting.bind(this))
     server.flaska.delete('/media/:filename', [server.queryHandler()], this.remove.bind(this))
+    server.flaska.options('/::path', [server.queryHandler(), this.security.verifyCorsEnabled], this.cors)
   }
 
   init(server) {
@@ -89,20 +91,32 @@ export default class MediaRoutes {
     }
   }
 
+  getSite(ctx) {
+    let site = this.security.verifyToken(ctx)
+    if (this.security.hasCors(site)) {
+      this.cors(ctx)
+    }
+    return site
+  }
+
   async listFiles(ctx) {
-    let site = await this.security.verifyToken(ctx)
+    let site = this.getSite(ctx)
 
     ctx.body = this.filesCacheGet(site)
   }
   
   async listPublicFiles(ctx) {
     this.security.throwIfNotPublic(ctx.params.site)
+    
+    if (this.security.hasCors(ctx.params.site)) {
+      this.cors(ctx)
+    }
 
     ctx.body = this.filesCacheGet(ctx.params.site)
   }
 
   async upload(ctx, noprefix = false) {
-    let site = await this.security.verifyToken(ctx)
+    let site = this.getSite(ctx)
     ctx.state.site = site
 
     let result = await this.formidable.uploadFile(ctx, ctx.state.site, noprefix)
@@ -181,7 +195,7 @@ export default class MediaRoutes {
   }
 
   async resizeExisting(ctx) {
-    let site = await this.security.verifyToken(ctx)
+    let site = this.getSite(ctx)
     ctx.state.site = site
     
     ctx.body = {}
@@ -195,7 +209,7 @@ export default class MediaRoutes {
   }
 
   async remove(ctx) {
-    let site = await this.security.verifyToken(ctx)
+    let site = this.getSite(ctx)
 
     this.filesCacheRemove(site, ctx.params.filename)
 

api/media/security.mjs

@@ -25,6 +25,24 @@ export function verifyToken(ctx) {
   }
 }
 
+export function hasCors(site) {
+  let sites = config.get('sites')
+  return sites[site]?.cors === true
+}
+
+export function verifyCorsEnabled(ctx) {
+  let site
+  try {
+    site = verifyToken(ctx)
+  } catch (err) {
+    throw new HttpError(404)
+  }
+
+  if (!hasCors(site)) {
+    throw new HttpError(404)
+  }
+}
+
 export function throwIfNotPublic(site) {
   let sites = config.get('sites')
   if (!sites[site] || sites[site].public !== true) {

appveyor.yml

@@ -66,7 +66,7 @@ on_success:
           https://git.nfp.is/api/v1/repos/$APPVEYOR_REPO_NAME/releases/$RELEASE_ID/assets
 
         echo "Deplying to production"
-        curl -X POST http://192.168.93.50:4010/update/storageupload
+        curl -X POST http://192.168.93.51:4010/update/storageupload
       fi
 
 # on build failure

package.json

@@ -1,6 +1,6 @@
 {
   "name": "storage-upload",
-  "version": "2.2.7",
+  "version": "2.2.12",
   "description": "Micro service for uploading and image resizing files to a storage server.",
   "main": "index.js",
   "scripts": {
@@ -42,7 +42,7 @@
   },
   "homepage": "https://github.com/nfp-projects/storage-upload#readme",
   "dependencies": {
-    "flaska": "^1.2.3",
+    "flaska": "^1.3.5",
     "formidable": "^1.2.2",
     "nconf-lite": "^2.0.0",
     "sharp": "^0.30.3"

test/helper.client.mjs

@@ -26,6 +26,9 @@ Client.prototype.customRequest = function(method = 'GET', path, body, options) {
       host: urlObj.hostname,
       port: Number(urlObj.port),
       path: urlObj.pathname + urlObj.search,
+      headers: {
+        origin: 'http://localhost'
+      }
     }))
 
     const req = http.request(opts)
@@ -44,7 +47,9 @@ Client.prototype.customRequest = function(method = 'GET', path, body, options) {
       })
 
       res.on('end', function () {
-        if (!output) return resolve(null)
+        let headers = opts.includeHeaders ? res.headers : null
+
+        if (!output) return resolve(headers ? { headers } : null)
         try {
           output = JSON.parse(output)
         } catch (e) {
@@ -56,29 +61,29 @@ Client.prototype.customRequest = function(method = 'GET', path, body, options) {
           err.status = output.status
           return reject(err)
         }
-        resolve(output)
+        resolve(headers ? { headers, output } : output)
       })
     })
     req.end()
   })
 }
 
-Client.prototype.get = function(url = '/') {
-  return this.customRequest('GET', url, null)
+Client.prototype.get = function(url = '/', options) {
+  return this.customRequest('GET', url, null, options)
 }
 
-Client.prototype.post = function(url = '/', body = {}) {
+Client.prototype.post = function(url = '/', body = {}, options) {
   let parsed = JSON.stringify(body)
-  return this.customRequest('POST', url, parsed, {
+  return this.customRequest('POST', url, parsed, defaults(options, {
     headers: {
       'Content-Type': 'application/json',
       'Content-Length': parsed.length,
     },
-  })
+  }))
 }
 
-Client.prototype.del = function(url = '/', body = {}) {
-  return this.customRequest('DELETE', url, JSON.stringify(body))
+Client.prototype.del = function(url = '/', body = {}, options) {
+  return this.customRequest('DELETE', url, JSON.stringify(body), options)
 }
 
 const random = (length = 8) => {
@@ -92,7 +97,7 @@ const random = (length = 8) => {
   return str;
 }
 
-Client.prototype.upload = function(url, file, method = 'POST', body = {}) {
+Client.prototype.upload = function(url, file, method = 'POST', body = {}, options) {
   return fs.readFile(file).then(data => {
     const crlf = '\r\n'
       const filename = path.basename(file)
@@ -114,12 +119,12 @@ Client.prototype.upload = function(url, file, method = 'POST', body = {}) {
         Buffer.from(`${crlf}--${boundary}--`),
       ])
 
-      return this.customRequest(method, url, multipartBody, {
+      return this.customRequest(method, url, multipartBody, defaults(options, {
         timeout: 5000,
         headers: {
           'Content-Type': 'multipart/form-data; boundary=' + boundary,
           'Content-Length': multipartBody.length,
         },
-      })
+      }))
   })
 }

test/helper.server.mjs

@@ -35,12 +35,25 @@ export function startServer() {
           "default@HS256": "asdf1234"
         }
       },
+      "development_cors": {
+        "keys": {
+          "default@HS256": "asdf1234"
+        },
+        "cors": true
+      },
       "existing": {
         "public": true,
         "keys": {
           "default@HS256": "asdf1234"
         }
-      }
+      },
+      "existing_cors": {
+        "public": true,
+        "keys": {
+          "default@HS256": "asdf1234"
+        },
+        "cors": true
+      },
     },
   })
 

test/media/api.test.mjs

@@ -15,7 +15,7 @@ function resolve(file) {
 
 const currYear = new Date().getFullYear().toString()
 
-t.describe('Media (API)', () => {
+t.timeout(10000).describe('Media (API)', () => {
   let client
   let secret = 'asdf1234'
   let testFiles = []
@@ -34,7 +34,82 @@ t.describe('Media (API)', () => {
     })
   })
 
-  t.timeout(10000).describe('POST /media', function temp() {
+  t.describe('OPTIONS', function() {
+    t.test('should fail options for every single route', async () => {
+      const testPaths = [
+        '/media',
+        '/media/:site',
+        '/media',
+        '/media/noprefix',
+        '/media/resize',
+        '/media/resize/:filename',
+        '/media/:filename',
+      ]
+
+      for (let path of testPaths) {
+        let err = await assert.isRejected(
+          client.customRequest('OPTIONS', path, null)
+        )
+        assert.strictEqual(err.status, 404)
+      }
+    })
+
+    t.test('should fail options for every single route even with token', async () => {
+      let token = encode(null, { iss: 'development' }, secret)
+      const testPaths = [
+        '/media',
+        '/media/:site',
+        '/media',
+        '/media/noprefix',
+        '/media/resize',
+        '/media/resize/:filename',
+        '/media/:filename',
+      ]
+
+      for (let path of testPaths) {
+        let err = await assert.isRejected(
+          client.customRequest('OPTIONS', path + `?token=${token}`, null)
+        )
+        assert.strictEqual(err.status, 404)
+      }
+    })
+
+    t.test('should work if specified has cors enabled', async () => {
+      const assertOrigin = 'http://localhost:9999'
+      let token = encode(null, { iss: 'development_cors' }, secret)
+      const testPaths = [
+        '/media',
+        '/media/:site',
+        '/media',
+        '/media/noprefix',
+        '/media/resize',
+        '/media/resize/:filename',
+        '/media/:filename',
+      ]
+
+      for (let path of testPaths) {
+        let res = await client.customRequest('OPTIONS', path + `?token=${token}`, null, {
+          includeHeaders: true,
+          headers: { origin: assertOrigin, 'access-control-request-method': 'POST' },
+        })
+        assert.strictEqual(res.headers['access-control-allow-origin'], assertOrigin)
+        assert.match(res.headers['access-control-allow-methods'], /post/i)
+      }
+    })
+  })
+
+  t.describe('POST /media', function temp() {
+    let config
+
+    t.before(async () => {
+      config = (await import('../../api/config.mjs')).default
+    })
+
+    t.afterEach(function() {
+      config.sources[2].store.fileSize = 524288000
+      delete config.sources[1].store.sites.development.fileSize
+    })
+
     t.test('should require authentication', async () => {
       resetLog()
       assert.strictEqual(log.error.callCount, 0)
@@ -88,27 +163,68 @@ t.describe('Media (API)', () => {
       let data = await assert.isFulfilled(
         client.upload(
           `/media?token=${token}`,
-          resolve('test.png')
+          resolve('test.png'),
+          'POST',
+          { },
+          { includeHeaders: true }
         )
       )
 
-      testFiles.push(data.path)
+      testFiles.push(data?.output?.path)
 
-      assert.ok(data)
-      assert.ok(data.filename)
-      assert.ok(data.filename.startsWith(currYear))
-      assert.ok(data.path)
+      assert.ok(data.output)
+      assert.ok(data.output.filename)
+      assert.ok(data.output.filename.startsWith(currYear))
+      assert.ok(data.output.path)
 
       let stats = await Promise.all([
         fs.stat(resolve('test.png')),
-        fs.stat(resolve(`../../public/${data.path}`)),
+        fs.stat(resolve(`../../public/${data.output.path}`)),
       ])
       assert.strictEqual(stats[0].size, stats[1].size)
 
-      let img = await sharp(resolve(`../../public/${data.path}`)).metadata()
+      let img = await sharp(resolve(`../../public/${data.output.path}`)).metadata()
       assert.strictEqual(img.width, 600)
       assert.strictEqual(img.height, 700)
       assert.strictEqual(img.format, 'png')
+      assert.notOk(data.headers['access-control-allow-origin'])
+    })
+
+    t.test('should upload file and create file and return cors if site has cors', async () => {
+      const assertOrigin = 'http://localhost:9000'
+      let token = encode(null, { iss: 'development_cors' }, secret)
+
+      let data = await assert.isFulfilled(
+        client.upload(
+          `/media?token=${token}`,
+          resolve('test.png'),
+          'POST',
+          { },
+          {
+            includeHeaders: true,
+            headers: { origin: assertOrigin, 'access-control-request-method': 'POST' },
+          }
+        )
+      )
+
+      testFiles.push(data.output?.path)
+
+      assert.ok(data.output)
+      assert.ok(data.output.filename)
+      assert.ok(data.output.filename.startsWith(currYear))
+      assert.ok(data.output.path)
+
+      let stats = await Promise.all([
+        fs.stat(resolve('test.png')),
+        fs.stat(resolve(`../../public/${data.output.path}`)),
+      ])
+      assert.strictEqual(stats[0].size, stats[1].size)
+
+      let img = await sharp(resolve(`../../public/${data.output.path}`)).metadata()
+      assert.strictEqual(img.width, 600)
+      assert.strictEqual(img.height, 700)
+      assert.strictEqual(img.format, 'png')
+      assert.strictEqual(data.headers['access-control-allow-origin'], assertOrigin)
     })
 
     t.test('should properly replace spaces and other stuff', async () => {
@@ -142,9 +258,67 @@ t.describe('Media (API)', () => {
       assert.strictEqual(img.height, 720)
       assert.strictEqual(img.format, 'png')
     })
+
+    t.test('should return cors for cors-enabled sites', async () => {
+      let token = encode(null, { iss: 'development' }, secret)
+
+      let data = await assert.isFulfilled(
+        client.upload(
+          `/media?token=${token}`,
+          resolve('test.png')
+        )
+      )
+
+      testFiles.push(data.path)
+
+      assert.ok(data)
+      assert.ok(data.filename)
+      assert.ok(data.filename.startsWith(currYear))
+      assert.ok(data.path)
+
+      let stats = await Promise.all([
+        fs.stat(resolve('test.png')),
+        fs.stat(resolve(`../../public/${data.path}`)),
+      ])
+      assert.strictEqual(stats[0].size, stats[1].size)
+
+      let img = await sharp(resolve(`../../public/${data.path}`)).metadata()
+      assert.strictEqual(img.width, 600)
+      assert.strictEqual(img.height, 700)
+      assert.strictEqual(img.format, 'png')
+    })
+    
+    t.test('should support site-specific fileSize', async () => {
+      let token = encode(null, { iss: 'development' }, secret)
+
+      config.sources[2].store.fileSize = 1000
+
+      let err = await assert.isRejected(
+        client.upload(
+          `/media?token=${token}`,
+          resolve('test.png')
+        )
+      )
+      assert.match(err.message, /maxFileSize exceeded/)
+
+      config.sources[1].store.sites.development.fileSize = 524288000
+      
+      let data = await assert.isFulfilled(
+        client.upload(
+          `/media?token=${token}`,
+          resolve('test.png')
+        )
+      )
+
+      testFiles.push(data.path)
+
+      assert.ok(data)
+      assert.ok(data.filename)
+      assert.ok(data.filename.startsWith(currYear))
+    })
   })
 
-  t.timeout(10000).describe('POST /media/noprefix', function temp() {
+  t.describe('POST /media/noprefix', function temp() {
     t.test('should require authentication', async () => {
       resetLog()
       assert.strictEqual(log.error.callCount, 0)
@@ -198,15 +372,18 @@ t.describe('Media (API)', () => {
       let data = await assert.isFulfilled(
         client.upload(
           `/media/noprefix?token=${token}`,
-          resolve('test.png')
+          resolve('test.png'),
+          'POST',
+          { },
+          { includeHeaders: true }
         )
       )
 
-      testFiles.push(data.path)
+      testFiles.push(data?.output?.path)
 
-      assert.ok(data)
-      assert.strictEqual(data.filename, 'test.png')
-      assert.ok(data.path)
+      assert.ok(data.output)
+      assert.strictEqual(data.output.filename, 'test.png')
+      assert.ok(data.output.path)
 
       let stats = await Promise.all([
         fs.stat(resolve('test.png')),
@@ -218,6 +395,44 @@ t.describe('Media (API)', () => {
       assert.strictEqual(img.width, 600)
       assert.strictEqual(img.height, 700)
       assert.strictEqual(img.format, 'png')
+      assert.notOk(data.headers['access-control-allow-origin'])
+    })
+    
+
+    t.test('should upload and create file with no prefix', async () => {
+      const assertOrigin = 'http://localhost:9000'
+      let token = encode(null, { iss: 'development_cors' }, secret)
+
+      let data = await assert.isFulfilled(
+        client.upload(
+          `/media/noprefix?token=${token}`,
+          resolve('test.png'),
+          'POST',
+          { },
+          {
+            includeHeaders: true,
+            headers: { origin: assertOrigin, 'access-control-request-method': 'POST' },
+          }
+        )
+      )
+
+      testFiles.push(data?.output?.path)
+
+      assert.ok(data.output)
+      assert.strictEqual(data.output.filename, 'test.png')
+      assert.ok(data.output.path)
+
+      let stats = await Promise.all([
+        fs.stat(resolve('test.png')),
+        fs.stat(resolve('../../public/development/test.png')),
+      ])
+      assert.strictEqual(stats[0].size, stats[1].size)
+
+      let img = await sharp(resolve('../../public/development/test.png')).metadata()
+      assert.strictEqual(img.width, 600)
+      assert.strictEqual(img.height, 700)
+      assert.strictEqual(img.format, 'png')
+      assert.strictEqual(data.headers['access-control-allow-origin'], assertOrigin)
     })
     
     t.test('should upload and create file with prefix if exists', async () => {
@@ -260,7 +475,7 @@ t.describe('Media (API)', () => {
     })
   })
 
-  t.timeout(10000).describe('POST /media/resize', function temp() {
+  t.describe('POST /media/resize', function temp() {
     t.test('should require authentication', async () => {
       resetLog()
       assert.strictEqual(log.error.callCount, 0)
@@ -316,27 +531,66 @@ t.describe('Media (API)', () => {
           `/media/resize?token=${token}`,
           resolve('test.png'),
           'POST',
-          { }
+          { },
+          { includeHeaders: true }
         )
       )
 
-      testFiles.push(data.path)
+      testFiles.push(data?.output?.path)
 
-      assert.ok(data)
-      assert.ok(data.filename)
-      assert.ok(data.filename.startsWith(currYear))
-      assert.ok(data.path)
+      assert.ok(data.output)
+      assert.ok(data.output.filename)
+      assert.ok(data.output.filename.startsWith(currYear))
+      assert.ok(data.output.path)
 
       let stats = await Promise.all([
         fs.stat(resolve('test.png')),
-        fs.stat(resolve(`../../public/${data.path}`)),
+        fs.stat(resolve(`../../public/${data.output.path}`)),
       ])
       assert.strictEqual(stats[0].size, stats[1].size)
 
-      let img = await sharp(resolve(`../../public/${data.path}`)).metadata()
+      let img = await sharp(resolve(`../../public/${data.output.path}`)).metadata()
       assert.strictEqual(img.width, 600)
       assert.strictEqual(img.height, 700)
       assert.strictEqual(img.format, 'png')
+      assert.notOk(data.headers['access-control-allow-origin'])
+    })
+
+    t.test('should upload file and create file and return cors if site has cors', async () => {
+      const assertOrigin = 'http://localhost:9000'
+      let token = encode(null, { iss: 'development_cors' }, secret)
+
+      let data = await assert.isFulfilled(
+        client.upload(
+          `/media/resize?token=${token}`,
+          resolve('test.png'),
+          'POST',
+          { },
+          {
+            includeHeaders: true,
+            headers: { origin: assertOrigin, 'access-control-request-method': 'POST' },
+          }
+        )
+      )
+
+      testFiles.push(data?.output?.path)
+
+      assert.ok(data.output)
+      assert.ok(data.output.filename)
+      assert.ok(data.output.filename.startsWith(currYear))
+      assert.ok(data.output.path)
+
+      let stats = await Promise.all([
+        fs.stat(resolve('test.png')),
+        fs.stat(resolve(`../../public/${data.output.path}`)),
+      ])
+      assert.strictEqual(stats[0].size, stats[1].size)
+
+      let img = await sharp(resolve(`../../public/${data.output.path}`)).metadata()
+      assert.strictEqual(img.width, 600)
+      assert.strictEqual(img.height, 700)
+      assert.strictEqual(img.format, 'png')
+      assert.strictEqual(data.headers['access-control-allow-origin'], assertOrigin)
     })
 
     t.test('should upload file and create multiple sizes for file', async () => {
@@ -500,31 +754,42 @@ t.describe('Media (API)', () => {
     })
   })
 
-  t.timeout(10000).describe('POST /media/resize/:filename', function temp() {
+  t.describe('POST /media/resize/:filename', function temp() {
     let sourceFilename
+    let sourceFilenameCors
     let sourcePath
+    let sourcePathCors
     
     t.before(async function() {
-      let token = encode(null, { iss: 'development' }, secret)
+      let issuers = ['development', 'development_cors']
 
-      let data = await assert.isFulfilled(
-        client.upload(
-          `/media/resize?token=${token}`,
-          resolve('test.png'),
-          'POST',
-          { }
+      for (let iss of issuers) {
+        let token = encode(null, { iss }, secret)
+
+        let data = await assert.isFulfilled(
+          client.upload(
+            `/media/resize?token=${token}`,
+            resolve('test.png'),
+            'POST',
+            { }
+          )
         )
-      )
 
-      testFiles.push(data.path)
+        testFiles.push(data.path)
 
-      assert.ok(data)
-      assert.ok(data.filename)
-      assert.ok(data.filename.startsWith(currYear))
-      assert.ok(data.path)
+        assert.ok(data)
+        assert.ok(data.filename)
+        assert.ok(data.filename.startsWith(currYear))
+        assert.ok(data.path)
 
-      sourceFilename = data.filename
-      sourcePath = data.path
+        if (iss === 'development') {
+          sourceFilename = data.filename
+          sourcePath = data.path
+        } else {
+          sourceFilenameCors = data.filename
+          sourcePathCors = data.path
+        }
+      }
     })
 
     t.test('should require authentication', async () => {
@@ -600,34 +865,75 @@ t.describe('Media (API)', () => {
                 compressionLevel: 9,
               }
             },
-          }
+          },
+          { includeHeaders: true }
         )
       )
       
-      testFiles.push(data.test1.path)
-      testFiles.push(data.test2.path)
+      testFiles.push(data.output.test1.path)
+      testFiles.push(data.output.test2.path)
 
-      assert.ok(data.test1.filename)
-      assert.ok(data.test1.filename.startsWith(currYear))
-      assert.ok(data.test1.path)
-      assert.ok(data.test2.filename)
-      assert.ok(data.test2.filename.startsWith(currYear))
-      assert.ok(data.test2.path)
+      assert.ok(data.output.test1.filename)
+      assert.ok(data.output.test1.filename.startsWith(currYear))
+      assert.ok(data.output.test1.path)
+      assert.ok(data.output.test2.filename)
+      assert.ok(data.output.test2.filename.startsWith(currYear))
+      assert.ok(data.output.test2.path)
 
       let img = await sharp(resolve(`../../public/${sourcePath}`)).metadata()
       assert.strictEqual(img.width, 600)
       assert.strictEqual(img.height, 700)
       assert.strictEqual(img.format, 'png')
       
-      img = await sharp(resolve(`../../public/${data.test1.path}`)).metadata()
+      img = await sharp(resolve(`../../public/${data.output.test1.path}`)).metadata()
       assert.strictEqual(img.width, 320)
       assert.strictEqual(img.height, 413)
       assert.strictEqual(img.format, 'jpeg')
       
-      img = await sharp(resolve(`../../public/${data.test2.path}`)).metadata()
+      img = await sharp(resolve(`../../public/${data.output.test2.path}`)).metadata()
       assert.strictEqual(img.width, 150)
       assert.strictEqual(img.height, 175)
       assert.strictEqual(img.format, 'png')
+      assert.notOk(data.headers['access-control-allow-origin'])
+    })
+
+    t.test('should create sizes for existing file and return cors if site has cors', async () => {
+      const assertOrigin = 'http://localhost:9000'
+      let token = encode(null, { iss: 'development_cors' }, secret)
+
+      let data = await assert.isFulfilled(
+        client.post(
+          `/media/resize/${sourceFilenameCors}?token=${token}`,
+          {
+            test2: {
+              format: 'png',
+              resize: { width: 150 },
+              png: { compressionLevel: 9 }
+            },
+          },
+          {
+            includeHeaders: true,
+            headers: { origin: assertOrigin, 'access-control-request-method': 'POST' },
+          }
+        )
+      )
+
+      testFiles.push(data.output.test2.path)
+
+      assert.ok(data.output.test2.filename)
+      assert.ok(data.output.test2.filename.startsWith(currYear))
+      assert.ok(data.output.test2.path)
+
+      let img = await sharp(resolve(`../../public/${sourcePath}`)).metadata()
+      assert.strictEqual(img.width, 600)
+      assert.strictEqual(img.height, 700)
+      assert.strictEqual(img.format, 'png')
+      
+      img = await sharp(resolve(`../../public/${data.output.test2.path}`)).metadata()
+      assert.strictEqual(img.width, 150)
+      assert.strictEqual(img.height, 175)
+      assert.strictEqual(img.format, 'png')
+      assert.strictEqual(data.headers['access-control-allow-origin'], assertOrigin)
     })
 
     t.test('should base64 output of existing file', async () => {
@@ -664,7 +970,7 @@ t.describe('Media (API)', () => {
     })
   })
 
-  t.timeout(10000).describe('DELETE /media/:filename', function temp() {
+  t.describe('DELETE /media/:filename', function temp() {
     t.test('should require authentication', async () => {
       resetLog()
       assert.strictEqual(log.error.callCount, 0)
@@ -713,7 +1019,7 @@ t.describe('Media (API)', () => {
     })
 
     t.test('should remove the file', async () => {
-      let token = encode(null, { iss: 'existing' }, secret)
+      let token = encode(null, { iss: 'development' }, secret)
 
       let data = await client.upload(
         `/media/noprefix?token=${token}`,
@@ -723,11 +1029,11 @@ t.describe('Media (API)', () => {
       let filepath = data.path
       testFiles.push(filepath)
 
-      let files = await client.get('/media/existing')
+      let files = await client.get('/media?token=' + token)
 
       let found = false
       for (let file of files) {
-        if (file.filename === 'test.png') {
+        if (file.filename === data.filename) {
           found = true
         }
       }
@@ -737,17 +1043,19 @@ t.describe('Media (API)', () => {
         fs.stat(resolve(`../../public/${filepath}`))
       )
 
-      await assert.isFulfilled(
-        client.del(`/media/test.png?token=${token}`)
+      let res = await assert.isFulfilled(
+        client.del(`/media/${data.filename}?token=${token}`, {}, {
+          includeHeaders: true
+        })
       )
 
       testFiles.splice(testFiles.length - 1)
 
-      files = await client.get('/media/existing')
+      files = await client.get('/media?token=' + token)
 
       found = false
       for (let file of files) {
-        if (file.filename === 'test.png') {
+        if (file.filename === data.filename) {
           found = true
         }
       }
@@ -756,6 +1064,58 @@ t.describe('Media (API)', () => {
       await assert.isRejected(
         fs.stat(resolve(`../../public/${filepath}`))
       )
+      assert.notOk(res.headers['access-control-allow-origin'])
+    })
+
+    t.test('should remove the file and return cors if site has cors', async () => {
+      const assertOrigin = 'http://localhost:9000'
+      let token = encode(null, { iss: 'development_cors' }, secret)
+
+      let data = await client.upload(
+        `/media/noprefix?token=${token}`,
+        resolve('test.png')
+      )
+
+      let filepath = data.path
+      testFiles.push(filepath)
+
+      let files = await client.get('/media?token=' + token)
+
+      let found = false
+      for (let file of files) {
+        if (file.filename === data.filename) {
+          found = true
+        }
+      }
+      assert.ok(found)
+      
+      await assert.isFulfilled(
+        fs.stat(resolve(`../../public/${filepath}`))
+      )
+
+      let res = await assert.isFulfilled(
+        client.del(`/media/${data.filename}?token=${token}`, {}, {
+          includeHeaders: true,
+          headers: { origin: assertOrigin, 'access-control-request-method': 'POST' },
+        })
+      )
+
+      testFiles.splice(testFiles.length - 1)
+
+      files = await client.get('/media?token=' + token)
+
+      found = false
+      for (let file of files) {
+        if (file.filename === data.filename) {
+          found = true
+        }
+      }
+      assert.notOk(found)
+
+      await assert.isRejected(
+        fs.stat(resolve(`../../public/${filepath}`))
+      )
+      assert.strictEqual(res.headers['access-control-allow-origin'], assertOrigin)
     })
   })
 
@@ -802,17 +1162,39 @@ t.describe('Media (API)', () => {
     t.test('should return list of files in specified folder', async () => {
       let token = encode(null, { iss: 'development' }, secret)
 
-      let data = await client.get('/media?token=' + token)
+      let data = await client.get('/media?token=' + token, { includeHeaders: true })
 
-      assert.ok(data.length)
+      assert.ok(data.output.length)
       let found = false
-      for (let file of data) {
+      for (let file of data.output) {
         if (file.filename === '.gitkeep') {
           found = true
           break
         }
       }
       assert.ok(found)
+      assert.notOk(data.headers['access-control-allow-origin'])
+    })
+
+    t.test('should return list of files in specified folder and return cors if site has cors', async () => {
+      const assertOrigin = 'http://localhost:9000'
+      let token = encode(null, { iss: 'development_cors' }, secret)
+
+      let data = await client.get('/media?token=' + token, { 
+        includeHeaders: true,
+        headers: { origin: assertOrigin, 'access-control-request-method': 'POST' },
+      })
+
+      assert.ok(data.output.length)
+      let found = false
+      for (let file of data.output) {
+        if (file.filename === '.gitkeep') {
+          found = true
+          break
+        }
+      }
+      assert.ok(found)
+      assert.strictEqual(data.headers['access-control-allow-origin'], assertOrigin)
     })
   })
 
@@ -832,11 +1214,25 @@ t.describe('Media (API)', () => {
     })
 
     t.test('should otherwise return list of files for a public site', async () => {
-      let files = await client.get('/media/existing')
-      assert.strictEqual(files[0].filename, '20220105_101610_test1.jpg')
-      assert.strictEqual(files[0].size, null)
-      assert.strictEqual(files[1].filename, '20220105_101610_test2.png')
-      assert.strictEqual(files[1].size, null)
+      let files = await client.get('/media/existing', { includeHeaders: true })
+      assert.strictEqual(files.output[0].filename, '20220105_101610_test1.jpg')
+      assert.strictEqual(files.output[0].size, null)
+      assert.strictEqual(files.output[1].filename, '20220105_101610_test2.png')
+      assert.strictEqual(files.output[1].size, null)
+      assert.notOk(files.headers['access-control-allow-origin'])
+    })
+
+    t.test('should otherwise return list of files for a public site and cors if site has cors', async () => {
+      const assertOrigin = 'http://localhost:9000'
+      let files = await client.get('/media/existing_cors', {
+        includeHeaders: true,
+        headers: { origin: assertOrigin, 'access-control-request-method': 'POST' },
+      })
+      assert.strictEqual(files.output[0].filename, '20220105_101610_test1.jpg')
+      assert.strictEqual(files.output[0].size, null)
+      assert.strictEqual(files.output[1].filename, '20220105_101610_test2.png')
+      assert.strictEqual(files.output[1].size, null)
+      assert.strictEqual(files.headers['access-control-allow-origin'], assertOrigin)
     })
   })
 })

test/media/routes.test.mjs

@@ -73,7 +73,7 @@ t.describe('#listFiles()', function() {
   const stubGetCache = stub()
   
   const routes = new MediaRoutes({
-    security: { verifyToken: stubVerify },
+    security: { verifyToken: stubVerify, hasCors: stub() },
   })
   routes.filesCacheGet = stubGetCache
 
@@ -87,7 +87,7 @@ t.describe('#listFiles()', function() {
 
     let ctx = createContext()
     const assertError = new Error('temp')
-    stubVerify.rejects(assertError)
+    stubVerify.throws(assertError)
 
     let err = await assert.isRejected(routes.listFiles(ctx))
 
@@ -102,7 +102,7 @@ t.describe('#listFiles()', function() {
     let ctx = createContext()
     const assertSiteName = 'benshapiro'
     const assertResult = { a: 1 }
-    stubVerify.resolves(assertSiteName)
+    stubVerify.returns(assertSiteName)
     stubGetCache.returns(assertResult)
 
     await routes.listFiles(ctx)
@@ -118,7 +118,7 @@ t.describe('#listPublicFiles()', function() {
   const stubGetCache = stub()
   
   const routes = new MediaRoutes({
-    security: { throwIfNotPublic: stubSitePublic },
+    security: { throwIfNotPublic: stubSitePublic, hasCors: stub() },
   })
   routes.filesCacheGet = stubGetCache
 
@@ -250,6 +250,7 @@ t.describe('#uploadNoPrefix', function() {
     security: {
       verifyToken: stubVerify,
       verifyBody: stub(),
+      hasCors: stub(),
     },
     formidable: { uploadFile: stubUpload, },
     fs: { stat: stubStat },
@@ -267,7 +268,7 @@ t.describe('#uploadNoPrefix', function() {
     let ctx = createContext()
     const assertSiteName = 'benshapiro'
     const assertError = new Error('hello')
-    stubVerify.resolves(assertSiteName)
+    stubVerify.returns(assertSiteName)
     stubUpload.rejects(assertError)
 
     let err = await assert.isRejected(routes.uploadNoPrefix(ctx))
@@ -300,6 +301,7 @@ t.describe('#resizeExisting', function() {
     security: {
       verifyToken: stubVerifyToken,
       verifyBody: stubVerifyBody,
+      hasCors: stub(),
     },
     fs: { stat: stubStat },
     sharp: stubSharp,
@@ -338,7 +340,7 @@ t.describe('#resizeExisting', function() {
 
     let ctx = createContext({ req: { body: { } } })
     const assertError = new Error('temp')
-    stubVerifyToken.rejects(assertError)
+    stubVerifyToken.throws(assertError)
 
     let err = await assert.isRejected(routes.resizeExisting(ctx))
 
@@ -367,7 +369,7 @@ t.describe('#resizeExisting', function() {
     const assertJpeg = { a: 1 }
     const assertFilename = 'asdfsafd'
     const assertSite = 'mario'
-    stubVerifyToken.resolves(assertSite)
+    stubVerifyToken.returns(assertSite)
 
     let ctx = createContext({ req: { body: {
       [assertKey]: {
@@ -418,7 +420,7 @@ t.describe('#resizeExisting', function() {
       const assertPayload = { a: 1 }
       const assertFilename = 'asdfsafd'
       const assertSite = 'mario'
-      stubVerifyToken.resolves(assertSite)
+      stubVerifyToken.returns(assertSite)
   
       let called = 0
       stubStat.returnWith(function() {
@@ -467,7 +469,7 @@ t.describe('#resizeExisting', function() {
     const assertValidKey2 = 'derp'
     const assertErrorKey = 'throwmyerr'
     const assertErrorMessage = 'some message here'
-    stubVerifyToken.resolves('asdf')
+    stubVerifyToken.returns('asdf')
 
     let called = 0
     stubStat.returnWith(function() {
@@ -512,7 +514,7 @@ t.describe('#resizeExisting', function() {
     const assertFilename = 'asdfsafd.png'
     const assertSite = 'mario'
     const assertBase64Data = 'asdf1234'
-    stubVerifyToken.resolves(assertSite)
+    stubVerifyToken.returns(assertSite)
 
     let ctx = createContext({ req: { body: {
       [assertKey]: {
@@ -562,6 +564,7 @@ t.describe('#resize', function() {
     security: {
       verifyToken: stubVerifyToken,
       verifyBody: stubVerifyBody,
+      hasCors: stub(),
     },
     fs: { stat: stubStat },
     formidable: { uploadFile: stubUpload },
@@ -618,7 +621,7 @@ t.describe('#resize', function() {
     const assertJpeg = { a: 1 }
     const assertFilename = 'asdfsafd'
     const assertSite = 'mario'
-    stubVerifyToken.resolves(assertSite)
+    stubVerifyToken.returns(assertSite)
     stubUpload.resolves({ filename: assertFilename + '.png' })
 
     let ctx = createContext({ req: { body: {
@@ -669,7 +672,7 @@ t.describe('#resize', function() {
       const assertPayload = { a: 1 }
       const assertFilename = 'asdfsafd'
       const assertSite = 'mario'
-      stubVerifyToken.resolves(assertSite)
+      stubVerifyToken.returns(assertSite)
       stubUpload.resolves({ filename: assertFilename + '.png' })
   
       let called = 0
@@ -720,7 +723,7 @@ t.describe('#resize', function() {
     const assertValidKey2 = 'derp'
     const assertErrorKey = 'throwmyerr'
     const assertErrorMessage = 'some message here'
-    stubVerifyToken.resolves('asdf')
+    stubVerifyToken.returns('asdf')
     stubUpload.resolves({ filename: 'file.png' })
 
     let called = 0
@@ -765,7 +768,7 @@ t.describe('#resize', function() {
     const assertFilename = 'asdfsafd.png'
     const assertSite = 'mario'
     const assertBase64Data = 'asdf1234'
-    stubVerifyToken.resolves(assertSite)
+    stubVerifyToken.returns(assertSite)
     stubUpload.resolves({ filename: assertFilename })
 
     let ctx = createContext({ req: { body: {
@@ -811,6 +814,7 @@ basicUploadTestRoutes.forEach(function(name) {
       security: {
         verifyToken: stubVerify,
         verifyBody: stub(),
+        hasCors: stub(),
       },
       formidable: { uploadFile: stubUpload, },
       fs: { stat: stubStat },
@@ -827,7 +831,7 @@ basicUploadTestRoutes.forEach(function(name) {
 
       let ctx = createContext({ req: { body: { } } })
       const assertError = new Error('temp')
-      stubVerify.rejects(assertError)
+      stubVerify.throws(assertError)
 
       let err = await assert.isRejected(routes[name](ctx))
 
@@ -842,7 +846,7 @@ basicUploadTestRoutes.forEach(function(name) {
       let ctx = createContext({ req: { body: { } } })
       const assertSiteName = 'benshapiro'
       const assertError = new Error('hello')
-      stubVerify.resolves(assertSiteName)
+      stubVerify.returns(assertSiteName)
       stubUpload.rejects(assertError)
 
       let err = await assert.isRejected(routes[name](ctx))
@@ -860,7 +864,7 @@ basicUploadTestRoutes.forEach(function(name) {
       const assertSize = 1241412
       const assertFilename = 'asdfsafd'
       const assertSite = 'mario'
-      stubVerify.resolves(assertSite)
+      stubVerify.returns(assertSite)
       stubUpload.resolves({ filename: assertFilename })
       stubStat.resolves({ size: assertSize })
       await routes[name](ctx)
@@ -887,6 +891,7 @@ t.describe('#remove()', function() {
   const routes = new MediaRoutes({
     security: {
       verifyToken: stubVerify,
+      hasCors: stub(),
     },
     fs: { unlink: stubUnlink },
   })
@@ -905,7 +910,7 @@ t.describe('#remove()', function() {
 
     let ctx = createContext({ req: { body: { } } })
     const assertError = new Error('temp')
-    stubVerify.rejects(assertError)
+    stubVerify.throws(assertError)
 
     let err = await assert.isRejected(routes.remove(ctx))
 
@@ -924,7 +929,7 @@ t.describe('#remove()', function() {
 
     let ctx = createContext({ req: { body: { } } })
     ctx.params.filename = assertFilename
-    stubVerify.resolves(assertSiteName)
+    stubVerify.returns(assertSiteName)
     stubUnlink.rejects(assertError)
 
     let err = await assert.isRejected(routes.remove(ctx))
@@ -948,7 +953,7 @@ t.describe('#remove()', function() {
 
     let ctx = createContext({ req: { body: { } } })
     ctx.params.filename = assertFilename
-    stubVerify.resolves(assertSiteName)
+    stubVerify.returns(assertSiteName)
 
     await routes.remove(ctx)
 

test/media/security.test.mjs

@@ -2,7 +2,7 @@ import { Eltro as t, assert} from 'eltro'
 import { HttpError } from 'flaska'
 
 import { createContext } from '../helper.server.mjs'
-import { verifyToken, verifyBody, throwIfNotPublic } from '../../api/media/security.mjs'
+import { verifyToken, verifyBody, throwIfNotPublic, verifyCorsEnabled } from '../../api/media/security.mjs'
 import encode from '../../api/jwt/encode.mjs'
 import config from '../../api/config.mjs'
 
@@ -61,6 +61,98 @@ t.describe('#throwIfNotPublic()', function() {
   })
 })
 
+t.describe('#verifyCorsEnabled()', function() {
+  let ctx
+  let backup = {}
+
+  t.beforeEach(function() {
+    ctx = createContext({ })
+  })
+
+  t.before(function() {
+    backup = config.sources[1].store
+    config.sources[1].store = {
+      sites: {
+        justatest: {
+          keys: {
+            'default@HS512': 'mysharedkey',
+          }
+        },
+        justatest2: {
+          keys: {
+            'default@HS512': 'mysharedkey',
+          },
+          cors: false,
+        },
+        justatest3: {
+          keys: {
+            'default@HS512': 'mysharedkey',
+          },
+          cors: true,
+        },
+      },
+    }
+  })
+
+  t.after(function() {
+    config.sources[1].store = backup
+  })
+
+  t.test('should throw 404 for sites that do not exist or are null', function() {
+    let tests = [
+      'justatest',
+      'justatest2',
+      'nonexisting1',
+      null,
+    ]
+
+    tests.forEach(function(test) {
+      ctx.query.set('token', encode({ typ: 'JWT', alg: 'HS512' }, { iss: test }, 'mysharedkey'))
+
+      assert.throws(function() { verifyCorsEnabled(ctx) }, function(err) {
+        assert.ok(err instanceof HttpError)
+        assert.ok(err instanceof Error)
+        assert.strictEqual(err.status, 404)
+        assert.notOk(err.message)
+        return true
+      }, `should throw with site ${test}`)
+    })
+  })
+
+  t.test('should throw 404 for sites that use wrong token', function() {
+    let tests = [
+      'justatest',
+      'justatest2',
+      'nonexisting1',
+      null,
+    ]
+
+    tests.forEach(function(test) {
+      ctx.query.set('token', encode({ typ: 'JWT', alg: 'HS512' }, { iss: test }, 'herp'))
+
+      assert.throws(function() { verifyCorsEnabled(ctx) }, function(err) {
+        assert.ok(err instanceof HttpError)
+        assert.ok(err instanceof Error)
+        assert.strictEqual(err.status, 404)
+        assert.notOk(err.message)
+        return true
+      }, `should throw with site ${test}`)
+    })
+  })
+
+  t.test('should pass for sites that have cors enabled', function() {
+    let tests = [
+      'justatest3',
+    ]
+
+    tests.forEach(function(test) {
+      ctx.query.set('token', encode({ typ: 'JWT', alg: 'HS512' }, { iss: test }, 'mysharedkey'))
+
+      assert.doesNotThrow(function() { verifyCorsEnabled(ctx) }, `should not throw with site ${test}`)
+    })
+  })
+})
+
 t.describe('#verifyToken()', function() {
   let backup = {}
   t.before(function() {