// Find dialog derived from an example from Qt Toolkit (license below (**))

////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2009-2022 The Octave Project Developers
//
// See the file COPYRIGHT.md in the top-level directory of this
// or <https://octave.org/copyright/>.
//
//  All rights reserved.
//  Contact: Nokia Corporation (qt-info@nokia.com)
//
// This file is part of Octave.
//
// Octave is free software: you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Octave is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Octave; see the file COPYING.  If not, see
// <https://www.gnu.org/licenses/>.
//
// ** This file is part of the examples of the Qt Toolkit.
// **
// ** $QT_BEGIN_LICENSE:LGPL$
// ** Commercial Usage
// ** Licensees holding valid Qt Commercial licenses may use this file in
// ** accordance with the Qt Commercial License Agreement provided with the
// ** Software or, alternatively, in accordance with the terms contained in
// ** a written agreement between you and Nokia.
// **
// ** GNU Lesser General Public License Usage
// ** Alternatively, this file may be used under the terms of the GNU Lesser
// ** General Public License version 2.1 as published by the Free Software
// ** Foundation and appearing in the file LICENSE.LGPL included in the
// ** packaging of this file.  Please review the following information to
// ** ensure the GNU Lesser General Public License version 2.1 requirements
// ** will be met: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
// **
// ** In addition, as a special exception, Nokia gives you certain additional
// ** rights.  These rights are described in the Nokia Qt LGPL Exception
// ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
// **
// ** GNU General Public License Usage
// ** Alternatively, this file may be used under the terms of the GNU
// ** General Public License version 3.0 as published by the Free Software
// ** Foundation and appearing in the file LICENSE.GPL included in the
// ** packaging of this file.  Please review the following information to
// ** ensure the GNU General Public License version 3.0 requirements will be
// ** met: https://www.gnu.org/copyleft/gpl.html.
// **
// ** If you have questions regarding the use of this file, please contact
// ** Nokia at qt-info@nokia.com.
// ** $QT_END_LICENSE$
//
////////////////////////////////////////////////////////////////////////

#if defined (HAVE_CONFIG_H)
#  include "config.h"
#endif

#if defined (HAVE_QSCINTILLA)

#include <QApplication>
#include <QCheckBox>
#include <QCheckBox>
#include <QCompleter>
#include <QDialogButtonBox>
#include <QGridLayout>
#include <QIcon>
#include <QLabel>
#include <QLineEdit>
#include <QMessageBox>
#include <QPushButton>
#include <QVBoxLayout>

#include "find-dialog.h"
#include "gui-preferences-ed.h"
#include "gui-utils.h"
#include "resource-manager.h"
#include "octave-qobject.h"

