Initial commit

Author: chrisfosterelli

.dockerignore

@@ -0,0 +1,2 @@
+node_modules
+.git

.gitignore

@@ -0,0 +1 @@
+node_modules

Dockerfile

@@ -0,0 +1,13 @@
+
+#
+# Dockerfile for docker-simple-file-upload
+#
+
+FROM node:8-alpine
+RUN mkdir -p /code
+WORKDIR /code
+ADD . /code
+RUN npm install --production
+ENV NODE_ENV production
+CMD [ "npm", "start" ]
+EXPOSE 3000

app.js

@@ -0,0 +1,50 @@
+
+/* Application server
+ */
+
+const express = require('express')
+const fileUpload = require('express-fileupload')
+
+const fileSize = process.env.FILESIZE || Infinity
+
+const files = 1
+const abortOnLimit = true
+const safeFileNames = true
+const limits = { fileSize, files }
+const options = { safeFileNames, limits, abortOnLimit }
+
+function handleUpload(req, res, next) {
+  const notProvided = new Error('File not provided')
+  const invalidKey = new Error('Invalid key provided')
+  if (!req.files || !req.files.data) throw notProvided
+  if (!req.query.key || !process.env['KEY_' + req.query.key]) throw invalidKey
+  req.files.data.mv(process.env['KEY_' + req.query.key], next)
+}
+
+function handleSuccess(req, res, next) {
+  console.log(`[${req.ip}] [201] Upload successful for ${req.query.key}`)
+  res.status(201).send('Upload successful') 
+}
+
+function handleInvalidRequests(req, res) {
+  throw new Error('That request is not supported')
+}
+
+function handleError(err, req, res, next) {
+  const code = 
+    err.message == 'File not provided' || 
+    err.message == 'Invalid key provided' ||
+    err.message == 'That request is not supported' ? 400 : 500
+  const response = code == 400 ? err.message : 'Could not process upload'
+  console.log(`[${req.ip}] [${code}] Upload failed with: ${err.message}`)
+  res.status(code).send(response)
+}
+
+const app = express()
+app.use(fileUpload(options))
+app.post('/upload', handleUpload)
+app.post('/upload', handleSuccess)
+app.all('*', handleInvalidRequests)
+app.use(handleError)
+
+module.exports = app

app.test.js

@@ -0,0 +1,44 @@
+
+/* Integration tests
+ */
+
+const fs = require('fs')
+const request = require('supertest')
+const app = require('./app')
+
+describe('/upload', () => {
+
+  test('it should accept a file upload', async () => {
+    process.env['KEY_TEST'] = '/tmp/AWEJA_testfile.txt'
+    const name = 'test.txt'
+    const url = '/upload?key=TEST'
+    const file = Buffer.from('This is a test file', 'utf8')
+    const res = await request(app).post(url).attach('data', file, name)
+    expect(res.status).toEqual(201)
+    expect(res.text).toEqual('Upload successful')
+    expect(fs.existsSync('/tmp/AWEJA_testfile.txt')).toEqual(true)
+    fs.unlinkSync('/tmp/AWEJA_testfile.txt')
+    delete process.env['KEY_TEST']
+  })
+
+  test('it should error for unknown requests', async () => {
+    const res = await request(app).get('/')
+    expect(res.status).toEqual(400)
+    expect(res.text).toEqual('That request is not supported')
+  })
+
+  test('it should fail when no file is provided', async () => {
+    const res = await request(app).post('/upload')
+    expect(res.status).toEqual(400)
+    expect(res.text).toEqual('File not provided')
+  })
+
+  test('it should fail when no key is provided', async () => {
+    const name = 'test.txt'
+    const file = Buffer.from('This is a test file', 'utf8')
+    const res = await request(app).post('/upload').attach('data', file, name)
+    expect(res.status).toEqual(400)
+    expect(res.text).toEqual('Invalid key provided')
+  })
+
+})

index.js

@@ -0,0 +1,10 @@
+
+/* Simple file upload API
+ */
+
+const app = require('./app')
+const port = process.env.PORT || 3000
+
+app.listen(port, () => {
+  console.log('Application started on port', port)
+})