in progress: move webtorrent into background BrowserWindow

This commit is contained in:
Feross Aboukhadijeh
2016-02-18 19:10:07 -08:00
parent d122ce9f36
commit bbe982d645
11 changed files with 259 additions and 257 deletions

View File

@@ -1,23 +0,0 @@
module.exports = captureVideoFrame
function captureVideoFrame (video, format) {
if (typeof video === 'string') video = document.querySelector(video)
if (!video || video.nodeName !== 'VIDEO') {
throw new Error('First argument must be a <video> element or selector')
}
if (format == null) format = 'png'
if (format !== 'png' && format !== 'jpg' && format !== 'webp') {
throw new Error('Second argument must be one of "png", "jpg", or "webp"')
}
var canvas = document.createElement('canvas')
canvas.width = video.videoWidth
canvas.height = video.videoHeight
canvas.getContext('2d').drawImage(video, 0, 0)
var dataUri = canvas.toDataURL('image/' + format)
var data = dataUri.split(',')[1]
return new Buffer(data, 'base64')
}

View File

@@ -1,26 +1,75 @@
var debug = require('debug/browser')('app:client')
var createElement = require('virtual-dom/create-element')
var diff = require('virtual-dom/diff')
var dragDrop = require('drag-drop')
var electron = require('electron')
var patch = require('virtual-dom/patch')
var ipc = electron.ipcRenderer
var App = require('./views/app')
var state = {
count: 0
torrents: [
{ name: 'Torrent 1' },
{ name: 'Torrent 2' }
]
}
// Init app
var currentVDom = App(state)
var currentVDom = App(state, handler)
var rootElement = createElement(currentVDom)
document.body.appendChild(rootElement)
function update () {
var newVDom = App(state)
debug('update')
var newVDom = App(state, handler)
var patches = diff(currentVDom, newVDom)
rootElement = patch(rootElement, patches)
currentVDom = newVDom
}
setInterval(function () {
state.count += 1
update()
}, 1000)
function handler (action, ...args) {
debug('handler: %s', action)
ipc.send('action', action, ...args)
}
// Seed via drag-and-drop
dragDrop('body', onFiles)
function onFiles (files) {
debug('got files:')
files.forEach(function (file) {
debug(' - %s (%s bytes)', file.name, file.size)
})
// .torrent file = start downloading the torrent
files.filter(isTorrentFile).forEach(downloadTorrentFile)
// everything else = seed these files
seed(files.filter(isNotTorrentFile))
}
function isTorrentFile (file) {
var extname = path.extname(file.name).toLowerCase()
return extname === '.torrent'
}
function isNotTorrentFile (file) {
return !isTorrentFile(file)
}
// Seed via upload input element
// var uploadElement = require('upload-element')
// var upload = document.querySelector('input[name=upload]')
// uploadElement(upload, function (err, files) {
// if (err) return onError(err)
// files = files.map(function (file) { return file.file })
// onFiles(files)
// })
// Download via input element
// document.querySelector('form').addEventListener('submit', function (e) {
// e.preventDefault()
// downloadTorrent(document.querySelector('form input[name=torrentId]').value.trim())
// })

View File

@@ -1,26 +0,0 @@
module.exports = torrentScreenshot
var captureVideoFrame = require('./capture-video-frame')
function torrentScreenshot (torrent, cb) {
if (torrent.ready) onReady()
else torrent.once('ready', onReady)
function onReady () {
// use largest file
var file = torrent.files.reduce(function (a, b) {
return a.length > b.length ? a : b
})
var video = document.createElement('video')
file.renderTo(video)
video.currentTime = 5
video.addEventListener('seeked', onSeeked)
function onSeeked (e) {
video.removeEventListener('seeked', onSeeked)
var buf = captureVideoFrame(video)
cb(null, buf)
}
}
}

View File

