feat: upload user image to digital ocean

Author: nalves599

package.json

@@ -40,8 +40,8 @@
     "@hapi/vision": "^6.1.0",
     "adm-zip": "^0.5.9",
     "async": "^3.2.3",
-    "axios": "^0.26.0",
     "aws-sdk": "^2.814.0",
+    "axios": "^0.26.0",
     "bunyan": "^1.8.15",
     "esprima": "^4.0.1",
     "good": "^2.3.0",

server/resources/user.js

@@ -1,3 +1,4 @@
+const aws = require('../plugins/aws')
 const Boom = require('@hapi/boom')
 const server = require('../').hapi
 const log = require('../helpers/logger')
@@ -6,6 +7,7 @@ const fieldsParser = require('../helpers/fieldsParser')
 const config = require('../../config')
 const User = require('../db/user')
 
+
 server.method('user.create', create, {})
 server.method('user.update', update, {})
 server.method('user.updateMe', updateMe, {})
@@ -55,6 +57,19 @@ async function update(filter, user, opts) {
     filter = { id: filter }
   }
 
+  if (user && user.img.startsWith("data:image/jpeg;base64,")) {
+    const base64Data = user.img.replace(/^data:image\/\w+;base64,/, "");
+    const buffer = Buffer.from(base64Data, "base64");
+
+    const imgUrl = await uploadUserImage(filter.id, buffer)
+    if (!imgUrl) {
+      log.error('Error setting image URL for user ' + filter.id)
+      throw Boom.internal('Error setting image URL for user ' + filter.id)
+    }
+
+    user.img = imgUrl
+  }
+
   if (user && user.company) {
     const user2 = Object.assign({}, user)
     if (!user.company.edition) {
@@ -404,3 +419,37 @@ async function getQRCode(user) {
   // FIXME: Implement this in a secure way using JWT
   return `sinfo://${btoa(JSON.stringify({ kind: "user", user: { id: user.id, name: user.name, img: user.img, role: user.role } }))}`
 }
+
+/* Image Functions */
+function getDataFromStream(stream) {
+  return new Promise((resolve, reject) => {
+    let data = []
+
+    stream.on('data', (chunk) => {
+      data.push(chunk)
+    })
+
+    stream.on('end', () => {
+      resolve(Buffer.concat(data))
+    })
+
+    stream.on('error', (err) => {
+      reject(err)
+    })
+  })
+}
+
+function getFileName(userId) {
+  return userId + '.jpg'
+}
+
+function getUserPath() {
+  return `static/users/`
+}
+
+async function uploadUserImage(userId, file) {
+  const path = getUserPath()
+  const fileName = getFileName(userId)
+  const data = await getDataFromStream(file)
+  return aws.upload(path, data, fileName, true)
+}