lemmy/ui/src/components/navbar.tsx

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, SortType, GetSiteResponse, Comment} from '../interfaces';
import { msgOp } from '../utils';
import { version } from '../version';
import { i18n } from '../i18next';
import { T } from 'inferno-i18next';

interface NavbarState {
  isLoggedIn: boolean;
  expanded: boolean;
  expandUserDropdown: boolean;
  replies: Array<Comment>,
  fetchCount: number,
  unreadCount: number;
  siteName: string;
}

export class Navbar extends Component<any, NavbarState> {
  private wsSub: Subscription;
  private userSub: Subscription;
  emptyState: NavbarState = {
    isLoggedIn: (UserService.Instance.user !== undefined),
    unreadCount: 0,
    fetchCount: 0,
    replies: [],
    expanded: false,
    expandUserDropdown: false,
    siteName: undefined
  }

  constructor(props: any, context: any) {
    super(props, context);
    this.state = this.emptyState;
    this.handleOverviewClick = this.handleOverviewClick.bind(this);

    this.keepFetchingReplies();

    // Subscribe to user changes
    this.userSub = UserService.Instance.sub.subscribe(user => {
      this.state.isLoggedIn = user.user !== undefined;
      this.state.unreadCount = 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();
    }

    WebSocketService.Instance.getSite();
  }

  render() {
    return (
      <div>{this.navbar()}</div>
    )
  }

  componentWillUnmount() {
    this.wsSub.unsubscribe();
    this.userSub.unsubscribe();
  }

  // TODO class active corresponding to current page
  navbar() {
    return (
      <nav class="container navbar navbar-expand-md navbar-light navbar-bg p-0 px-3">
        <Link title={version} class="navbar-brand" to="/">
          <svg class="icon mr-2 mouse-icon"><use xlinkHref="#icon-mouse"></use></svg>
          {this.state.siteName}
        </Link>
        <button class="navbar-toggler" type="button" onClick={linkEvent(this, this.expandNavbar)}>
          <span class="navbar-toggler-icon"></span>
        </button>
        <div className={`${!this.state.expanded && 'collapse'} navbar-collapse`}>
          <ul class="navbar-nav mr-auto">
            <li class="nav-item">
              <Link class="nav-link" to="/communities"><T i18nKey="communities">#</T></Link>
            </li>
            <li class="nav-item">
              <Link class="nav-link" to="/search"><T i18nKey="search">#</T></Link>
            </li>
            <li class="nav-item">
              <Link class="nav-link" to={{pathname: '/create_post', state: { prevPath: this.currentLocation }}}><T i18nKey="create_post">#</T></Link>
            </li>
            <li class="nav-item">
              <Link class="nav-link" to="/create_community"><T i18nKey="create_community">#</T></Link>
            </li>
          </ul>
          <ul class="navbar-nav ml-auto mr-2">
            {this.state.isLoggedIn ? 
            <>
              {
                <li className="nav-item">
                  <Link class="inbox nav-link" to="/inbox">
                    <svg class="icon"><use xlinkHref="#icon-mail"></use></svg>
                    {this.state.unreadCount> 0 && <span class="ml-1 badge badge-light">{this.state.unreadCount}</span>}
                  </Link>
                </li>
              }
              <li className={`nav-item dropdown ${this.state.expandUserDropdown && 'show'}`}>
                <a class="pointer nav-link dropdown-toggle" onClick={linkEvent(this, this.expandUserDropdown)} role="button">
                  {UserService.Instance.user.username}
                </a>
                <div className={`dropdown-menu dropdown-menu-right ${this.state.expandUserDropdown && 'show'}`}>
                  <a role="button" class="dropdown-item pointer" onClick={linkEvent(this, this.handleOverviewClick)}><T i18nKey="overview">#</T></a>
                  <a role="button" class="dropdown-item pointer" onClick={ linkEvent(this, this.handleLogoutClick) }><T i18nKey="logout">#</T></a>
                </div>
              </li> 
            </>
              : 
              <Link class="nav-link" to="/login"><T i18nKey="login_sign_up">#</T></Link>
            }
          </ul>
        </div>
      </nav>
    );
  }

  expandUserDropdown(i: Navbar) {
    i.state.expandUserDropdown = !i.state.expandUserDropdown;
    i.setState(i.state);
  }

  handleLogoutClick(i: Navbar) {
    i.state.expandUserDropdown = false;
    UserService.Instance.logout();
    i.context.router.history.push('/');
  }

  handleOverviewClick(i: Navbar) {
    i.state.expandUserDropdown = false;
    i.setState(i.state);
    let userPage = `/u/${UserService.Instance.user.username}`;
    i.context.router.history.push(userPage);
  }

  expandNavbar(i: Navbar) {
    i.state.expanded = !i.state.expanded;
    i.setState(i.state);
  }

  parseMessage(msg: any) {
    let op: UserOperation = msgOp(msg);
    if (msg.error) {
      // TODO
      if (msg.error == "Not logged in.") {
        UserService.Instance.logout();
        location.reload();
      }
      return;
    } else if (op == UserOperation.GetReplies) {
      let res: GetRepliesResponse = msg;
      let unreadReplies = res.replies.filter(r => !r.read);
      if (unreadReplies.length > 0 && this.state.fetchCount > 1 && 
          (JSON.stringify(this.state.replies) !== JSON.stringify(unreadReplies))) {
        this.notify(unreadReplies);
      }

      this.state.replies = unreadReplies;
      this.sendRepliesCount(res);
    } else if (op == UserOperation.GetSite) {
      let res: GetSiteResponse = msg;

      if (res.site) {
        this.state.siteName = res.site.name;
        WebSocketService.Instance.site = res.site;
        this.setState(this.state);
      }
    } 
  }

  keepFetchingReplies() {
    this.fetchReplies();
    setInterval(() => this.fetchReplies(), 30000);
  }

  fetchReplies() {
    if (this.state.isLoggedIn) {
      let repliesForm: GetRepliesForm = {
        sort: SortType[SortType.New],
        unread_only: true,
        page: 1,
        limit: 9999,
      };
      WebSocketService.Instance.getReplies(repliesForm);
      this.state.fetchCount++;
    }
  }

  get currentLocation() {
    return this.context.router.history.location.pathname;
  }

  sendRepliesCount(res: GetRepliesResponse) {
    UserService.Instance.sub.next({user: UserService.Instance.user, unreadCount: res.replies.filter(r => !r.read).length});
  }

  requestNotificationPermission() {
    if (UserService.Instance.user) {
    document.addEventListener('DOMContentLoaded', function () {
      if (!Notification) {
        alert(i18n.t('notifications_error')); 
        return;
      }

      if (Notification.permission !== 'granted')
        Notification.requestPermission();
    });
    }
  }

  notify(replies: Array<Comment>) {
    let recentReply = replies[0];
    if (Notification.permission !== 'granted')
      Notification.requestPermission();
    else {
      var notification = new Notification(`${replies.length} ${i18n.t('unread_messages')}`, {
        icon: `${window.location.protocol}//${window.location.host}/static/assets/apple-touch-icon.png`,
        body: `${recentReply.creator_name}: ${recentReply.content}`
      });

      notification.onclick = () => {
        this.context.router.history.push(`/post/${recentReply.post_id}/comment/${recentReply.id}`);
      };

    }
  }
}