forked from cfsworkload/blue-messenger
-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.js
130 lines (99 loc) · 3.95 KB
/
app.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
// Copyright (c) 2016 IBM Corp. All rights reserved.
// Use of this source code is governed by the Apache License,
// Version 2.0, a copy of which can be found in the LICENSE file.
// This application uses express as it's web server
// for more info, see: http://expressjs.com
var express = require('express');
var http = require('http');
var mosca = require('mosca');
var fs = require('fs');
var db;
var cloudant;
var dbCredentials = {
dbName : 'my_sample_db'
};
//Get the port and host name from the environment variables
var port = (process.env.VCAP_APP_PORT || 3000);
var host = (process.env.VCAP_APP_HOST || '0.0.0.0');
//setup cloudant db
function initDBConnection() {
if(process.env.VCAP_SERVICES) {
var vcapServices = JSON.parse(process.env.VCAP_SERVICES);
if(vcapServices.cloudantNoSQLDB) {
dbCredentials.host = vcapServices.cloudantNoSQLDB[0].credentials.host;
dbCredentials.port = vcapServices.cloudantNoSQLDB[0].credentials.port;
dbCredentials.user = vcapServices.cloudantNoSQLDB[0].credentials.username;
dbCredentials.password = vcapServices.cloudantNoSQLDB[0].credentials.password;
dbCredentials.url = vcapServices.cloudantNoSQLDB[0].credentials.url;
}
console.log('VCAP Services: '+JSON.stringify(process.env.VCAP_SERVICES));
}
else{
dbCredentials.host = "ffe37731-0505-4683-96a8-87d02a33e03e-bluemix.cloudant.com";
dbCredentials.port = 443;
dbCredentials.user = "ffe37731-0505-4683-96a8-87d02a33e03e-bluemix";
dbCredentials.password = "c7003d0b156d9c4ce856c4e6b4427f3b576c7ea6229235f0369ada1ed47b159c";
dbCredentials.url = "https://ffe37731-0505-4683-96a8-87d02a33e03e-bluemix:c7003d0b156d9c4ce856c4e6b4427f3b576c7ea6229235f0369ada1ed47b159c@ffe37731-0505-4683-96a8-87d02a33e03e-bluemix.cloudant.com";
}
cloudant = require('cloudant')(dbCredentials.url);
//check if DB exists if not create
cloudant.db.create(dbCredentials.dbName, function (err, res) {
if (err) { console.log('could not create db ', err); }
});
db = cloudant.use(dbCredentials.dbName);
}
initDBConnection();
// create a new express server
var app = express();
app.set('port', port);
app.set('view engine', 'ejs');
// Enable reverse proxy support in Express. This causes the
// the "X-Forwarded-Proto" header field to be trusted so its
// value can be used to determine the protocol. See
// http://expressjs.com/api#app-settings for more details.
app.enable('trust proxy');
// Add a handler to inspect the req.secure flag (see
// http://expressjs.com/api#req.secure). This allows us
// to know whether the request was via http or https.
app.use (function (req, res, next) {
if (req.secure) {
// request was via https
res.redirect('http://' + req.headers.host + req.url);
} else {
// request was via http
next();
}
});
// serve the files out of ./public as our main files
app.use(express.static(__dirname + '/public'));
/**
* This is our home route. This gets called when we visit our
* base address http://MYSERVERNAME.mybluemix.net/
**/
app.get('/', function(req, res) {
res.sendfile(__dirname + '/public/index.html');
});
// Create the MQTT server
var mqttServe = new mosca.Server({});
mqttServe.on('clientConnected', function(client) {
console.log('client connected', client.id);
});
mqttServe.on('published', function(packet, client){
console.log('Message: ', packet.payload.toString("utf8"));
fs.appendFile("../logs/mqtt.log", packet.topic + ": " + packet.payload.toString("utf8") + "\n", function(err) {
if(err) {
return console.log(err);
}
});
db.insert({"Topic": packet.topic, "Message": packet.payload.toString("utf8")}, function(err, body) {
if (!err)
console.log(body);
});
});
//Create http server and attach the MQTT server to the websocket
var httpServer = http.createServer(app);
mqttServe.attachHttpServer(httpServer);
//begin listening
httpServer.listen(app.get('port'), function(){
console.log('Express server listening on port ' + app.get('port'));
});