import { Component, linkEvent } from 'inferno'; import { Link } from 'inferno-router'; import { Subscription } from 'rxjs'; import { retryWhen, delay, take } from 'rxjs/operators'; import { WebSocketService, UserService } from '../services'; import { UserOperation, GetRepliesForm, GetRepliesResponse, GetUserMentionsForm, GetUserMentionsResponse, GetPrivateMessagesForm, PrivateMessagesResponse, SortType, GetSiteResponse, Comment, CommentResponse, PrivateMessage, PrivateMessageResponse, WebSocketJsonResponse, } from '../interfaces'; import { wsJsonToRes, pictshareAvatarThumbnail, showAvatars, fetchLimit, isCommentType, toast, } from '../utils'; import { version } from '../version'; import { i18n } from '../i18next'; interface NavbarState { isLoggedIn: boolean; expanded: boolean; replies: Array; mentions: Array; messages: Array; unreadCount: number; siteName: string; } export class Navbar extends Component { private wsSub: Subscription; private userSub: Subscription; emptyState: NavbarState = { isLoggedIn: UserService.Instance.user !== undefined, unreadCount: 0, replies: [], mentions: [], messages: [], expanded: false, siteName: undefined, }; constructor(props: any, context: any) { super(props, context); this.state = this.emptyState; // Subscribe to user changes this.userSub = UserService.Instance.sub.subscribe(user => { this.state.isLoggedIn = user.user !== undefined; if (this.state.isLoggedIn) { this.state.unreadCount = user.user.unreadCount; this.requestNotificationPermission(); } this.setState(this.state); }); this.wsSub = WebSocketService.Instance.subject .pipe(retryWhen(errors => errors.pipe(delay(3000), take(10)))) .subscribe( msg => this.parseMessage(msg), err => console.error(err), () => console.log('complete') ); if (this.state.isLoggedIn) { this.requestNotificationPermission(); // TODO couldn't get re-logging in to re-fetch unreads this.fetchUnreads(); } WebSocketService.Instance.getSite(); } render() { return this.navbar(); } componentWillUnmount() { this.wsSub.unsubscribe(); this.userSub.unsubscribe(); } // TODO class active corresponding to current page navbar() { return ( ); } expandNavbar(i: Navbar) { i.state.expanded = !i.state.expanded; i.setState(i.state); } parseMessage(msg: WebSocketJsonResponse) { let res = wsJsonToRes(msg); if (msg.error) { if (msg.error == 'not_logged_in') { UserService.Instance.logout(); location.reload(); } return; } else if (msg.reconnect) { this.fetchUnreads(); } else if (res.op == UserOperation.GetReplies) { let data = res.data as GetRepliesResponse; let unreadReplies = data.replies.filter(r => !r.read); this.state.replies = unreadReplies; this.state.unreadCount = this.calculateUnreadCount(); this.setState(this.state); this.sendUnreadCount(); } else if (res.op == UserOperation.GetUserMentions) { let data = res.data as GetUserMentionsResponse; let unreadMentions = data.mentions.filter(r => !r.read); this.state.mentions = unreadMentions; this.state.unreadCount = this.calculateUnreadCount(); this.setState(this.state); this.sendUnreadCount(); } else if (res.op == UserOperation.GetPrivateMessages) { let data = res.data as PrivateMessagesResponse; let unreadMessages = data.messages.filter(r => !r.read); this.state.messages = unreadMessages; this.state.unreadCount = this.calculateUnreadCount(); this.setState(this.state); this.sendUnreadCount(); } else if (res.op == UserOperation.CreateComment) { let data = res.data as CommentResponse; if (this.state.isLoggedIn) { if (data.recipient_ids.includes(UserService.Instance.user.id)) { this.state.replies.push(data.comment); this.state.unreadCount++; this.setState(this.state); this.sendUnreadCount(); this.notify(data.comment); } } } else if (res.op == UserOperation.CreatePrivateMessage) { let data = res.data as PrivateMessageResponse; if (this.state.isLoggedIn) { if (data.message.recipient_id == UserService.Instance.user.id) { this.state.messages.push(data.message); this.state.unreadCount++; this.setState(this.state); this.sendUnreadCount(); this.notify(data.message); } } } else if (res.op == UserOperation.GetSite) { let data = res.data as GetSiteResponse; if (data.site && !this.state.siteName) { this.state.siteName = data.site.name; WebSocketService.Instance.site = data.site; this.setState(this.state); } } } fetchUnreads() { if (this.state.isLoggedIn) { let repliesForm: GetRepliesForm = { sort: SortType[SortType.New], unread_only: true, page: 1, limit: fetchLimit, }; let userMentionsForm: GetUserMentionsForm = { sort: SortType[SortType.New], unread_only: true, page: 1, limit: fetchLimit, }; let privateMessagesForm: GetPrivateMessagesForm = { unread_only: true, page: 1, limit: fetchLimit, }; if (this.currentLocation !== '/inbox') { WebSocketService.Instance.getReplies(repliesForm); WebSocketService.Instance.getUserMentions(userMentionsForm); WebSocketService.Instance.getPrivateMessages(privateMessagesForm); } } } get currentLocation() { return this.context.router.history.location.pathname; } sendUnreadCount() { UserService.Instance.user.unreadCount = this.state.unreadCount; UserService.Instance.sub.next({ user: UserService.Instance.user, }); } calculateUnreadCount(): number { return ( this.state.replies.filter(r => !r.read).length + this.state.mentions.filter(r => !r.read).length + this.state.messages.filter(r => !r.read).length ); } requestNotificationPermission() { if (UserService.Instance.user) { document.addEventListener('DOMContentLoaded', function() { if (!Notification) { toast(i18n.t('notifications_error'), 'danger'); return; } if (Notification.permission !== 'granted') Notification.requestPermission(); }); } } notify(reply: Comment | PrivateMessage) { if (Notification.permission !== 'granted') Notification.requestPermission(); else { var notification = new Notification(reply.creator_name, { icon: reply.creator_avatar ? reply.creator_avatar : `${window.location.protocol}//${window.location.host}/static/assets/apple-touch-icon.png`, body: `${reply.content}`, }); notification.onclick = () => { this.context.router.history.push( isCommentType(reply) ? `/post/${reply.post_id}/comment/${reply.id}` : `/inbox` ); }; } } }