Check quota in PATCH too

Author: kjetilk

lib/handlers/patch.js

@@ -10,6 +10,7 @@ const utils = require('../utils.js')
 const error = require('../http-error')
 const $rdf = require('rdflib')
 const crypto = require('crypto')
+const overQuota = require('../utils').overQuota
 
 const DEFAULT_TARGET_TYPE = 'text/turtle'
 
@@ -53,7 +54,7 @@ function patchHandler (req, res, next) {
   ])
   // Patch the graph and write it back to the file
   .then(([graph, patchObject]) => applyPatch(patchObject, graph, target))
-  .then(graph => writeGraph(graph, target))
+    .then(graph => writeGraph(graph, target, root, ldp.serverUri))
   // Send the result to the client
   .then(result => { res.send(result) })
   .then(next, next)
@@ -137,19 +138,27 @@ function applyPatch (patchObject, graph, target) {
 }
 
 // Writes the RDF graph to the given resource
-function writeGraph (graph, resource) {
+function writeGraph (graph, resource, root, serverUri) {
   debug('PATCH -- Writing patched file')
   return new Promise((resolve, reject) => {
     const resourceSym = graph.sym(resource.uri)
     const serialized = $rdf.serialize(resourceSym, graph, resource.uri, resource.contentType)
 
-    fs.writeFile(resource.file, serialized, {encoding: 'utf8'}, function (err) {
-      if (err) {
-        return reject(error(500, `Failed to write file after patch: ${err}`))
+    // First check if we are above quota
+    overQuota(root, serverUri).then((isOverQuota) => {
+      if (isOverQuota) {
+        return reject(error(413,
+          'User has exceeded their storage quota'))
       }
-      debug('PATCH -- applied successfully')
-      resolve('Patch applied successfully.\n')
-    })
+
+      fs.writeFile(resource.file, serialized, {encoding: 'utf8'}, function (err) {
+        if (err) {
+          return reject(error(500, `Failed to write file after patch: ${err}`))
+        }
+        debug('PATCH -- applied successfully')
+        resolve('Patch applied successfully.\n')
+      })
+    }).catch(() => reject(error(500, 'Error finding user quota')))
   })
 }