From d296e5ca4384b1e47cae9a3c114b7710a588c39c Mon Sep 17 00:00:00 2001
From: Sean Clarke <me@seanland.ca>
Date: Mon, 6 Jan 2020 16:42:13 -0500
Subject: [PATCH] visit model created

---
 controllers/visits.js |  1 +
 model/Visit.js        | 22 ++++++++++++++++++++++
 routes/index.js       |  5 -----
 routes/sites.js       |  1 +
 4 files changed, 24 insertions(+), 5 deletions(-)
 create mode 100644 controllers/visits.js

diff --git a/controllers/visits.js b/controllers/visits.js
new file mode 100644
index 0000000..f2c02a1
--- /dev/null
+++ b/controllers/visits.js
@@ -0,0 +1 @@
+var Visit = require('../model/Visit');
diff --git a/model/Visit.js b/model/Visit.js
index e69de29..5151c87 100644
--- a/model/Visit.js
+++ b/model/Visit.js
@@ -0,0 +1,22 @@
+var con = require('../database/db');
+
+var Visit = function(data){
+  this.id = data.id;
+  this.date = data.date;
+  this.img = data.img;
+  this.site_id = data.site_id;
+};
+
+Visit.findBySite = function(id, callback){
+  var sql = "SELECT * FROM visits WHERE site_id=?";
+
+  con.query(sql, id, function(err, result){
+    if (err) return callback(err);
+    // if not site is found.
+    if (result[0] == undefined) callback(404);
+    else {
+      console.log("Visit.findBySite: " + result.length + "visits retrieved!");
+      callback(err, result);
+    }
+  });
+};
diff --git a/routes/index.js b/routes/index.js
index 5314356..fe6d848 100644
--- a/routes/index.js
+++ b/routes/index.js
@@ -3,11 +3,6 @@ var router = express.Router();
 
 var sitesController = require('../controllers/sites')
 
-/* GET home page. */
-// router.get('/', function(req, res, next) {
-//   res.render('index', { title: 'UNESCO Tracker - Sean Clarke' });
-// });
-
 router.get('/', sitesController.getAll);
 
 module.exports = router;
diff --git a/routes/sites.js b/routes/sites.js
index e1cef3f..ad6376d 100644
--- a/routes/sites.js
+++ b/routes/sites.js
@@ -3,6 +3,7 @@ var router = express.Router();
 
 var siteController = require('../controllers/sites');
 
+router.get('/', siteController.getAll);
 router.get('/:id', siteController.findById);
 
 module.exports = router;