hexbear-frontend/src/services/WebSocketService.ts

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();
};