/////////////////////////////////////////////////////////////////////////////
// Name:        src/gtk/filedlg.cpp
// Purpose:     native implementation of wxFileDialog
// Author:      Robert Roebling, Zbigniew Zagorski, Mart Raudsepp
// Copyright:   (c) 1998 Robert Roebling, 2004 Zbigniew Zagorski, 2005 Mart Raudsepp
// Licence:     wxWindows licence
/////////////////////////////////////////////////////////////////////////////

// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"

#if wxUSE_FILEDLG

#include "wx/filedlg.h"

#ifndef WX_PRECOMP
    #include "wx/intl.h"
    #include "wx/log.h"
    #include "wx/msgdlg.h"
#endif

#include "wx/gtk/private.h"
#include "wx/gtk/private/error.h"
#include "wx/gtk/private/mnemonics.h"
#include "wx/gtk/private/gtk3-compat.h"

#ifdef __UNIX__
#include <unistd.h> // chdir
#endif

#include "wx/filename.h" // wxFilename
#include "wx/tokenzr.h" // wxStringTokenizer
#include "wx/filefn.h" // ::wxGetCwd
#include "wx/modalhook.h"

//-----------------------------------------------------------------------------
// "clicked" for OK-button
//-----------------------------------------------------------------------------

extern "C" {
static void gtk_filedialog_ok_callback(GtkWidget *widget, wxFileDialog *dialog)
{
    int style = dialog->GetWindowStyle();
    wxGtkString filename(gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(widget)));

    // gtk version numbers must be identical with the one in ctor (that calls set_do_overwrite_confirmation)
#ifndef __WXGTK3__
#if GTK_CHECK_VERSION(2,7,3)
    if (!wx_is_at_least_gtk2(8))
#endif
    {
        if ((style & wxFD_SAVE) && (style & wxFD_OVERWRITE_PROMPT))
        {
            if ( g_file_test(filename, G_FILE_TEST_EXISTS) )
            {
                wxString msg;

                msg.Printf(
                    _("File '%s' already exists, do you really want to overwrite it?"),
                    wxString::FromUTF8(filename));

                wxMessageDialog dlg(dialog, msg, _("Confirm"),
                                   wxYES_NO | wxICON_QUESTION);
                if (dlg.ShowModal() != wxID_YES)
                    return;
            }
        }
    }
#endif

    if (style & wxFD_FILE_MUST_EXIST)
    {
        if ( !g_file_test(filename, G_FILE_TEST_EXISTS) )
        {
            wxMessageDialog dlg( dialog, _("Please choose an existing file."),
                                 _("Error"), wxOK| wxICON_ERROR);
            dlg.ShowModal();
            return;
        }
    }

    // change to the directory where the user went if asked
    if (style & wxFD_CHANGE_DIR)
    {
        // Use chdir to not care about filename encodings
        wxGtkString folder(g_path_get_dirname(filename));
        if ( chdir(folder) != 0 )
        {
            wxLogSysError(_("Changing current directory to \"%s\" failed"),
                          wxString::FromUTF8(folder));
        }
    }

    wxCommandEvent event(wxEVT_BUTTON, wxID_OK);
    event.SetEventObject(dialog);
    dialog->HandleWindowEvent(event);
}
}

//-----------------------------------------------------------------------------
// "clicked" for Cancel-button
//-----------------------------------------------------------------------------

