added data model and user model

This commit is contained in:
Mohammed Ashab Uddin
2020-07-21 13:58:04 +04:00
parent 95dc1ae218
commit 0efda2dc11
2 changed files with 93 additions and 0 deletions

View File

@@ -0,0 +1,43 @@
const {Website} = require('./website');
class User{
constructor(id){
this.id = id;
//uses key value pair to store website
//key is the companyName of the website
//value is the website itself
this.websites = {};
}
/**
*
* @param {Website} website
*/
addWebsite(website){
if(this.websites[website.companyName] != undefined)
{
return false;
}else{
this.websites[website.companyName] = website;
return true;
}
}
deleteWebsite(companyName){
if(this.websites[companyName] != undefined)
{
return delete this.websites[companyName];
}else{
return false
}
}
getWebsite(companyName){
return this.websites[companyName];
}
}
module.exports.User = User;