namespace octave
{
  find_dialog::find_dialog (base_qobject& oct_qobj,
                            octave_dock_widget *ed, QWidget *p)
    : QDialog (p), m_octave_qobj (oct_qobj), m_editor (ed),
      m_in_sel (false), m_sel_beg (-1), m_sel_end (-1)
  {
    setWindowTitle (tr ("Editor: Find and Replace"));

    m_search_label = new QLabel (tr ("Find &what:"));
    m_search_line_edit = new QComboBox (this);
    m_search_line_edit->setToolTip (tr ("Enter text to search for"));
    m_search_line_edit->setEditable (true);
    m_search_line_edit->setMaxCount (m_mru_length);
    m_search_line_edit->completer ()->setCaseSensitivity (Qt::CaseSensitive);
    m_search_label->setBuddy (m_search_line_edit);

    m_replace_label = new QLabel (tr ("Re&place with:"));
    m_replace_line_edit = new QComboBox (this);
    m_replace_line_edit->setToolTip (tr ("Enter new text replacing search hits"));
    m_replace_line_edit->setEditable (true);
    m_replace_line_edit->setMaxCount (m_mru_length);
    m_replace_line_edit->completer ()->setCaseSensitivity (Qt::CaseSensitive);
    m_replace_label->setBuddy (m_replace_line_edit);

     int width = QFontMetrics (m_search_line_edit->font ()).averageCharWidth();
     m_search_line_edit->setFixedWidth (20*width);
     m_replace_line_edit->setFixedWidth (20*width);

    m_case_check_box = new QCheckBox (tr ("Match &case"));
    m_from_start_check_box = new QCheckBox (tr ("Search from &start"));
    m_wrap_check_box = new QCheckBox (tr ("&Wrap while searching"));
    m_wrap_check_box->setChecked (true);
    m_find_next_button = new QPushButton (tr ("&Find Next"));
    m_find_prev_button = new QPushButton (tr ("Find &Previous"));
    m_replace_button = new QPushButton (tr ("&Replace"));
    m_replace_all_button = new QPushButton (tr ("Replace &All"));

    m_more_button = new QPushButton (tr ("&More..."));
    m_more_button->setCheckable (true);
    m_more_button->setAutoDefault (false);

    m_button_box = new QDialogButtonBox (Qt::Vertical);
    m_button_box->addButton (m_find_next_button, QDialogButtonBox::ActionRole);
    m_button_box->addButton (m_find_prev_button, QDialogButtonBox::ActionRole);
    m_button_box->addButton (m_replace_button, QDialogButtonBox::ActionRole);
    m_button_box->addButton (m_replace_all_button, QDialogButtonBox::ActionRole);
    m_button_box->addButton (m_more_button, QDialogButtonBox::ActionRole);
    m_button_box->addButton (QDialogButtonBox::Close);

    m_extension = new QWidget (this);
    m_whole_words_check_box = new QCheckBox (tr ("&Whole words"));
    m_regex_check_box = new QCheckBox (tr ("Regular E&xpressions"));
    m_backward_check_box = new QCheckBox (tr ("Search &backward"));
    m_search_selection_check_box = new QCheckBox (tr ("Search se&lection"));
    m_search_selection_check_box->setCheckable (true);

    connect (m_find_next_button, &QPushButton::clicked,
             this, &find_dialog::find_next);
    connect (m_find_prev_button, &QPushButton::clicked,
             this, &find_dialog::find_prev);
    connect (m_more_button, &QPushButton::toggled,
             m_extension, &QWidget::setVisible);
    connect (m_replace_button, &QPushButton::clicked,
             this, &find_dialog::replace);
    connect (m_replace_all_button, &QPushButton::clicked,
             this, &find_dialog::replace_all);
    connect (m_backward_check_box, &QCheckBox::stateChanged,
             this, &find_dialog::handle_backward_search_changed);
    connect (m_button_box, &QDialogButtonBox::rejected,
             this, &find_dialog::close);

    connect (m_search_selection_check_box, &QCheckBox::stateChanged,
             this, &find_dialog::handle_sel_search_changed);

    QVBoxLayout *extension_layout = new QVBoxLayout ();
    extension_layout->setMargin (0);
    extension_layout->addWidget (m_whole_words_check_box);
    extension_layout->addWidget (m_backward_check_box);
    extension_layout->addWidget (m_search_selection_check_box);
    m_extension->setLayout (extension_layout);

    QGridLayout *top_left_layout = new QGridLayout;
    top_left_layout->addWidget (m_search_label, 1, 1);
    top_left_layout->addWidget (m_search_line_edit, 1, 2);
    top_left_layout->addWidget (m_replace_label, 2, 1);
    top_left_layout->addWidget (m_replace_line_edit, 2, 2);

    QVBoxLayout *left_layout = new QVBoxLayout;
    left_layout->addLayout (top_left_layout);
    left_layout->insertStretch (1, 5);
    left_layout->addWidget (m_case_check_box);
    left_layout->addWidget (m_from_start_check_box);
    left_layout->addWidget (m_wrap_check_box);
    left_layout->addWidget (m_regex_check_box);

    QGridLayout *main_layout = new QGridLayout;
    main_layout->setSizeConstraint (QLayout::SetFixedSize);
    main_layout->addLayout (left_layout, 0, 0);
    main_layout->addWidget (m_button_box, 0, 1);
    main_layout->addWidget (m_extension, 1, 0);
    setLayout (main_layout);

    m_extension->hide ();
    m_find_next_button->setDefault (true);
    m_find_result_available = false;
    m_rep_all = 0;
    m_rep_active = false;

    // Connect required external signals
    connect (ed, SIGNAL (edit_area_changed (octave_qscintilla *)),
             this, SLOT (update_edit_area (octave_qscintilla *)));

    setWindowModality (Qt::NonModal);

    setAttribute(Qt::WA_ShowWithoutActivating);
    setAttribute(Qt::WA_DeleteOnClose);
  }

  // The edit_area has changed: update relevant data of the file dialog
  void find_dialog::update_edit_area (octave_qscintilla *edit_area)
  {
    m_edit_area = edit_area;
    m_search_selection_check_box->setEnabled (edit_area->hasSelectedText ());

    connect (m_edit_area, SIGNAL (copyAvailable (bool)),
             this,       SLOT (handle_selection_changed (bool)),
             Qt::UniqueConnection);
  }

