mirror of
https://git.joinsharkey.org/Sharkey/Sharkey.git
synced 2024-11-23 01:43:09 +02:00
upd: change way of fetching descendants
This commit is contained in:
parent
73b086b6b8
commit
84fcf9efc7
1 changed files with 43 additions and 1 deletions
|
@ -1203,7 +1203,7 @@ export default class Misskey implements MegalodonInterface {
|
|||
);
|
||||
const context: Entity.Context = {
|
||||
ancestors: parents.reverse(),
|
||||
descendants: res.data.map(n => MisskeyAPI.Converter.note(n, this.baseUrl))
|
||||
descendants: this.dfs(await Promise.all(res.data.map(n => MisskeyAPI.Converter.note(n, this.baseUrl))))
|
||||
}
|
||||
return {
|
||||
...res,
|
||||
|
@ -1212,6 +1212,48 @@ export default class Misskey implements MegalodonInterface {
|
|||
})
|
||||
}
|
||||
|
||||
private dfs(graph: Entity.Status[]) {
|
||||
// we don't need to run dfs if we have zero or one elements
|
||||
if (graph.length <= 1) {
|
||||
return graph;
|
||||
}
|
||||
|
||||
// sort the graph first, so we can grab the correct starting point
|
||||
graph = graph.sort((a, b) => {
|
||||
if (a.id < b.id) return -1;
|
||||
if (a.id > b.id) return 1;
|
||||
return 0;
|
||||
});
|
||||
|
||||
const initialPostId = graph[0].in_reply_to_id;
|
||||
|
||||
// populate stack with all top level replies
|
||||
const stack = graph
|
||||
.filter((reply) => reply.in_reply_to_id === initialPostId)
|
||||
.reverse();
|
||||
const visited = new Set();
|
||||
const result = [];
|
||||
|
||||
while (stack.length) {
|
||||
const currentPost = stack.pop();
|
||||
|
||||
if (currentPost === undefined) return result;
|
||||
|
||||
if (!visited.has(currentPost)) {
|
||||
visited.add(currentPost);
|
||||
result.push(currentPost);
|
||||
|
||||
for (const reply of graph
|
||||
.filter((reply) => reply.in_reply_to_id === currentPost.id)
|
||||
.reverse()) {
|
||||
stack.push(reply);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public async getStatusSource(_id: string): Promise<Response<Entity.StatusSource>> {
|
||||
return new Promise((_, reject) => {
|
||||
const err = new NoImplementedError('misskey does not support')
|
||||
|
|
Loading…
Reference in a new issue