From 596f56b708688de2fb278733fa50334be0dabd82 Mon Sep 17 00:00:00 2001
From: Mohammed Ashab Uddin <mu15@hw.ac.uk>
Date: Wed, 22 Jul 2020 10:43:52 +0400
Subject: [PATCH] StructuredData uses entity to store data

---
 whatsapp-web.js/src/models/data.js | 48 ++++++++++++++++++------------
 1 file changed, 29 insertions(+), 19 deletions(-)

diff --git a/whatsapp-web.js/src/models/data.js b/whatsapp-web.js/src/models/data.js
index 2ec48b1..6b672e9 100644
--- a/whatsapp-web.js/src/models/data.js
+++ b/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){
-        if(this.data[id] != undefined)
-        {
-            this.data[id] = new User(id);
-            return true
-        }
-        
-        return false;
+    /**
+     * Adds a user 
+     * @param {User} user
+     */
+    addUser(user){
+        return this.entity.addData(user.id, user);
     }
     
     /**
@@ -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 delete this.data[id];
-            
-        }
-        
-        return false;
+        return this.entity.deleteData(id)
+    }
+
+    /**
+     * Check if a user with the given id exists
+     * @param {string} id 
+     */
+    isUser(id){
+        return this.entity.isData(id);
+    }
+
+    /**
+     * Clears all the data stored
+     */
+    clearData(){
+        this.entity.clearData();
     }