import { Login, LoginResponse, CreatePost, EditPost, CreateComment, DeletePost, RemovePost, StickyPost, LockPost, PostResponse, SearchResponse, FollowCommunity, CommunityResponse, GetFollowedCommunitiesResponse, GetPostResponse, Register, Comment, EditComment, DeleteComment, RemoveComment, Search, CommentResponse, GetCommunity, CreateCommunity, DeleteCommunity, RemoveCommunity, GetUserMentions, CreateCommentLike, CreatePostLike, EditPrivateMessage, DeletePrivateMessage, GetFollowedCommunities, GetPrivateMessages, GetSite, GetPost, PrivateMessageResponse, PrivateMessagesResponse, GetUserMentionsResponse, SaveUserSettings, SortType, ListingType, GetSiteResponse, SearchType, LemmyHttp, BanUserResponse, BanUser, BanFromCommunity, BanFromCommunityResponse, Post, CreatePrivateMessage, } from 'lemmy-js-client'; export interface API { client: LemmyHttp; auth?: string; } export let alpha: API = { client: new LemmyHttp('http://localhost:8541/api/v2'), }; export let beta: API = { client: new LemmyHttp('http://localhost:8551/api/v2'), }; export let gamma: API = { client: new LemmyHttp('http://localhost:8561/api/v2'), }; export let delta: API = { client: new LemmyHttp('http://localhost:8571/api/v2'), }; export let epsilon: API = { client: new LemmyHttp('http://localhost:8581/api/v2'), }; export async function setupLogins() { let formAlpha: Login = { username_or_email: 'lemmy_alpha', password: 'lemmy', }; let resAlpha = alpha.client.login(formAlpha); let formBeta = { username_or_email: 'lemmy_beta', password: 'lemmy', }; let resBeta = beta.client.login(formBeta); let formGamma = { username_or_email: 'lemmy_gamma', password: 'lemmy', }; let resGamma = gamma.client.login(formGamma); let formDelta = { username_or_email: 'lemmy_delta', password: 'lemmy', }; let resDelta = delta.client.login(formDelta); let formEpsilon = { username_or_email: 'lemmy_epsilon', password: 'lemmy', }; let resEpsilon = epsilon.client.login(formEpsilon); let res = await Promise.all([ resAlpha, resBeta, resGamma, resDelta, resEpsilon, ]); alpha.auth = res[0].jwt; beta.auth = res[1].jwt; gamma.auth = res[2].jwt; delta.auth = res[3].jwt; epsilon.auth = res[4].jwt; } export async function createPost( api: API, community_id: number ): Promise { let name = randomString(5); let body = randomString(10); let url = 'https://google.com/'; let form: CreatePost = { name, url, body, auth: api.auth, community_id, nsfw: false, }; return api.client.createPost(form); } export async function editPost(api: API, post: Post): Promise { let name = 'A jest test federated post, updated'; let form: EditPost = { name, edit_id: post.id, auth: api.auth, nsfw: false, }; return api.client.editPost(form); } export async function deletePost( api: API, deleted: boolean, post: Post ): Promise { let form: DeletePost = { edit_id: post.id, deleted: deleted, auth: api.auth, }; return api.client.deletePost(form); } export async function removePost( api: API, removed: boolean, post: Post ): Promise { let form: RemovePost = { edit_id: post.id, removed, auth: api.auth, }; return api.client.removePost(form); } export async function stickyPost( api: API, stickied: boolean, post: Post ): Promise { let form: StickyPost = { edit_id: post.id, stickied, auth: api.auth, }; return api.client.stickyPost(form); } export async function lockPost( api: API, locked: boolean, post: Post ): Promise { let form: LockPost = { edit_id: post.id, locked, auth: api.auth, }; return api.client.lockPost(form); } export async function searchPost( api: API, post: Post ): Promise { let form: Search = { q: post.ap_id, type_: SearchType.Posts, sort: SortType.TopAll, }; return api.client.search(form); } export async function searchPostLocal( api: API, post: Post ): Promise { let form: Search = { q: post.name, type_: SearchType.Posts, sort: SortType.TopAll, }; return api.client.search(form); } export async function getPost( api: API, post_id: number ): Promise { let form: GetPost = { id: post_id, }; return api.client.getPost(form); } export async function searchComment( api: API, comment: Comment ): Promise { let form: Search = { q: comment.ap_id, type_: SearchType.Comments, sort: SortType.TopAll, }; return api.client.search(form); } export async function searchForBetaCommunity( api: API ): Promise { // Make sure lemmy-beta/c/main is cached on lemmy_alpha // Use short-hand search url let form: Search = { q: '!main@lemmy-beta:8551', type_: SearchType.Communities, sort: SortType.TopAll, }; return api.client.search(form); } export async function searchForCommunity( api: API, q: string ): Promise { // Use short-hand search url let form: Search = { q, type_: SearchType.Communities, sort: SortType.TopAll, }; return api.client.search(form); } export async function searchForUser( api: API, apShortname: string ): Promise { // Make sure lemmy-beta/c/main is cached on lemmy_alpha // Use short-hand search url let form: Search = { q: apShortname, type_: SearchType.Users, sort: SortType.TopAll, }; return api.client.search(form); } export async function banUserFromSite( api: API, user_id: number, ban: boolean ): Promise { // Make sure lemmy-beta/c/main is cached on lemmy_alpha // Use short-hand search url let form: BanUser = { user_id, ban, remove_data: false, auth: api.auth, }; return api.client.banUser(form); } export async function banUserFromCommunity( api: API, user_id: number, community_id: number, ban: boolean ): Promise { // Make sure lemmy-beta/c/main is cached on lemmy_alpha // Use short-hand search url let form: BanFromCommunity = { user_id, community_id, remove_data: false, ban, auth: api.auth, }; return api.client.banFromCommunity(form); } export async function followCommunity( api: API, follow: boolean, community_id: number ): Promise { let form: FollowCommunity = { community_id, follow, auth: api.auth, }; return api.client.followCommunity(form); } export async function checkFollowedCommunities( api: API ): Promise { let form: GetFollowedCommunities = { auth: api.auth, }; return api.client.getFollowedCommunities(form); } export async function likePost( api: API, score: number, post: Post ): Promise { let form: CreatePostLike = { post_id: post.id, score: score, auth: api.auth, }; return api.client.likePost(form); } export async function createComment( api: API, post_id: number, parent_id?: number, content = 'a jest test comment' ): Promise { let form: CreateComment = { content, post_id, parent_id, auth: api.auth, }; return api.client.createComment(form); } export async function editComment( api: API, edit_id: number, content = 'A jest test federated comment update' ): Promise { let form: EditComment = { content, edit_id, auth: api.auth, }; return api.client.editComment(form); } export async function deleteComment( api: API, deleted: boolean, edit_id: number ): Promise { let form: DeleteComment = { edit_id, deleted, auth: api.auth, }; return api.client.deleteComment(form); } export async function removeComment( api: API, removed: boolean, edit_id: number ): Promise { let form: RemoveComment = { edit_id, removed, auth: api.auth, }; return api.client.removeComment(form); } export async function getMentions(api: API): Promise { let form: GetUserMentions = { sort: SortType.New, unread_only: false, auth: api.auth, }; return api.client.getUserMentions(form); } export async function likeComment( api: API, score: number, comment: Comment ): Promise { let form: CreateCommentLike = { comment_id: comment.id, score, auth: api.auth, }; return api.client.likeComment(form); } export async function createCommunity( api: API, name_: string = randomString(5) ): Promise { let description = 'a sample description'; let icon = 'https://image.flaticon.com/icons/png/512/35/35896.png'; let banner = 'https://image.flaticon.com/icons/png/512/35/35896.png'; let form: CreateCommunity = { name: name_, title: name_, description, icon, banner, category_id: 1, nsfw: false, auth: api.auth, }; return api.client.createCommunity(form); } export async function getCommunity( api: API, id: number ): Promise { let form: GetCommunity = { id, }; return api.client.getCommunity(form); } export async function deleteCommunity( api: API, deleted: boolean, edit_id: number ): Promise { let form: DeleteCommunity = { edit_id, deleted, auth: api.auth, }; return api.client.deleteCommunity(form); } export async function removeCommunity( api: API, removed: boolean, edit_id: number ): Promise { let form: RemoveCommunity = { edit_id, removed, auth: api.auth, }; return api.client.removeCommunity(form); } export async function createPrivateMessage( api: API, recipient_id: number ): Promise { let content = 'A jest test federated private message'; let form: CreatePrivateMessage = { content, recipient_id, auth: api.auth, }; return api.client.createPrivateMessage(form); } export async function editPrivateMessage( api: API, edit_id: number ): Promise { let updatedContent = 'A jest test federated private message edited'; let form: EditPrivateMessage = { content: updatedContent, edit_id, auth: api.auth, }; return api.client.editPrivateMessage(form); } export async function deletePrivateMessage( api: API, deleted: boolean, edit_id: number ): Promise { let form: DeletePrivateMessage = { deleted, edit_id, auth: api.auth, }; return api.client.deletePrivateMessage(form); } export async function registerUser( api: API, username: string = randomString(5) ): Promise { let form: Register = { username, password: 'test', password_verify: 'test', admin: false, show_nsfw: true, }; return api.client.register(form); } export async function saveUserSettingsBio( api: API, auth: string ): Promise { let form: SaveUserSettings = { show_nsfw: true, theme: 'darkly', default_sort_type: SortType.Active, default_listing_type: ListingType.All, lang: 'en', show_avatars: true, send_notifications_to_email: false, bio: 'a changed bio', auth, }; return saveUserSettings(api, form); } export async function saveUserSettings( api: API, form: SaveUserSettings ): Promise { return api.client.saveUserSettings(form); } export async function getSite( api: API, auth: string ): Promise { let form: GetSite = { auth, }; return api.client.getSite(form); } export async function listPrivateMessages( api: API ): Promise { let form: GetPrivateMessages = { auth: api.auth, unread_only: false, limit: 999, }; return api.client.getPrivateMessages(form); } export async function unfollowRemotes( api: API ): Promise { // Unfollow all remote communities let followed = await checkFollowedCommunities(api); let remoteFollowed = followed.communities.filter( c => c.community.local == false ); for (let cu of remoteFollowed) { await followCommunity(api, false, cu.community.id); } let followed2 = await checkFollowedCommunities(api); return followed2; } export async function followBeta(api: API): Promise { // Cache it let search = await searchForBetaCommunity(api); let com = search.communities.find(c => c.community.local == false); if (com) { let follow = await followCommunity(api, true, com.community.id); return follow; } } export function delay(millis: number = 500) { return new Promise(resolve => setTimeout(resolve, millis)); } export function longDelay() { return delay(10000); } export function wrapper(form: any): string { return JSON.stringify(form); } function randomString(length: number): string { var result = ''; var characters = 'abcdefghijklmnopqrstuvwxyz0123456789_'; var charactersLength = characters.length; for (var i = 0; i < length; i++) { result += characters.charAt(Math.floor(Math.random() * charactersLength)); } return result; }