Commit 842b574ea8edb049a6403e9b4434d95823a39caa
Add option to only seed for people/channels I follow
Anders Rune Jensen committed on 9/19/2017, 1:18:08 PMParent: 1b9093405deadf26da6cab9aefb5cdf7ec500421
Files changed
| index.js | changed |
| index.js | |||
|---|---|---|---|
| @@ -6,43 +6,109 @@ | |||
| 6 | 6 … | var pull = require('pull-stream') | |
| 7 | 7 … | ||
| 8 | 8 … | program | |
| 9 | 9 … | .option('-f, --folder [value]', 'Folder for sharing') | |
| 10 … | + .option('-i, --only-people-i-follow', 'Only seed urls from people or channels I follow') | ||
| 10 | 11 … | .parse(process.argv); | |
| 11 | 12 … | ||
| 12 | -require('ssb-client')((err, sbot) => { | ||
| 13 … | +function extractLinksAndSeed(err, logs) { | ||
| 13 | 14 … | if (err) throw err; | |
| 14 | 15 … | ||
| 15 | - console.log("Looking for dat links") | ||
| 16 … | + console.log("Found " + logs.length) | ||
| 16 | 17 … | ||
| 18 … | + logs.forEach(msg => { | ||
| 19 … | + uri.withinString(msg.value.content.text, (datLink) => { | ||
| 20 … | + if (!datLink.startsWith("dat://")) return | ||
| 21 … | + | ||
| 22 … | + console.log("Saving to:", program.folder + "/" + datLink.substring(6)) | ||
| 23 … | + | ||
| 24 … | + Dat(program.folder + "/" + datLink.substring(6), { | ||
| 25 … | + key: datLink | ||
| 26 … | + }, function (err, dat) { | ||
| 27 … | + if (err) throw err | ||
| 28 … | + | ||
| 29 … | + console.log("sharing:", datLink) | ||
| 30 … | + dat.joinNetwork() | ||
| 31 … | + }) | ||
| 32 … | + }) | ||
| 33 … | + }) | ||
| 34 … | +} | ||
| 35 … | + | ||
| 36 … | +function getAll(sbot) { | ||
| 37 … | + console.log("Looking for dat links in all feeds") | ||
| 38 … | + | ||
| 17 | 39 … | pull( | |
| 18 | 40 … | sbot.createLogStream({ reverse: true, limit: 10000 }), | |
| 19 | 41 … | pull.filter((msg) => { | |
| 20 | 42 … | return !msg.value || | |
| 21 | 43 … | msg.value.content.type == 'post' && | |
| 22 | 44 … | typeof msg.value.content.text == "string" && | |
| 23 | 45 … | msg.value.content.text.indexOf("dat://") != -1 | |
| 24 | 46 … | }), | |
| 25 | - pull.collect((err, log) => { | ||
| 26 | - if (err) throw err; | ||
| 47 … | + pull.collect(extractLinksAndSeed) | ||
| 48 … | + ) | ||
| 49 … | +} | ||
| 27 | 50 … | ||
| 28 | - console.log("Found " + log.length) | ||
| 51 … | +function messagesFromPeopleIFollow(sbot, following, channelSubscriptions) { | ||
| 52 … | + console.log("users:", following) | ||
| 53 … | + console.log("channels:", channelSubscriptions) | ||
| 54 … | + pull( | ||
| 55 … | + sbot.createLogStream({ reverse: true, limit: 10000 }), | ||
| 56 … | + pull.filter((msg) => { | ||
| 57 … | + return !msg.value || | ||
| 58 … | + ((msg.value.author in following || | ||
| 59 … | + msg.value.content.channel in channelSubscriptions) | ||
| 60 … | + && msg.value.content.type == 'post' && | ||
| 61 … | + typeof msg.value.content.text == "string" && | ||
| 62 … | + msg.value.content.text.indexOf("dat://") != -1) | ||
| 63 … | + }), | ||
| 64 … | + pull.collect(extractLinksAndSeed) | ||
| 65 … | + ) | ||
| 66 … | +} | ||
| 29 | 67 … | ||
| 30 | - log.forEach(msg => { | ||
| 31 | - uri.withinString(msg.value.content.text, (datLink) => { | ||
| 32 | - if (!datLink.startsWith("dat://")) return | ||
| 68 … | +function getFromPeopleIFollow(sbot) { | ||
| 69 … | + var following = [] | ||
| 70 … | + var channelSubscriptions = [] | ||
| 33 | 71 … | ||
| 34 | - console.log("Saving to:", program.folder + "/" + datLink.substring(6)) | ||
| 72 … | + console.log("Looking for dat links in people i follow") | ||
| 35 | 73 … | ||
| 36 | - Dat(program.folder + "/" + datLink.substring(6), { | ||
| 37 | - key: datLink | ||
| 38 | - }, function (err, dat) { | ||
| 39 | - if (err) throw err | ||
| 40 | - | ||
| 41 | - console.log("sharing:", datLink) | ||
| 42 | - dat.joinNetwork() | ||
| 43 | - }) | ||
| 44 | - }) | ||
| 74 … | + sbot.whoami((err, feed) => { | ||
| 75 … | + pull( | ||
| 76 … | + sbot.createUserStream({ id: feed.id }), | ||
| 77 … | + pull.filter((msg) => { | ||
| 78 … | + return !msg.value || | ||
| 79 … | + msg.value.content.type == 'contact' || | ||
| 80 … | + (msg.value.content.type == 'channel' && | ||
| 81 … | + typeof msg.value.content.subscribed != 'undefined') | ||
| 82 … | + }), | ||
| 83 … | + pull.collect(function (err, msgs) { | ||
| 84 … | + msgs.forEach((msg) => { | ||
| 85 … | + if (msg.value.content.type == 'contact') | ||
| 86 … | + { | ||
| 87 … | + if (msg.value.content.following) | ||
| 88 … | + following[msg.value.content.contact] = 1 | ||
| 89 … | + else | ||
| 90 … | + delete following[msg.value.content.contact] | ||
| 91 … | + } | ||
| 92 … | + else // channel subscription | ||
| 93 … | + { | ||
| 94 … | + if (msg.value.content.subscribed) | ||
| 95 … | + channelSubscriptions[msg.value.content.channel] = 1 | ||
| 96 … | + else | ||
| 97 … | + delete channelSubscriptions[msg.value.content.channel] | ||
| 98 … | + } | ||
| 99 … | + }) | ||
| 100 … | + | ||
| 101 … | + messagesFromPeopleIFollow(sbot, following, channelSubscriptions) | ||
| 45 | 102 … | }) | |
| 46 | - }) | ||
| 47 | - ) | ||
| 103 … | + ) | ||
| 104 … | + }) | ||
| 105 … | +} | ||
| 106 … | + | ||
| 107 … | +require('ssb-client')((err, sbot) => { | ||
| 108 … | + if (err) throw err; | ||
| 109 … | + | ||
| 110 … | + if (program.onlyPeopleIFollow) | ||
| 111 … | + getFromPeopleIFollow(sbot) | ||
| 112 … | + else | ||
| 113 … | + getAll(sbot) | ||
| 48 | 114 … | }) | |
Built with git-ssb-web