extern "C"
{

static void
gtk_filedialog_cancel_callback(GtkWidget * WXUNUSED(w), wxFileDialog *dialog)
{
    wxCommandEvent event(wxEVT_COMMAND_BUTTON_CLICKED, wxID_CANCEL);
    event.SetEventObject(dialog);
    dialog->HandleWindowEvent(event);
}

static void gtk_filedialog_response_callback(GtkWidget *w,
                                             gint response,
                                             wxFileDialog *dialog)
{
    if (response == GTK_RESPONSE_ACCEPT)
        gtk_filedialog_ok_callback(w, dialog);
    else    // GTK_RESPONSE_CANCEL or GTK_RESPONSE_NONE
        gtk_filedialog_cancel_callback(w, dialog);
}

static void gtk_filedialog_selchanged_callback(GtkFileChooser *chooser,
                                               wxFileDialog *dialog)
{
    wxGtkString filename(gtk_file_chooser_get_preview_filename(chooser));

    dialog->GTKSelectionChanged(wxString::FromUTF8(filename));
}


static void gtk_filedialog_update_preview_callback(GtkFileChooser *chooser,
                                                   gpointer user_data)
{
    GtkWidget *preview = GTK_WIDGET(user_data);

    wxGtkString filename(gtk_file_chooser_get_preview_filename(chooser));

    if ( !filename )
        return;

    GdkPixbuf *pixbuf = gdk_pixbuf_new_from_file_at_size(filename, 128, 128, nullptr);
    gboolean have_preview = pixbuf != nullptr;

    gtk_image_set_from_pixbuf(GTK_IMAGE(preview), pixbuf);
    if ( pixbuf )
        g_object_unref (pixbuf);

    gtk_file_chooser_set_preview_widget_active(chooser, have_preview);
}

#if GTK_CHECK_VERSION(3,20,0)
static void wx_filedialog_show(GtkWidget*, wxFileDialog* win)
{
    // If m_widget is shown, then GtkFileChooserNative is not being used.
    // This happens when using wxFilePickerCtrl, for example.
    win->GTKDropNative();
}
#endif

} // extern "C"

void wxFileDialog::AddChildGTK(wxWindowGTK* child)
{
    // allow dialog to be resized smaller horizontally
    gtk_widget_set_size_request(
        child->m_widget, child->GetMinWidth(), child->m_height);

    gtk_file_chooser_set_extra_widget(
        GTK_FILE_CHOOSER(m_widget), child->m_widget);
}

//-----------------------------------------------------------------------------
// wxFileDialog
//-----------------------------------------------------------------------------

wxIMPLEMENT_DYNAMIC_CLASS(wxFileDialog, wxFileDialogBase);

wxBEGIN_EVENT_TABLE(wxFileDialog,wxFileDialogBase)
    EVT_BUTTON(wxID_OK, wxFileDialog::OnFakeOk)
    EVT_SIZE(wxFileDialog::OnSize)
wxEND_EVENT_TABLE()

wxFileDialog::wxFileDialog(wxWindow *parent, const wxString& message,
                           const wxString& defaultDir,
                           const wxString& defaultFileName,
                           const wxString& wildCard,
                           long style, const wxPoint& pos,
                           const wxSize& sz,
                           const wxString& name)
    : wxFileDialogBase()
{
    Create(parent, message, defaultDir, defaultFileName, wildCard, style, pos, sz, name);
}