  void find_dialog::save_settings ()
  {
    gui_settings settings;

    // Save position
    QPoint dlg_pos = pos ();

#if defined (Q_OS_WIN32)
    int y = dlg_pos.y ();
#else
    int y = dlg_pos.y () - geometry ().height () + frameGeometry ().height ();
#endif

    m_last_position = QPoint (dlg_pos.x (), y);

    settings.setValue (ed_fdlg_pos.key, m_last_position);

    // Is current search/replace text in the mru list?
    mru_update (m_search_line_edit);
    mru_update (m_replace_line_edit);

    // Store mru lists
    QStringList mru;
    for (int i = 0; i < m_search_line_edit->count (); i++)
      mru.append (m_search_line_edit->itemText (i));
    settings.setValue (ed_fdlg_search.key, mru);

    mru.clear ();
    for (int i = 0; i < m_replace_line_edit->count (); i++)
      mru.append (m_replace_line_edit->itemText (i));
    settings.setValue (ed_fdlg_replace.key, mru);

    // Store dialog's options
    int opts = 0
               + m_extension->isVisible () * FIND_DLG_MORE
               + m_case_check_box->isChecked () * FIND_DLG_CASE
               + m_from_start_check_box->isChecked () * FIND_DLG_START
               + m_wrap_check_box->isChecked () * FIND_DLG_WRAP
               + m_regex_check_box->isChecked () * FIND_DLG_REGX
               + m_whole_words_check_box->isChecked () * FIND_DLG_WORDS
               + m_backward_check_box->isChecked () * FIND_DLG_BACK
               + m_search_selection_check_box->isChecked () * FIND_DLG_SEL;
    settings.setValue (ed_fdlg_opts.key, opts);

    settings.sync ();
  }

  void find_dialog::restore_settings (QPoint ed_bottom_right)
  {
    gui_settings settings;

    // Get mru lists for search and replace text
    QStringList mru = settings.value (ed_fdlg_search.key).toStringList ();
    while (mru.length () > m_mru_length)
      mru.removeLast ();
    m_search_line_edit->addItems (mru);

    mru = settings.value (ed_fdlg_replace.key).toStringList ();
    while (mru.length () > m_mru_length)
      mru.removeLast ();
    m_replace_line_edit->addItems (mru);

    // Get the dialog's options
    int opts = settings.value (ed_fdlg_opts.key, ed_fdlg_opts.def).toInt ();

    m_extension->setVisible (FIND_DLG_MORE & opts);
    m_case_check_box->setChecked (FIND_DLG_CASE & opts);
    m_from_start_check_box->setChecked (FIND_DLG_START & opts);
    m_wrap_check_box->setChecked (FIND_DLG_WRAP & opts);
    m_regex_check_box->setChecked (FIND_DLG_REGX & opts);
    m_whole_words_check_box->setChecked (FIND_DLG_WORDS & opts);
    m_backward_check_box->setChecked (FIND_DLG_BACK & opts);
    m_search_selection_check_box->setChecked (FIND_DLG_SEL & opts);

    // Default position:  lower right of editor's position
    int xp = ed_bottom_right.x () - sizeHint ().width ();
    int yp = ed_bottom_right.y () - sizeHint ().height ();
    QRect default_geometry (xp, yp, sizeHint ().width (), sizeHint ().height ());

    // Last position from settings
    m_last_position = settings.value (ed_fdlg_pos.key, QPoint (xp, yp)).toPoint ();
    QRect last_geometry (m_last_position,
                         QSize (sizeHint ().width (), sizeHint ().height ()));

    // Make sure we are on the screen
    adjust_to_screen (last_geometry, default_geometry);
    m_last_position = last_geometry.topLeft ();

    move (m_last_position);
  }

  // set text of "search from start" depending on backward search
  void find_dialog::handle_backward_search_changed (int backward)
  {
    if (backward)
      m_from_start_check_box->setText (tr ("Search from end"));
    else
      m_from_start_check_box->setText (tr ("Search from start"));
  }

  // search text has changed: reset the search
  void find_dialog::handle_search_text_changed (void)
  {
    // Return if nothing has changed
    if (m_search_line_edit->currentText () == m_search_line_edit->itemText (0))
      return;

    if (m_search_selection_check_box->isChecked ())
      m_find_result_available = false;

    mru_update (m_search_line_edit);
  }

