43 lines
873 B
JavaScript
43 lines
873 B
JavaScript
|
|
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; |