bool wxFileDialog::Create(wxWindow *parent, const wxString& message,
                           const wxString& defaultDir,
                           const wxString& defaultFileName,
                           const wxString& wildCard,
                           long style, const wxPoint& pos,
                           const wxSize& sz,
                           const wxString& name)
{
    parent = GetParentForModalDialog(parent, style);

    if (!BaseType::Create(parent, message, defaultDir, defaultFileName,
                                  wildCard, style, pos, sz, name))
    {
        return false;
    }

    if (!PreCreation(parent, pos, wxDefaultSize) ||
        !CreateBase(parent, wxID_ANY, pos, wxDefaultSize, style,
                wxDefaultValidator, wxT("filedialog")))
    {
        wxFAIL_MSG( wxT("wxFileDialog creation failed") );
        return false;
    }

    GtkFileChooserAction gtk_action;
    GtkWindow* gtk_parent = nullptr;
    if (parent)
        gtk_parent = GTK_WINDOW( gtk_widget_get_toplevel(parent->m_widget) );

    wxString ok_btn_stock;
    if ( style & wxFD_SAVE )
    {
        gtk_action = GTK_FILE_CHOOSER_ACTION_SAVE;
#ifdef __WXGTK4__
        ok_btn_stock = wxConvertMnemonicsToGTK(wxGetStockLabel(wxID_SAVE));
#else
        ok_btn_stock = "gtk-save";
#endif
    }
    else
    {
        gtk_action = GTK_FILE_CHOOSER_ACTION_OPEN;
#ifdef __WXGTK4__
        ok_btn_stock = wxConvertMnemonicsToGTK(wxGetStockLabel(wxID_OPEN));
#else
        ok_btn_stock = "gtk-open";
#endif
    }

    m_widget = gtk_file_chooser_dialog_new(
                   m_message.utf8_str(),
                   gtk_parent,
                   gtk_action,
#ifdef __WXGTK4__
                   static_cast<const gchar*>(wxConvertMnemonicsToGTK(wxGetStockLabel(wxID_CANCEL)).utf8_str()),
#else
                   "gtk-cancel",
#endif
                   GTK_RESPONSE_CANCEL,
                   static_cast<const gchar*>(ok_btn_stock.utf8_str()), GTK_RESPONSE_ACCEPT,
                   nullptr);

    g_object_ref(m_widget);
    GtkFileChooser* file_chooser = GTK_FILE_CHOOSER(m_widget);

    m_fc.SetWidget(file_chooser);

#if GTK_CHECK_VERSION(3,20,0)
    if (wx_is_at_least_gtk3(20) && (style & wxFD_PREVIEW) == 0)
    {
        m_fileChooserNative = GTK_FILE_CHOOSER(gtk_file_chooser_native_new(
            m_message.utf8_str(), gtk_parent, gtk_action, nullptr, nullptr));
        m_fcNative = new wxGtkFileChooser;
        m_fcNative->SetWidget(m_fileChooserNative);
        g_signal_connect(m_widget, "show", G_CALLBACK(wx_filedialog_show), this);
    }
#endif

    gtk_dialog_set_default_response(GTK_DIALOG(m_widget), GTK_RESPONSE_ACCEPT);

    if ( style & wxFD_MULTIPLE )
    {
        gtk_file_chooser_set_select_multiple(file_chooser, true);
        if (m_fileChooserNative)
            gtk_file_chooser_set_select_multiple(m_fileChooserNative, true);
    }

    // local-only property could be set to false to allow non-local files to be
    // loaded. In that case get/set_uri(s) should be used instead of
    // get/set_filename(s) everywhere and the GtkFileChooserDialog should
    // probably also be created with a backend, e.g. "gnome-vfs", "default", ...
    // (gtk_file_chooser_dialog_new_with_backend). Currently local-only is kept
    // as the default - true:
    // gtk_file_chooser_set_local_only(GTK_FILE_CHOOSER(m_widget), true);

    g_signal_connect (m_widget, "response",
        G_CALLBACK (gtk_filedialog_response_callback), this);

    g_signal_connect (m_widget, "selection-changed",
        G_CALLBACK (gtk_filedialog_selchanged_callback), this);

     // deal with extensions/filters
    SetWildcard(wildCard);

    wxString defaultFileNameWithExt = defaultFileName;
    if ( !wildCard.empty() && !defaultFileName.empty() &&
            !wxFileName(defaultFileName).HasExt() )
    {
        // append the default extension, if any, to the initial file name: GTK
        // won't do it for us by default (unlike e.g. MSW)
        const wxFileName fnWC(m_fc.GetCurrentWildCard());
        if ( fnWC.HasExt() )
        {
            // Notice that we shouldn't append the extension if it's a wildcard
            // because this is not useful: the user would need to change it to use
            // some fixed extension anyhow.
            const wxString& ext = fnWC.GetExt();
            if ( ext.find_first_of("?*") == wxString::npos )
                defaultFileNameWithExt << "." << ext;
        }
    }


    // if defaultDir is specified it should contain the directory and
    // defaultFileName should contain the default name of the file, however if
    // directory is not given, defaultFileName contains both
    wxFileName fn;
    if ( defaultDir.empty() )
        fn.Assign(defaultFileNameWithExt);
    else if ( !defaultFileNameWithExt.empty() )
        fn.Assign(defaultDir, defaultFileNameWithExt);
    else
        fn.AssignDir(defaultDir);

    // set the initial file name and/or directory
    fn.MakeAbsolute(); // GTK+ needs absolute path
    const wxString dir = fn.GetPath();
    if ( !dir.empty() )
    {
        const auto folder(wxGTK_CONV_FN(dir));
        gtk_file_chooser_set_current_folder(file_chooser, folder);
        if (m_fileChooserNative)
            gtk_file_chooser_set_current_folder(m_fileChooserNative, folder);
    }

    const wxString fname = fn.GetFullName();
    if ( style & wxFD_SAVE )
    {
        if ( !fname.empty() )
        {
            const auto curName(wxGTK_CONV_FN(fname));
            gtk_file_chooser_set_current_name(file_chooser, curName);
            if (m_fileChooserNative)
                gtk_file_chooser_set_current_name(m_fileChooserNative, curName);
        }

#if GTK_CHECK_VERSION(2,7,3)
        if ((style & wxFD_OVERWRITE_PROMPT) && wx_is_at_least_gtk2(8))
        {
            gtk_file_chooser_set_do_overwrite_confirmation(file_chooser, true);
            if (m_fileChooserNative)
                gtk_file_chooser_set_do_overwrite_confirmation(m_fileChooserNative, true);
        }
#endif
    }
    else // wxFD_OPEN
    {
        if ( !fname.empty() )
        {
            const auto filename(wxGTK_CONV_FN(fn.GetFullPath()));
            gtk_file_chooser_set_filename(file_chooser, filename);
            if (m_fileChooserNative)
                gtk_file_chooser_set_filename(m_fileChooserNative, filename);
        }
    }

    if ( style & wxFD_PREVIEW )
    {
        GtkWidget *previewImage = gtk_image_new();

        gtk_file_chooser_set_preview_widget(file_chooser, previewImage);
        g_signal_connect(m_widget, "update-preview",
                         G_CALLBACK(gtk_filedialog_update_preview_callback),
                         previewImage);
    }

    if (style & wxFD_SHOW_HIDDEN)
    {
        gtk_file_chooser_set_show_hidden(file_chooser, true);
        if (m_fileChooserNative)
            gtk_file_chooser_set_show_hidden(m_fileChooserNative, true);
    }

    return true;
}