@@ -1,38 +0,0 @@
var logElem = exports.logElem = document.querySelector('.log')
var speed = document.querySelector('.speed')
var logHeading = document.querySelector('#logHeading')
exports.log = function log (item) {
logHeading.style.display = 'block'
if (typeof item === 'string') {
var p = document.createElement('p')
p.innerHTML = item
logElem.appendChild(p)
return p
} else {
logElem.appendChild(item)
exports.lineBreak()
return item
}
}
exports.lineBreak = function lineBreak () {
logElem.appendChild(document.createElement('br'))
}
// replace the last P in the log
exports.updateSpeed = function updateSpeed (str) {
speed.innerHTML = str
}
exports.warning = function warning (err) {
console.error(err.stack || err.message || err)
exports.log(err.message || err)
}
exports.error = function error (err) {
console.error(err.stack || err.message || err)
var p = exports.log(err.message || err)
p.style.color = 'red'
p.style.fontWeight = 'bold'
}

View File

@@ -2,15 +2,26 @@ module.exports = App
var h = require('virtual-dom/h')
function App (state) {
var count = state.count
return h('div', {
style: {
textAlign: 'center',
lineHeight: (100 + count) + 'px',
border: '1px solid red',
width: (100 + count) + 'px',
height: (100 + count) + 'px'
}
}, [ String(count) ])
function App (state, handler) {
var torrents = state.torrents
var list = torrents.map(function (torrent) {
return h('.torrent', [
h('.name', torrent.name)
])
})
return h('.app', [
h('h1', 'WebTorrent'),
h('.torrent-list', list),
h('.add', [
h('button', {
onclick: onAddTorrent
}, 'Add New Torrent')
])
])
function onAddTorrent (e) {
handler('addTorrent', '6a9759bffd5c0af65319979fb7832189f4f3c35d')
}
}

View File

@@ -1,194 +0,0 @@
/* global URL, Blob */
var createTorrent = require('create-torrent')
var debug = require('debug')('instant.io')
var dragDrop = require('drag-drop')
var path = require('path')
var prettyBytes = require('pretty-bytes')
var throttle = require('throttleit')
var thunky = require('thunky')
var torrentPoster = require('./torrent-poster')
var uploadElement = require('upload-element')
var WebTorrent = require('webtorrent')
var xhr = require('xhr')
var util = require('./util')
global.WEBTORRENT_ANNOUNCE = createTorrent.announceList
.map(function (arr) {
return arr[0]
})
.filter(function (url) {
return url.indexOf('wss://') === 0 || url.indexOf('ws://') === 0
})
var getClient = thunky(function (cb) {
getRtcConfig('https://instant.io/rtcConfig', function (err, rtcConfig) {
if (err) util.error(err)
createClient(rtcConfig)
})
function createClient (rtcConfig) {
var client = window.client = new WebTorrent({ rtcConfig: rtcConfig })
client.on('warning', util.warning)
client.on('error', util.error)
cb(null, client)
}
})
// For performance, create the client immediately
getClient(function () {})
// Seed via upload input element
var upload = document.querySelector('input[name=upload]')
uploadElement(upload, function (err, files) {
if (err) return util.error(err)
files = files.map(function (file) { return file.file })
onFiles(files)
})
// Seed via drag-and-drop
dragDrop('body', onFiles)
// Download via input element
document.querySelector('form').addEventListener('submit', function (e) {
e.preventDefault()
downloadTorrent(document.querySelector('form input[name=torrentId]').value.trim())
})
// Download by URL hash
onHashChange()
window.addEventListener('hashchange', onHashChange)
function onHashChange () {
var hash = decodeURIComponent(window.location.hash.substring(1)).trim()
if (hash !== '') downloadTorrent(hash)
}
// Register a protocol handler for "magnet:" (will prompt the user)
// navigator.registerProtocolHandler('magnet', window.location.origin + '#%s', 'Instant.io')
function getRtcConfig (url, cb) {
xhr(url, function (err, res) {
if (err || res.statusCode !== 200) {
cb(new Error('Could not get WebRTC config from server. Using default (without TURN).'))
} else {
var rtcConfig
try {
rtcConfig = JSON.parse(res.body)
} catch (err) {
return cb(new Error('Got invalid WebRTC config from server: ' + res.body))
}
debug('got rtc config: %o', rtcConfig)
cb(null, rtcConfig)
}
})
}
function onFiles (files) {
debug('got files:')
files.forEach(function (file) {
debug(' - %s (%s bytes)', file.name, file.size)
})
// .torrent file = start downloading the torrent
files.filter(isTorrentFile).forEach(downloadTorrentFile)
// everything else = seed these files
seed(files.filter(isNotTorrentFile))
}
function isTorrentFile (file) {
var extname = path.extname(file.name).toLowerCase()
return extname === '.torrent'
}
function isNotTorrentFile (file) {
return !isTorrentFile(file)
}
function downloadTorrent (torrentId) {
util.log('Downloading torrent from ' + torrentId)
getClient(function (err, client) {
if (err) return util.error(err)
client.add(torrentId, onTorrent)
})
}
function downloadTorrentFile (file) {
util.log('Downloading torrent from <strong>' + file.name + '</strong>')
getClient(function (err, client) {
if (err) return util.error(err)
client.add(file, onTorrent)
})
}
function seed (files) {
if (files.length === 0) return
util.log('Seeding ' + files.length + ' files')
// Seed from WebTorrent
getClient(function (err, client) {
if (err) return util.error(err)
client.seed(files, onTorrent)
})
}
function onTorrent (torrent) {
upload.value = upload.defaultValue // reset upload element
var torrentFileName = path.basename(torrent.name, path.extname(torrent.name)) + '.torrent'
util.log('"' + torrentFileName + '" contains ' + torrent.files.length + ' files:')
torrent.files.forEach(function (file) {
util.log('&nbsp;&nbsp;- ' + file.name + ' (' + prettyBytes(file.length) + ')')
})
util.log(
'Torrent info hash: ' + torrent.infoHash + ' ' +
'<a href="/#' + torrent.infoHash + '" onclick="prompt(\'Share this link with anyone you want to download this torrent:\', this.href);return false;">[Share link]</a> ' +
'<a href="' + torrent.magnetURI + '" target="_blank">[Magnet URI]</a> ' +
'<a href="' + torrent.torrentFileBlobURL + '" target="_blank" download="' + torrentFileName + '">[Download .torrent]</a>'
)
function updateSpeed () {
var progress = (100 * torrent.progress).toFixed(1)
util.updateSpeed(
'<b>Peers:</b> ' + torrent.swarm.wires.length + ' ' +
'<b>Progress:</b> ' + progress + '% ' +
'<b>Download speed:</b> ' + prettyBytes(window.client.downloadSpeed) + '/s ' +
'<b>Upload speed:</b> ' + prettyBytes(window.client.uploadSpeed) + '/s'
)
}
torrent.on('download', throttle(updateSpeed, 250))
torrent.on('upload', throttle(updateSpeed, 250))
setInterval(updateSpeed, 5000)
updateSpeed()
torrentPoster(torrent, function (err, buf) {
if (err) return util.error(err)
var img = document.createElement('img')
img.src = URL.createObjectURL(new Blob([ buf ], { type: 'image/png' }))
document.body.appendChild(img)
})
// TODO: play torrent when user clicks button
// torrent.files.forEach(function (file) {
// // append file
// file.appendTo(util.logElem, function (err, elem) {
// if (err) return util.error(err)
// })
// // append download link
// file.getBlobURL(function (err, url) {
// if (err) return util.error(err)
// var a = document.createElement('a')
// a.target = '_blank'
// a.download = file.name
// a.href = url
// a.textContent = 'Download ' + file.name
// util.log(a)
// })
// })
}