ui/src/services/WebSocketService.ts

   1 import { wsUri } from '../env';
   2 import {
   3   LoginForm,
   4   RegisterForm,
   5   UserOperation,
   6   CommunityForm,
   7   DeleteCommunityForm,
   8   RemoveCommunityForm,
   9   PostForm,
  10   DeletePostForm,
  11   RemovePostForm,
  12   LockPostForm,
  13   StickyPostForm,
  14   SavePostForm,
  15   CommentForm,
  16   DeleteCommentForm,
  17   RemoveCommentForm,
  18   MarkCommentAsReadForm,
  19   SaveCommentForm,
  20   CommentLikeForm,
  21   GetPostForm,
  22   GetPostsForm,
  23   CreatePostLikeForm,
  24   GetCommunityForm,
  25   FollowCommunityForm,
  26   GetFollowedCommunitiesForm,
  27   GetUserDetailsForm,
  28   ListCommunitiesForm,
  29   GetModlogForm,
  30   BanFromCommunityForm,
  31   AddModToCommunityForm,
  32   TransferCommunityForm,
  33   AddAdminForm,
  34   TransferSiteForm,
  35   BanUserForm,
  36   SiteForm,
  37   UserView,
  38   GetRepliesForm,
  39   GetUserMentionsForm,
  40   MarkUserMentionAsReadForm,
  41   SearchForm,
  42   UserSettingsForm,
  43   DeleteAccountForm,
  44   PasswordResetForm,
  45   PasswordChangeForm,
  46   PrivateMessageForm,
  47   EditPrivateMessageForm,
  48   DeletePrivateMessageForm,
  49   MarkPrivateMessageAsReadForm,
  50   GetPrivateMessagesForm,
  51   GetCommentsForm,
  52   UserJoinForm,
  53   GetSiteConfig,
  54   GetSiteForm,
  55   SiteConfigForm,
  56   MessageType,
  57   WebSocketJsonResponse,
  58 } from '../interfaces';
  59 import { UserService } from './';
  60 import { i18n } from '../i18next';
  61 import { toast } from '../utils';
  62 import { Observable } from 'rxjs';
  63 import { share } from 'rxjs/operators';
  64 import ReconnectingWebSocket from 'reconnecting-websocket';
  65
  66 export class WebSocketService {
  67   private static _instance: WebSocketService;
  68   public ws: ReconnectingWebSocket;
  69   public subject: Observable<any>;
  70
  71   public admins: Array<UserView>;
  72   public banned: Array<UserView>;
  73
  74   private constructor() {
  75     this.ws = new ReconnectingWebSocket(wsUri);
  76     let firstConnect = true;
  77
  78     this.subject = Observable.create((obs: any) => {
  79       this.ws.onmessage = e => {
  80         obs.next(JSON.parse(e.data));
  81       };
  82       this.ws.onopen = () => {
  83         console.log(`Connected to ${wsUri}`);
  84
  85         if (!firstConnect) {
  86           let res: WebSocketJsonResponse = {
  87             reconnect: true,
  88           };
  89           obs.next(res);
  90         }
  91
  92         firstConnect = false;
  93       };
  94     }).pipe(share());
  95   }
  96
  97   public static get Instance() {
  98     return this._instance || (this._instance = new this());
  99   }
 100
 101   public userJoin() {
 102     let form: UserJoinForm = { auth: UserService.Instance.auth };
 103     this.ws.send(this.wsSendWrapper(UserOperation.UserJoin, form));
 104   }
 105
 106   public login(loginForm: LoginForm) {
 107     this.ws.send(this.wsSendWrapper(UserOperation.Login, loginForm));
 108   }
 109
 110   public register(registerForm: RegisterForm) {
 111     this.ws.send(this.wsSendWrapper(UserOperation.Register, registerForm));
 112   }
 113
 114   public getCaptcha() {
 115     this.ws.send(this.wsSendWrapper(UserOperation.GetCaptcha, {}));
 116   }
 117
 118   public createCommunity(form: CommunityForm) {
 119     this.setAuth(form);
 120     this.ws.send(this.wsSendWrapper(UserOperation.CreateCommunity, form));
 121   }
 122
 123   public editCommunity(form: CommunityForm) {
 124     this.setAuth(form);
 125     this.ws.send(this.wsSendWrapper(UserOperation.EditCommunity, form));
 126   }
 127
 128   public deleteCommunity(form: DeleteCommunityForm) {
 129     this.setAuth(form);
 130     this.ws.send(this.wsSendWrapper(UserOperation.DeleteCommunity, form));
 131   }
 132
 133   public removeCommunity(form: RemoveCommunityForm) {
 134     this.setAuth(form);
 135     this.ws.send(this.wsSendWrapper(UserOperation.RemoveCommunity, form));
 136   }
 137
 138   public followCommunity(followCommunityForm: FollowCommunityForm) {
 139     this.setAuth(followCommunityForm);
 140     this.ws.send(
 141       this.wsSendWrapper(UserOperation.FollowCommunity, followCommunityForm)
 142     );
 143   }
 144
 145   public listCommunities(form: ListCommunitiesForm) {
 146     this.setAuth(form, false);
 147     this.ws.send(this.wsSendWrapper(UserOperation.ListCommunities, form));
 148   }
 149
 150   public getFollowedCommunities() {
 151     let form: GetFollowedCommunitiesForm = { auth: UserService.Instance.auth };
 152     this.ws.send(
 153       this.wsSendWrapper(UserOperation.GetFollowedCommunities, form)
 154     );
 155   }
 156
 157   public listCategories() {
 158     this.ws.send(this.wsSendWrapper(UserOperation.ListCategories, {}));
 159   }
 160
 161   public createPost(form: PostForm) {
 162     this.setAuth(form);
 163     this.ws.send(this.wsSendWrapper(UserOperation.CreatePost, form));
 164   }
 165
 166   public getPost(form: GetPostForm) {
 167     this.setAuth(form, false);
 168     this.ws.send(this.wsSendWrapper(UserOperation.GetPost, form));
 169   }
 170
 171   public getCommunity(form: GetCommunityForm) {
 172     this.setAuth(form, false);
 173     this.ws.send(this.wsSendWrapper(UserOperation.GetCommunity, form));
 174   }
 175
 176   public createComment(form: CommentForm) {
 177     this.setAuth(form);
 178     this.ws.send(this.wsSendWrapper(UserOperation.CreateComment, form));
 179   }
 180
 181   public editComment(form: CommentForm) {
 182     this.setAuth(form);
 183     this.ws.send(this.wsSendWrapper(UserOperation.EditComment, form));
 184   }
 185
 186   public deleteComment(form: DeleteCommentForm) {
 187     this.setAuth(form);
 188     this.ws.send(this.wsSendWrapper(UserOperation.DeleteComment, form));
 189   }
 190
 191   public removeComment(form: RemoveCommentForm) {
 192     this.setAuth(form);
 193     this.ws.send(this.wsSendWrapper(UserOperation.RemoveComment, form));
 194   }
 195
 196   public markCommentAsRead(form: MarkCommentAsReadForm) {
 197     this.setAuth(form);
 198     this.ws.send(this.wsSendWrapper(UserOperation.MarkCommentAsRead, form));
 199   }
 200
 201   public likeComment(form: CommentLikeForm) {
 202     this.setAuth(form);
 203     this.ws.send(this.wsSendWrapper(UserOperation.CreateCommentLike, form));
 204   }
 205
 206   public saveComment(form: SaveCommentForm) {
 207     this.setAuth(form);
 208     this.ws.send(this.wsSendWrapper(UserOperation.SaveComment, form));
 209   }
 210
 211   public getPosts(form: GetPostsForm) {
 212     this.setAuth(form, false);
 213     this.ws.send(this.wsSendWrapper(UserOperation.GetPosts, form));
 214   }
 215
 216   public getComments(form: GetCommentsForm) {
 217     this.setAuth(form, false);
 218     this.ws.send(this.wsSendWrapper(UserOperation.GetComments, form));
 219   }
 220
 221   public likePost(form: CreatePostLikeForm) {
 222     this.setAuth(form);
 223     this.ws.send(this.wsSendWrapper(UserOperation.CreatePostLike, form));
 224   }
 225
 226   public editPost(form: PostForm) {
 227     this.setAuth(form);
 228     this.ws.send(this.wsSendWrapper(UserOperation.EditPost, form));
 229   }
 230
 231   public deletePost(form: DeletePostForm) {
 232     this.setAuth(form);
 233     this.ws.send(this.wsSendWrapper(UserOperation.DeletePost, form));
 234   }
 235
 236   public removePost(form: RemovePostForm) {
 237     this.setAuth(form);
 238     this.ws.send(this.wsSendWrapper(UserOperation.RemovePost, form));
 239   }
 240
 241   public lockPost(form: LockPostForm) {
 242     this.setAuth(form);
 243     this.ws.send(this.wsSendWrapper(UserOperation.LockPost, form));
 244   }
 245
 246   public stickyPost(form: StickyPostForm) {
 247     this.setAuth(form);
 248     this.ws.send(this.wsSendWrapper(UserOperation.StickyPost, form));
 249   }
 250
 251   public savePost(form: SavePostForm) {
 252     this.setAuth(form);
 253     this.ws.send(this.wsSendWrapper(UserOperation.SavePost, form));
 254   }
 255
 256   public banFromCommunity(form: BanFromCommunityForm) {
 257     this.setAuth(form);
 258     this.ws.send(this.wsSendWrapper(UserOperation.BanFromCommunity, form));
 259   }
 260
 261   public addModToCommunity(form: AddModToCommunityForm) {
 262     this.setAuth(form);
 263     this.ws.send(this.wsSendWrapper(UserOperation.AddModToCommunity, form));
 264   }
 265
 266   public transferCommunity(form: TransferCommunityForm) {
 267     this.setAuth(form);
 268     this.ws.send(this.wsSendWrapper(UserOperation.TransferCommunity, form));
 269   }
 270
 271   public transferSite(form: TransferSiteForm) {
 272     this.setAuth(form);
 273     this.ws.send(this.wsSendWrapper(UserOperation.TransferSite, form));
 274   }
 275
 276   public banUser(form: BanUserForm) {
 277     this.setAuth(form);
 278     this.ws.send(this.wsSendWrapper(UserOperation.BanUser, form));
 279   }
 280
 281   public addAdmin(form: AddAdminForm) {
 282     this.setAuth(form);
 283     this.ws.send(this.wsSendWrapper(UserOperation.AddAdmin, form));
 284   }
 285
 286   public getUserDetails(form: GetUserDetailsForm) {
 287     this.setAuth(form, false);
 288     this.ws.send(this.wsSendWrapper(UserOperation.GetUserDetails, form));
 289   }
 290
 291   public getReplies(form: GetRepliesForm) {
 292     this.setAuth(form);
 293     this.ws.send(this.wsSendWrapper(UserOperation.GetReplies, form));
 294   }
 295
 296   public getUserMentions(form: GetUserMentionsForm) {
 297     this.setAuth(form);
 298     this.ws.send(this.wsSendWrapper(UserOperation.GetUserMentions, form));
 299   }
 300
 301   public markUserMentionAsRead(form: MarkUserMentionAsReadForm) {
 302     this.setAuth(form);
 303     this.ws.send(this.wsSendWrapper(UserOperation.MarkUserMentionAsRead, form));
 304   }
 305
 306   public getModlog(form: GetModlogForm) {
 307     this.ws.send(this.wsSendWrapper(UserOperation.GetModlog, form));
 308   }
 309
 310   public createSite(siteForm: SiteForm) {
 311     this.setAuth(siteForm);
 312     this.ws.send(this.wsSendWrapper(UserOperation.CreateSite, siteForm));
 313   }
 314
 315   public editSite(siteForm: SiteForm) {
 316     this.setAuth(siteForm);
 317     this.ws.send(this.wsSendWrapper(UserOperation.EditSite, siteForm));
 318   }
 319
 320   public getSite(form: GetSiteForm = {}) {
 321     this.setAuth(form, false);
 322     this.ws.send(this.wsSendWrapper(UserOperation.GetSite, form));
 323   }
 324
 325   public getSiteConfig() {
 326     let siteConfig: GetSiteConfig = {};
 327     this.setAuth(siteConfig);
 328     this.ws.send(this.wsSendWrapper(UserOperation.GetSiteConfig, siteConfig));
 329   }
 330
 331   public search(form: SearchForm) {
 332     this.setAuth(form, false);
 333     this.ws.send(this.wsSendWrapper(UserOperation.Search, form));
 334   }
 335
 336   public markAllAsRead() {
 337     let form = {};
 338     this.setAuth(form);
 339     this.ws.send(this.wsSendWrapper(UserOperation.MarkAllAsRead, form));
 340   }
 341
 342   public saveUserSettings(userSettingsForm: UserSettingsForm) {
 343     this.setAuth(userSettingsForm);
 344     this.ws.send(
 345       this.wsSendWrapper(UserOperation.SaveUserSettings, userSettingsForm)
 346     );
 347   }
 348
 349   public deleteAccount(form: DeleteAccountForm) {
 350     this.setAuth(form);
 351     this.ws.send(this.wsSendWrapper(UserOperation.DeleteAccount, form));
 352   }
 353
 354   public passwordReset(form: PasswordResetForm) {
 355     this.ws.send(this.wsSendWrapper(UserOperation.PasswordReset, form));
 356   }
 357
 358   public passwordChange(form: PasswordChangeForm) {
 359     this.ws.send(this.wsSendWrapper(UserOperation.PasswordChange, form));
 360   }
 361
 362   public createPrivateMessage(form: PrivateMessageForm) {
 363     this.setAuth(form);
 364     this.ws.send(this.wsSendWrapper(UserOperation.CreatePrivateMessage, form));
 365   }
 366
 367   public editPrivateMessage(form: EditPrivateMessageForm) {
 368     this.setAuth(form);
 369     this.ws.send(this.wsSendWrapper(UserOperation.EditPrivateMessage, form));
 370   }
 371
 372   public deletePrivateMessage(form: DeletePrivateMessageForm) {
 373     this.setAuth(form);
 374     this.ws.send(this.wsSendWrapper(UserOperation.DeletePrivateMessage, form));
 375   }
 376
 377   public markPrivateMessageAsRead(form: MarkPrivateMessageAsReadForm) {
 378     this.setAuth(form);
 379     this.ws.send(
 380       this.wsSendWrapper(UserOperation.MarkPrivateMessageAsRead, form)
 381     );
 382   }
 383
 384   public getPrivateMessages(form: GetPrivateMessagesForm) {
 385     this.setAuth(form);
 386     this.ws.send(this.wsSendWrapper(UserOperation.GetPrivateMessages, form));
 387   }
 388
 389   public saveSiteConfig(form: SiteConfigForm) {
 390     this.setAuth(form);
 391     this.ws.send(this.wsSendWrapper(UserOperation.SaveSiteConfig, form));
 392   }
 393
 394   private wsSendWrapper(op: UserOperation, data: MessageType) {
 395     let send = { op: UserOperation[op], data: data };
 396     console.log(send);
 397     return JSON.stringify(send);
 398   }
 399
 400   private setAuth(obj: any, throwErr: boolean = true) {
 401     obj.auth = UserService.Instance.auth;
 402     if (obj.auth == null && throwErr) {
 403       toast(i18n.t('not_logged_in'), 'danger');
 404       throw 'Not logged in';
 405     }
 406   }
 407 }
 408
 409 window.onbeforeunload = () => {
 410   WebSocketService.Instance.ws.close();
 411 };