wxFileDialog::~wxFileDialog()
{
    if (m_extraControl)
    {
        // get chooser to drop its reference right now, allowing wxWindow dtor
        // to verify that ref count drops to zero
        gtk_file_chooser_set_extra_widget(
            GTK_FILE_CHOOSER(m_widget), nullptr);
    }
    delete m_fcNative;
    if (m_fileChooserNative)
        g_object_unref(m_fileChooserNative);
}

void wxFileDialog::OnFakeOk(wxCommandEvent& WXUNUSED(event))
{
    // Update the current directory from here, accessing it later may not work
    // due to the strange way GtkFileChooser works.
    wxGtkString
        str(gtk_file_chooser_get_current_folder(GTK_FILE_CHOOSER(m_widget)));
    m_dir = wxString::FromUTF8(str);

    TransferDataFromExtraControl();

    EndDialog(wxID_OK);
}

void wxFileDialog::GTKDropNative()
{
    if (m_fileChooserNative)
    {
        delete m_fcNative;
        m_fcNative = nullptr;
        g_object_unref(m_fileChooserNative);
        m_fileChooserNative = nullptr;
    }
}

int wxFileDialog::ShowModal()
{
    WX_HOOK_MODAL_DIALOG();

    CreateExtraControl();

#if GTK_CHECK_VERSION(3,20,0)
    if (m_extraControl)
    {
        // GtkFileChooserNative does not support extra controls
        GTKDropNative();
    }
    if (m_fileChooserNative)
    {
        m_returnCode = 0;
        int res = gtk_native_dialog_run(GTK_NATIVE_DIALOG(m_fileChooserNative));
        if (res == GTK_RESPONSE_ACCEPT)
        {
            if (HasFlag(wxFD_CHANGE_DIR))
            {
                wxGtkString filename(gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(m_fileChooserNative)));
                wxGtkString dir(g_path_get_dirname(filename));
                if ( chdir(dir) != 0 )
                {
                    wxLogSysError(_("Changing current directory to \"%s\" failed"),
                                  wxString::FromUTF8(dir));
                }
            }
            m_returnCode = wxID_OK;
        }
        else if (m_returnCode == 0)
            m_returnCode = wxID_CANCEL;

        return m_returnCode;
    }
#endif

    return BaseType::ShowModal();
}

void wxFileDialog::EndModal(int retCode)
{
#if GTK_CHECK_VERSION(3,20,0)
    if (m_fileChooserNative)
    {
        m_returnCode = retCode;
        gtk_native_dialog_hide(GTK_NATIVE_DIALOG(m_fileChooserNative));
    }
    else
#endif
    {
        BaseType::EndModal(retCode);
    }
}

void wxFileDialog::DoSetSize(int WXUNUSED(x), int WXUNUSED(y),
                             int WXUNUSED(width), int WXUNUSED(height),
                             int WXUNUSED(sizeFlags))
{
}

void wxFileDialog::OnSize(wxSizeEvent&)
{
    // avoid calling DoLayout(), which will set the (wrong) size of
    // m_extraControl, its size is managed by GtkFileChooser
}

