81 lines
2.3 KiB
TypeScript
81 lines
2.3 KiB
TypeScript
import { api } from "misskey-js";
|
|
import type { Note } from "misskey-js/entities.js";
|
|
import { sample } from "./util";
|
|
|
|
export const misskey = new api.APIClient({
|
|
origin: Bun.env["MISSKEY_ORIGIN"] || "https://misskey.cannorin.net",
|
|
credential: Bun.env["MISSKEY_CREDENTIAL"],
|
|
});
|
|
|
|
export const me = await misskey.request("i", {});
|
|
|
|
/** check if a note is suitable as an input */
|
|
export const isSuitableAsInput = (n: Note) =>
|
|
!n.user.isBot &&
|
|
!n.replyId &&
|
|
(!n.mentions || n.mentions.length === 0) &&
|
|
n.text?.length &&
|
|
["public", "home"].includes(n.visibility) &&
|
|
!n.cw &&
|
|
n.text.length > 0;
|
|
|
|
/** randomly sample some notes from the timeline */
|
|
export async function getNotes(
|
|
followNotesCount: number,
|
|
localNotesCount: number,
|
|
globalNotesCount: number,
|
|
) {
|
|
// randomly sample N following notes
|
|
const followNotes = (count: number) =>
|
|
misskey
|
|
.request("notes/timeline", { limit: 100 })
|
|
.then((xs) => xs.filter(isSuitableAsInput))
|
|
.then((xs) => sample(xs, count));
|
|
|
|
// randomly sample N local notes
|
|
const localNotes = (count: number) =>
|
|
misskey
|
|
.request("notes/local-timeline", { limit: 100 })
|
|
.then((xs) => xs.filter(isSuitableAsInput))
|
|
.then((xs) => sample(xs, count));
|
|
|
|
// randomly sample N global notes
|
|
const globalNotes = (count: number) =>
|
|
misskey
|
|
.request("notes/global-timeline", { limit: 100 })
|
|
.then((xs) => xs.filter(isSuitableAsInput))
|
|
.then((xs) => sample(xs, count));
|
|
|
|
const notes = await Promise.all([
|
|
followNotes(followNotesCount),
|
|
localNotes(localNotesCount),
|
|
globalNotes(globalNotesCount),
|
|
]);
|
|
return sample(notes.flat());
|
|
}
|
|
|
|
/** fetch the whole reply tree */
|
|
export async function expandReplyTree(
|
|
note: Note,
|
|
cutoff = 5,
|
|
): Promise<{ last: Note; history: Note[] }> {
|
|
let current = note;
|
|
let count = 0;
|
|
const history: Note[] = [];
|
|
while (current.replyId && count < cutoff) {
|
|
const parent = await misskey.request("notes/show", {
|
|
noteId: current.replyId,
|
|
});
|
|
history.push(parent);
|
|
current = parent;
|
|
count++;
|
|
}
|
|
return { last: current, history: history.reverse() };
|
|
}
|
|
|
|
export const sanitizeText = (text: string) =>
|
|
text
|
|
.replaceAll(/(\r\n|\r|\n)\s+/g, "\n\n") // remove extra newlines
|
|
.replaceAll("@", "") // remove mentions
|
|
.replaceAll("#", ""); // remove hashtags
|