StructuredData uses entity to store data

whatsapp-web.js/src/models/data.js

@@ -1,21 +1,18 @@
+const { Entity } = require("./entity");
 const { User } = require("./user");
-const { Website } = require("./website");
 
 
 class StructuredData {
     
     constructor() {
-        this.data={};
+        this.entity = new Entity();
     }
-    
+    /**
-    addUser(id){
+     * Adds a user 
-        if(this.data[id] != undefined)
+     * @param {User} user
-        {
+     */
-            this.data[id] = new User(id);
+    addUser(user){
-            return true
+        return this.entity.addData(user.id, user);
-        }
-        
-        return false;
     }
     
     /**
@@ -24,17 +21,30 @@ class StructuredData {
      * @returns {User}
      */
     getUser(id){
-        return this.data[id];
+        return this.entity.getData(id);
     }
     
+    /**
+     * Delete the user with a specific id
+     * @param {string} id 
+     */
     deleteUser(id){
-        if(this.data[id] != undefined)
+        return this.entity.deleteData(id)
-        {
+    }
-            return delete this.data[id];
+
-            
+    /**
-        }
+     * Check if a user with the given id exists
-        
+     * @param {string} id 
-        return false;
+     */
+    isUser(id){
+        return this.entity.isData(id);
+    }
+
+    /**
+     * Clears all the data stored
+     */
+    clearData(){
+        this.entity.clearData();
     }