  // replaced text has changed: reset the search
  void find_dialog::handle_replace_text_changed (void)
  {
    // Return if nothing has changed
    if (m_replace_line_edit->currentText () == m_replace_line_edit->itemText (0))
      return;

    mru_update (m_replace_line_edit);
  }

  // Update the mru list
  void find_dialog::mru_update (QComboBox *mru)
  {
    // Remove possible empty entries from the mru list
    int index;
    while ((index = mru->findText (QString ())) >= 0)
      mru->removeItem (index);

    // Get current text and return if it is empty
    QString text = mru->currentText ();

    if (text.isEmpty ())
      return;

    // Remove occurrences of the current text in the mru list
    while ((index = mru->findText (text)) >= 0)
      mru->removeItem (index);

    // Remove the last entry from the end if the list is full
    if (mru->count () == m_mru_length)
      mru->removeItem (m_mru_length -1);

    // Insert new item at the beginning and set it as current item
    mru->insertItem (0, text);
    mru->setCurrentIndex (0);
  }

  void find_dialog::handle_sel_search_changed (int selected)
  {
    m_from_start_check_box->setEnabled (! selected);
    m_find_result_available = false;
  }

  void find_dialog::handle_selection_changed (bool has_selected)
  {
    if (m_rep_active)
      return;

    m_search_selection_check_box->setEnabled (has_selected);
    m_find_result_available = false;
  }

  // initialize search text with selected text if this is in one single line
  void find_dialog::init_search_text (void)
  {
    if (m_edit_area && m_edit_area->hasSelectedText ())
      {
        int lbeg, lend, cbeg, cend;
        m_edit_area->getSelection (&lbeg, &cbeg, &lend, &cend);
        if (lbeg == lend)
          m_search_line_edit->setCurrentText (m_edit_area->selectedText ());
      }

    // set focus to "Find what" and select all text
    m_search_line_edit->setFocus ();
    m_search_line_edit->lineEdit ()->selectAll ();

    // Default to "find" next time.
    // Otherwise, it defaults to the last action, which may be "replace all".
    m_find_next_button->setDefault (true);
  }

  void find_dialog::find_next (void)
  {
    find (! m_backward_check_box->isChecked ());
  }

  void find_dialog::find_prev (void)
  {
    find (m_backward_check_box->isChecked ());
  }

  void find_dialog::find (bool forward)
  {
    if (! m_edit_area)
      return;

    handle_search_text_changed ();

    // line adn col: -1 means search starts at current position
    int line = -1, col = -1;

    bool do_wrap = m_wrap_check_box->isChecked ();
    bool do_forward = forward;

    // Initialize the selection begin and end if it is the first search
    if (! m_find_result_available)
      {
        if (m_search_selection_check_box->isChecked ()
            && m_edit_area->hasSelectedText ())
          {
            int l1, c1, l2, c2;
            m_edit_area->getSelection (&l1, &c1, &l2, &c2);

            // Store the position of the selection
            m_sel_beg = m_edit_area->positionFromLineIndex (l1, c1);
            m_sel_end = m_edit_area->positionFromLineIndex (l2, c2);
            m_in_sel = true;
          }
        else
          m_in_sel = false;
      }

    // Get the correct line/col for beginning the search
    if (m_rep_all)
      {
        // Replace All
        if (m_rep_all == 1)
          {
            // Start at the beginning of file/sel if it is the first try
            if (m_in_sel)
              m_edit_area->lineIndexFromPosition (m_sel_beg, &line, &col);
            else
              {
                line = 0;
                col = 0;
              }
          }
        do_wrap = false;  // Never wrap when replacing all
      }
    else
      {
        // Normal search (not replace all): calculate start position of
        // search (in file or selection)
        if (m_from_start_check_box->isChecked ()
            || (m_in_sel && (! m_find_result_available)))
          {
            // From the beginning or the end of file/sel
            if (do_forward)
              {
                // From the beginning
                if (m_in_sel)
                  m_edit_area->lineIndexFromPosition (m_sel_beg, &line, &col);
                else
                  {
                    line = 0;
                    col = 0;
                  }
              }
            else
              {
                // From the end
                if (m_in_sel)
                  m_edit_area->lineIndexFromPosition (m_sel_end, &line, &col);
                else
                  {
                    line = m_edit_area->lines () - 1;
                    col  = m_edit_area->text (line).length () - 1;
                    if (col == -1)
                      col = 0;
                  }
              }
          }
        else if (! do_forward)
          {
            // Start from where the cursor is.  Fix QScintilla's cursor
            // positioning
            m_edit_area->getCursorPosition (&line, &col);
            if (m_find_result_available && m_edit_area->hasSelectedText ())
              {
                int currpos = m_edit_area->positionFromLineIndex (line, col);
                currpos -= (m_search_line_edit->currentText ().length ());
                if (currpos < 0)
                  currpos = 0;
                m_edit_area->lineIndexFromPosition (currpos, &line, &col);
              }
          }
      }

    // Do the search
    m_find_result_available
      = m_edit_area->findFirst (m_search_line_edit->currentText (),
                                m_regex_check_box->isChecked (),
                                m_case_check_box->isChecked (),
                                m_whole_words_check_box->isChecked (),
                                do_wrap,
                                do_forward,
                                line, col,
                                true
#if defined (HAVE_QSCI_VERSION_2_6_0)
                                , true
#endif
                               );

    if (m_find_result_available)
      {
        // Search successful: reset search-from-start box and check for
        // the current selection
        m_from_start_check_box->setChecked (0);

        if (m_in_sel)
          {
            m_edit_area->getCursorPosition (&line, &col);
            int pos = m_edit_area->positionFromLineIndex (line, col);

            int l1, c1, l2, c2;
            m_edit_area->lineIndexFromPosition (m_sel_beg, &l1, &c1);
            m_edit_area->lineIndexFromPosition (m_sel_end, &l2, &c2);
            m_edit_area->show_selection_markers (l1, c1, l2, c2);

            // Check if new start position is still within the selection
            m_find_result_available =  pos >= m_sel_beg && pos <= m_sel_end;
          }
      }

    // No more search hits
    if (! m_find_result_available)
      {
        if (m_in_sel)
          {
            // Restore real selection and remove marker for selection
            int l1, c1, l2, c2;
            m_edit_area->lineIndexFromPosition (m_sel_beg, &l1, &c1);
            m_edit_area->lineIndexFromPosition (m_sel_end, &l2, &c2);
            m_edit_area->setSelection (l1, c1, l2, c2);
            m_edit_area->clear_selection_markers ();
          }

        // Display message if not replace all
        if (! m_rep_all)
          no_matches_message ();
      }

  }

