mirror of https://codeberg.org/pzp/pzp-hub.git
116 lines
3.0 KiB
JavaScript
116 lines
3.0 KiB
JavaScript
const cat = require('pull-cat')
|
|
const Notify = require('pull-notify')
|
|
const pull = require('pull-stream')
|
|
const debug = require('debug')('ppppp:hub')
|
|
const Tokens = require('./tokens.cjs')
|
|
const Members = require('./members.cjs')
|
|
|
|
function ErrorDuplex(message) {
|
|
const err = new Error(message)
|
|
return {
|
|
source(_abort, cb) {
|
|
cb(err)
|
|
},
|
|
sink(read) {
|
|
read(err, () => {})
|
|
},
|
|
}
|
|
}
|
|
|
|
module.exports = {
|
|
name: 'hub',
|
|
version: '1.0.0',
|
|
manifest: {
|
|
createTunnel: 'duplex',
|
|
ping: 'sync',
|
|
attendants: 'source',
|
|
createToken: 'sync',
|
|
},
|
|
permissions: {
|
|
anonymous: {
|
|
allow: ['createTunnel', 'ping', 'attendants', 'createToken'],
|
|
},
|
|
},
|
|
init(sstack, config) {
|
|
if (!sstack.conn || !sstack.conn.connect) {
|
|
throw new Error('tunnel plugin is missing the required ssb-conn plugin')
|
|
}
|
|
debug('running multiserver at %s', sstack.getAddress('public'))
|
|
|
|
Tokens.load(config.path)
|
|
Members.load(config.path)
|
|
|
|
// Ensure that client connections are only from members or to-be members
|
|
sstack.auth.hook(function (fn, args) {
|
|
const [clientMetadata, cb] = args
|
|
const {pubkey, extra} = clientMetadata
|
|
if (Members.has(pubkey)) {
|
|
cb(null, true)
|
|
} else if (extra && Tokens.has(extra)) {
|
|
Tokens.delete(extra)
|
|
Members.add(pubkey)
|
|
cb(null, true)
|
|
} else {
|
|
debug('prevented stranger %s from connecting to us', pubkey)
|
|
cb(new Error('client is a stranger'))
|
|
}
|
|
})
|
|
|
|
const attendants = new Map()
|
|
const notifyAttendants = Notify()
|
|
|
|
pull(
|
|
sstack.conn.hub().listen(),
|
|
pull.filter(
|
|
({ type }) => type === 'connecting-failed' || type === 'disconnected'
|
|
),
|
|
pull.filter(({ key }) => !!key && attendants.has(key)),
|
|
pull.drain(({ key }) => {
|
|
debug('farewell %s', key)
|
|
attendants.delete(key)
|
|
notifyAttendants([...attendants.keys()])
|
|
})
|
|
)
|
|
|
|
setInterval(() => {
|
|
notifyAttendants([...attendants.keys()])
|
|
}, 10e3)
|
|
|
|
return {
|
|
attendants() {
|
|
const clientId = this.id
|
|
if (clientId && clientId !== sstack.id) {
|
|
debug('welcome %s', clientId)
|
|
if (!attendants.has(clientId)) {
|
|
attendants.set(clientId, sstack.peers[clientId][0])
|
|
notifyAttendants([...attendants.keys()])
|
|
}
|
|
}
|
|
|
|
const initial = pull.values([[...attendants.keys()]])
|
|
return cat([initial, notifyAttendants.listen()])
|
|
},
|
|
|
|
createTunnel(target) {
|
|
if (attendants.has(target)) {
|
|
// prettier-ignore
|
|
debug('received tunnel request for target %s from %s', target, this.id)
|
|
const origin = this.id
|
|
return attendants.get(target).hubClient.connect(origin, () => {})
|
|
} else {
|
|
// prettier-ignore
|
|
return ErrorDuplex(`Cannot createTunnel with ${target} who appears to be offline`)
|
|
}
|
|
},
|
|
|
|
ping() {
|
|
return Date.now()
|
|
},
|
|
|
|
createToken() {
|
|
return Tokens.create()
|
|
},
|
|
}
|
|
},
|
|
}
|