var fs = require('fs') var homedir = require('os').homedir() var path = homedir + '/.bogbook/' var bogdir = path + 'bogs/' var confpath = path + 'config.json' if (!fs.existsSync(homedir + '/.bogbook/')) {fs.mkdirSync(homedir + '/.bogbook/')} if (!fs.existsSync(bogdir)){fs.mkdirSync(bogdir)} if (fs.existsSync(confpath)) { console.log('loading config from ' + confpath) var config = require(confpath) } else { var config = { port: '8089', wsport: '8080', url: 'localhost', author: '@Q++V5BbvWIg8B+TqtC9ZKFhetruuw+nOgxEqfjlOZI0=' } config.fullurl = 'http://' + config.url + ':' + config.port + '/' fs.writeFileSync(confpath, JSON.stringify(config), 'utf-8') } console.log(config) if (process.argv[2] === 'verbose') { var VERBOSE = true } else { var VERBOSE = false } console.log('Verbose output is ' + VERBOSE + ' run with `node server verbose` to see all output') // log messages function printAsk(req, unboxedreq) { if (VERBOSE) { console.log(req.requester + ' asked for feed ' + unboxedreq.author + ' after sequence ' + unboxedreq.seq) } } function printNewFeed (msg, req) { if (VERBOSE) console.log('Saved full log of ' + msg.author + ' sent by ' + req.requester) else console.log('NEW FEED from ' + msg.author) } function printUpdateFeed (msg, req) { if (VERBOSE) console.log('combined existing feed of ' + msg.author + ' sent from ' + req.requester + ' with diff and saved to server') else console.log('NEW UPDATE from ' + msg.author) } function printNoFeed (msg, req) { if (VERBOSE) { console.log('We don\'t have the log on the server, requesting log from ' + req.requester ) } } function printClientLonger (msg, req) { if (VERBOSE) { console.log(req.requester + '\'s feed of ' + msg.author + ' is longer, requesting diff from ' + req.requester) } } function printClientShorter (msg, req, baserange, endrange) { if (VERBOSE) { console.log(req.requester + ' feed of ' + msg.author + ' is shorter, sending from ' + baserange + ' to ' + endrange + ' to ' + req.requester) } } function printFeedIdentical (msg, req) { if (VERBOSE) { console.log(msg.author + '\'s feed sent from ' + req.requester + ' is identical') } } // static server (8089) var serve = require('koa-static-server') var koa = require('koa') var open = require('open') var app = new koa() app.use(serve({rootDir: '.', notFoundFile: 'index.html'})) app.listen(config.port) open(config.fullurl) console.log('Bogbook is running at: ' + config.fullurl) var bog = require('./bog') var WS = require('ws') var nacl = require('tweetnacl') nacl.util = require('tweetnacl-util') var wserve = new WS.Server({ port: config.wsport }) bog.keys().then(key => { wserve.on('connection', function (ws) { ws.on('message', function (message) { var req = JSON.parse(message) if (req.sendpub) { ws.send(key.publicKey) } else { bog.unbox(req.box, req.requester, key).then(unboxed => { var unboxedreq = JSON.parse(unboxed) //console.log(unboxedreq) if (unboxedreq.seq >= 0) { printAsk(req, unboxedreq) fs.readFile(bogdir + unboxedreq.author, 'UTF-8', function (err, data) { if (data) { var feed = JSON.parse(data) bog.open(feed[0]).then(msg => { if (unboxedreq.seq === msg.seq) { printFeedIdentical(msg, req) } if (unboxedreq.seq > msg.seq) { printClientLonger(msg, req) var reqdiff = JSON.stringify({author: unboxedreq.author, seq: msg.seq}) bog.box(reqdiff, req.requester, key).then(boxed => { var obj = { requester: key.publicKey, box: boxed } ws.send(JSON.stringify(obj)) }) } if (unboxedreq.seq < msg.seq) { var endrange = feed.length - unboxedreq.seq - Math.floor(Math.random() * 50 + 1) console.log(endrange) if (endrange < 0) { endrange = 0 } var baserange = feed.length - unboxedreq.seq printClientShorter(msg, req, baserange, endrange) var diff = JSON.stringify( feed.slice( endrange, baserange) ) bog.box(diff, req.requester, key).then(boxed => { var obj = { requester: key.publicKey, box: boxed } ws.send(JSON.stringify(obj)) ws.close() }) } }) } else { printNoFeed(unboxedreq, req) var reqwhole = JSON.stringify({author: unboxedreq.author, seq: 0}) bog.box(reqwhole, req.requester, key).then(boxed => { var obj = { requester: key.publicKey, box: boxed } ws.send(JSON.stringify(obj)) }) } }) } else if (Array.isArray(unboxedreq)) { bog.open(unboxedreq[0]).then(msg => { if (msg.seq === unboxedreq.length) { fs.writeFile(bogdir + msg.author, JSON.stringify(unboxedreq), 'UTF-8', function (err, success) { printNewFeed(msg, req) }) } if (msg.seq > unboxedreq.length) { fs.readFile(bogdir + msg.author, 'UTF-8', function (err, data) { var feed = JSON.parse(data) bog.open(feed[0]).then(lastmsg => { if (unboxedreq.length + lastmsg.seq === msg.seq) { var newlog = unboxedreq.concat(feed) fs.writeFile(bogdir + msg.author, JSON.stringify(newlog), 'UTF-8', function (err, success) { printUpdateFeed(msg, req) }) } }) }) } }) } }) } }) }) })