Fix bug

1 files changed, 19 insertions, 0 deletions

diff --git a/src/misc/extract-mentions.ts b/src/misc/extract-mentions.ts
new file mode 100644
index 0000000000..1d844211c0
--- /dev/null
+++ b/src/misc/extract-mentions.ts
@@ -0,0 +1,19 @@
+import parse from '../mfm/parse';
+import { Node, IMentionNode } from '../mfm/parser';
+
+export default function(tokens: ReturnType<typeof parse>): IMentionNode['props'][] {
+	const mentions: IMentionNode['props'][] = [];
+
+	const extract = (tokens: Node[]) => {
+		for (const x of tokens.filter(x => x.name === 'mention')) {
+			mentions.push(x.props);
+		}
+		for (const x of tokens.filter(x => x.children)) {
+			extract(x.children);
+		}
+	};
+
+	extract(tokens);
+
+	return mentions;
+}