Add basic way to guess check type from url

Refs fzaninotto#4, fzaninotto#3
Masatoshi · Apr 10, 2012 · 1f847d2 · 1f847d2
1 parent d390e3c
commit 1f847d2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/app/dashboard/app.js b/app/dashboard/app.js
@@ -43,6 +43,7 @@ app.post('/check', function(req, res) {
   var check = new Check(req.body.check);
   check.tags = Check.convertTags(req.body.check.tags);
   check.interval = req.body.check.interval * 1000;
+  check.type = Check.guessType(check.url);
   check.save(function(err) {
     req.flash('info', 'New check has been created');
     res.redirect('/check/' + check._id);
@@ -61,6 +62,7 @@ app.put('/check/:id', function(req, res, next) {
   var check = req.body.check;
   check.tags = Check.convertTags(check.tags);
   check.interval = req.body.check.interval * 1000;
+  check.type = Check.guessType(check.url);
   Check.update({ _id: req.params.id }, { $set: check }, { upsert: true }, function(err) {
     if (err) return next(err);
     req.flash('info', 'Changes have been saved');

diff --git a/models/check.js b/models/check.js
@@ -13,6 +13,7 @@ var CheckMonthlyStat = require('../models/checkMonthlyStat');
 // main model
 var Check = new Schema({
     name        : String
+  , type        : String
   , url         : String
   , interval    : { type: Number, default: 60000 }  // interval between two pings
   , maxTime     : { type: Number, default: 1500 }   // time under which a ping is considered responsive
@@ -231,6 +232,15 @@ Check.statics.convertTags = function(tags) {
   return tags;
 }
 
+Check.statics.guessType = function(url) {
+  if (url.search(/^http:\/\//) != -1) {
+    return 'http';
+  }
+  if (url.search(/^https:\/\//) != -1) {
+    return 'https';
+  }
+}
+
 /**
  * Calls a function for all checks that need to be polled.
  *