adding wit.getSamples & wit.backup

Author: bashz

.gitignore

@@ -3,3 +3,4 @@ node_modules/
 run.js
 npm-debug.log
 .nyc_output
+backup.zip

package.json

@@ -1,6 +1,6 @@
 {
   "name": "wit-api",
-  "version": "0.0.5",
+  "version": "0.0.6",
   "description": "Implementation of wit api, provides methods to easily manage your wit app",
   "main": "index.js",
   "keywords": [

src/Sample.js

@@ -1,18 +1,47 @@
+const fs = require('fs')
+const req = require('request')
+
 const actions = function(id) {
   return {
+    get: {
+      method: 'GET',
+      uri: '/samples'
+    },
     add: {
       method: 'POST',
       uri: '/samples'
     },
     delete: {
       method: 'DELETE',
       uri: '/samples'
+    },
+    export: {
+      method: 'GET',
+      uri: 'export'
     }
   }
 }
 
 module.exports = function (request) {
   return {
+    get: function (limit = 10, offset = 0, entities = [], values = [], isNegative = false) {
+      let payload = actions().get
+      payload.qs = {
+        limit,
+        offset,
+        entity_ids: entities,
+        entity_values: values,
+        negative: isNegative
+      }
+      return new Promise((resolve, reject) => {
+        request(payload, (err, res) => {
+          if (err) {
+            return reject(err)
+          }
+          return resolve(res)
+        })
+      })
+    },
     add: function (...args) {
       let payload = actions().add
       let body = Array.isArray(args[0]) ? args[0] : [{text: args[0], entities: args[1]}]
@@ -46,6 +75,24 @@ module.exports = function (request) {
           return resolve(res)
         })
       })
+    },
+    export: function (filePath = './backup.zip') {
+      let payload = actions().export
+      return new Promise((resolve, reject) => {
+        request(payload, (err, res) => {
+          if (err) {
+            return reject(err)
+          }
+          req(res)
+          .on('error', (err) => {
+            return reject(err)
+          })
+          .on('response', (response) => {
+            return resolve(response)
+          })
+          .pipe(fs.createWriteStream(filePath))
+        })
+      })
     }
   }
 }

src/Wit.js

@@ -48,6 +48,8 @@ const Wit = function(token, timeout) {
   const trainer = sample(this.doRequest)
   this.train = trainer.add
   this.forget = trainer.delete
+  this.getSamples = trainer.get
+  this.backup = trainer.export
 
   this.app = function(name, data) {
     return new App(name, this.doRequest, data)

test/test.js

@@ -421,7 +421,7 @@ describe('Wit', function () {
         }).catch(e => {
           expect(e).to.be.an(Error)
           expect(e).to.have.property('code')
-          expect(e.code).to.be('unknown')
+          expect(e.code).to.be('no-auth')
           expect(e.message).not.to.be.empty()
         })
       })