/* 
 * File:   KeyringIO.h
 * Author: Saleem Edah-Tally - nmset@yandex.com
 * License : GPL v2
 * Copyright Saleem Edah-Tally - © 2019
 *
 * Created on November 13, 2020, 6:46 PM
 */

#ifndef KEYRINGIO_H
#define KEYRINGIO_H

#include "K7Main.h"
#include <Wt/WComboBox.h>
#include <Wt/WStreamResource.h>
#include "PopupUploader.h"
#include "PopupDeleter.h"
#include "PopupCreate.h"
#include "PopupExportSecretKey.h"

using namespace Wt;

class K7Main;

/**
 * Manages the keyring : import, create and delete keys.
 * Is a pseudo-extension of K7Main. Both classes are friends to each other, and
 * everything is private here.
 * Does not edit keys.
 * @param owner
 */
class KeyringIO : public WObject
{
    friend class K7Main;
public:

private:
    KeyringIO(K7Main * owner);
    virtual ~KeyringIO();

    K7Main * m_owner;
    PopupUpload * m_popupUpload;
    PopupDelete * m_popupDelete;
    PopupCreate * m_popupCreate;
    PopupExportSecretKey * m_popupExportSecretKey;

    // Variables from m_owner mapped here.
    AppConfig * m_config;
    TransientMessageWidget * m_tmwMessage;
    WPushButton * m_btnUpload;
    WPushButton * m_btnImport;
    WPushButton * m_btnDelete;
    WPushButton * m_btnCreate;
    WPushButton * m_btnExport;
    WLineEdit * m_leSearch;

    // Used to disconnect m_btnExport from previous slot.
    Signals::connection m_exportSecretConnection;

    /**
     * Shows a non-blocking popup to upload a key,
     * with forward confirmation for upload.
     */
    void ShowPopupUpload();
    /**
     * Actually import the uploaded file.
     */
    void DoImportKey();
    /**
     * Shows a non-blocking popup to delete a key
     */
    void ShowPopupDelete();
    /**
     * Shows a non-blocking popup to export a secret key
     */
    void ShowPopupExportSecretKey(const WString& fpr);
    /**
     * All public keys can be deleted.
     * Private keys can be deleted only if the user
     * manages that key, as declared in the configuration file.
     * @param fullKeyID
     * @return 
     */
    bool CanKeyBeDeleted(const WString& fullKeyID);
    /**
     * Actually delete the key, with forward confirmation in the popup.
     */
    void DoDeleteKey();
    /**
     * Shows a non-blocking popup to create keys.
     */
    void ShowPopupCreate();
    /**
     * Validates input and creates the keys.
     */
    void DoCreateKey();
    /**
     * Shows a button to import an uploaded key
     * @param spool
     */
    void OnUploadCompleted(const WString& spool);
    /**
     * If isSecret is true, shows a popup for passphrase input befort exporting.
     * Else, adds a link with a shared resource to the export button.
     * @param fpr
     * @param isSecret
     */
    void PrepareExport(const WString& fpr, bool isSecret);
    /**
     * Adds a link with a shared resource to the export button of the popup.
     */
    void OnPreExportSecretKey(const WString& fpr);
};

/**
 * Export a public key for download.
 * \n Secret keys cannot be exported and are excluded.
 * @param fpr
 * @param isSecret
 * @param tmw
 */
class ExportKeyStreamResource : public WStreamResource
{
public:
    ExportKeyStreamResource(const string& keyData,
                            const WString& fpr,
                            TransientMessageWidget * tmw);
    ExportKeyStreamResource(const string& keyData,
                            const WString& fpr,
                            const string& mimeType,
                            TransientMessageWidget * tmw);
    virtual ~ExportKeyStreamResource();
    /**
     * Actually exports the public key and starts download on success.
     * @param request
     * @param response
     */
    void handleRequest(const Http::Request& request,
                       Http::Response& response) override;

private:
    WString m_fpr;
    string m_keyData;
    TransientMessageWidget * m_tmwMessage;
};

#endif /* KEYRINGIO_H */