refactor(api): moved to seperate file
This commit is contained in:
Alexey Kasyanchuk
639
src/background/api.js
Normal file
639
src/background/api.js
Normal file
@ -0,0 +1,639 @@
|
|||||||
|
const ipaddr = require('ipaddr.js');
|
||||||
|
|
||||||
|
module.exports = ({
|
||||||
|
sphinx,
|
||||||
|
send,
|
||||||
|
recive,
|
||||||
|
p2p,
|
||||||
|
config,
|
||||||
|
baseRowData,
|
||||||
|
torrentClient
|
||||||
|
}) => {
|
||||||
|
let torrentClientHashMap = {}
|
||||||
|
|
||||||
|
let topCache = {};
|
||||||
|
setInterval(() => {
|
||||||
|
topCache = {};
|
||||||
|
}, 24 * 60 * 60 * 1000);
|
||||||
|
|
||||||
|
recive('recentTorrents', function(callback)
|
||||||
|
{
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
sphinx.query('SELECT * FROM `torrents` ORDER BY added DESC LIMIT 0,10', function (error, rows, fields) {
|
||||||
|
if(!rows) {
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let torrents = [];
|
||||||
|
rows.forEach((row) => {
|
||||||
|
torrents.push(baseRowData(row));
|
||||||
|
});
|
||||||
|
|
||||||
|
callback(torrents)
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
recive('statistic', function(callback)
|
||||||
|
{
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
sphinx.query('SELECT count(*) AS torrents, sum(size) AS sz FROM `torrents`', function (error, rows, fields) {
|
||||||
|
if(!rows) {
|
||||||
|
console.error(error)
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let result = {torrents: rows[0].torrents || 0, size: rows[0].sz || 0}
|
||||||
|
|
||||||
|
sphinx.query('SELECT count(*) AS files FROM `files`', function (error, rows, fields) {
|
||||||
|
if(!rows) {
|
||||||
|
console.error(error)
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
result.files = rows[0].files || 0
|
||||||
|
|
||||||
|
callback(result)
|
||||||
|
})
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
const onTorrent = (hash, options, callback) => {
|
||||||
|
if(hash.length != 40)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
// remote request
|
||||||
|
if(options.peer)
|
||||||
|
{
|
||||||
|
console.log('remote torrent request to peer')
|
||||||
|
const peer = p2p.find(options.peer)
|
||||||
|
if(!peer)
|
||||||
|
{
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
delete options.peer;
|
||||||
|
peer.emit('torrent', {hash, options}, (data) => {
|
||||||
|
console.log('remote torrent result')
|
||||||
|
callback(data)
|
||||||
|
})
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
sphinx.query('SELECT * FROM `torrents` WHERE `hash` = ?', hash, function (error, rows, fields) {
|
||||||
|
if(!rows || rows.length == 0) {
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
let torrent = rows[0];
|
||||||
|
|
||||||
|
if(options.files)
|
||||||
|
{
|
||||||
|
sphinx.query('SELECT * FROM `files` WHERE `hash` = ? LIMIT 50000', hash, function (error, rows, fields) {
|
||||||
|
torrent.filesList = rows;
|
||||||
|
callback(baseRowData(torrent))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
callback(baseRowData(torrent))
|
||||||
|
}
|
||||||
|
|
||||||
|
if(torrentClientHashMap[hash])
|
||||||
|
{
|
||||||
|
const torrent = torrentClient.get(torrentClientHashMap[hash])
|
||||||
|
if(!torrent)
|
||||||
|
return
|
||||||
|
|
||||||
|
send('downloading', torrent.infoHash)
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
recive('torrent', onTorrent);
|
||||||
|
p2p.on('torrent', ({hash, options} = {}, callback) => {
|
||||||
|
if(!hash)
|
||||||
|
return;
|
||||||
|
|
||||||
|
onTorrent(hash, options, (data) => callback(data))
|
||||||
|
})
|
||||||
|
|
||||||
|
if(config.p2pReplication)
|
||||||
|
{
|
||||||
|
console.log('p2p replication enabled')
|
||||||
|
|
||||||
|
p2p.on('randomTorrents', (nil, callback) => {
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
sphinx.query('SELECT * FROM `torrents` ORDER BY rand() limit 5', (error, torrents) => {
|
||||||
|
if(!torrents || torrents.length == 0) {
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let hashes = {}
|
||||||
|
for(const torrent of torrents)
|
||||||
|
{
|
||||||
|
delete torrent.id
|
||||||
|
hashes[torrent.hash] = torrent
|
||||||
|
}
|
||||||
|
|
||||||
|
const inSql = Object.keys(hashes).map(hash => sphinx.escape(hash)).join(',');
|
||||||
|
sphinx.query(`SELECT * FROM files WHERE hash IN(${inSql}) limit 50000`, (error, files) => {
|
||||||
|
if(!files)
|
||||||
|
{
|
||||||
|
files = []
|
||||||
|
}
|
||||||
|
|
||||||
|
files.forEach((file) => {
|
||||||
|
if(!hashes[file.hash].filesList)
|
||||||
|
hashes[file.hash].filesList = []
|
||||||
|
hashes[file.hash].filesList.push(file)
|
||||||
|
})
|
||||||
|
callback(Object.values(hashes))
|
||||||
|
})
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
const getReplicationTorrents = (nextTimeout = 5000) => {
|
||||||
|
console.log('call replication')
|
||||||
|
let gotTorrents = 0
|
||||||
|
|
||||||
|
p2p.emit('randomTorrents', null, (torrents) => {
|
||||||
|
if(!torrents || torrents.length == 0)
|
||||||
|
return
|
||||||
|
|
||||||
|
gotTorrents += torrents.length
|
||||||
|
|
||||||
|
torrents.forEach((torrent) => {
|
||||||
|
console.log('replicate remote torrent', torrent)
|
||||||
|
insertTorrentToDB(torrent)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
setTimeout(() => getReplicationTorrents(gotTorrents > 8 ? gotTorrents * 600 : 10000), nextTimeout)
|
||||||
|
}
|
||||||
|
// start
|
||||||
|
getReplicationTorrents()
|
||||||
|
}
|
||||||
|
|
||||||
|
const searchTorrentCall = function(text, navigation, callback)
|
||||||
|
{
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
if(!text || text.length <= 2) {
|
||||||
|
callback(undefined);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const safeSearch = !!navigation.safeSearch;
|
||||||
|
|
||||||
|
const index = navigation.index || 0;
|
||||||
|
const limit = navigation.limit || 10;
|
||||||
|
let args = [text, index, limit];
|
||||||
|
const orderBy = navigation.orderBy;
|
||||||
|
let order = '';
|
||||||
|
let where = '';
|
||||||
|
if(orderBy && orderBy.length > 0)
|
||||||
|
{
|
||||||
|
const orderDesc = navigation.orderDesc ? 'DESC' : 'ASC';
|
||||||
|
args.splice(1, 0, orderBy);
|
||||||
|
order = 'ORDER BY ?? ' + orderDesc;
|
||||||
|
}
|
||||||
|
if(safeSearch)
|
||||||
|
{
|
||||||
|
where += " and contentCategory != 'xxx' ";
|
||||||
|
}
|
||||||
|
if(navigation.type && navigation.type.length > 0)
|
||||||
|
{
|
||||||
|
where += ' and contentType = ' + sphinx.escape(navigation.type) + ' ';
|
||||||
|
}
|
||||||
|
if(navigation.size)
|
||||||
|
{
|
||||||
|
if(navigation.size.max > 0)
|
||||||
|
where += ' and size < ' + sphinx.escape(navigation.size.max) + ' ';
|
||||||
|
if(navigation.size.min > 0)
|
||||||
|
where += ' and size > ' + sphinx.escape(navigation.size.min) + ' ';
|
||||||
|
}
|
||||||
|
if(navigation.files)
|
||||||
|
{
|
||||||
|
if(navigation.files.max > 0)
|
||||||
|
where += ' and files < ' + sphinx.escape(navigation.files.max) + ' ';
|
||||||
|
if(navigation.files.min > 0)
|
||||||
|
where += ' and files > ' + sphinx.escape(navigation.files.min) + ' ';
|
||||||
|
}
|
||||||
|
|
||||||
|
let searchList = [];
|
||||||
|
//args.splice(orderBy && orderBy.length > 0 ? 1 : 0, 1);
|
||||||
|
//sphinx.query('SELECT * FROM `torrents` WHERE `name` like \'%' + text + '%\' ' + where + ' ' + order + ' LIMIT ?,?', args, function (error, rows, fields) {
|
||||||
|
sphinx.query('SELECT * FROM `torrents` WHERE MATCH(?) ' + where + ' ' + order + ' LIMIT ?,?', args, function (error, rows, fields) {
|
||||||
|
if(!rows) {
|
||||||
|
console.log(error)
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
rows.forEach((row) => {
|
||||||
|
searchList.push(baseRowData(row));
|
||||||
|
});
|
||||||
|
callback(searchList);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
recive('searchTorrent', (text, navigation, callback) => {
|
||||||
|
searchTorrentCall(text, navigation, callback)
|
||||||
|
p2p.emit('searchTorrent', {text, navigation}, (remote, socketObject) => {
|
||||||
|
console.log('remote search results', remote && remote.length)
|
||||||
|
if(remote && remote.length > 0)
|
||||||
|
{
|
||||||
|
const { _socket: socket } = socketObject
|
||||||
|
const peer = { address: socket.remoteAddress, port: socket.remotePort }
|
||||||
|
remote = remote.map(torrent => Object.assign(torrent, {peer}))
|
||||||
|
}
|
||||||
|
send('remoteSearchTorrent', remote)
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
p2p.on('searchTorrent', ({text, navigation} = {}, callback) => {
|
||||||
|
if(!text)
|
||||||
|
return;
|
||||||
|
|
||||||
|
searchTorrentCall(text, navigation, (data) => callback(data))
|
||||||
|
})
|
||||||
|
|
||||||
|
const searchFilesCall = function(text, navigation, callback)
|
||||||
|
{
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
if(!text || text.length <= 2) {
|
||||||
|
callback(undefined);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const safeSearch = !!navigation.safeSearch;
|
||||||
|
|
||||||
|
const index = navigation.index || 0;
|
||||||
|
const limit = navigation.limit || 10;
|
||||||
|
let args = [text, index, limit];
|
||||||
|
const orderBy = navigation.orderBy;
|
||||||
|
let order = '';
|
||||||
|
let where = '';
|
||||||
|
|
||||||
|
if(orderBy && orderBy.length > 0)
|
||||||
|
{
|
||||||
|
const orderDesc = navigation.orderDesc ? 'DESC' : 'ASC';
|
||||||
|
args.splice(1, 0, orderBy);
|
||||||
|
order = 'ORDER BY ?? ' + orderDesc;
|
||||||
|
}
|
||||||
|
/*
|
||||||
|
if(safeSearch)
|
||||||
|
{
|
||||||
|
where += " and contentCategory != 'xxx' ";
|
||||||
|
}
|
||||||
|
if(navigation.type && navigation.type.length > 0)
|
||||||
|
{
|
||||||
|
where += ' and contentType = ' + sphinx.escape(navigation.type) + ' ';
|
||||||
|
}
|
||||||
|
if(navigation.size)
|
||||||
|
{
|
||||||
|
if(navigation.size.max > 0)
|
||||||
|
where += ' and torrentSize < ' + sphinx.escape(navigation.size.max) + ' ';
|
||||||
|
if(navigation.size.min > 0)
|
||||||
|
where += ' and torrentSize > ' + sphinx.escape(navigation.size.min) + ' ';
|
||||||
|
}
|
||||||
|
if(navigation.files)
|
||||||
|
{
|
||||||
|
if(navigation.files.max > 0)
|
||||||
|
where += ' and files < ' + sphinx.escape(navigation.files.max) + ' ';
|
||||||
|
if(navigation.files.min > 0)
|
||||||
|
where += ' and files > ' + sphinx.escape(navigation.files.min) + ' ';
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
let search = {};
|
||||||
|
//args.splice(orderBy && orderBy.length > 0 ? 1 : 0, 1);
|
||||||
|
//sphinx.query('SELECT * FROM `files` inner join torrents on(torrents.hash = files.hash) WHERE files.path like \'%' + text + '%\' ' + where + ' ' + order + ' LIMIT ?,?', args, function (error, rows, fields) {
|
||||||
|
sphinx.query('SELECT * FROM `files` WHERE MATCH(?) ' + where + ' ' + order + ' LIMIT ?,?', args, function (error, files, fields) {
|
||||||
|
if(!files) {
|
||||||
|
console.log(error)
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if(files.length === 0)
|
||||||
|
{
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
for(const file of files)
|
||||||
|
{
|
||||||
|
if(!search[file.hash])
|
||||||
|
{
|
||||||
|
search[file.hash] = { path: [] }
|
||||||
|
}
|
||||||
|
search[file.hash].path.push(file.path)
|
||||||
|
}
|
||||||
|
const inSql = Object.keys(search).map(hash => sphinx.escape(hash)).join(',');
|
||||||
|
sphinx.query(`SELECT * FROM torrents WHERE hash IN(${inSql})`, (err, torrents) => {
|
||||||
|
if(!torrents) {
|
||||||
|
console.log(err)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
for(const torrent of torrents)
|
||||||
|
{
|
||||||
|
search[torrent.hash] = Object.assign(torrent, search[torrent.hash])
|
||||||
|
}
|
||||||
|
|
||||||
|
callback(Object.values(search));
|
||||||
|
})
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
recive('searchFiles', (text, navigation, callback) => {
|
||||||
|
searchFilesCall(text, navigation, callback)
|
||||||
|
p2p.emit('searchFiles', {text, navigation}, (remote, socketObject) => {
|
||||||
|
console.log('remote search files results', remote && remote.length)
|
||||||
|
if(remote && remote.length > 0)
|
||||||
|
{
|
||||||
|
const { _socket: socket } = socketObject
|
||||||
|
const peer = { address: socket.remoteAddress, port: socket.remotePort }
|
||||||
|
remote = remote.map(torrent => Object.assign(torrent, {peer}))
|
||||||
|
}
|
||||||
|
send('remoteSearchFiles', remote)
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
p2p.on('searchFiles', ({text, navigation} = {}, callback) => {
|
||||||
|
if(!text)
|
||||||
|
return;
|
||||||
|
|
||||||
|
searchFilesCall(text, navigation, (data) => callback(data))
|
||||||
|
})
|
||||||
|
|
||||||
|
recive('checkTrackers', function(hash)
|
||||||
|
{
|
||||||
|
if(hash.length != 40)
|
||||||
|
return;
|
||||||
|
|
||||||
|
updateTorrentTrackers(hash);
|
||||||
|
});
|
||||||
|
|
||||||
|
recive('topTorrents', function(type, callback)
|
||||||
|
{
|
||||||
|
let where = '';
|
||||||
|
let max = 20;
|
||||||
|
if(type && type.length > 0)
|
||||||
|
{
|
||||||
|
where += ' and contentType = ' + sphinx.escape(type) + ' ';
|
||||||
|
max = 15;
|
||||||
|
|
||||||
|
if(type == 'hours')
|
||||||
|
{
|
||||||
|
where = ' and `added` > ' + Math.floor(Date.now() / 1000) - (60 * 60 * 24)
|
||||||
|
}
|
||||||
|
if(type == 'week')
|
||||||
|
{
|
||||||
|
where = ' and `added` > ' + Math.floor(Date.now() / 1000) - (60 * 60 * 24 * 7)
|
||||||
|
}
|
||||||
|
if(type == 'month')
|
||||||
|
{
|
||||||
|
where = ' and `added` > ' + Math.floor(Date.now() / 1000) - (60 * 60 * 24 * 30)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const query = `SELECT * FROM torrents WHERE seeders > 0 and contentCategory != 'xxx' ${where} ORDER BY seeders DESC LIMIT ${max}`;
|
||||||
|
if(topCache[query])
|
||||||
|
{
|
||||||
|
callback(topCache[query]);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
sphinx.query(query, function (error, rows) {
|
||||||
|
if(!rows || rows.length == 0) {
|
||||||
|
callback(undefined)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
rows = rows.map((row) => {
|
||||||
|
return baseRowData(row);
|
||||||
|
});
|
||||||
|
topCache[query] = rows;
|
||||||
|
callback(rows);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
recive('peers', (callback) =>
|
||||||
|
{
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
callback(p2p.size)
|
||||||
|
});
|
||||||
|
|
||||||
|
recive('p2pStatus', (callback) =>
|
||||||
|
{
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
callback(p2p.p2pStatus)
|
||||||
|
});
|
||||||
|
|
||||||
|
recive('config', (callback) =>
|
||||||
|
{
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
callback(config)
|
||||||
|
});
|
||||||
|
|
||||||
|
recive('setConfig', (options, callback) =>
|
||||||
|
{
|
||||||
|
if(typeof options !== 'object')
|
||||||
|
return;
|
||||||
|
|
||||||
|
if(typeof options.indexer !== 'undefined')
|
||||||
|
{
|
||||||
|
const upSpider = () => {
|
||||||
|
if(options.indexer)
|
||||||
|
spider.listen(config.spiderPort)
|
||||||
|
else
|
||||||
|
spider.close()
|
||||||
|
}
|
||||||
|
|
||||||
|
if(options.spiderPort !== config.spiderPort)
|
||||||
|
spider.close(upSpider)
|
||||||
|
else
|
||||||
|
upSpider()
|
||||||
|
}
|
||||||
|
|
||||||
|
if(upnp)
|
||||||
|
upnp.ratsUnmap()
|
||||||
|
|
||||||
|
for(const option in options)
|
||||||
|
{
|
||||||
|
if(option in config)
|
||||||
|
config[option] = options[option]
|
||||||
|
}
|
||||||
|
|
||||||
|
if(upnp)
|
||||||
|
upnp.ratsMap()
|
||||||
|
|
||||||
|
if(config.p2p)
|
||||||
|
{
|
||||||
|
spider.announceHashes = [crypto.createHash('sha1').update('degrats-v1').digest()]
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
spider.announceHashes = []
|
||||||
|
}
|
||||||
|
|
||||||
|
if(typeof callback === 'function')
|
||||||
|
callback(true)
|
||||||
|
});
|
||||||
|
|
||||||
|
recive('download', (torrentObject) =>
|
||||||
|
{
|
||||||
|
const magnet = `magnet:?xt=urn:btih:${torrentObject.hash}`
|
||||||
|
console.log('download', magnet)
|
||||||
|
if(torrentClient.get(magnet)) {
|
||||||
|
console.log('aready added')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
torrentClient.add(magnet, {path: config.client.downloadPath}, (torrent) =>{
|
||||||
|
torrentClientHashMap[torrent.infoHash] = magnet
|
||||||
|
torrent.torrentObject = torrentObject
|
||||||
|
console.log('start downloading', torrent.infoHash)
|
||||||
|
send('downloading', torrent.infoHash)
|
||||||
|
|
||||||
|
torrent.on('done', () => {
|
||||||
|
console.log('download done', torrent.infoHash)
|
||||||
|
delete torrentClientHashMap[torrent.infoHash]
|
||||||
|
send('downloadDone', torrent.infoHash)
|
||||||
|
})
|
||||||
|
|
||||||
|
let now = Date.now()
|
||||||
|
torrent.on('download', (bytes) => {
|
||||||
|
if(Date.now() - now < 100)
|
||||||
|
return
|
||||||
|
now = Date.now()
|
||||||
|
|
||||||
|
send('downloadProgress', torrent.infoHash, {
|
||||||
|
received: bytes,
|
||||||
|
downloaded: torrent.downloaded,
|
||||||
|
speed: torrent.downloadSpeed,
|
||||||
|
progress: torrent.progress
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
recive('downloadCancel', (hash, callback) =>
|
||||||
|
{
|
||||||
|
const id = torrentClientHashMap[hash]
|
||||||
|
if(!id)
|
||||||
|
{
|
||||||
|
if(callback)
|
||||||
|
callback(false)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
torrentClient.remove(id, (err) => {
|
||||||
|
if(err)
|
||||||
|
{
|
||||||
|
if(callback)
|
||||||
|
callback(false)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
delete torrentClientHashMap[hash]
|
||||||
|
send('downloadDone', hash, true)
|
||||||
|
|
||||||
|
if(callback)
|
||||||
|
callback(true)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
recive('downloads', (callback) =>
|
||||||
|
{
|
||||||
|
callback(torrentClient.torrents.map(torrent => ({
|
||||||
|
torrentObject: torrent.torrentObject,
|
||||||
|
received: torrent.received,
|
||||||
|
downloaded: torrent.downloaded,
|
||||||
|
progress: torrent.progress,
|
||||||
|
speed: torrent.downloadSpeed
|
||||||
|
})))
|
||||||
|
})
|
||||||
|
|
||||||
|
let socketIPV4 = () => {
|
||||||
|
let ip = socket.request.connection.remoteAddress;
|
||||||
|
if (ipaddr.IPv4.isValid(ip)) {
|
||||||
|
// all ok
|
||||||
|
} else if (ipaddr.IPv6.isValid(ip)) {
|
||||||
|
let ipv6 = ipaddr.IPv6.parse(ip);
|
||||||
|
if (ipv6.isIPv4MappedAddress()) {
|
||||||
|
ip = ipv6.toIPv4Address().toString();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ip
|
||||||
|
};
|
||||||
|
|
||||||
|
recive('vote', function(hash, isGood, callback)
|
||||||
|
{
|
||||||
|
if(hash.length != 40)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if(typeof callback != 'function')
|
||||||
|
return;
|
||||||
|
|
||||||
|
const ip = socketIPV4();
|
||||||
|
isGood = !!isGood;
|
||||||
|
|
||||||
|
sphinx.query('SELECT * FROM `torrents_actions` WHERE `hash` = ? AND (`action` = \'good\' OR `action` = \'bad\') AND ipv4 = ?', [hash, ip], function (error, rows, fields) {
|
||||||
|
if(!rows) {
|
||||||
|
console.error(error);
|
||||||
|
}
|
||||||
|
if(rows.length > 0) {
|
||||||
|
callback(false)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
sphinx.query('SELECT good, bad FROM `torrents` WHERE `hash` = ?', hash, function (error, rows, fields) {
|
||||||
|
if(!rows || rows.length == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
let {good, bad} = rows[0];
|
||||||
|
const action = isGood ? 'good' : 'bad';
|
||||||
|
sphinx.query('INSERT INTO `torrents_actions` SET ?', {hash, action, ipv4: ip}, function(err, result) {
|
||||||
|
if(!result) {
|
||||||
|
console.error(err);
|
||||||
|
}
|
||||||
|
sphinx.query('UPDATE torrents SET ' + action + ' = ' + action + ' + 1 WHERE hash = ?', hash, function(err, result) {
|
||||||
|
if(!result) {
|
||||||
|
console.error(err);
|
||||||
|
}
|
||||||
|
if(isGood) {
|
||||||
|
good++;
|
||||||
|
} else {
|
||||||
|
bad++;
|
||||||
|
}
|
||||||
|
send('vote', {
|
||||||
|
hash, good, bad
|
||||||
|
});
|
||||||
|
callback(true)
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
@ -9,13 +9,13 @@ const P2PServer = require('./p2p')
|
|||||||
const stun = require('stun')
|
const stun = require('stun')
|
||||||
const natUpnp = require('nat-upnp');
|
const natUpnp = require('nat-upnp');
|
||||||
const http = require('https')
|
const http = require('https')
|
||||||
|
const API = require('./api')
|
||||||
//var express = require('express');
|
//var express = require('express');
|
||||||
//var app = express();
|
//var app = express();
|
||||||
//var server = http.Server(app);
|
//var server = http.Server(app);
|
||||||
//var io = require('socket.io')(server);
|
//var io = require('socket.io')(server);
|
||||||
//var sm = require('sitemap');
|
//var sm = require('sitemap');
|
||||||
//var phantomjs = require('phantomjs-prebuilt')
|
//var phantomjs = require('phantomjs-prebuilt')
|
||||||
var ipaddr = require('ipaddr.js');
|
|
||||||
//const disk = require('diskusage');
|
//const disk = require('diskusage');
|
||||||
const encryptor = require('simple-encryptor')('rats-on-the-boat-enc-v0');
|
const encryptor = require('simple-encryptor')('rats-on-the-boat-enc-v0');
|
||||||
const os = require('os');
|
const os = require('os');
|
||||||
@ -32,7 +32,6 @@ const checkInternet = require('./checkInternet')
|
|||||||
const {torrentTypeDetect} = require('../app/content');
|
const {torrentTypeDetect} = require('../app/content');
|
||||||
|
|
||||||
const torrentClient = require('./torrentClient')
|
const torrentClient = require('./torrentClient')
|
||||||
let torrentClientHashMap = {}
|
|
||||||
|
|
||||||
// Start server
|
// Start server
|
||||||
//server.listen(config.httpPort);
|
//server.listen(config.httpPort);
|
||||||
@ -253,11 +252,6 @@ function baseRowData(row)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let topCache = {};
|
|
||||||
setInterval(() => {
|
|
||||||
topCache = {};
|
|
||||||
}, 24 * 60 * 60 * 1000);
|
|
||||||
|
|
||||||
const p2p = new P2PServer(send)
|
const p2p = new P2PServer(send)
|
||||||
p2p.encryptor = encryptor
|
p2p.encryptor = encryptor
|
||||||
p2p.listen()
|
p2p.listen()
|
||||||
@ -299,629 +293,17 @@ if(config.p2pBootstrap)
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
//io.on('connection', function(socket)
|
// setup api
|
||||||
//{
|
console.log(API)
|
||||||
recive('recentTorrents', function(callback)
|
API({
|
||||||
{
|
sphinx,
|
||||||
if(typeof callback != 'function')
|
recive,
|
||||||
return;
|
send,
|
||||||
|
p2p,
|
||||||
sphinx.query('SELECT * FROM `torrents` ORDER BY added DESC LIMIT 0,10', function (error, rows, fields) {
|
config,
|
||||||
if(!rows) {
|
baseRowData,
|
||||||
callback(undefined)
|
torrentClient
|
||||||
return;
|
})
|
||||||
}
|
|
||||||
|
|
||||||
let torrents = [];
|
|
||||||
rows.forEach((row) => {
|
|
||||||
torrents.push(baseRowData(row));
|
|
||||||
});
|
|
||||||
|
|
||||||
callback(torrents)
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
recive('statistic', function(callback)
|
|
||||||
{
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
sphinx.query('SELECT count(*) AS torrents, sum(size) AS sz FROM `torrents`', function (error, rows, fields) {
|
|
||||||
if(!rows) {
|
|
||||||
console.error(error)
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let result = {torrents: rows[0].torrents || 0, size: rows[0].sz || 0}
|
|
||||||
|
|
||||||
sphinx.query('SELECT count(*) AS files FROM `files`', function (error, rows, fields) {
|
|
||||||
if(!rows) {
|
|
||||||
console.error(error)
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
result.files = rows[0].files || 0
|
|
||||||
|
|
||||||
callback(result)
|
|
||||||
})
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
const onTorrent = (hash, options, callback) => {
|
|
||||||
if(hash.length != 40)
|
|
||||||
return;
|
|
||||||
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
// remote request
|
|
||||||
if(options.peer)
|
|
||||||
{
|
|
||||||
console.log('remote torrent request to peer')
|
|
||||||
const peer = p2p.find(options.peer)
|
|
||||||
if(!peer)
|
|
||||||
{
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
delete options.peer;
|
|
||||||
peer.emit('torrent', {hash, options}, (data) => {
|
|
||||||
console.log('remote torrent result')
|
|
||||||
callback(data)
|
|
||||||
})
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
sphinx.query('SELECT * FROM `torrents` WHERE `hash` = ?', hash, function (error, rows, fields) {
|
|
||||||
if(!rows || rows.length == 0) {
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
let torrent = rows[0];
|
|
||||||
|
|
||||||
if(options.files)
|
|
||||||
{
|
|
||||||
sphinx.query('SELECT * FROM `files` WHERE `hash` = ? LIMIT 50000', hash, function (error, rows, fields) {
|
|
||||||
torrent.filesList = rows;
|
|
||||||
callback(baseRowData(torrent))
|
|
||||||
});
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
callback(baseRowData(torrent))
|
|
||||||
}
|
|
||||||
|
|
||||||
if(torrentClientHashMap[hash])
|
|
||||||
{
|
|
||||||
const torrent = torrentClient.get(torrentClientHashMap[hash])
|
|
||||||
if(!torrent)
|
|
||||||
return
|
|
||||||
|
|
||||||
send('downloading', torrent.infoHash)
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
recive('torrent', onTorrent);
|
|
||||||
p2p.on('torrent', ({hash, options} = {}, callback) => {
|
|
||||||
if(!hash)
|
|
||||||
return;
|
|
||||||
|
|
||||||
onTorrent(hash, options, (data) => callback(data))
|
|
||||||
})
|
|
||||||
|
|
||||||
if(config.p2pReplication)
|
|
||||||
{
|
|
||||||
console.log('p2p replication enabled')
|
|
||||||
|
|
||||||
p2p.on('randomTorrents', (nil, callback) => {
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
sphinx.query('SELECT * FROM `torrents` ORDER BY rand() limit 5', (error, torrents) => {
|
|
||||||
if(!torrents || torrents.length == 0) {
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let hashes = {}
|
|
||||||
for(const torrent of torrents)
|
|
||||||
{
|
|
||||||
delete torrent.id
|
|
||||||
hashes[torrent.hash] = torrent
|
|
||||||
}
|
|
||||||
|
|
||||||
const inSql = Object.keys(hashes).map(hash => sphinx.escape(hash)).join(',');
|
|
||||||
sphinx.query(`SELECT * FROM files WHERE hash IN(${inSql}) limit 50000`, (error, files) => {
|
|
||||||
if(!files)
|
|
||||||
{
|
|
||||||
files = []
|
|
||||||
}
|
|
||||||
|
|
||||||
files.forEach((file) => {
|
|
||||||
if(!hashes[file.hash].filesList)
|
|
||||||
hashes[file.hash].filesList = []
|
|
||||||
hashes[file.hash].filesList.push(file)
|
|
||||||
})
|
|
||||||
callback(Object.values(hashes))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
});
|
|
||||||
|
|
||||||
const getReplicationTorrents = (nextTimeout = 5000) => {
|
|
||||||
console.log('call replication')
|
|
||||||
let gotTorrents = 0
|
|
||||||
|
|
||||||
p2p.emit('randomTorrents', null, (torrents) => {
|
|
||||||
if(!torrents || torrents.length == 0)
|
|
||||||
return
|
|
||||||
|
|
||||||
gotTorrents += torrents.length
|
|
||||||
|
|
||||||
torrents.forEach((torrent) => {
|
|
||||||
console.log('replicate remote torrent', torrent)
|
|
||||||
insertTorrentToDB(torrent)
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
setTimeout(() => getReplicationTorrents(gotTorrents > 8 ? gotTorrents * 600 : 10000), nextTimeout)
|
|
||||||
}
|
|
||||||
// start
|
|
||||||
getReplicationTorrents()
|
|
||||||
}
|
|
||||||
|
|
||||||
const searchTorrentCall = function(text, navigation, callback)
|
|
||||||
{
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
if(!text || text.length <= 2) {
|
|
||||||
callback(undefined);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const safeSearch = !!navigation.safeSearch;
|
|
||||||
|
|
||||||
const index = navigation.index || 0;
|
|
||||||
const limit = navigation.limit || 10;
|
|
||||||
let args = [text, index, limit];
|
|
||||||
const orderBy = navigation.orderBy;
|
|
||||||
let order = '';
|
|
||||||
let where = '';
|
|
||||||
if(orderBy && orderBy.length > 0)
|
|
||||||
{
|
|
||||||
const orderDesc = navigation.orderDesc ? 'DESC' : 'ASC';
|
|
||||||
args.splice(1, 0, orderBy);
|
|
||||||
order = 'ORDER BY ?? ' + orderDesc;
|
|
||||||
}
|
|
||||||
if(safeSearch)
|
|
||||||
{
|
|
||||||
where += " and contentCategory != 'xxx' ";
|
|
||||||
}
|
|
||||||
if(navigation.type && navigation.type.length > 0)
|
|
||||||
{
|
|
||||||
where += ' and contentType = ' + sphinx.escape(navigation.type) + ' ';
|
|
||||||
}
|
|
||||||
if(navigation.size)
|
|
||||||
{
|
|
||||||
if(navigation.size.max > 0)
|
|
||||||
where += ' and size < ' + sphinx.escape(navigation.size.max) + ' ';
|
|
||||||
if(navigation.size.min > 0)
|
|
||||||
where += ' and size > ' + sphinx.escape(navigation.size.min) + ' ';
|
|
||||||
}
|
|
||||||
if(navigation.files)
|
|
||||||
{
|
|
||||||
if(navigation.files.max > 0)
|
|
||||||
where += ' and files < ' + sphinx.escape(navigation.files.max) + ' ';
|
|
||||||
if(navigation.files.min > 0)
|
|
||||||
where += ' and files > ' + sphinx.escape(navigation.files.min) + ' ';
|
|
||||||
}
|
|
||||||
|
|
||||||
let searchList = [];
|
|
||||||
//args.splice(orderBy && orderBy.length > 0 ? 1 : 0, 1);
|
|
||||||
//sphinx.query('SELECT * FROM `torrents` WHERE `name` like \'%' + text + '%\' ' + where + ' ' + order + ' LIMIT ?,?', args, function (error, rows, fields) {
|
|
||||||
sphinx.query('SELECT * FROM `torrents` WHERE MATCH(?) ' + where + ' ' + order + ' LIMIT ?,?', args, function (error, rows, fields) {
|
|
||||||
if(!rows) {
|
|
||||||
console.log(error)
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
rows.forEach((row) => {
|
|
||||||
searchList.push(baseRowData(row));
|
|
||||||
});
|
|
||||||
callback(searchList);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
recive('searchTorrent', (text, navigation, callback) => {
|
|
||||||
searchTorrentCall(text, navigation, callback)
|
|
||||||
p2p.emit('searchTorrent', {text, navigation}, (remote, socketObject) => {
|
|
||||||
console.log('remote search results', remote && remote.length)
|
|
||||||
if(remote && remote.length > 0)
|
|
||||||
{
|
|
||||||
const { _socket: socket } = socketObject
|
|
||||||
const peer = { address: socket.remoteAddress, port: socket.remotePort }
|
|
||||||
remote = remote.map(torrent => Object.assign(torrent, {peer}))
|
|
||||||
}
|
|
||||||
send('remoteSearchTorrent', remote)
|
|
||||||
})
|
|
||||||
});
|
|
||||||
|
|
||||||
p2p.on('searchTorrent', ({text, navigation} = {}, callback) => {
|
|
||||||
if(!text)
|
|
||||||
return;
|
|
||||||
|
|
||||||
searchTorrentCall(text, navigation, (data) => callback(data))
|
|
||||||
})
|
|
||||||
|
|
||||||
const searchFilesCall = function(text, navigation, callback)
|
|
||||||
{
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
if(!text || text.length <= 2) {
|
|
||||||
callback(undefined);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const safeSearch = !!navigation.safeSearch;
|
|
||||||
|
|
||||||
const index = navigation.index || 0;
|
|
||||||
const limit = navigation.limit || 10;
|
|
||||||
let args = [text, index, limit];
|
|
||||||
const orderBy = navigation.orderBy;
|
|
||||||
let order = '';
|
|
||||||
let where = '';
|
|
||||||
|
|
||||||
if(orderBy && orderBy.length > 0)
|
|
||||||
{
|
|
||||||
const orderDesc = navigation.orderDesc ? 'DESC' : 'ASC';
|
|
||||||
args.splice(1, 0, orderBy);
|
|
||||||
order = 'ORDER BY ?? ' + orderDesc;
|
|
||||||
}
|
|
||||||
/*
|
|
||||||
if(safeSearch)
|
|
||||||
{
|
|
||||||
where += " and contentCategory != 'xxx' ";
|
|
||||||
}
|
|
||||||
if(navigation.type && navigation.type.length > 0)
|
|
||||||
{
|
|
||||||
where += ' and contentType = ' + sphinx.escape(navigation.type) + ' ';
|
|
||||||
}
|
|
||||||
if(navigation.size)
|
|
||||||
{
|
|
||||||
if(navigation.size.max > 0)
|
|
||||||
where += ' and torrentSize < ' + sphinx.escape(navigation.size.max) + ' ';
|
|
||||||
if(navigation.size.min > 0)
|
|
||||||
where += ' and torrentSize > ' + sphinx.escape(navigation.size.min) + ' ';
|
|
||||||
}
|
|
||||||
if(navigation.files)
|
|
||||||
{
|
|
||||||
if(navigation.files.max > 0)
|
|
||||||
where += ' and files < ' + sphinx.escape(navigation.files.max) + ' ';
|
|
||||||
if(navigation.files.min > 0)
|
|
||||||
where += ' and files > ' + sphinx.escape(navigation.files.min) + ' ';
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
|
|
||||||
let search = {};
|
|
||||||
//args.splice(orderBy && orderBy.length > 0 ? 1 : 0, 1);
|
|
||||||
//sphinx.query('SELECT * FROM `files` inner join torrents on(torrents.hash = files.hash) WHERE files.path like \'%' + text + '%\' ' + where + ' ' + order + ' LIMIT ?,?', args, function (error, rows, fields) {
|
|
||||||
sphinx.query('SELECT * FROM `files` WHERE MATCH(?) ' + where + ' ' + order + ' LIMIT ?,?', args, function (error, files, fields) {
|
|
||||||
if(!files) {
|
|
||||||
console.log(error)
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if(files.length === 0)
|
|
||||||
{
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
for(const file of files)
|
|
||||||
{
|
|
||||||
if(!search[file.hash])
|
|
||||||
{
|
|
||||||
search[file.hash] = { path: [] }
|
|
||||||
}
|
|
||||||
search[file.hash].path.push(file.path)
|
|
||||||
}
|
|
||||||
const inSql = Object.keys(search).map(hash => sphinx.escape(hash)).join(',');
|
|
||||||
sphinx.query(`SELECT * FROM torrents WHERE hash IN(${inSql})`, (err, torrents) => {
|
|
||||||
if(!torrents) {
|
|
||||||
console.log(err)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
for(const torrent of torrents)
|
|
||||||
{
|
|
||||||
search[torrent.hash] = Object.assign(torrent, search[torrent.hash])
|
|
||||||
}
|
|
||||||
|
|
||||||
callback(Object.values(search));
|
|
||||||
})
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
recive('searchFiles', (text, navigation, callback) => {
|
|
||||||
searchFilesCall(text, navigation, callback)
|
|
||||||
p2p.emit('searchFiles', {text, navigation}, (remote, socketObject) => {
|
|
||||||
console.log('remote search files results', remote && remote.length)
|
|
||||||
if(remote && remote.length > 0)
|
|
||||||
{
|
|
||||||
const { _socket: socket } = socketObject
|
|
||||||
const peer = { address: socket.remoteAddress, port: socket.remotePort }
|
|
||||||
remote = remote.map(torrent => Object.assign(torrent, {peer}))
|
|
||||||
}
|
|
||||||
send('remoteSearchFiles', remote)
|
|
||||||
})
|
|
||||||
});
|
|
||||||
|
|
||||||
p2p.on('searchFiles', ({text, navigation} = {}, callback) => {
|
|
||||||
if(!text)
|
|
||||||
return;
|
|
||||||
|
|
||||||
searchFilesCall(text, navigation, (data) => callback(data))
|
|
||||||
})
|
|
||||||
|
|
||||||
recive('checkTrackers', function(hash)
|
|
||||||
{
|
|
||||||
if(hash.length != 40)
|
|
||||||
return;
|
|
||||||
|
|
||||||
updateTorrentTrackers(hash);
|
|
||||||
});
|
|
||||||
|
|
||||||
recive('topTorrents', function(type, callback)
|
|
||||||
{
|
|
||||||
let where = '';
|
|
||||||
let max = 20;
|
|
||||||
if(type && type.length > 0)
|
|
||||||
{
|
|
||||||
where += ' and contentType = ' + sphinx.escape(type) + ' ';
|
|
||||||
max = 15;
|
|
||||||
|
|
||||||
if(type == 'hours')
|
|
||||||
{
|
|
||||||
where = ' and `added` > ' + Math.floor(Date.now() / 1000) - (60 * 60 * 24)
|
|
||||||
}
|
|
||||||
if(type == 'week')
|
|
||||||
{
|
|
||||||
where = ' and `added` > ' + Math.floor(Date.now() / 1000) - (60 * 60 * 24 * 7)
|
|
||||||
}
|
|
||||||
if(type == 'month')
|
|
||||||
{
|
|
||||||
where = ' and `added` > ' + Math.floor(Date.now() / 1000) - (60 * 60 * 24 * 30)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const query = `SELECT * FROM torrents WHERE seeders > 0 and contentCategory != 'xxx' ${where} ORDER BY seeders DESC LIMIT ${max}`;
|
|
||||||
if(topCache[query])
|
|
||||||
{
|
|
||||||
callback(topCache[query]);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
sphinx.query(query, function (error, rows) {
|
|
||||||
if(!rows || rows.length == 0) {
|
|
||||||
callback(undefined)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
rows = rows.map((row) => {
|
|
||||||
return baseRowData(row);
|
|
||||||
});
|
|
||||||
topCache[query] = rows;
|
|
||||||
callback(rows);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
recive('peers', (callback) =>
|
|
||||||
{
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
callback(p2p.size)
|
|
||||||
});
|
|
||||||
|
|
||||||
recive('p2pStatus', (callback) =>
|
|
||||||
{
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
callback(p2p.p2pStatus)
|
|
||||||
});
|
|
||||||
|
|
||||||
recive('config', (callback) =>
|
|
||||||
{
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
callback(config)
|
|
||||||
});
|
|
||||||
|
|
||||||
recive('setConfig', (options, callback) =>
|
|
||||||
{
|
|
||||||
if(typeof options !== 'object')
|
|
||||||
return;
|
|
||||||
|
|
||||||
if(typeof options.indexer !== 'undefined')
|
|
||||||
{
|
|
||||||
const upSpider = () => {
|
|
||||||
if(options.indexer)
|
|
||||||
spider.listen(config.spiderPort)
|
|
||||||
else
|
|
||||||
spider.close()
|
|
||||||
}
|
|
||||||
|
|
||||||
if(options.spiderPort !== config.spiderPort)
|
|
||||||
spider.close(upSpider)
|
|
||||||
else
|
|
||||||
upSpider()
|
|
||||||
}
|
|
||||||
|
|
||||||
if(upnp)
|
|
||||||
upnp.ratsUnmap()
|
|
||||||
|
|
||||||
for(const option in options)
|
|
||||||
{
|
|
||||||
if(option in config)
|
|
||||||
config[option] = options[option]
|
|
||||||
}
|
|
||||||
|
|
||||||
if(upnp)
|
|
||||||
upnp.ratsMap()
|
|
||||||
|
|
||||||
if(config.p2p)
|
|
||||||
{
|
|
||||||
spider.announceHashes = [crypto.createHash('sha1').update('degrats-v1').digest()]
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
spider.announceHashes = []
|
|
||||||
}
|
|
||||||
|
|
||||||
if(typeof callback === 'function')
|
|
||||||
callback(true)
|
|
||||||
});
|
|
||||||
|
|
||||||
recive('download', (torrentObject) =>
|
|
||||||
{
|
|
||||||
const magnet = `magnet:?xt=urn:btih:${torrentObject.hash}`
|
|
||||||
console.log('download', magnet)
|
|
||||||
if(torrentClient.get(magnet)) {
|
|
||||||
console.log('aready added')
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
torrentClient.add(magnet, {path: config.client.downloadPath}, (torrent) =>{
|
|
||||||
torrentClientHashMap[torrent.infoHash] = magnet
|
|
||||||
torrent.torrentObject = torrentObject
|
|
||||||
console.log('start downloading', torrent.infoHash)
|
|
||||||
send('downloading', torrent.infoHash)
|
|
||||||
|
|
||||||
torrent.on('done', () => {
|
|
||||||
console.log('download done', torrent.infoHash)
|
|
||||||
delete torrentClientHashMap[torrent.infoHash]
|
|
||||||
send('downloadDone', torrent.infoHash)
|
|
||||||
})
|
|
||||||
|
|
||||||
let now = Date.now()
|
|
||||||
torrent.on('download', (bytes) => {
|
|
||||||
if(Date.now() - now < 100)
|
|
||||||
return
|
|
||||||
now = Date.now()
|
|
||||||
|
|
||||||
send('downloadProgress', torrent.infoHash, {
|
|
||||||
received: bytes,
|
|
||||||
downloaded: torrent.downloaded,
|
|
||||||
speed: torrent.downloadSpeed,
|
|
||||||
progress: torrent.progress
|
|
||||||
})
|
|
||||||
})
|
|
||||||
})
|
|
||||||
});
|
|
||||||
|
|
||||||
recive('downloadCancel', (hash, callback) =>
|
|
||||||
{
|
|
||||||
const id = torrentClientHashMap[hash]
|
|
||||||
if(!id)
|
|
||||||
{
|
|
||||||
if(callback)
|
|
||||||
callback(false)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
torrentClient.remove(id, (err) => {
|
|
||||||
if(err)
|
|
||||||
{
|
|
||||||
if(callback)
|
|
||||||
callback(false)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
delete torrentClientHashMap[hash]
|
|
||||||
send('downloadDone', hash, true)
|
|
||||||
|
|
||||||
if(callback)
|
|
||||||
callback(true)
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
recive('downloads', (callback) =>
|
|
||||||
{
|
|
||||||
callback(torrentClient.torrents.map(torrent => ({
|
|
||||||
torrentObject: torrent.torrentObject,
|
|
||||||
received: torrent.received,
|
|
||||||
downloaded: torrent.downloaded,
|
|
||||||
progress: torrent.progress,
|
|
||||||
speed: torrent.downloadSpeed
|
|
||||||
})))
|
|
||||||
})
|
|
||||||
|
|
||||||
let socketIPV4 = () => {
|
|
||||||
let ip = socket.request.connection.remoteAddress;
|
|
||||||
if (ipaddr.IPv4.isValid(ip)) {
|
|
||||||
// all ok
|
|
||||||
} else if (ipaddr.IPv6.isValid(ip)) {
|
|
||||||
let ipv6 = ipaddr.IPv6.parse(ip);
|
|
||||||
if (ipv6.isIPv4MappedAddress()) {
|
|
||||||
ip = ipv6.toIPv4Address().toString();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ip
|
|
||||||
};
|
|
||||||
|
|
||||||
recive('vote', function(hash, isGood, callback)
|
|
||||||
{
|
|
||||||
if(hash.length != 40)
|
|
||||||
return;
|
|
||||||
|
|
||||||
if(typeof callback != 'function')
|
|
||||||
return;
|
|
||||||
|
|
||||||
const ip = socketIPV4();
|
|
||||||
isGood = !!isGood;
|
|
||||||
|
|
||||||
sphinx.query('SELECT * FROM `torrents_actions` WHERE `hash` = ? AND (`action` = \'good\' OR `action` = \'bad\') AND ipv4 = ?', [hash, ip], function (error, rows, fields) {
|
|
||||||
if(!rows) {
|
|
||||||
console.error(error);
|
|
||||||
}
|
|
||||||
if(rows.length > 0) {
|
|
||||||
callback(false)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
sphinx.query('SELECT good, bad FROM `torrents` WHERE `hash` = ?', hash, function (error, rows, fields) {
|
|
||||||
if(!rows || rows.length == 0)
|
|
||||||
return;
|
|
||||||
|
|
||||||
let {good, bad} = rows[0];
|
|
||||||
const action = isGood ? 'good' : 'bad';
|
|
||||||
sphinx.query('INSERT INTO `torrents_actions` SET ?', {hash, action, ipv4: ip}, function(err, result) {
|
|
||||||
if(!result) {
|
|
||||||
console.error(err);
|
|
||||||
}
|
|
||||||
sphinx.query('UPDATE torrents SET ' + action + ' = ' + action + ' + 1 WHERE hash = ?', hash, function(err, result) {
|
|
||||||
if(!result) {
|
|
||||||
console.error(err);
|
|
||||||
}
|
|
||||||
if(isGood) {
|
|
||||||
good++;
|
|
||||||
} else {
|
|
||||||
bad++;
|
|
||||||
}
|
|
||||||
send('vote', {
|
|
||||||
hash, good, bad
|
|
||||||
});
|
|
||||||
callback(true)
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
//});
|
|
||||||
|
|
||||||
|
|
||||||
let undoneQueries = 0;
|
let undoneQueries = 0;
|
||||||
|
|||||||
Reference in New Issue
Block a user