xo-server/src/main.js
2013-11-02 15:09:05 +00:00

484 lines
10 KiB
JavaScript

var _ = require('underscore');
var connect = require('connect');
var fs = require('fs');
var Q = require('q');
var Session = require('./session');
var WSServer = require('ws').Server;
//--------------------------------------
var xo = require('./xo')();
var Api = require('./api');
var api = new Api(xo);
var http_servers = [];
// Modified by Martin Dobrev @ 2013-10-22
var ldap = require('ldapjs');
var Knex = require('knex');
ldap.Attribute.settings.guid_format = ldap.GUID_FORMAT_B;
// End of mod by Martin Dobrev
//////////////////////////////////////////////////////////////////////
function json_api_call(session, message)
{
/* jshint newcap:false */
var req = {
'id': null,
};
function format_error(error)
{
return JSON.stringify({
'jsonrpc': '2.0',
'error': error,
'id': req.id,
});
}
try
{
req = JSON.parse(message.toString());
}
catch (e)
{
if (e instanceof SyntaxError)
{
return Q(format_error(Api.err.INVALID_JSON));
}
return Q(format_error(Api.err.SERVER_ERROR));
}
/* jshint laxbreak: true */
if (!req.method || !req.params
|| (undefined === req.id)
|| ('2.0' !== req.jsonrpc))
{
return Q(format_error(Api.err.INVALID_REQUEST));
}
return api.exec(
session,
{
'method': req.method,
'params': req.params,
}
).then(
function (result) {
return JSON.stringify({
'jsonrpc': '2.0',
'result': result,
'id': req.id,
});
},
function (error) {
if (!_.isObject(error) || (error instanceof Error))
{
console.error(error.stack || error);
return format_error(Api.err.SERVER_ERROR);
}
return format_error(error);
}
);
}
//////////////////////////////////////////////////////////////////////
// Static file serving (for XO-Web for instance).
//////////////////////////////////////////////////////////////////////
xo.on('started', function () {
http_servers.forEach(function (http_server) {
http_server.on('request', connect()
// Compresses reponses using GZip.
.use(connect.compress())
// Caches the responses in memory.
//.use(connect.staticCache())
// Serve static files.
.use(connect.static(__dirname +'/../public/http'))
);
});
});
//////////////////////////////////////////////////////////////////////
// WebSocket to TCP proxy (used for consoles).
//////////////////////////////////////////////////////////////////////
// Protocol:
//
// 1. The web browser connects to the server via WebSocket.
//
// 2. It sends a first message containing the “host” and “port” to
// connect to in a JSON object.
//
// 3. All messages to send to the TCP server and received from it will
// be encoded using Base64.
// @todo Avoid Base64 encoding and directly use binary streams.
// xo.on('started', function () {
// function on_connection(socket)
// {
// // Parses the first message which SHOULD contains the host and
// // port of the host to connect to.
// socket.once('message', function (message) {
// try
// {
// message = JSON.parse(message);
// }
// catch (e)
// {
// socket.close();
// return;
// }
// if (!message.host && !message.port)
// {
// socket.close();
// return;
// }
// var target = tcp.createConnection(message.port, message.host);
// target.on('data', function (data) {
// socket.send(data.toString('base64'));
// });
// target.on('end', function () {
// socket.close();
// });
// target.on('error', function () {
// target.end();
// });
// socket.on('message', function (message) {
// target.write(new Buffer(message, 'base64'));
// });
// socket.on('close', function () {
// target.end();
// });
// });
// socket.on('error', function () {
// socket.close();
// });
// }
// http_servers.forEach(function (http_server) {
// new WSServer({
// 'server': http_server,
// 'path': '/websockify',
// })
// .on('connection', on_connection);
// });
// });
//////////////////////////////////////////////////////////////////////
// JSON-RPC over WebSocket.
//////////////////////////////////////////////////////////////////////
xo.on('started', function () {
function on_connection(socket)
{
var session = new Session(xo);
session.once('close', function () {
socket.close();
});
socket.on('message', function (request) {
json_api_call(session, request).then(function (response) {
// Send response if session still open.
if (socket.readyState === socket.OPEN)
{
socket.send(response);
}
}).done();
});
// @todo Ugly inter dependency.
socket.once('close', function () {
session.close();
});
}
http_servers.forEach(function (http_server) {
new WSServer({
'server': http_server,
'path': '/api/',
}).on('connection', on_connection);
});
});
//////////////////////////////////////////////////////////////////////
// JSON-RPC over TCP.
//////////////////////////////////////////////////////////////////////
// xo.on('started', function () {
// require('net').createServer(function (socket) {
// var session = new Session(xo);
// session.on('close', function () {
// socket.end(); // @todo Check it is enough.
// });
// var length = null; // Expected message length.
// var buffer = new Buffer(1024); // @todo I hate hardcoded values!
// socket.on('data', function (data) {
// data.copy(buffer);
// // Read the message length.
// if (!length)
// {
// var i = _.indexOf(buffer, 10);
// if (-1 === i)
// {
// return;
// }
// length = +(buffer.toString('ascii', 0, i));
// // If the length is NaN, we cannot do anything except
// // closing the connection.
// if (length !== length)
// {
// session.close();
// return;
// }
// buffer = buffer.slice(i + 1);
// }
// // We do not have received everything.
// if (buffer.length < length)
// {
// return;
// }
// json_api_call(
// session,
// buffer.slice(0, length).toString()
// ).then(function (response) {
// // @todo Handle long messages.
// socket.write(response.length +'\n'+ response);
// }).done();
// // @todo Check it frees the memory.
// buffer = buffer.slice(length);
// length = null;
// });
// // @todo Ugly inter dependency.
// socket.once('close', function () {
// session.close();
// });
// }).listen(1024); // @todo Should be configurable.
// });
//////////////////////////////////////////////////////////////////////
var cfg = {
'data': {},
'get': function (path) {
/* jshint noempty: false */
if (!_.isArray(path))
{
path = Array.prototype.slice.call(arguments);
}
var current = this.data;
for (
var i = 0, n = path.length;
(i < n) && (undefined !== (current = current[path[i]]));
++i
)
{}
if (i < n)
{
return undefined;
}
return current;
},
'merge': function (data) {
var helper = function (target, source) {
if (null === source) // Special case.
{
return target;
}
if (!_.isObject(target) || !_.isObject(source))
{
return source;
}
if (_.isArray(target) && _.isArray(source))
{
target.push.apply(target, source);
return target;
}
for (var prop in source)
{
target[prop] = helper(target[prop], source[prop]);
}
return target;
};
helper(this.data, data);
return this;
},
};
// Defaults values.
cfg.merge({
'http': {
'enabled': true,
'host': '0.0.0.0',
'port': 80,
},
'https': {
'enabled': false,
'host': '0.0.0.0',
'port': 443,
'certificate': './certificate.pem',
'key': './key.pem',
},
'redis': {
'uri': 'tcp://127.0.0.1:6379',
},
// Modified by Martin Dobrev @ 2013-10-22
'mysql': {
'enabled' : false,
'username' : 'xoauser',
'password' : 'xoapass',
'database' : 'xoa',
'host' : 'localhost',
'port' : 3306,
},
'ldap': {
'enabled' : false,
'host' : 'localhost',
'binddn' : 'dc=example,dc=com',
},
'userdb': {
'type' : 'redis' // Accepted values 'redis', 'mysql' and 'ldap'. More to come later on
},
// End of mod by Martin Dobrev
});
function read_file(file)
{
return Q.ninvoke(fs, 'readFile', file, {'encoding': 'utf-8'});
}
read_file(__dirname +'/../config/local.yaml').then(
function (data) {
data = require('js-yaml').safeLoad(data);
cfg.merge(data);
},
function (e) {
console.error('[Warning] Reading config file: '+ e);
}
).then(function () {
if (cfg.get('users'))
{
console.warn('[Warn] Users in config file are no longer supported.');
}
if (cfg.get('servers'))
{
console.warn('[Warn] Servers in config file are no longer supported.');
}
if (cfg.get('http', 'enabled'))
{
var host = cfg.get('http', 'host');
var port = cfg.get('http', 'port');
http_servers.push(
require('http').createServer().listen(port, host)
.on('listening', function () {
console.info(
'XO-Server HTTP server is listening on %s:%s',
host, port
);
})
);
}
// Modified by Martin Dobrev
if (cfg.get('mysql', 'enabled'))
{
var host = cfg.get('mysql', 'host').toString();
var port = cfg.get('mysql', 'port');
var database = cfg.get('mysql', 'database').toString();
var username = cfg.get('mysql', 'username').toString();
var password = cfg.get('mysql', 'password').toString();
// Initialize Knex
//
// /!\ Initialize it only once
//
// on later stage use knex = require('knex').knex for DB queries
Knex.knex = Knex.initialize(
{
client: 'mysql',
connection:
{
host: host,
user: username,
password: password,
database: database,
port: port,
charset: 'utf8',
}
});
}
// End of mod by Martin Dobrev
if (cfg.get('https', 'enabled'))
{
return Q.all([
read_file(cfg.get('https', 'certificate')),
read_file(cfg.get('https', 'key')),
]).spread(function (certificate, key) {
var host = cfg.get('https', 'host');
var port = cfg.get('https', 'port');
http_servers.push(
require('https').createServer({
'cert': certificate,
'key': key,
}).listen(port, host).on('listening', function () {
console.info(
'XO-Server HTTPS server is listening on %s:%s',
host, port
);
})
);
});
}
}).then(function () {
xo.start(cfg);
}).done();
// Create an initial user if there are none.
xo.on('started', function () {
xo.users.exists().then(function (success) {
if (success)
{
return;
}
console.warn('[Warning] No users, creating “admin@admin.net” with password “admin”');
return xo.users.create('admin@admin.net', 'admin', 'admin');
}).done();
});