You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

505 lines
14 KiB

import { wsUri } from '../env';
import {
LogoutForm,
RegisterForm,
UserOperation,
CommunityForm,
PostForm,
SavePostForm,
CommentForm,
SaveCommentForm,
CommentLikeForm,
GetPostsForm,
CreatePostLikeForm,
GetCommunityForm,
FollowCommunityForm,
GetFollowedCommunitiesForm,
GetModlogForm,
BanFromCommunityForm,
AddModToCommunityForm,
TransferCommunityForm,
AddAdminForm,
AddSitemodForm,
TransferSiteForm,
BanUserForm,
SiteForm,
UserView,
GetRepliesForm,
GetUserMentionsForm,
SearchForm,
UserSettingsForm,
DeleteAccountForm,
PasswordResetForm,
PasswordChangeForm,
PrivateMessageForm,
EditPrivateMessageForm,
GetPrivateMessagesForm,
GetCommentsForm,
UserJoinForm,
GetSiteConfig,
GetSiteForm,
SiteConfigForm,
MessageType,
WebSocketJsonResponse,
ListCommentReportsForm,
GetReportCountForm,
ResolveCommentReportForm,
ResolvePostReportForm,
CreateCommentReportForm,
CreatePostReportForm,
ListPostReportsForm,
GetCommunitySettingsForm,
EditCommunitySettingsForm,
GetUserTagForm,
SetUserTagForm,
MarkPrivateMessageReadForm,
DeletePrivateMessageForm,
DeleteCommentForm,
RemoveCommentForm,
DeletePostForm,
RemovePostForm,
StickyPostForm,
LockPostForm,
MarkCommentReadForm,
RemoveUserContentForm,
FeaturePostForm,
MarkUserMentionAsReadForm,
PostJoinForm,
} from '../interfaces';
import { UserService } from './';
import { i18n } from '../i18next';
import { toast } from '../utils';
import { Observable } from 'rxjs';
import { share } from 'rxjs/operators';
import ReconnectingWebSocket from 'reconnecting-websocket';
export class WebSocketService {
private static _instance: WebSocketService;
public ws: ReconnectingWebSocket;
public subject: Observable<any>;
public admins: Array<UserView>;
public sitemods: Array<UserView>;
public banned: Array<UserView>;
private constructor() {
this.ws = new ReconnectingWebSocket(wsUri);
let firstConnect = true;
this.subject = Observable.create((obs: any) => {
this.ws.onmessage = e => {
try {
obs.next(JSON.parse(e.data));
} catch (error) {
console.warn('Failed to parse websocket message');
console.warn(e.data);
throw new Error(error);
}
};
this.ws.onopen = () => {
console.log(`Connected to ${wsUri}`);
if (UserService.Instance.user) {
this.userJoin();
}
if (!firstConnect) {
let res: WebSocketJsonResponse = {
reconnect: true,
};
obs.next(res);
}
firstConnect = false;
};
}).pipe(share());
}
public static get Instance() {
return this._instance || (this._instance = new this());
}
public userJoin() {
let form: UserJoinForm = { auth: UserService.Instance.auth };
this.ws.send(this.wsSendWrapper(UserOperation.UserJoin, form));
}
public postJoin(id: number) {
let form: PostJoinForm = {
post_id: id,
auth: UserService.Instance.auth,
};
this.ws.send(this.wsSendWrapper(UserOperation.PostJoin, form));
}
public leaveAll() {
this.ws.send(this.wsSendWrapper(UserOperation.LeaveRooms, {}));
}
public logout(loginForm: LogoutForm) {
this.ws.send(this.wsSendWrapper(UserOperation.Logout, loginForm));
}
// TODO setup uses this, remove it
public register(registerForm: RegisterForm) {
this.ws.send(this.wsSendWrapper(UserOperation.Register, registerForm));
}
public getCaptcha() {
this.ws.send(this.wsSendWrapper(UserOperation.GetCaptcha, {}));
}
public createCommunity(communityForm: CommunityForm) {
this.setAuth(communityForm);
this.ws.send(
this.wsSendWrapper(UserOperation.CreateCommunity, communityForm)
);
}
public editCommunity(communityForm: CommunityForm) {
this.setAuth(communityForm);
this.ws.send(
this.wsSendWrapper(UserOperation.EditCommunity, communityForm)
);
}
public followCommunity(followCommunityForm: FollowCommunityForm) {
this.setAuth(followCommunityForm);
this.ws.send(
this.wsSendWrapper(UserOperation.FollowCommunity, followCommunityForm)
);
}
public getFollowedCommunities() {
let form: GetFollowedCommunitiesForm = { auth: UserService.Instance.auth };
this.ws.send(
this.wsSendWrapper(UserOperation.GetFollowedCommunities, form)
);
}
public listCategories() {
this.ws.send(this.wsSendWrapper(UserOperation.ListCategories, {}));
}
public createPost(postForm: PostForm) {
this.setAuth(postForm);
this.ws.send(this.wsSendWrapper(UserOperation.CreatePost, postForm));
}
public getCommunity(form: GetCommunityForm) {
this.setAuth(form, false);
this.ws.send(this.wsSendWrapper(UserOperation.GetCommunity, form));
}
public createComment(commentForm: CommentForm) {
this.setAuth(commentForm);
this.ws.send(this.wsSendWrapper(UserOperation.CreateComment, commentForm));
}
public editComment(commentForm: CommentForm) {
this.setAuth(commentForm);
this.ws.send(this.wsSendWrapper(UserOperation.EditComment, commentForm));
}
public deleteComment(commentForm: DeleteCommentForm) {
this.setAuth(commentForm);
this.ws.send(this.wsSendWrapper(UserOperation.DeleteComment, commentForm));
}
public removeComment(commentForm: RemoveCommentForm) {
this.setAuth(commentForm);
this.ws.send(this.wsSendWrapper(UserOperation.RemoveComment, commentForm));
}
public markCommentAsRead(commentForm: MarkCommentReadForm) {
this.setAuth(commentForm);
this.ws.send(
this.wsSendWrapper(UserOperation.MarkCommentAsRead, commentForm)
);
}
public likeComment(form: CommentLikeForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.CreateCommentLike, form));
}
public saveComment(form: SaveCommentForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.SaveComment, form));
}
public getPosts(form: GetPostsForm) {
this.setAuth(form, false);
this.ws.send(this.wsSendWrapper(UserOperation.GetPosts, form));
}
public getComments(form: GetCommentsForm) {
this.setAuth(form, false);
this.ws.send(this.wsSendWrapper(UserOperation.GetComments, form));
}
public likePost(form: CreatePostLikeForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.CreatePostLike, form));
}
public editPost(postForm: PostForm) {
this.setAuth(postForm);
this.ws.send(this.wsSendWrapper(UserOperation.EditPost, postForm));
}
public deletePost(postForm: DeletePostForm) {
this.setAuth(postForm);
this.ws.send(this.wsSendWrapper(UserOperation.DeletePost, postForm));
}
public removePost(postForm: RemovePostForm) {
this.setAuth(postForm);
this.ws.send(this.wsSendWrapper(UserOperation.RemovePost, postForm));
}
public lockPost(postForm: LockPostForm) {
this.setAuth(postForm);
this.ws.send(this.wsSendWrapper(UserOperation.LockPost, postForm));
}
public stickyPost(postForm: StickyPostForm) {
this.setAuth(postForm);
this.ws.send(this.wsSendWrapper(UserOperation.StickyPost, postForm));
}
public featurePost(postForm: FeaturePostForm) {
this.setAuth(postForm);
this.ws.send(this.wsSendWrapper(UserOperation.FeaturePost, postForm));
}
public savePost(form: SavePostForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.SavePost, form));
}
public banFromCommunity(form: BanFromCommunityForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.BanFromCommunity, form));
}
public removeUserContent(form: RemoveUserContentForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.RemoveUserContent, form));
}
public addModToCommunity(form: AddModToCommunityForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.AddModToCommunity, form));
}
public transferCommunity(form: TransferCommunityForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.TransferCommunity, form));
}
public transferSite(form: TransferSiteForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.TransferSite, form));
}
public banUser(form: BanUserForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.BanUser, form));
}
public addAdmin(form: AddAdminForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.AddAdmin, form));
}
public addSitemod(form: AddSitemodForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.AddSitemod, form));
}
public getReplies(form: GetRepliesForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.GetReplies, form));
}
public getUserMentions(form: GetUserMentionsForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.GetUserMentions, form));
}
public markUserMentionAsRead(form: MarkUserMentionAsReadForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.MarkUserMentionAsRead, form));
}
public getModlog(form: GetModlogForm) {
this.setAuth(form, false);
this.ws.send(this.wsSendWrapper(UserOperation.GetModlog, form));
}
public createSite(siteForm: SiteForm) {
this.setAuth(siteForm);
this.ws.send(this.wsSendWrapper(UserOperation.CreateSite, siteForm));
}
public editSite(siteForm: SiteForm) {
this.setAuth(siteForm);
this.ws.send(this.wsSendWrapper(UserOperation.EditSite, siteForm));
}
public getSite(form: GetSiteForm = {}) {
this.setAuth(form, false);
this.ws.send(this.wsSendWrapper(UserOperation.GetSite, form));
}
public getSiteConfig() {
let siteConfig: GetSiteConfig = {};
this.setAuth(siteConfig);
this.ws.send(this.wsSendWrapper(UserOperation.GetSiteConfig, siteConfig));
}
public search(form: SearchForm) {
this.setAuth(form, false);
this.ws.send(this.wsSendWrapper(UserOperation.Search, form));
}
public markAllAsRead() {
let form = {};
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.MarkAllAsRead, form));
}
public saveUserSettings(userSettingsForm: UserSettingsForm) {
this.setAuth(userSettingsForm);
this.ws.send(
this.wsSendWrapper(UserOperation.SaveUserSettings, userSettingsForm)
);
}
public deleteAccount(form: DeleteAccountForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.DeleteAccount, form));
}
public passwordReset(form: PasswordResetForm) {
this.ws.send(this.wsSendWrapper(UserOperation.PasswordReset, form));
}
public passwordChange(form: PasswordChangeForm) {
this.ws.send(this.wsSendWrapper(UserOperation.PasswordChange, form));
}
public createPrivateMessage(form: PrivateMessageForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.CreatePrivateMessage, form));
}
public editPrivateMessage(form: EditPrivateMessageForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.EditPrivateMessage, form));
}
public markPrivateMessageRead(form: MarkPrivateMessageReadForm) {
this.setAuth(form);
this.ws.send(
this.wsSendWrapper(UserOperation.MarkPrivateMessageAsRead, form)
);
}
public deletePrivateMessage(form: DeletePrivateMessageForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.DeletePrivateMessage, form));
}
public getPrivateMessages(form: GetPrivateMessagesForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.GetPrivateMessages, form));
}
public saveSiteConfig(form: SiteConfigForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.SaveSiteConfig, form));
}
public listCommentReports(form: ListCommentReportsForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.ListCommentReports, form));
}
public listPostReports(form: ListPostReportsForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.ListPostReports, form));
}
public getReportCount(form: GetReportCountForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.GetReportCount, form));
}
public resolveCommentReport(form: ResolveCommentReportForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.ResolveCommentReport, form));
}
public resolvePostReport(form: ResolvePostReportForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.ResolvePostReport, form));
}
public createCommentReport(form: CreateCommentReportForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.CreateCommentReport, form));
}
public createPostReport(form: CreatePostReportForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.CreatePostReport, form));
}
public getCommunitySettings(form: GetCommunitySettingsForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.GetCommunitySettings, form));
}
public editCommunitySettings(form: EditCommunitySettingsForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.EditCommunitySettings, form));
}
public getSiteModerators() {
this.ws.send(
this.wsSendWrapper(UserOperation.GetSiteModerators, {
// figure this out later !
limit: 1000,
})
);
}
public getUserTags(form: GetUserTagForm) {
this.ws.send(this.wsSendWrapper(UserOperation.GetUserTag, form));
}
public setUserTags(form: SetUserTagForm) {
this.setAuth(form);
this.ws.send(this.wsSendWrapper(UserOperation.SetUserTag, form));
}
private wsSendWrapper(op: UserOperation, data: MessageType) {
let send = { op: UserOperation[op], data: data };
console.log(send);
return JSON.stringify(send);
}
private setAuth(obj: any, throwErr = true) {
obj.auth = UserService.Instance.auth;
if (obj.auth == null && throwErr) {
toast(i18n.t('not_logged_in'), 'danger');
throw 'Not logged in';
}
}
}
window.onbeforeunload = () => {
WebSocketService.Instance.ws.close();
};