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