feat: allow STRU to take name suggesstion

QuorumDMS · Mar 10, 2017 · b7e17af · b7e17af
1 parent 9276f7f
commit b7e17af
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/commands/registration/stou.js b/src/commands/registration/stou.js
@@ -4,10 +4,16 @@ module.exports = {
   directive: 'STOU',
   handler: function (args) {
     if (!this.fs) return this.reply(550, 'File system not instantiated');
-    if (!this.fs.getUniqueName) return this.reply(402, 'Not supported by file system');
+    if (!this.fs.get || !this.fs.getUniqueName) return this.reply(402, 'Not supported by file system');
 
-    args.command._[1] = this.fs.getUniqueName();
-    return stor.call(this, args);
+    const fileName = args.command._[1];
+    return this.fs.get(fileName)
+    .catch(() => fileName)                // does not exist, name is unique
+    .then(() => this.fs.getUniqueName())  // exists, must create new unique name
+    .then(name => {
+      args.command._[1] = name;
+      return stor.call(this, args);
+    });
   },
   syntax: '{{cmd}}',
   description: 'Store file uniquely'