From 3bdf43d11b4ce25bd0efc97e3f3a6fa277cabd67 Mon Sep 17 00:00:00 2001
From: Alexis <35.alexis.pele@gmail.com>
Date: Wed, 30 Dec 2020 21:52:06 +0100
Subject: [PATCH] Updated models to include roles

---
 api/models/role-model.js | 7 +++++++
 api/models/user-model.js | 9 +++++++--
 2 files changed, 14 insertions(+), 2 deletions(-)
 create mode 100644 api/models/role-model.js

diff --git a/api/models/role-model.js b/api/models/role-model.js
new file mode 100644
index 0000000..23fcd1d
--- /dev/null
+++ b/api/models/role-model.js
@@ -0,0 +1,7 @@
+const bookshelf = require('../database/bookshelf').bookshelf
+
+let Role = bookshelf.Model.extend({
+    tableName: 'role',
+})
+
+module.exports = bookshelf.model('Role', Role);
\ No newline at end of file
diff --git a/api/models/user-model.js b/api/models/user-model.js
index 9d7c019..a7407c4 100644
--- a/api/models/user-model.js
+++ b/api/models/user-model.js
@@ -1,9 +1,14 @@
 'use strict'
-const bookshelf = require('../database/bookshelf').bookshelf
+const bookshelf = require('../database/bookshelf').bookshelf;
+
+require('./role-model');
 
 let User = bookshelf.Model.extend({
     tableName: 'user',
-    hidden: ['id', 'password']
+    hidden: [ 'password' ],
+    role() {
+        return this.belongsTo( 'Role' );
+    },
 })
 
 module.exports = bookshelf.model('User', User)
\ No newline at end of file