mirror of
https://git.joinsharkey.org/Sharkey/Sharkey.git
synced 2024-11-12 21:43:08 +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 = {
|
const context: Entity.Context = {
|
||||||
ancestors: parents.reverse(),
|
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 {
|
return {
|
||||||
...res,
|
...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>> {
|
public async getStatusSource(_id: string): Promise<Response<Entity.StatusSource>> {
|
||||||
return new Promise((_, reject) => {
|
return new Promise((_, reject) => {
|
||||||
const err = new NoImplementedError('misskey does not support')
|
const err = new NoImplementedError('misskey does not support')
|
||||||
|
|
Loading…
Reference in a new issue