ui/src/services/WebSocketService.ts

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