mirror of
https://iceshrimp.dev/limepotato/jormungandr-bite.git
synced 2025-03-04 07:18:50 -07:00
530 lines
22 KiB
TypeScript
530 lines
22 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 { UserProfile } from "@/models/entities/user-profile.js";
|
|
import { verifyLink } from "@/services/fetch-rel-me.js";
|
|
import { MastoApiError } from "@/server/api/mastodon/middleware/catch-errors.js";
|
|
import { MastoContext } from "@/server/api/mastodon/index.js";
|
|
|
|
export type AccountCache = {
|
|
locks: AsyncLock;
|
|
accounts: MastodonEntity.Account[];
|
|
users: User[];
|
|
};
|
|
|
|
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, reblogs: boolean, notify: boolean, ctx: MastoContext): Promise<MastodonEntity.Relationship> {
|
|
//FIXME: implement reblogs & notify params
|
|
const localUser = ctx.user as ILocalUser;
|
|
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, ctx: MastoContext): Promise<MastodonEntity.Relationship> {
|
|
const localUser = ctx.user as ILocalUser;
|
|
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, ctx: MastoContext): Promise<MastodonEntity.Relationship> {
|
|
const localUser = ctx.user as ILocalUser;
|
|
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, ctx: MastoContext): Promise<MastodonEntity.Relationship> {
|
|
const localUser = ctx.user as ILocalUser;
|
|
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, notifications: boolean = true, duration: number = 0, ctx: MastoContext): Promise<MastodonEntity.Relationship> {
|
|
//FIXME: respect notifications parameter
|
|
const localUser = ctx.user as ILocalUser;
|
|
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, ctx: MastoContext): Promise<MastodonEntity.Relationship> {
|
|
const localUser = ctx.user as ILocalUser;
|
|
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, ctx: MastoContext): Promise<MastodonEntity.Relationship> {
|
|
const localUser = ctx.user as ILocalUser;
|
|
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, ctx: MastoContext): Promise<MastodonEntity.Relationship> {
|
|
const localUser = ctx.user as ILocalUser;
|
|
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(ctx: MastoContext): Promise<MastodonEntity.Account> {
|
|
const user = ctx.user as ILocalUser;
|
|
const files = (ctx.request as any).files as Files | undefined;
|
|
const formData = (ctx.request as any).body as updateCredsData;
|
|
|
|
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(avatar, ctx);
|
|
updates.avatarId = file.id;
|
|
}
|
|
|
|
if (header) {
|
|
const file = await MediaHelpers.uploadMediaBasic(header, ctx);
|
|
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(ctx);
|
|
}
|
|
|
|
public static async verifyCredentials(ctx: MastoContext): Promise<MastodonEntity.Account> {
|
|
const user = ctx.user as ILocalUser;
|
|
const acct = UserConverter.encode(user, ctx);
|
|
const profile = UserProfiles.findOneByOrFail({ userId: user.id });
|
|
const privacy = this.getDefaultNoteVisibility(ctx);
|
|
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> {
|
|
const split = acct.toLowerCase().split('@');
|
|
if (split.length > 2) throw new Error('Invalid acct');
|
|
return Users.findOneBy({ usernameLower: split[0], host: split[1] ?? IsNull() })
|
|
.then(p => {
|
|
if (p) return p;
|
|
throw new MastoApiError(404);
|
|
});
|
|
}
|
|
|
|
public static async getUserMutes(maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40, ctx: MastoContext): Promise<MastodonEntity.MutedAccount[]> {
|
|
if (limit > 80) limit = 80;
|
|
|
|
const user = ctx.user as ILocalUser;
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
Mutings.createQueryBuilder("muting"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
);
|
|
|
|
query.andWhere("muting.muterId = :userId", { userId: user.id })
|
|
.innerJoinAndSelect("muting.mutee", "mutee");
|
|
|
|
return PaginationHelpers.execQueryLinkPagination(query, limit, minId !== undefined, ctx)
|
|
.then(async mutes => {
|
|
const users = mutes
|
|
.map(p => p.mutee)
|
|
.filter(p => p) as User[];
|
|
|
|
return await UserConverter.encodeMany(users, ctx)
|
|
.then(res => res.map(m => {
|
|
const muting = mutes.find(acc => acc.muteeId === m.id);
|
|
return {
|
|
...m,
|
|
mute_expires_at: muting?.expiresAt?.toISOString() ?? null
|
|
} as MastodonEntity.MutedAccount
|
|
}));
|
|
});
|
|
}
|
|
|
|
public static async getUserBlocks(maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40, ctx: MastoContext): Promise<User[]> {
|
|
if (limit > 80) limit = 80;
|
|
|
|
const user = ctx.user as ILocalUser;
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
Blockings.createQueryBuilder("blocking"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
);
|
|
|
|
query.andWhere("blocking.blockerId = :userId", { userId: user.id })
|
|
.innerJoinAndSelect("blocking.blockee", "blockee");
|
|
|
|
return PaginationHelpers.execQueryLinkPagination(query, limit, minId !== undefined, ctx)
|
|
.then(blocks => {
|
|
return blocks
|
|
.map(p => p.blockee)
|
|
.filter(p => p) as User[];
|
|
});
|
|
}
|
|
|
|
public static async getUserFollowRequests(maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40, ctx: MastoContext): Promise<User[]> {
|
|
if (limit > 80) limit = 80;
|
|
|
|
const user = ctx.user as ILocalUser;
|
|
const query = PaginationHelpers.makePaginationQuery(
|
|
FollowRequests.createQueryBuilder("request"),
|
|
sinceId,
|
|
maxId,
|
|
minId
|
|
);
|
|
|
|
query.andWhere("request.followeeId = :userId", { userId: user.id })
|
|
.innerJoinAndSelect("request.follower", "follower");
|
|
|
|
return PaginationHelpers.execQueryLinkPagination(query, limit, minId !== undefined, ctx)
|
|
.then(requests => {
|
|
return requests
|
|
.map(p => p.follower)
|
|
.filter(p => p) as User[];
|
|
});
|
|
}
|
|
|
|
public static async getUserStatuses(user: User, 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, ctx: MastoContext): Promise<Note[]> {
|
|
if (limit > 40) limit = 40;
|
|
const localUser = ctx.user as ILocalUser | null;
|
|
|
|
if (tagged !== undefined && tagged.length > 0) {
|
|
//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.execQueryLinkPagination(query, limit, minId !== undefined, ctx);
|
|
}
|
|
|
|
public static async getUserBookmarks(maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 20, ctx: MastoContext): Promise<Note[]> {
|
|
if (limit > 40) limit = 40;
|
|
|
|
const localUser = ctx.user as ILocalUser;
|
|
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.execQueryLinkPagination(query, limit, minId !== undefined, ctx)
|
|
.then(res => res.map(p => p.note as Note));
|
|
}
|
|
|
|
public static async getUserFavorites(maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 20, ctx: MastoContext): Promise<Note[]> {
|
|
if (limit > 40) limit = 40;
|
|
|
|
const localUser = ctx.user as ILocalUser;
|
|
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.execQueryLinkPagination(query, limit, minId !== undefined, ctx)
|
|
.then(res => res.map(p => p.note as Note));
|
|
}
|
|
|
|
private static async getUserRelationships(type: RelationshipType, user: User, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40, ctx: MastoContext): Promise<User[]> {
|
|
if (limit > 80) limit = 80;
|
|
|
|
const localUser = ctx.user as ILocalUser | null;
|
|
const profile = await UserProfiles.findOneByOrFail({ userId: user.id });
|
|
if (profile.ffVisibility === "private") {
|
|
if (!localUser || user.id !== localUser.id) return [];
|
|
} else if (profile.ffVisibility === "followers") {
|
|
if (!localUser) return [];
|
|
if (user.id !== localUser.id) {
|
|
const isFollowed = await Followings.exist({
|
|
where: {
|
|
followeeId: user.id,
|
|
followerId: localUser.id,
|
|
},
|
|
});
|
|
if (!isFollowed) return [];
|
|
}
|
|
}
|
|
|
|
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 PaginationHelpers.execQueryLinkPagination(query, limit, minId !== undefined, ctx)
|
|
.then(relations => relations
|
|
.map(p => type === "followers" ? p.follower : p.followee)
|
|
.filter(p => p) as User[]);
|
|
}
|
|
|
|
public static async getUserFollowers(user: User, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40, ctx: MastoContext): Promise<User[]> {
|
|
return this.getUserRelationships('followers', user, maxId, sinceId, minId, limit, ctx);
|
|
}
|
|
|
|
public static async getUserFollowing(user: User, maxId: string | undefined, sinceId: string | undefined, minId: string | undefined, limit: number = 40, ctx: MastoContext): Promise<User[]> {
|
|
return this.getUserRelationships('following', user, maxId, sinceId, minId, limit, ctx);
|
|
}
|
|
|
|
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, ctx: MastoContext): Promise<User> {
|
|
const cache = ctx.cache as AccountCache;
|
|
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 async getUserCachedOr404(id: string, ctx: MastoContext): Promise<User> {
|
|
return this.getUserCached(id, ctx).catch(_ => {
|
|
throw new MastoApiError(404);
|
|
});
|
|
}
|
|
|
|
public static async getUserOr404(id: string): Promise<User> {
|
|
return getUser(id).catch(_ => {
|
|
throw new MastoApiError(404);
|
|
});
|
|
}
|
|
|
|
public static getFreshAccountCache(): AccountCache {
|
|
return {
|
|
locks: new AsyncLock(),
|
|
accounts: [],
|
|
users: [],
|
|
};
|
|
}
|
|
|
|
public static async getDefaultNoteVisibility(ctx: MastoContext): Promise<IceshrimpVisibility> {
|
|
const user = ctx.user as ILocalUser;
|
|
return RegistryItems.findOneBy({
|
|
domain: IsNull(),
|
|
userId: user.id,
|
|
key: 'defaultNoteVisibility',
|
|
scope: '{client,base}'
|
|
}).then(p => p?.value ?? 'public')
|
|
}
|
|
}
|