1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
var { promisify } = require("util");
var pull = require("pull-stream");
var Pushable = require("pull-pushable");
var many = require("./pull-many-v2");
// Change to 'true' to get debugging output
DEBUG = false
DEFAULT_CHANNEL_OBJECT = {
messages: []
}
DEFAULT_CHANNEL_OBJECT_SIZE = 15
FRIENDS_POLL_TIME = 1 // ms
GRAPH_TYPE = "follow"
MAX_HOPS = 1
class StreamData {
constructor() {
this.finished = false,
this.oldestTimestampSeen = Number.MAX_SAFE_INTEGER,
this.waitingMessages = []
this.markFinished = function() {
this.finished = true,
this.oldestTimestampSeen = 0
}
}
}
class StreamController {
constructor() {
this.outputStream = Pushable(),
this.streamData = {
channelStream: new StreamData(),
hashtagStream: new StreamData(),
},
this.pushMessage = function(source, newMsg) {
var streamData = source == "hashtag" ? this.streamData.hashtagStream : this.streamData.channelStream;
streamData.oldestTimestampSeen = newMsg.value.timestamp;
streamData.waitingMessages.push(newMsg);
this.pushNewlySafeMessages();
},
this.pushNewlySafeMessages = function() {
var newlySafeMessages = [];
var oldestSafeTimestamp = Math.max(...Object.values(this.streamData).map(datum => datum.oldestTimestampSeen));
debug("oldest safe timestamp:" + oldestSafeTimestamp);
for(var streamDataIndex in this.streamData) {
var streamDatum = this.streamData[streamDataIndex]
debug("length of waiting messages: " + streamDatum.waitingMessages.length);
if(streamDatum.waitingMessages.length) {
debug("timestamp of first waiting message: " + streamDatum.waitingMessages[0].value.timestamp);
}
while(streamDatum.waitingMessages.length && streamDatum.waitingMessages[0].value.timestamp >= oldestSafeTimestamp) {
var safeMsg = streamDatum.waitingMessages.shift(); // pop the newest waiting message
newlySafeMessages.push(safeMsg);
}
}
// in general, newlySafeMessages might not be in order, we should sort before appending
newlySafeMessages.sort((msg1, msg2) => (msg1.value.timestamp > msg2.value.timestamp ? -1 : 1));
for(var msgIndex in newlySafeMessages) {
debug("pushing safe message...");
this.outputStream.push(newlySafeMessages[msgIndex]);
}
},
this.finish = function() {
for(var datumIndex in this.streamData) {
this.streamData[datumIndex].oldestTimestampSeen = 0;
}
this.pushNewlySafeMessages();
}
}
}
module.exports = {
getMessages: function(client, channelName, preserve, cb, hops=MAX_HOPS) {
client.friends.hops({
dunbar: Number.MAX_SAFE_INTEGER,
max: hops
}, function(error, friends) {
if(error) {
throw "Couldn't get a list of friends from scuttlebot:\n" + error;
}
var followedIds = Object.keys(friends).map(id => id.toLowerCase());
getMessagesFrom(client, channelName, followedIds, preserve, cb);
});
}
}
function getMessagesFrom(client, channelName, followedIds, preserve, cb) {
debug("Fetching messages from IDs in " + JSON.stringify(followedIds));
var channelTag = "#" + channelName;
var streamController = new StreamController();
var hashtagStream = createHashtagStream(client, channelName);
var channelStream = createChannelStream(client, channelName);
var stream = many([hashtagStream, channelStream]);
pull(stream, pull.filter(function(msg) {
return followedIds.includes(msg.data.value.author.toLowerCase());
}), pull.filter(function(msg) {
var hasHashtag = msg.data.value.content && msg.data.value.content.text && typeof(msg.data.value.content.text) == "string" && msg.data.value.content.text.includes(channelTag);
if(msg.source == "hashtag") {
return hasHashtag;
}
else {
return !hasHashtag; // prevents us from double-counting messages with both the hashtag and the channel header
}
}), pull.drain(function(msg) {
streamController.pushMessage(msg.source, msg.data);
}, function() {
streamController.finish();
}));
cb(streamController.outputStream, preserve);
}
function createHashtagStream(client, channelName) {
var search = client.search && client.search.query;
if(!search) {
console.log("[FATAL] ssb-search plugin must be installed to us channels");
process.exit(5);
}
var query = search({
query: channelName
});
query.streamName = "hashtag"; // mark the stream object so we can tell which stream an object came from
return query;
}
function createChannelStream(client, channelName) {
var query = client.query.read({
query: [{
"$filter": {
value: {
content: {
channel: channelName
}
}
}
}],
reverse: true
});
query.streamName = "channel"; // mark the stream object so we can tell which stream a message came from later
return query;
}
function debug(message) {
if(DEBUG) {
var timestamp = new Date();
console.log("[channels-lib] [" + timestamp.toISOString() + "] " + message);
}
}
|
