import { Component, linkEvent, createRef, RefObject } 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, pictrsAvatarThumbnail, showAvatars, fetchLimit, isCommentType, toast, messageToastify, md, setTheme, getLanguage, } from '../utils'; import { i18n } from '../i18next'; interface NavbarState { isLoggedIn: boolean; expanded: boolean; replies: Array; mentions: Array; messages: Array; unreadCount: number; searchParam: string; toggleSearch: boolean; siteLoading: boolean; siteRes: GetSiteResponse; onSiteBanner?(url: string): any; } export class Navbar extends Component { private wsSub: Subscription; private userSub: Subscription; private unreadCountSub: Subscription; private searchTextField: RefObject; emptyState: NavbarState = { isLoggedIn: false, unreadCount: 0, replies: [], mentions: [], messages: [], expanded: false, siteRes: { site: { id: null, name: null, creator_id: null, creator_name: null, published: null, number_of_users: null, number_of_posts: null, number_of_comments: null, number_of_communities: null, enable_downvotes: null, open_registration: null, enable_nsfw: null, icon: null, banner: null, creator_preferred_username: null, }, my_user: null, admins: [], banned: [], online: null, version: null, federated_instances: null, }, searchParam: '', toggleSearch: false, siteLoading: true, }; constructor(props: any, context: any) { super(props, context); this.state = this.emptyState; 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') ); WebSocketService.Instance.getSite(); this.searchTextField = createRef(); } componentDidMount() { // Subscribe to jwt changes this.userSub = UserService.Instance.jwtSub.subscribe(res => { // A login if (res !== undefined) { this.requestNotificationPermission(); } else { this.state.isLoggedIn = false; } WebSocketService.Instance.getSite(); this.setState(this.state); }); // Subscribe to unread count changes this.unreadCountSub = UserService.Instance.unreadCountSub.subscribe(res => { this.setState({ unreadCount: res }); }); } handleSearchParam(i: Navbar, event: any) { i.state.searchParam = event.target.value; i.setState(i.state); } updateUrl() { const searchParam = this.state.searchParam; this.setState({ searchParam: '' }); this.setState({ toggleSearch: false }); if (searchParam === '') { this.context.router.history.push(`/search/`); } else { this.context.router.history.push( `/search/q/${searchParam}/type/all/sort/topall/page/1` ); } } handleSearchSubmit(i: Navbar, event: any) { event.preventDefault(); i.updateUrl(); } handleSearchBtn(i: Navbar, event: any) { event.preventDefault(); i.setState({ toggleSearch: true }); i.searchTextField.current.focus(); const offsetWidth = i.searchTextField.current.offsetWidth; if (i.state.searchParam && offsetWidth > 100) { i.updateUrl(); } } handleSearchBlur(i: Navbar, event: any) { if (!(event.relatedTarget && event.relatedTarget.name !== 'search-btn')) { i.state.toggleSearch = false; i.setState(i.state); } } render() { return this.navbar(); } componentWillUnmount() { this.wsSub.unsubscribe(); this.userSub.unsubscribe(); this.unreadCountSub.unsubscribe(); } // TODO class active corresponding to current page navbar() { let user = UserService.Instance.user; 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; this.state.siteRes = data; // The login if (data.my_user) { UserService.Instance.user = data.my_user; WebSocketService.Instance.userJoin(); // On the first load, check the unreads if (this.state.isLoggedIn == false) { this.requestNotificationPermission(); this.fetchUnreads(); setTheme(data.my_user.theme, true); i18n.changeLanguage(getLanguage()); } this.state.isLoggedIn = true; } this.state.siteLoading = false; this.setState(this.state); } } fetchUnreads() { console.log('Fetching unreads...'); 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.unreadCountSub.next(this.state.unreadCount); } 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 ); } get canAdmin(): boolean { return ( UserService.Instance.user && this.state.siteRes.admins .map(a => a.id) .includes(UserService.Instance.user.id) ); } 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) { let creator_name = reply.creator_name; let creator_avatar = reply.creator_avatar ? reply.creator_avatar : `${window.location.protocol}//${window.location.host}/static/assets/apple-touch-icon.png`; let link = isCommentType(reply) ? `/post/${reply.post_id}/comment/${reply.id}` : `/inbox`; let htmlBody = md.render(reply.content); let body = reply.content; // Unfortunately the notifications API can't do html messageToastify( creator_name, creator_avatar, htmlBody, link, this.context.router ); if (Notification.permission !== 'granted') Notification.requestPermission(); else { var notification = new Notification(reply.creator_name, { icon: creator_avatar, body: body, }); notification.onclick = () => { event.preventDefault(); this.context.router.history.push(link); }; } } }