KeyhoteeMainWindow.hpp

#pragma once
#include <memory>
#include <unordered_map>
#include <bts/addressbook/addressbook.hpp>
#include <bts/application.hpp>

#include "qtreusable/selfsizingmainwindow.h"

#include "dataaccessimpl.h"
#include "mailprocessorimpl.hpp"
#include "ch/ModificationsChecker.hpp"
#include "ATopLevelWindowsContainer.hpp"

namespace Ui { class KeyhoteeMainWindow; }

class QTreeWidgetItem;
class QLineEdit;
class QCompleter;

class AddressBookModel;
class Contact;
class ContactView;
class ContactsTable;
class InboxView;
class Mailbox;
class MailboxModel;
class Mailbox;
class MenuEditControl;
class KeyhoteeMainWindow;
class TKeyhoteeApplication;

/**
 *  GUI widgets and GUI state for a contact.
 *  Not all contacts have this, only contacts "active" in GUI.
 */
class ContactGui
{
  friend KeyhoteeMainWindow;

private:
  unsigned int     _unread_msg_count;
  QTreeWidgetItem* _tree_item;
  ContactView*     _view;

public:
  ContactGui() {}
  ContactGui(QTreeWidgetItem* tree_item, ContactView* view)
    : _tree_item(tree_item), _view(view), _unread_msg_count(0) {}

  void updateTreeItemDisplay();
  void setUnreadMsgCount(unsigned int count);
  bool isChatVisible();
  void receiveChatMessage(const QString& from, const QString& msg, const QDateTime& dateTime);
private:
};

class KeyhoteeMainWindow  : public ATopLevelWindowsContainer,
                            protected bts::application_delegate,
                            protected IMailProcessor::IUpdateSink,
                            public IModificationsChecker
{
  Q_OBJECT
public:

  void newMailMessage();
  void newMailMessageTo(const Contact& contact);
  void addContact();
  void addToContacts(const bts::addressbook::wallet_contact* wallet_contact);
  void onSidebarSelectionChanged();
  void onSidebarDoubleClicked();
  void selectContactItem(QTreeWidgetItem* item);
  void selectIdentityItem(QTreeWidgetItem* item);
  void sideBarSplitterMoved(int pos, int index);

  void openContactGui(int contact_id);
  ContactGui* getContactGui(int contact_id);
  ContactGui* createContactGuiIfNecessary(int contact_id);
  bool isSelectedContactGui(ContactGui* contactGui);

  void openDraft(int draft_id);
  void openMail(int message_id);
  void openSent(int message_id);
  virtual bool canContinue() const;

  void displayDiagnosticLog();
  void setEnabledAttachmentSaveOption(bool enable);
  void setEnabledDeleteOption( bool enable ) const;
  void refreshDeleteContactOption() const;
  void refreshEditMenu() const;
  void setEnabledMailActions(bool enable);
  void setMailSettings (MailSettings& mailSettings);  
  ContactsTable* getContactsPage();

  AddressBookModel* getAddressBookModel() { return _addressbook_model; }

signals:
  void checkSendMailSignal();
protected:
  virtual void closeEvent(QCloseEvent *);
  virtual void keyPressEvent(QKeyEvent *);

private:
  /// application_delegate interface implementation
  virtual void received_text(const bts::bitchat::decrypted_message& msg);
  virtual void received_email(const bts::bitchat::decrypted_message& msg);

private:
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMessageSaving() override;
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMessageSaved(const TStoredMailMessage& msg,
    const TStoredMailMessage* overwrittenOne) override;
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMessageGroupPending(unsigned int count) override;
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMessagePending(const TStoredMailMessage& msg,
    const TStoredMailMessage* savedDraftMsg) override;
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMessageGroupPendingEnd() override;
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMessageSendingStart() override;
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMessageSent(const TStoredMailMessage& pendingMsg,
    const TStoredMailMessage& sentMsg) override;
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMessageSendingEnd() override;
  /// \see IMessageProcessor::IUpdateSink interface description.
  virtual void OnMissingSenderIdentity(const TRecipientPublicKey& senderId,
    const TPhysicalMailMessage& msg) override;

  /// Only TKeyhoteeApplication can build main window object.
  friend class TKeyhoteeApplication;
  KeyhoteeMainWindow(const TKeyhoteeApplication& mainApp);
  virtual ~KeyhoteeMainWindow();
  /// Helper method to simplify onSidebarSelectionChanged code.
  void activateMailboxPage(Mailbox* mailBox);

private slots:
  // ---------- MenuBar
  // File
  void onExit();
  // Edit
  void onCopy();
  void onCut();
  void onPaste();
  void onSelectAll();
  // Identity
  void onNewIdentity();
  void enableNewMessageIcon();
  void onEnableMining(bool enabled);
  // Mail
  void onSaveAttachement();
  // Contact
  void onSetIcon();
  // Help
  void onDiagnostic();
  void onAbout();

  void onCanceledNewContact();
  void onSavedNewContact(int idxNewContact);
  void onItemContactRemoved (QTreeWidgetItem&);
  void onRemoveContact ();
  void onClipboardChanged();
  void onFocusChanged(QWidget *old, QWidget *now);

private:
  bool isIdentityPresent();
  void addressBookDataChanged(const QModelIndex& top_left, const QModelIndex& bottom_right,
                              const QVector<int>& roles);
  void searchEditChanged(QString search_string);

  void createContactGui(int contact_id);
  void showContactGui(ContactGui& contact_gui);
  void deleteContactGui(int contact_id);
  void setupStatusBar();
  void notSupported();
  void enableMenu(bool enable);
  bool checkSaving() const;
  void showContacts() const;

  /// Class attributes:

  //QCompleter*                             _contact_completer;
  QTreeWidgetItem*                        _identities_root;
  QTreeWidgetItem*                        _mailboxes_root;
  QTreeWidgetItem*                        _wallets_root;
  QTreeWidgetItem*                        _contacts_root;
  QTreeWidgetItem*                        _inbox_root;
  QTreeWidgetItem*                        _drafts_root;
  QTreeWidgetItem*                        _out_box_root;
  QTreeWidgetItem*                        _sent_root;
  QTreeWidgetItem*                        _bitcoin_root;
  QTreeWidgetItem*                        _bitshares_root;
  QTreeWidgetItem*                        _litecoin_root;

  MailboxModel*                           _inbox_model;
  MailboxModel*                           _draft_model;
  MailboxModel*                           _pending_model;
  MailboxModel*                           _sent_model;

  AddressBookModel*                       _addressbook_model;
  bts::addressbook::addressbook_ptr       _addressbook;
  std::unordered_map<int, ContactGui>     _contact_guis;

  QLineEdit*                              _search_edit;
  Ui::KeyhoteeMainWindow*                 ui;
  TConnectionStatusDS                     ConnectionStatusDS;
  TMailProcessor                          MailProcessor;
  Mailbox*                                _currentMailbox;
  MenuEditControl*                        menuEdit;
}; //KeyhoteeMainWindow

KeyhoteeMainWindow* getKeyhoteeWindow();