89 lines
2.3 KiB
Dart
89 lines
2.3 KiB
Dart
import 'package:lemmy_api_client/v2.dart';
|
|
|
|
import 'util/hot_rank.dart';
|
|
|
|
enum CommentSortType {
|
|
hot,
|
|
top,
|
|
// ignore: constant_identifier_names
|
|
new_,
|
|
old,
|
|
chat,
|
|
}
|
|
|
|
extension on CommentSortType {
|
|
/// returns a compare function for sorting a CommentTree according
|
|
/// to the comment sort type
|
|
int Function(CommentTree a, CommentTree b) get sortFunction {
|
|
switch (this) {
|
|
case CommentSortType.chat:
|
|
throw Exception('Sorting a CommentTree in chat mode is not supported'
|
|
' because it would restructure the whole tree');
|
|
|
|
case CommentSortType.hot:
|
|
return (b, a) =>
|
|
a.comment.computedHotRank.compareTo(b.comment.computedHotRank);
|
|
|
|
case CommentSortType.new_:
|
|
return (b, a) =>
|
|
a.comment.comment.published.compareTo(b.comment.comment.published);
|
|
|
|
case CommentSortType.old:
|
|
return (b, a) =>
|
|
b.comment.comment.published.compareTo(a.comment.comment.published);
|
|
|
|
case CommentSortType.top:
|
|
return (b, a) =>
|
|
a.comment.counts.score.compareTo(b.comment.counts.score);
|
|
}
|
|
|
|
throw Exception('unreachable');
|
|
}
|
|
}
|
|
|
|
class CommentTree {
|
|
CommentView comment;
|
|
List<CommentTree> children;
|
|
|
|
CommentTree(this.comment, [this.children]) {
|
|
children ??= [];
|
|
}
|
|
|
|
/// takes raw linear comments and turns them into a CommentTree
|
|
static List<CommentTree> fromList(List<CommentView> comments) {
|
|
CommentTree gatherChildren(CommentTree parent) {
|
|
for (final el in comments) {
|
|
if (el.comment.parentId == parent.comment.comment.id) {
|
|
parent.children.add(gatherChildren(CommentTree(el)));
|
|
}
|
|
}
|
|
return parent;
|
|
}
|
|
|
|
final topLevelParents = comments
|
|
.where((e) => e.comment.parentId == null)
|
|
.map((e) => CommentTree(e));
|
|
|
|
final result = topLevelParents.map(gatherChildren).toList();
|
|
return result;
|
|
}
|
|
|
|
/// recursive sorter
|
|
void _sort(int compare(CommentTree a, CommentTree b)) {
|
|
children.sort(compare);
|
|
for (final el in children) {
|
|
el._sort(compare);
|
|
}
|
|
}
|
|
|
|
/// Sorts in-place a list of CommentTrees according to a given sortType
|
|
static List<CommentTree> sortList(
|
|
CommentSortType sortType, List<CommentTree> comms) {
|
|
comms.sort(sortType.sortFunction);
|
|
for (final el in comms) {
|
|
el._sort(sortType.sortFunction);
|
|
}
|
|
return comms;
|
|
}
|
|
}
|