87 lines
3.1 KiB
TypeScript
87 lines
3.1 KiB
TypeScript
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
import {Database, Model, Q, Query, Relation} from '@nozbe/watermelondb';
|
|
|
|
import {MM_TABLES} from '@constants/database';
|
|
|
|
import type PostModel from '@typings/database/models/servers/post';
|
|
import type PostInChannelModel from '@typings/database/models/servers/posts_in_channel';
|
|
|
|
const {SERVER: {POST, POSTS_IN_CHANNEL}} = MM_TABLES;
|
|
|
|
export const prepareDeletePost = async (post: PostModel): Promise<Model[]> => {
|
|
const preparedModels: Model[] = [post.prepareDestroyPermanently()];
|
|
const relations: Array<Relation<Model> | Query<Model>> = [post.drafts, post.postsInThread];
|
|
for await (const relation of relations) {
|
|
try {
|
|
const model = await relation.fetch();
|
|
if (model) {
|
|
if (Array.isArray(model)) {
|
|
model.forEach((m) => preparedModels.push(m.prepareDestroyPermanently()));
|
|
} else {
|
|
preparedModels.push(model.prepareDestroyPermanently());
|
|
}
|
|
}
|
|
} catch {
|
|
// Record not found, do nothing
|
|
}
|
|
}
|
|
|
|
const associatedChildren: Array<Query<any>> = [post.files, post.reactions];
|
|
for await (const children of associatedChildren) {
|
|
const models = await children.fetch() as Model[];
|
|
models.forEach((model) => preparedModels.push(model.prepareDestroyPermanently()));
|
|
}
|
|
|
|
return preparedModels;
|
|
};
|
|
|
|
export const queryPostById = async (database: Database, postId: string) => {
|
|
try {
|
|
const userRecord = (await database.collections.get(MM_TABLES.SERVER.POST).find(postId)) as PostModel;
|
|
return userRecord;
|
|
} catch {
|
|
return undefined;
|
|
}
|
|
};
|
|
|
|
export const queryPostsInChannel = (database: Database, channelId: string): Promise<PostInChannelModel[]> => {
|
|
try {
|
|
return database.get(POSTS_IN_CHANNEL).query(
|
|
Q.where('channel_id', channelId),
|
|
Q.sortBy('latest', Q.desc),
|
|
).fetch() as Promise<PostInChannelModel[]>;
|
|
} catch {
|
|
return Promise.resolve([] as PostInChannelModel[]);
|
|
}
|
|
};
|
|
|
|
export const queryPostsChunk = (database: Database, channelId: string, earliest: number, latest: number): Promise<PostModel[]> => {
|
|
try {
|
|
return database.get(POST).query(
|
|
Q.and(
|
|
Q.where('channel_id', channelId),
|
|
Q.where('create_at', Q.between(earliest, latest)),
|
|
Q.where('delete_at', Q.eq(0)),
|
|
),
|
|
Q.sortBy('create_at', Q.desc),
|
|
).fetch() as Promise<PostModel[]>;
|
|
} catch {
|
|
return Promise.resolve([] as PostModel[]);
|
|
}
|
|
};
|
|
|
|
export const queryRecentPostsInChannel = async (database: Database, channelId: string): Promise<PostModel[]> => {
|
|
try {
|
|
const chunks = await queryPostsInChannel(database, channelId);
|
|
if (chunks.length) {
|
|
const recent = chunks[0];
|
|
return queryPostsChunk(database, channelId, recent.earliest, recent.latest);
|
|
}
|
|
return Promise.resolve([] as PostModel[]);
|
|
} catch {
|
|
return Promise.resolve([] as PostModel[]);
|
|
}
|
|
};
|
|
|