mirror of
https://iceshrimp.dev/limepotato/jormungandr-bite.git
synced 2025-03-04 07:18:50 -07:00
527 lines
21 KiB
TypeScript
527 lines
21 KiB
TypeScript
import { Note } from "@/models/entities/note.js";
|
|
import { ILocalUser, User } from "@/models/entities/user.js";
|
|
import {
|
|
Blockings,
|
|
Followings,
|
|
FollowRequests,
|
|
Mutings,
|
|
NoteFavorites,
|
|
NoteReactions,
|
|
Notes,
|
|
NoteWatchings,
|
|
RegistryItems,
|
|
UserNotePinings,
|
|
UserProfiles,
|
|
Users
|
|
} from "@/models/index.js";
|
|
import { generateVisibilityQuery } from "@/server/api/common/generate-visibility-query.js";
|
|
import { generateMutedUserQuery } from "@/server/api/common/generate-muted-user-query.js";
|
|
import { generateBlockedUserQuery } from "@/server/api/common/generate-block-query.js";
|
|
import AsyncLock from "async-lock";
|
|
import { getUser } from "@/server/api/common/getters.js";
|
|
import { PaginationHelpers } from "@/server/api/mastodon/helpers/pagination.js";
|
|
import { awaitAll } from "@/prelude/await-all.js";
|
|
import createFollowing from "@/services/following/create.js";
|
|
import deleteFollowing from "@/services/following/delete.js";
|
|
import cancelFollowRequest from "@/services/following/requests/cancel.js";
|
|
import createBlocking from "@/services/blocking/create.js";
|
|
import deleteBlocking from "@/services/blocking/delete.js";
|
|
import { genId } from "@/misc/gen-id.js";
|
|
import { Muting } from "@/models/entities/muting.js";
|
|
import { publishUserEvent } from "@/services/stream.js";
|
|
import { UserConverter } from "@/server/api/mastodon/converters/user.js";
|
|
import acceptFollowRequest from "@/services/following/requests/accept.js";
|
|
import { rejectFollowRequest } from "@/services/following/reject.js";
|
|
import { Brackets, IsNull } from "typeorm";
|
|
import { IceshrimpVisibility, VisibilityConverter } from "@/server/api/mastodon/converters/visibility.js";
|
|
import { Files } from "formidable";
|
|
import { toSingleLast } from "@/prelude/array.js";
|
|
import { MediaHelpers } from "@/server/api/mastodon/helpers/media.js";
|
|
import { FileConverter } from "@/server/api/mastodon/converters/file.js";
|
|
import { UserProfile } from "@/models/entities/user-profile.js";
|
|
import { verifyLink } from "@/services/fetch-rel-me.js";
|
|
|
|
export type AccountCache = {
|
|
locks: AsyncLock;
|
|
accounts: MastodonEntity.Account[];
|
|
users: User[];
|
|
};
|
|
|
|
export type LinkPaginationObject<T> = {
|
|
data: T;
|
|
maxId?: string | undefined;
|
|
minId?: string | undefined;
|
|
}
|
|
|
|
export type updateCredsData = {
|
|
display_name: string;
|
|
note: string;
|
|
locked: boolean;
|
|
bot: boolean;
|
|
discoverable: boolean;
|
|
fields_attributes?: { name: string, value: string }[];
|
|
}
|
|
|
|
type RelationshipType = 'followers' | 'following';
|
|
|
|
export class UserHelpers {
|
|
public static async followUser(target: User, localUser: ILocalUser, reblogs: boolean, notify: boolean): Promise<MastodonEntity.Relationship> {
|
|
//FIXME: implement reblogs & notify params
|
|
const following = await Followings.exist({where: {followerId: localUser.id, followeeId: target.id}});
|
|
const requested = await FollowRequests.exist({where: {followerId: localUser.id, followeeId: target.id}});
|
|
if (!following && !requested)
|
|
await createFollowing(localUser, target);
|
|
|
|
return this.getUserRelationshipTo(target.id, localUser.id);
|
|
}
|
|
|
|
public static async unfollowUser(target: User, localUser: ILocalUser): Promise<MastodonEntity.Relationship> {
|
|
const following = await Followings.exist({where: {followerId: localUser.id, followeeId: target.id}});
|
|
const requested = await FollowRequests.exist({where: {followerId: localUser.id, followeeId: target.id}});
|
|
if (following)
|
|
await deleteFollowing(localUser, target);
|
|
if (requested)
|
|
await cancelFollowRequest(target, localUser);
|
|
|
|
return this.getUserRelationshipTo(target.id, localUser.id);
|
|
}
|
|
|
|
public static async blockUser(target: User, localUser: ILocalUser): Promise<MastodonEntity.Relationship> {
|
|
const blocked = await Blockings.exist({where: {blockerId: localUser.id, blockeeId: target.id}});
|
|
if (!blocked)
|
|
await createBlocking(localUser, target);
|
|
|
|
return this.getUserRelationshipTo(target.id, localUser.id);
|
|
}
|
|
|
|
public static async unblockUser(target: User, localUser: ILocalUser): Promise<MastodonEntity.Relationship> {
|
|
const blocked = await Blockings.exist({where: {blockerId: localUser.id, blockeeId: target.id}});
|
|
if (blocked)
|
|
await deleteBlocking(localUser, target);
|
|
|
|
return this.getUserRelationshipTo(target.id, localUser.id);
|
|
}
|
|
|
|
public static async muteUser(target: User, localUser: ILocalUser, notifications: boolean = true, duration: number = 0): Promise<MastodonEntity.Relationship> {
|
|
//FIXME: respect notifications parameter
|
|
const muted = await Mutings.exist({where: {muterId: localUser.id, muteeId: target.id}});
|
|
if (!muted) {
|
|
await Mutings.insert({
|
|
id: genId(),
|
|
createdAt: new Date(),
|
|
expiresAt: duration === 0 ? null : new Date(new Date().getTime() + (duration * 1000)),
|
|
muterId: localUser.id,
|
|
muteeId: target.id,
|
|
} as Muting);
|
|
|
|
publishUserEvent(localUser.id, "mute", target);
|
|
|
|
NoteWatchings.delete({
|
|
userId: localUser.id,
|
|
noteUserId: target.id,
|
|
});
|
|
}
|
|
|
|
return this.getUserRelationshipTo(target.id, localUser.id);
|
|
}
|
|
|
|
public static async unmuteUser(target: User, localUser: ILocalUser): Promise<MastodonEntity.Relationship> {
|
|
const muting = await Mutings.findOneBy({muterId: localUser.id, muteeId: target.id});
|
|
if (muting) {
|
|
await Mutings.delete({
|
|
id: muting.id,
|
|
});
|
|
|
|
publishUserEvent(localUser.id, "unmute", target);
|
|
}
|
|
|
|
return this.getUserRelationshipTo(target.id, localUser.id);
|
|
}
|
|
|
|
public static async acceptFollowRequest(target: User, localUser: ILocalUser): Promise<MastodonEntity.Relationship> {
|
|
const pending = await FollowRequests.exist({where: {followerId: target.id, followeeId: localUser.id}});
|
|
if (pending)
|
|
await acceptFollowRequest(localUser, target);
|
|
return this.getUserRelationshipTo(target.id, localUser.id);
|
|
}
|
|
|
|
public static async rejectFollowRequest(target: User, localUser: ILocalUser): Promise<MastodonEntity.Relationship> {
|
|
const pending = await FollowRequests.exist({where: {followerId: target.id, followeeId: localUser.id}});
|
|
if (pending)
|
|
await rejectFollowRequest(localUser, target);
|
|
return this.getUserRelationshipTo(target.id, localUser.id);
|
|
}
|
|
|
|
public static async updateCredentials(user: ILocalUser, formData: updateCredsData, files: Files | undefined): Promise<MastodonEntity.Account> {
|
|
const updates: Partial<User> = {};
|
|
const profileUpdates: Partial<UserProfile> = {};
|
|
|
|
const avatar = toSingleLast(files?.avatar);
|
|
const header = toSingleLast(files?.header);
|
|
|
|
if (avatar) {
|
|
const file = await MediaHelpers.uploadMediaBasic(user, avatar);
|
|
updates.avatarId = file.id;
|
|
}
|
|
|
|
if (header) {
|
|
const file = await MediaHelpers.uploadMediaBasic(user, header);
|
|
updates.bannerId = file.id;
|
|
}
|
|
|
|
if (formData.fields_attributes) {
|
|
profileUpdates.fields = await Promise.all(formData.fields_attributes.map(async field => {
|
|
const verified = field.value.startsWith("http") ? await verifyLink(field.value, user.username) : undefined;
|
|
return {
|
|
...field,
|
|
verified
|
|
};
|
|
}));
|
|
}
|
|
|
|
if (formData.display_name) updates.name = formData.display_name;
|
|
if (formData.note) profileUpdates.description = formData.note;
|
|
if (formData.locked) updates.isLocked = formData.locked;
|
|
if (formData.bot) updates.isBot = formData.bot;
|
|
if (formData.discoverable) updates.isExplorable = formData.discoverable;
|
|
|
|
if (Object.keys(updates).length > 0) await Users.update(user.id, updates);
|
|
if (Object.keys(profileUpdates).length > 0) await UserProfiles.update({ userId: user.id }, profileUpdates);
|
|
|
|
return this.verifyCredentials(user);
|
|
}
|
|
|
|
public static async verifyCredentials(user: ILocalUser): Promise<MastodonEntity.Account> {
|
|
const acct = UserConverter.encode(user);
|
|
const profile = UserProfiles.findOneByOrFail({userId: user.id});
|
|
const privacy = this.getDefaultNoteVisibility(user);
|
|
const fields = profile.then(profile => profile.fields.map(field => {
|
|
return {
|
|
name: field.name,
|
|
value: field.value
|
|
} as MastodonEntity.Field;
|
|
}));
|
|
return acct.then(acct => {
|
|
const source = {
|
|
note: profile.then(profile => profile.description ?? ''),
|
|
fields: fields,
|
|
privacy: privacy.then(p => VisibilityConverter.encode(p)),
|
|
sensitive: profile.then(p => p.alwaysMarkNsfw),
|
|
language: profile.then(p => p.lang ?? ''),
|
|
};
|
|
|
|
const result = {
|
|
...acct,
|
|
source: awaitAll(source)
|
|
};
|
|
|
|
return awaitAll(result);
|
|
});
|
|
}
|
|
|
|
public static async getUserFromAcct(acct: string): Promise<User | null> {
|
|
const split = acct.toLowerCase().split('@');
|
|
if (split.length > 2) throw new Error('Invalid acct');
|
|
return Users.findOneBy({usernameLower: split[0], host: split[1] ?? IsNull()});
|
|
}
|
|
|
|
public static async getUserMutes(user: ILocalUser, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40, cache: AccountCache = UserHelpers.getFreshAccountCache()): Promise<LinkPaginationObject<MastodonEntity.MutedAccount[]>> {
|
|
if (limit > 80) limit = 80;
|
|
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
Mutings.createQueryBuilder("muting"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
);
|
|
|
|
query.andWhere("muting.muterId = :userId", {userId: user.id})
|
|
.innerJoinAndSelect("muting.mutee", "mutee");
|
|
|
|
return query.take(limit).getMany().then(async p => {
|
|
if (minId !== undefined) p = p.reverse();
|
|
const users = p
|
|
.map(p => p.mutee)
|
|
.filter(p => p) as User[];
|
|
|
|
const result = await UserConverter.encodeMany(users, cache)
|
|
.then(res => res.map(m => {
|
|
const muting = p.find(acc => acc.muteeId === m.id);
|
|
return {
|
|
...m,
|
|
mute_expires_at: muting?.expiresAt?.toISOString() ?? null
|
|
} as MastodonEntity.MutedAccount
|
|
}));
|
|
|
|
return {
|
|
data: result,
|
|
maxId: p.map(p => p.id).at(-1),
|
|
minId: p.map(p => p.id)[0],
|
|
};
|
|
});
|
|
}
|
|
|
|
public static async getUserBlocks(user: ILocalUser, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40): Promise<LinkPaginationObject<User[]>> {
|
|
if (limit > 80) limit = 80;
|
|
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
Blockings.createQueryBuilder("blocking"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
);
|
|
|
|
query.andWhere("blocking.blockerId = :userId", {userId: user.id})
|
|
.innerJoinAndSelect("blocking.blockee", "blockee");
|
|
|
|
return query.take(limit).getMany().then(p => {
|
|
if (minId !== undefined) p = p.reverse();
|
|
const users = p
|
|
.map(p => p.blockee)
|
|
.filter(p => p) as User[];
|
|
|
|
return {
|
|
data: users,
|
|
maxId: p.map(p => p.id).at(-1),
|
|
minId: p.map(p => p.id)[0],
|
|
};
|
|
});
|
|
}
|
|
|
|
public static async getUserFollowRequests(user: ILocalUser, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40): Promise<LinkPaginationObject<User[]>> {
|
|
if (limit > 80) limit = 80;
|
|
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
FollowRequests.createQueryBuilder("request"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
);
|
|
|
|
query.andWhere("request.followeeId = :userId", {userId: user.id})
|
|
.innerJoinAndSelect("request.follower", "follower");
|
|
|
|
return query.take(limit).getMany().then(p => {
|
|
if (minId !== undefined) p = p.reverse();
|
|
const users = p
|
|
.map(p => p.follower)
|
|
.filter(p => p) as User[];
|
|
|
|
return {
|
|
data: users,
|
|
maxId: p.map(p => p.id).at(-1),
|
|
minId: p.map(p => p.id)[0],
|
|
};
|
|
});
|
|
}
|
|
|
|
public static async getUserStatuses(user: User, localUser: ILocalUser | null, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 20, onlyMedia: boolean = false, excludeReplies: boolean = false, excludeReblogs: boolean = false, pinned: boolean = false, tagged: string | undefined): Promise<Note[]> {
|
|
if (limit > 40) limit = 40;
|
|
|
|
if (tagged !== undefined) {
|
|
//FIXME respect tagged
|
|
return [];
|
|
}
|
|
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
Notes.createQueryBuilder("note"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
)
|
|
.andWhere("note.userId = :userId");
|
|
|
|
if (pinned) {
|
|
const sq = UserNotePinings.createQueryBuilder("pin")
|
|
.select("pin.noteId")
|
|
.where("pin.userId = :userId");
|
|
query.andWhere(`note.id IN (${sq.getQuery()})`);
|
|
}
|
|
|
|
if (excludeReblogs) {
|
|
query.andWhere(
|
|
new Brackets(qb => {
|
|
qb.where('note.renoteId IS NULL')
|
|
.orWhere('note.text IS NOT NULL');
|
|
}));
|
|
}
|
|
|
|
if (excludeReplies) {
|
|
query.leftJoin("note", "thread", "note.threadId = thread.id")
|
|
.andWhere(
|
|
new Brackets(qb => {
|
|
qb.where("note.replyId IS NULL")
|
|
.orWhere(new Brackets(qb => {
|
|
qb.where('note.mentions = :mentions', {mentions: []})
|
|
.andWhere('thread.userId = :userId')
|
|
}));
|
|
}));
|
|
}
|
|
|
|
query.leftJoinAndSelect("note.renote", "renote");
|
|
|
|
generateVisibilityQuery(query, localUser);
|
|
if (localUser) {
|
|
generateMutedUserQuery(query, localUser, user);
|
|
generateBlockedUserQuery(query, localUser);
|
|
}
|
|
|
|
if (onlyMedia) query.andWhere("note.fileIds != '{}'");
|
|
|
|
query.andWhere("note.visibility != 'hidden'");
|
|
query.andWhere("note.visibility != 'specified'");
|
|
|
|
query.setParameters({userId: user.id});
|
|
|
|
return PaginationHelpers.execQuery(query, limit, minId !== undefined);
|
|
}
|
|
|
|
public static async getUserBookmarks(localUser: ILocalUser, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 20): Promise<LinkPaginationObject<Note[]>> {
|
|
if (limit > 40) limit = 40;
|
|
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
NoteFavorites.createQueryBuilder("favorite"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
)
|
|
.andWhere("favorite.userId = :meId", {meId: localUser.id})
|
|
.leftJoinAndSelect("favorite.note", "note");
|
|
|
|
generateVisibilityQuery(query, localUser);
|
|
|
|
return PaginationHelpers.execQuery(query, limit, minId !== undefined)
|
|
.then(res => {
|
|
return {
|
|
data: res.map(p => p.note as Note),
|
|
maxId: res.map(p => p.id).at(-1),
|
|
minId: res.map(p => p.id)[0],
|
|
};
|
|
});
|
|
}
|
|
|
|
public static async getUserFavorites(localUser: ILocalUser, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 20): Promise<LinkPaginationObject<Note[]>> {
|
|
if (limit > 40) limit = 40;
|
|
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
NoteReactions.createQueryBuilder("reaction"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
)
|
|
.andWhere("reaction.userId = :meId", {meId: localUser.id})
|
|
.leftJoinAndSelect("reaction.note", "note");
|
|
|
|
generateVisibilityQuery(query, localUser);
|
|
|
|
return PaginationHelpers.execQuery(query, limit, minId !== undefined)
|
|
.then(res => {
|
|
return {
|
|
data: res.map(p => p.note as Note),
|
|
maxId: res.map(p => p.id).at(-1),
|
|
minId: res.map(p => p.id)[0],
|
|
};
|
|
});
|
|
}
|
|
|
|
private static async getUserRelationships(type: RelationshipType, user: User, localUser: ILocalUser | null, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40): Promise<LinkPaginationObject<User[]>> {
|
|
if (limit > 80) limit = 80;
|
|
|
|
const profile = await UserProfiles.findOneByOrFail({userId: user.id});
|
|
if (profile.ffVisibility === "private") {
|
|
if (!localUser || user.id !== localUser.id) return {data: []};
|
|
} else if (profile.ffVisibility === "followers") {
|
|
if (!localUser) return {data: []};
|
|
if (user.id !== localUser.id) {
|
|
const isFollowed = await Followings.exist({
|
|
where: {
|
|
followeeId: user.id,
|
|
followerId: localUser.id,
|
|
},
|
|
});
|
|
if (!isFollowed) return {data: []};
|
|
}
|
|
}
|
|
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
Followings.createQueryBuilder("following"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
);
|
|
|
|
if (type === "followers") {
|
|
query.andWhere("following.followeeId = :userId", {userId: user.id})
|
|
.innerJoinAndSelect("following.follower", "follower");
|
|
} else {
|
|
query.andWhere("following.followerId = :userId", {userId: user.id})
|
|
.innerJoinAndSelect("following.followee", "followee");
|
|
}
|
|
|
|
return query.take(limit).getMany().then(p => {
|
|
if (minId !== undefined) p = p.reverse();
|
|
|
|
return {
|
|
data: p.map(p => type === "followers" ? p.follower : p.followee).filter(p => p) as User[],
|
|
maxId: p.map(p => p.id).at(-1),
|
|
minId: p.map(p => p.id)[0],
|
|
};
|
|
});
|
|
}
|
|
|
|
public static async getUserFollowers(user: User, localUser: ILocalUser | null, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40): Promise<LinkPaginationObject<User[]>> {
|
|
return this.getUserRelationships('followers', user, localUser, maxId, sinceId, minId, limit);
|
|
}
|
|
|
|
public static async getUserFollowing(user: User, localUser: ILocalUser | null, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40): Promise<LinkPaginationObject<User[]>> {
|
|
return this.getUserRelationships('following', user, localUser, maxId, sinceId, minId, limit);
|
|
}
|
|
|
|
public static async getUserRelationhipToMany(targetIds: string[], localUserId: string): Promise<MastodonEntity.Relationship[]> {
|
|
return Promise.all(targetIds.map(targetId => this.getUserRelationshipTo(targetId, localUserId)));
|
|
}
|
|
|
|
public static async getUserRelationshipTo(targetId: string, localUserId: string): Promise<MastodonEntity.Relationship> {
|
|
const relation = await Users.getRelation(localUserId, targetId);
|
|
const response = {
|
|
id: targetId,
|
|
following: relation.isFollowing,
|
|
followed_by: relation.isFollowed,
|
|
blocking: relation.isBlocking,
|
|
blocked_by: relation.isBlocked,
|
|
muting: relation.isMuted,
|
|
muting_notifications: relation.isMuted,
|
|
requested: relation.hasPendingFollowRequestFromYou,
|
|
domain_blocking: false, //FIXME
|
|
showing_reblogs: !relation.isRenoteMuted,
|
|
endorsed: false,
|
|
notifying: false, //FIXME
|
|
note: '' //FIXME
|
|
}
|
|
|
|
return awaitAll(response);
|
|
}
|
|
|
|
public static async getUserCached(id: string, cache: AccountCache = UserHelpers.getFreshAccountCache()): Promise<User> {
|
|
return cache.locks.acquire(id, async () => {
|
|
const cacheHit = cache.users.find(p => p.id == id);
|
|
if (cacheHit) return cacheHit;
|
|
return getUser(id).then(p => {
|
|
cache.users.push(p);
|
|
return p;
|
|
});
|
|
});
|
|
}
|
|
|
|
public static getFreshAccountCache(): AccountCache {
|
|
return {
|
|
locks: new AsyncLock(),
|
|
accounts: [],
|
|
users: [],
|
|
};
|
|
}
|
|
|
|
public static async getDefaultNoteVisibility(user: ILocalUser): Promise<IceshrimpVisibility> {
|
|
return RegistryItems.findOneBy({domain: IsNull(), userId: user.id, key: 'defaultNoteVisibility', scope: '{client,base}'}).then(p => p?.value ?? 'public')
|
|
}
|
|
}
|