  void find_dialog::do_replace (void)
  {
    if (m_edit_area)
      {
        m_rep_active = true;  // changes in selection not made by the user

        m_edit_area->replace (m_replace_line_edit->currentText ());
        if (m_in_sel)
          {
            // Update the length of the selection
            m_sel_end = m_sel_end
                        - m_search_line_edit->currentText ().toUtf8 ().size ()
                        + m_replace_line_edit->currentText ().toUtf8 ().size ();
          }

        m_rep_active = false;
      }
  }

  void find_dialog::replace (void)
  {
    if (m_edit_area)
      {
        handle_replace_text_changed ();

        // Do the replace if we have selected text
        if (m_find_result_available && m_edit_area->hasSelectedText ())
          do_replace ();

        find_next ();
      }
  }

  void find_dialog::replace_all (void)
  {
    int line, col;

    if (m_edit_area)
      {
        handle_replace_text_changed ();

        m_edit_area->getCursorPosition (&line, &col);

        m_rep_all = 1;
        find_next ();  // find first occurrence (forward)

        m_edit_area->beginUndoAction ();
        while (m_find_result_available)   // while search string is found
          {
            do_replace ();
            m_rep_all++;                                          // inc counter
            find_next ();                                        // find next
          }
        m_edit_area->endUndoAction ();

        QMessageBox msg_box (QMessageBox::Information, tr ("Replace Result"),
                             tr ("%1 items replaced").arg (m_rep_all-1),
                             QMessageBox::Ok, this);
        msg_box.exec ();

        m_rep_all = 0;
        m_find_result_available = false;

        if (! m_search_selection_check_box->isChecked ())
          m_edit_area->setCursorPosition (line, col);
      }
  }

  void find_dialog::no_matches_message (void)
  {
    QMessageBox msg_box (QMessageBox::Information, tr ("Find Result"),
                         tr ("No more matches found"), QMessageBox::Ok, this);
    msg_box.exec ();
  }

  void find_dialog::reject ()
  {
    close ();
  }

  void find_dialog::closeEvent (QCloseEvent *e)
  {
    save_settings ();
    e->accept ();
  }

  // Show and hide with (re-)storing position, otherwise there is always
  // a small shift each time the dialog is shown again
  void find_dialog::set_visible (bool visible)
  {
    if (visible)
      {
        show ();
        move (m_last_position);
      }
    else
      {
        m_last_position = pos ();
        hide ();
      }
  }

}
#endif