wxString wxFileDialog::GetPath() const
{
    wxCHECK_MSG( !HasFlag(wxFD_MULTIPLE), wxString(), "When using wxFD_MULTIPLE, must call GetPaths() instead" );
    return GetFileChooser().GetPath();
}

void wxFileDialog::GetFilenames(wxArrayString& files) const
{
    GetFileChooser().GetFilenames(files);
}

void wxFileDialog::GetPaths(wxArrayString& paths) const
{
    GetFileChooser().GetPaths(paths);
}

wxString wxFileDialog::GetDirectory() const
{
    return GetFileChooser().GetDirectory();
}

void wxFileDialog::SetMessage(const wxString& message)
{
    m_message = message;
    SetTitle(message);
}

void wxFileDialog::SetPath(const wxString& path)
{
    BaseType::SetPath(path);

    // Don't do anything if no path is specified, in particular don't set the
    // path to m_dir below as this would result in opening the dialog in the
    // parent directory of this one instead of m_dir itself.
    if ( path.empty() )
        return;

    // we need an absolute path for GTK native chooser so ensure that we have
    // it: use the initial directory if it was set or just CWD otherwise (this
    // is the default behaviour if m_dir is empty)
    const wxString str(wxFileName(path).GetAbsolutePath(m_dir));
    m_fc.SetPath(str);
    if (m_fcNative)
        m_fcNative->SetPath(str);
}

void wxFileDialog::SetDirectory(const wxString& dir)
{
    BaseType::SetDirectory(dir);

    m_fc.SetDirectory(dir);
    if (m_fcNative)
        m_fcNative->SetDirectory(dir);
}

void wxFileDialog::SetFilename(const wxString& name)
{
    BaseType::SetFilename(name);

    if (HasFdFlag(wxFD_SAVE))
    {
        const auto curName(name.utf8_str());
        gtk_file_chooser_set_current_name(GTK_FILE_CHOOSER(m_widget), curName);
        if (m_fileChooserNative)
            gtk_file_chooser_set_current_name(m_fileChooserNative, curName);
    }

    else
    {
        wxString path( GetDirectory() );
        if (path.empty())
        {
            // SetPath() fires an assert if fed other than filepaths
            return;
        }
        SetPath(wxFileName(path, name).GetFullPath());
    }
}

wxString wxFileDialog::GetFilename() const
{
    wxCHECK_MSG( !HasFlag(wxFD_MULTIPLE), wxString(), "When using wxFD_MULTIPLE, must call GetFilenames() instead" );

    wxString currentFilename(GetFileChooser().GetFilename());
    if (currentFilename.empty())
    {
        // m_fc.GetFilename() will return empty until the dialog has been shown
        // in which case use any previously provided value
        currentFilename = m_fileName;
    }
    return currentFilename;
}

void wxFileDialog::SetWildcard(const wxString& wildCard)
{
    BaseType::SetWildcard(wildCard);
    m_fc.SetWildcard( GetWildcard() );
    if (m_fcNative)
        m_fcNative->SetWildcard(wildCard);
}

void wxFileDialog::SetFilterIndex(int filterIndex)
{
    m_fc.SetFilterIndex( filterIndex);
    if (m_fcNative)
        m_fcNative->SetFilterIndex(filterIndex);
}

int wxFileDialog::GetFilterIndex() const
{
    return GetFileChooser().GetFilterIndex();
}

void wxFileDialog::GTKSelectionChanged(const wxString& filename)
{
    m_currentlySelectedFilename = filename;

    UpdateExtraControlUI();
}

bool wxFileDialog::AddShortcut(const wxString& directory, int WXUNUSED(flags))
{
    wxGtkError error;

    const auto folder(directory.utf8_str());
    if ( !gtk_file_chooser_add_shortcut_folder(GTK_FILE_CHOOSER(m_widget),
                                               folder,
                                               error.Out()) )
    {
        wxLogDebug("Failed to add shortcut \"%s\": %s",
                   directory, error.GetMessage());

        return false;
    }
    if (m_fileChooserNative)
    {
        gtk_file_chooser_add_shortcut_folder(
            GTK_FILE_CHOOSER(m_fileChooserNative), folder, nullptr);
    }

    return true;
}

#endif // wxUSE_FILEDLG
