Blob Blame Raw
#pragma once

#ifndef OVERWRITEPOPUP_H
#define OVERWRITEPOPUP_H

// TnzQt includes
#include "toonzqt/dvdialog.h"

//===========================================================

//    Forward declarations

class ToonzScene;
class TFilePath;

namespace DVGui {
class CheckBox;
class LineEdit;
}

class QLabel;
class QRadioButton;
class QPushButton;

//===========================================================

/*!
  This dialog offers a standard 'overwrite file' resolution Dialog.

  Sometimes, the system attempts to write to the file path of an
  already existing file. This dialog presents a standard interface
  to let users choose the conflict resolution manually.

  There are 3 possible resolution choices:

  \li \b Overwrite the existing file
  \li \b Keep the existing file (aborting the write operation)
  \li \b Rename the file to write adding a suffix

  Additionally, the dialog could be \t CANCELED, either by closing it
  or pressing the "Cancel" button.
*/
class OverwriteDialog : public DVGui::Dialog {
  Q_OBJECT

public:
  enum Resolution {
    CANCELED        = 0x0,
    KEEP_OLD        = 0x1,
    OVERWRITE       = 0x2,
    RENAME          = 0x4,
    ALL_RESOLUTIONS = KEEP_OLD | OVERWRITE | RENAME
  };
  enum Flags { NO_FLAG = 0x0, APPLY_TO_ALL_FLAG = 0x1 };

public:
  struct ExistsFunc {
    virtual QString conflictString(const TFilePath &fp) const = 0;
    virtual bool operator()(const TFilePath &fp) const        = 0;
  };

  struct DecodeFileExistsFunc : public ExistsFunc {
    ToonzScene *m_scene;
    DecodeFileExistsFunc(ToonzScene *scene) : m_scene(scene) {}

    QString conflictString(const TFilePath &fp) const;
    bool operator()(const TFilePath &fp) const;
  };

public:
  OverwriteDialog();

  bool cancelPressed() const { return m_cancelPressed; }

  Resolution execute(TFilePath &filePath, const ExistsFunc &exists,
                     Resolution acceptedRes = ALL_RESOLUTIONS,
                     Flags flags            = NO_FLAG);

  //--------------------- Legacy Functions ------------------------

  // The following functions are deprecated and retained for backward
  // compatibility only

  //! Returns the resolution type chosen by the user
  Resolution getChoice() const { return m_choice; }

  //! Returns the suffix to be added at the base file name when the chosen
  //! resolution type is \t KEEP_OLD.
  std::wstring getSuffix();

  //! Resets state variables
  void reset();

  /*! this method has to be called for each filepath to be imported. Only if
necessary, it opens a popup.
put parameter multiLoad to true only if you are importing more then one level
(so that the button 'apply to all' appears in the dialog)*/
  std::wstring execute(ToonzScene *scene, const TFilePath &levelPath,
                       bool multiLoad);

protected slots:

  void applyToAll();
  void cancel();
  void onButtonClicked(int);

private:
  bool m_applyToAll;
  bool m_cancelPressed;

  Resolution m_choice;

  QLabel *m_label;
  QRadioButton *m_overwrite;
  QRadioButton *m_keep;
  QRadioButton *m_rename;
  DVGui::LineEdit *m_suffix;
  QPushButton *m_okBtn, *m_okToAllBtn, *m_cancelBtn;

private:
  TFilePath addSuffix(const TFilePath &src) const;
};

#endif  // OVERWRITEPOPUP_H