spice-couchbase 2.0.13 → 2.0.16

package/build/index.js

@@ -1,72 +1,241 @@
 "use strict";
 
+function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (enumerableOnly) symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; }); keys.push.apply(keys, symbols); } return keys; }
+
+function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(Object(source), true).forEach(function (key) { _defineProperty(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } } return target; }
+
+function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
+
 function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
 
 function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
 
-var _ = require('lodash');
+var _ = require("lodash");
+
+var couchbase = require("couchbase");
 
-var couchbase = require('couchbase');
+var {
+  default: axios
+} = require("axios");
+
+var qs = require("qs");
+
+if (!Promise.allSettled) {
+  Promise.allSettled = promises => Promise.all(promises.map((promise, i) => promise.then(value => ({
+    status: "fulfilled",
+    value
+  })).catch(reason => ({
+    status: "rejected",
+    reason
+  }))));
+}
 
 module.exports = class Database {
-  constructor(connection, full_text_index) {
+  constructor(connection) {
+    var {
+      full_text_index,
+      collection,
+      scope,
+      query_scope = "cluster",
+      auto_create_collection
+    } = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
+
     if (connection) {
       this.connection_name = connection;
       this.connection = spice.database.connections[connection];
+      this.query_scope = query_scope;
       this.cluster = spice.database.clusters[connection];
       this.bucket = this.connection._scope._bucket._name;
+      this.db_config = spice.config.database.connections[connection];
+      this.collection_manager = spice.database.buckets[connection].collections();
+      this.auto_create_collection = auto_create_collection || (collection || spice.config.database.connections[connection].collection) != undefined && (collection || spice.config.database.connections[connection].collection) != "default" && (scope || spice.config.database.connections[connection].scope) != "_default";
+      this.collection_name = collection || spice.config.database.connections[connection].collection || "_default";
+      this.scope_name = scope || spice.config.database.connections[connection].scope || "_default";
+      this.scope = spice.database.scopes[connection];
       this.full_text_index = spice.config.database.connections[connection].full_text_index || full_text_index;
       this.N1qlQuery = couchbase.N1qlQuery;
       this.SearchQuery = couchbase.SearchQuery;
+      this.connection = this.scope.collection(this.collection_name);
+      this.userManager = this.getUserManager(connection);
+
+      if (this.auto_create_collection) {
+        this.upsertCollection();
+      }
     }
   }
 
-  connect(key, value) {
-    return new Promise((resolve, reject) => {
+  connect(connection_name, config) {
+    var _this = this;
+
+    return _asyncToGenerator(function* () {
+      var cluster = yield couchbase.connect(config.nodes, {
+        username: config.username,
+        password: config.password
+      });
+      var bucket = cluster.bucket(config.bucket);
+      var collection;
+      var scope_name = _this.scope_name || config.scope || "_default";
+      var collection_name = _this.collection_name || config.collection || "_default";
+      var scope = bucket.scope(scope_name);
+      collection = bucket.scope(scope_name).collection(collection_name || "_default");
+      return {
+        connection: connection_name,
+        bucket,
+        cluster,
+        scope,
+        collection
+      };
+    })();
+  }
+
+  getUserManager(connection) {
+    return spice.database.clusters[connection].users();
+  }
+
+  assignUserPermission() {
+    var _this2 = this;
+
+    return _asyncToGenerator(function* () {
       try {
-        var cluster = new couchbase.Cluster(value.nodes, {
-          username: value.username,
-          password: value.password
+        var allUsers = yield _this2.userManager.getAllUsers({
+          timeout: 6000
         });
-        var bucket = cluster.bucket(value.bucket);
-        var collection = bucket.defaultCollection();
-        resolve({
-          connection: key,
-          bucket,
-          cluster,
-          collection
+        var update_user_response = yield _this2.userManager.upsertUser(_this2.db_config.username, {
+          roles: [{
+            name: "scope_admin",
+            bucket: _this2.db_config.bucket
+          }, {
+            name: "data_reader",
+            bucket: _this2.db_config.bucket
+          }]
         });
       } catch (e) {
-        spice.logger.debug(e, e.stack);
-        reject(e);
+        console.log(e);
       }
-    });
+    })();
   }
 
-  get(doc_id) {
-    var that = this;
-    return new Promise(
-    /*#__PURE__*/
-    function () {
-      var _ref = _asyncToGenerator(function* (resolve, reject) {
-        that.connection.get(doc_id, function (err, res) {
-          var return_value = {};
+  getCouchbaseHost() {
+    var default_port = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 8091;
+    return (this.db_config.api_base || "http://".concat(_.last(_.first(this.db_config.nodes.split(",")).split("://")))) + ":" + default_port;
+  } //create wait function
+
+
+  wait(ms) {
+    return _asyncToGenerator(function* () {
+      return new Promise(resolve => setTimeout(resolve, ms));
+    })();
+  }
+
+  upsertCollection() {
+    var _this3 = this;
 
-          if (err) {
-            reject(err);
-          } else {
-            _.defaults(return_value, res.value);
+    return _asyncToGenerator(function* () {
+      try {
+        var collection_exists = yield _this3.collectionExists(_this3.collection_name, _this3.scope_name);
 
-            return_value.id = doc_id;
-            resolve(return_value);
+        if (!collection_exists) {
+          yield _this3.createCollection(_this3.collection_name, _this3.scope_name);
+          yield _this3.wait(2000);
+          yield _this3.createPrimaryIndex(_this3.scope_name, _this3.collection_name);
+        }
+      } catch (e) {
+        console.log(e);
+      }
+    })();
+  }
+
+  collectionExists(collection_name, scope_name) {
+    var _this4 = this;
+
+    return _asyncToGenerator(function* () {
+      try {
+        var _$find;
+
+        var all_scopes = yield axios.get("".concat(_this4.getCouchbaseHost(), "/pools/default/buckets/").concat(_this4.db_config.bucket, "/scopes"), {
+          auth: {
+            username: _this4.db_config.username,
+            password: _this4.db_config.password
           }
         });
-      });
+        var collection_e = _.find(((_$find = _.find(all_scopes.data.scopes, {
+          name: scope_name
+        })) === null || _$find === void 0 ? void 0 : _$find.collections) || [], {
+          name: collection_name
+        }) != undefined;
+        return collection_e;
+      } catch (e) {
+        console.log("Exists error happened", e);
+      }
+    })();
+  }
 
-      return function (_x, _x2) {
-        return _ref.apply(this, arguments);
-      };
-    }());
+  createCollection(connection_name, scope_name) {
+    var _this5 = this;
+
+    return _asyncToGenerator(function* () {
+      try {
+        var token = "".concat(_this5.db_config.username, ":").concat(_this5.db_config.password);
+        var encodedToken = Buffer.from(token).toString("base64");
+        var headers = {
+          Authorization: "Basic " + encodedToken,
+          "Content-Type": "application/x-www-form-urlencoded"
+        };
+        var response = yield axios({
+          method: "post",
+          url: "".concat(_this5.getCouchbaseHost(), "/pools/default/buckets/").concat(_this5.db_config.bucket, "/scopes/").concat(scope_name, "/collections"),
+          headers,
+          data: qs.stringify({
+            name: connection_name
+          })
+        }).data;
+        return response;
+      } catch (e) {}
+    })();
+  }
+
+  createPrimaryIndex(scope_name, col_name) {
+    var _this6 = this;
+
+    return _asyncToGenerator(function* () {
+      try {
+        var response = yield _this6.query("CREATE PRIMARY INDEX ON default:`".concat(_this6.db_config.bucket, "`.`").concat(scope_name, "`.`").concat(col_name, "`"));
+        return response;
+      } catch (e) {
+        console.log("CREATE Index :::", e);
+      }
+    })();
+  }
+
+  createScope(scope_name) {
+    var _this7 = this;
+
+    return _asyncToGenerator(function* () {
+      try {
+        yield _this7.collection_manager.createScope(scope_name);
+      } catch (e) {
+        if (e instanceof couchbase.ScopeExistsError) {
+          console.log("The scope already exists");
+        } else {
+          throw e;
+        }
+      }
+    })();
+  }
+
+  get(doc_id) {
+    var _this8 = this;
+
+    return _asyncToGenerator(function* () {
+      try {
+        var response = yield _this8.connection.get(doc_id);
+        return _objectSpread({
+          id: doc_id
+        }, response.content);
+      } catch (e) {
+        console.error(e);
+      }
+    })();
   }
 
   insert(doc_id, document, expiry) {
@@ -74,13 +243,13 @@ module.exports = class Database {
     return new Promise(
     /*#__PURE__*/
     function () {
-      var _ref2 = _asyncToGenerator(function* (resolve, reject) {
+      var _ref = _asyncToGenerator(function* (resolve, reject) {
         that.connection.insert(doc_id, document, {
           expiry: expiry
         },
         /*#__PURE__*/
         function () {
-          var _ref3 = _asyncToGenerator(function* (err, res) {
+          var _ref2 = _asyncToGenerator(function* (err, res) {
             if (err) {
               reject(err);
             } else {
@@ -93,120 +262,59 @@ module.exports = class Database {
             }
           });
 
-          return function (_x5, _x6) {
-            return _ref3.apply(this, arguments);
+          return function (_x3, _x4) {
+            return _ref2.apply(this, arguments);
           };
         }());
       });
 
-      return function (_x3, _x4) {
-        return _ref2.apply(this, arguments);
+      return function (_x, _x2) {
+        return _ref.apply(this, arguments);
       };
     }());
   }
 
   touch(doc_id) {
-    var expiry = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 0;
-    var that = this;
-    return new Promise(
-    /*#__PURE__*/
-    function () {
-      var _ref4 = _asyncToGenerator(function* (resolve, reject) {
-        that.connection.touch(doc_id, expiry,
-        /*#__PURE__*/
-        function () {
-          var _ref5 = _asyncToGenerator(function* (err, res) {
-            if (err) {
-              reject(err);
-            } else {
-              try {
-                var created = yield that.get(doc_id);
-                resolve(created);
-              } catch (e) {
-                reject(e);
-              }
-            }
-          });
-
-          return function (_x9, _x10) {
-            return _ref5.apply(this, arguments);
-          };
-        }());
-      });
+    var _this9 = this;
 
-      return function (_x7, _x8) {
-        return _ref4.apply(this, arguments);
-      };
-    }());
+    var expiry = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 0;
+    return _asyncToGenerator(function* () {
+      yield _this9.connection.touch(doc_id, expiry);
+      return yield that.get(doc_id);
+    })();
   }
 
   count(doc_id, adjustment, initial_value) {
-    var that = this;
-    return new Promise(
-    /*#__PURE__*/
-    function () {
-      var _ref6 = _asyncToGenerator(function* (resolve, reject) {
-        that.connection.counter(doc_id, adjustment, {
-          initial: initial_value
-        }, function (err, res) {
-          if (err) {
-            reject(err);
-          } else {
-            resolve(res.value);
-          }
-        });
-      });
+    var _this10 = this;
 
-      return function (_x11, _x12) {
-        return _ref6.apply(this, arguments);
-      };
-    }());
+    return _asyncToGenerator(function* () {
+      var res = yield _this10.connection.counter(doc_id, adjustment, {
+        initial: initial_value
+      });
+      return res.value;
+    })();
   }
 
   update(doc_id, document) {
-    var that = this;
-    return new Promise(
-    /*#__PURE__*/
-    function () {
-      var _ref7 = _asyncToGenerator(function* (resolve, reject) {
-        that.connection.replace(doc_id, document, function (err, res) {
-          if (err) {
-            reject(err);
-          } else {
-            resolve(res.value);
-          }
-        });
-      });
+    var _this11 = this;
 
-      return function (_x13, _x14) {
-        return _ref7.apply(this, arguments);
-      };
-    }());
+    return _asyncToGenerator(function* () {
+      var res = yield _this11.connection.replace(doc_id, document);
+      return res.value;
+    })();
   }
 
   delete(doc_id) {
-    var that = this;
-    return new Promise(
-    /*#__PURE__*/
-    function () {
-      var _ref8 = _asyncToGenerator(function* (resolve, reject) {
-        that.connection.remove(doc_id, function (err, res) {
-          if (err) {
-            reject(err);
-          } else {
-            resolve(res.value);
-          }
-        });
-      });
+    var _this12 = this;
 
-      return function (_x15, _x16) {
-        return _ref8.apply(this, arguments);
-      };
-    }());
+    return _asyncToGenerator(function* () {
+      var res = yield _this12.connection.remove(doc_id);
+      return res.value;
+    })();
   }
 
   multi_get(keys, keep_type) {
-    var _this = this;
+    var _this13 = this;
 
     return _asyncToGenerator(function* () {
       /* var that = this; */
@@ -218,9 +326,14 @@ module.exports = class Database {
 
 
       if (_.isArray(keys) && keys.length > 0) {
-        var functions = _.map(keys, key => _this.connection.get(key));
+        var functions = _.map(keys, key => _this13.connection.get(key));
+
+        var returned_all = yield Promise.allSettled(functions);
+
+        var responses = _.flatten(_.compact(_.map(returned_all, returned_obj => {
+          if (returned_obj.status == "fulfilled") return returned_obj.value;
+        })));
 
-        var responses = yield Promise.all(functions);
         return _.map(responses, (res, index) => {
           if (!keep_type) {
             delete res.value.type;
@@ -229,23 +342,6 @@ module.exports = class Database {
           res.value.id = keys[index];
           return res.value;
         });
-        /* that.connection.getMulti(keys, function (err, res) {
-            if (res) {
-              let return_value = [];
-              for (var i in res) {
-                if (res[i].value) {
-                  if (!keep_type) {
-                    delete res[i].value['type']
-                  }
-                  res[i].value.id = i;
-                  return_value.push(res[i].value);
-                }
-              }
-              resolve(return_value);
-            } else if (err) {
-              reject(err);
-            }
-          }); */
       } else {
         return [];
       }
@@ -257,7 +353,7 @@ module.exports = class Database {
     return new Promise(
     /*#__PURE__*/
     function () {
-      var _ref9 = _asyncToGenerator(function* (resolve, reject) {
+      var _ref3 = _asyncToGenerator(function* (resolve, reject) {
         var ViewQuery = couchbase.ViewQuery;
         var query = ViewQuery.from(design_doc, view);
 
@@ -284,7 +380,7 @@ module.exports = class Database {
         that.connection.query(query,
         /*#__PURE__*/
         function () {
-          var _ref10 = _asyncToGenerator(function* (err, results, matrix) {
+          var _ref4 = _asyncToGenerator(function* (err, results, matrix) {
             var return_value = [];
 
             if (err) {
@@ -307,100 +403,112 @@ module.exports = class Database {
             }
           });
 
-          return function (_x19, _x20, _x21) {
-            return _ref10.apply(this, arguments);
+          return function (_x7, _x8, _x9) {
+            return _ref4.apply(this, arguments);
           };
         }());
       });
 
-      return function (_x17, _x18) {
-        return _ref9.apply(this, arguments);
+      return function (_x5, _x6) {
+        return _ref3.apply(this, arguments);
       };
     }());
   }
 
-  search(type, columns, query, limit, offset, order_by, do_count, statement_consistent) {
-    var that = this;
-    return new Promise(
-    /*#__PURE__*/
-    function () {
-      var _ref11 = _asyncToGenerator(function* (resolve, reject) {
-        try {
-          if (do_count == undefined) {
-            do_count = "false";
-          }
+  search(type, columns, query, limit, offset, order_by, do_count, query_scope) {
+    var _this14 = this;
 
-          var fix = "";
-          var count_query = "";
+    return _asyncToGenerator(function* () {
+      var use_bucket_name = !(spice.config.database.connections[_this14.connection_name].scope && spice.config.database.connections[_this14.connection_name].collection);
+      var is_legacy = _this14.scope_name !== "_default" && _this14.collection_name !== "_default";
+      var alias = "_reference";
+      var bucket = "`".concat(use_bucket_name ? _this14.bucket : _this14.collection_name, "`");
 
-          if (!limit) {
-            limit = 0;
-          }
+      try {
+        //let uses_scopes = this.scope this.scope_name !== "_default";
+        if (do_count == undefined) {
+          do_count = "false";
+        }
 
-          if (!offset) {
-            offset = 0;
-          }
+        var fix = "";
+        var count_query = "";
 
-          if (columns == "*") {
-            columns = "(" + that.bucket + ").*";
-            fix = ", META(" + that.bucket + ").id  AS id";
-          }
+        if (!limit) {
+          limit = 0;
+        }
 
-          if (!columns) {
-            columns = "(" + that.bucket + ").*";
-            fix = ", META(" + that.bucket + ").id AS id";
-          }
+        if (!offset) {
+          offset = 0;
+        }
 
-          if (columns == "id") {
-            fix = ", META(" + that.bucket + ").id AS id";
-            columns = "";
-          }
+        if (columns == "*") {
+          columns = "(" + bucket + ").*";
+          fix = ", META(" + bucket + ").id  AS id";
+        }
 
-          var front_query = "SELECT " + columns + fix + " FROM " + that.bucket + " WHERE type='" + type + "'";
-          count_query = "SELECT count(meta().id) as count FROM " + that.bucket + " WHERE type='" + type + "'";
+        if (!columns) {
+          columns = "(" + bucket + ").*";
+          fix = ", META(" + bucket + ").id AS id";
+        }
 
-          if (query) {
-            count_query = count_query + " AND " + query;
-            query = front_query + " AND " + query;
-          } else {
-            query = front_query;
-          }
+        if (columns == "id") {
+          fix = ", META(" + bucket + ").id AS id";
+          columns = "";
+        }
 
-          if (order_by) {
-            query = "".concat(query, " ORDER BY ").concat(order_by);
-          }
+        var front_query = "SELECT ".concat(columns, " ").concat(fix, " FROM ").concat(bucket);
 
-          if (limit > 0) {
-            query = query + " LIMIT " + limit;
-          }
+        if (is_legacy || query) {
+          front_query += " WHERE ";
+        }
 
-          if (offset > 0) {
-            query = query + " OFFSET " + offset;
-          }
+        var conditions = [];
 
-          var query_element = query;
-          var count_query_element = count_query;
-          var queries = [that.getData(query_element)];
+        if (use_bucket_name) {
+          conditions.push("type = '".concat(type, "'"));
+        }
 
-          if (do_count === "true") {
-            queries.push(that.getCount(count_query_element));
-          }
+        if (query) {
+          conditions.push(query);
+        }
 
-          var responses = yield Promise.all(queries);
-          resolve({
-            data: responses[0],
-            total: responses.length > 1 ? responses[1] : null
-          });
-        } catch (error) {
-          console.log(error.stack);
-          reject(error);
+        query = front_query + conditions.join(" AND ");
+        count_query = "SELECT count(meta().id) as count FROM ".concat(bucket);
+
+        if (use_bucket_name || query) {
+          count_query += " WHERE ";
         }
-      });
 
-      return function (_x22, _x23) {
-        return _ref11.apply(this, arguments);
-      };
-    }());
+        query = front_query + conditions.join(" AND ");
+
+        if (order_by) {
+          query = "".concat(query, " ORDER BY ").concat(order_by);
+        }
+
+        if (limit > 0) {
+          query = query + " LIMIT " + limit;
+        }
+
+        if (offset > 0) {
+          query = query + " OFFSET " + offset;
+        }
+
+        var active_query_space = query_scope || !use_bucket_name ? "scope" : "cluster";
+        var queries = [_this14.getData(query, active_query_space)];
+
+        if (do_count === "true") {
+          queries.push(_this14.getCount(count_query, active_query_space));
+        }
+
+        var responses = yield Promise.all(queries);
+        return {
+          data: responses[0],
+          total: responses.length > 1 ? responses[1] : null
+        };
+      } catch (error) {
+        throw error;
+      }
+    })();
   }
 
   full_text_search(type, search_query, limit) {
@@ -410,7 +518,7 @@ module.exports = class Database {
     return new Promise(
     /*#__PURE__*/
     function () {
-      var _ref12 = _asyncToGenerator(function* (resolve, reject) {
+      var _ref5 = _asyncToGenerator(function* (resolve, reject) {
         try {
           var full_query = search_query + " AND type:" + type;
           var query = that.SearchQuery.new(that.full_text_index || full_text_index, that.SearchQuery.queryString(full_query));
@@ -431,7 +539,7 @@ module.exports = class Database {
           that.cluster.query(query,
           /*#__PURE__*/
           function () {
-            var _ref13 = _asyncToGenerator(function* (err, results, matrix) {
+            var _ref6 = _asyncToGenerator(function* (err, results, matrix) {
               var return_value = [];
 
               if (err) {
@@ -457,8 +565,8 @@ module.exports = class Database {
               }
             });
 
-            return function (_x26, _x27, _x28) {
-              return _ref13.apply(this, arguments);
+            return function (_x12, _x13, _x14) {
+              return _ref6.apply(this, arguments);
             };
           }());
         } catch (e) {
@@ -466,35 +574,44 @@ module.exports = class Database {
         }
       });
 
-      return function (_x24, _x25) {
-        return _ref12.apply(this, arguments);
+      return function (_x10, _x11) {
+        return _ref5.apply(this, arguments);
       };
     }());
   }
 
   getCount(count_query_element) {
-    var _this2 = this;
+    var _this15 = this;
 
+    var query_scope = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : "cluster";
     return _asyncToGenerator(function* () {
-      var response = yield _this2.cluster.query(count_query_element);
+      var response = yield _this15.query(count_query_element, query_scope);
       return response.rows[0].count;
     })();
   }
 
   getData(query_element) {
-    var _this3 = this;
+    var _this16 = this;
 
+    var query_scope = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : "cluster";
     return _asyncToGenerator(function* () {
-      var response = yield _this3.cluster.query(query_element);
+      var response = yield _this16.query(query_element, query_scope);
       return response.rows;
     })();
   }
 
   query(query) {
-    var _this4 = this;
+    var _this17 = this;
 
+    var query_scope = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : "cluster";
     return _asyncToGenerator(function* () {
-      return _this4.cluster.query(query);
+      var scope = query_scope || _this17.query_scope;
+
+      if (scope == "cluster") {
+        return yield _this17.cluster.query(query);
+      } else {
+        return yield _this17.scope.query(query);
+      }
     })();
   }