//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
//
// copyright            : (C) 2014 Eran Ifrah
// file name            : codelitediff.cpp
//
// -------------------------------------------------------------------------
// A
//              _____           _      _     _ _
//             /  __ \         | |    | |   (_) |
//             | /  \/ ___   __| | ___| |    _| |_ ___
//             | |    / _ \ / _  |/ _ \ |   | | __/ _ )
//             | \__/\ (_) | (_| |  __/ |___| | ||  __/
//              \____/\___/ \__,_|\___\_____/_|\__\___|
//
//                                                  F i l e
//
//    This program 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 2 of the License, or
//    (at your option) any later version.
//
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////

#include "codelitediff.h"

#include "Diff/DiffFoldersFrame.h"
#include "Diff/DiffSideBySidePanel.h"
#include "Diff/clDiffFrame.h"
#include "Keyboard/clKeyboardManager.h"
#include "NewFileComparison.h"
#include "event_notifier.h"
#include "file_logger.h"
#include "macros.h"

#include <wx/ffile.h>
#include <wx/menu.h>
#include <wx/xrc/xmlres.h>

namespace
{
/// Return the window associated with the event or the active window
wxWindow* GetWindowFromEvent(wxEvent& event)
{
    // Check if the menu contains the window to close, if not close the active tab
    auto menu = dynamic_cast<wxMenu*>(event.GetEventObject());
    while (menu && menu->GetParent()) {
        menu = menu->GetParent();
    }
    auto win = clGetManager()->GetMainNotebook()->GetCurrentPage();
    if (menu && menu->GetClientData()) {
        win = reinterpret_cast<wxWindow*>(menu->GetClientData());
    }
    return win;
}

/// Return the editor associated with the event or the active editor
IEditor* GetEditorFromEvent(wxEvent& event)
{
    auto win = GetWindowFromEvent(event);
    CHECK_PTR_RET_NULL(win);

    return clGetManager()->GetEditorFromWindow(win);
}
} // namespace

// Define the plugin entry point
CL_PLUGIN_API IPlugin* CreatePlugin(IManager* manager) { return new CodeLiteDiff(manager); }

CL_PLUGIN_API PluginInfo* GetPluginInfo()
{
    static PluginInfo info;
    info.SetAuthor("Eran Ifrah");
    info.SetName("Diff Plugin");
    info.SetDescription(_("CodeLite Diff Plugin"));
    info.SetVersion("v1.0");
    return &info;
}

CL_PLUGIN_API int GetPluginInterfaceVersion() { return PLUGIN_INTERFACE_VERSION; }

CodeLiteDiff::CodeLiteDiff(IManager* manager)
    : IPlugin(manager)
{
    m_longName = _("CodeLite Diff Plugin");
    m_shortName = "Diff Plugin";

    Bind(wxEVT_MENU, &CodeLiteDiff::OnDiff, this, XRCID("diff_compare_with"));
    EventNotifier::Get()->Bind(wxEVT_CONTEXT_MENU_TAB_LABEL, &CodeLiteDiff::OnTabContextMenu, this);
    wxTheApp->Bind(wxEVT_MENU, &CodeLiteDiff::OnNewDiff, this, XRCID("diff_new_comparison"));
    wxTheApp->Bind(wxEVT_MENU, &CodeLiteDiff::OnNewDiffFolder, this, XRCID("diff_new_folder"));

    clKeyboardManager::Get()->AddAccelerator(_("Diff Tool"),
                                             { { "diff_new_comparison", _("New File Comparison"), "Ctrl-Shift-C" },
                                               { "diff_new_folder", _("New Folder Comparison"), "Ctrl-Alt-F" } });
}

CodeLiteDiff::~CodeLiteDiff() {}

void CodeLiteDiff::CreateToolBar(clToolBarGeneric* toolbar) { wxUnusedVar(toolbar); }

void CodeLiteDiff::CreatePluginMenu(wxMenu* pluginsMenu)
{
    wxMenu* menu = new wxMenu;
    menu->Append(XRCID("diff_new_comparison"), _("New File Comparison"), _("Start new diff"));
    menu->Append(XRCID("diff_new_folder"), _("New Folder Comparison"), _("Start new folders diff"));
    pluginsMenu->Append(wxID_ANY, _("Diff Tool"), menu);
}

void CodeLiteDiff::UnPlug()
{
    wxTheApp->Unbind(wxEVT_MENU, &CodeLiteDiff::OnNewDiff, this, XRCID("diff_new_comparison"));
    EventNotifier::Get()->Unbind(wxEVT_CONTEXT_MENU_TAB_LABEL, &CodeLiteDiff::OnTabContextMenu, this);
    Unbind(wxEVT_MENU, &CodeLiteDiff::OnDiff, this, XRCID("diff_compare_with"));
}

void CodeLiteDiff::OnNewDiff(wxCommandEvent& e)
{
    clDiffFrame* diff = new clDiffFrame(EventNotifier::Get()->TopFrame());
    diff->Show();
}

void CodeLiteDiff::OnTabContextMenu(clContextMenuEvent& event)
{
    event.Skip();
    DoClear();

    auto activeWin = dynamic_cast<wxWindow*>(event.GetEditor());
    CHECK_PTR_RET(activeWin);

    auto activeEditor = clGetManager()->GetEditorFromWindow(activeWin);
    CHECK_PTR_RET(activeEditor);

    m_leftFile = activeEditor->GetFileName();

    // Edit the context menu
    wxMenuItem* mi = new wxMenuItem(event.GetMenu(), XRCID("diff_compare_with"), _("Compare with..."), "");
    mi->SetBitmap(m_mgr->GetStdIcons()->LoadBitmap("diff"));
    event.GetMenu()->AppendSeparator();
    event.GetMenu()->Append(mi);
    event.GetMenu()->Bind(wxEVT_MENU, &CodeLiteDiff::OnDiff, this, XRCID("diff_compare_with"));
}

void CodeLiteDiff::DoClear() { m_leftFile.Clear(); }

void CodeLiteDiff::OnDiff(wxCommandEvent& event)
{
    bool tempfile(false);
    NewFileComparison dlg(EventNotifier::Get()->TopFrame(), m_leftFile);
    if (dlg.ShowModal() == wxID_OK) {
        if (m_leftFile.GetName().StartsWith(_("Untitled"))) {
            tempfile = true;
            auto leftEditor = GetEditorFromEvent(event);
            m_leftFile = SaveEditorToTmpfile(leftEditor);
            if (!m_leftFile.IsOk()) {
                clDEBUG() << "CodeLiteDiff::OnDiff: call to SaveEditorToTmpfile() failed for m_leftFile" << endl;
                return;
            }
        }
        wxString secondFile = dlg.GetTextCtrlFileName()->GetValue();
        if (secondFile.StartsWith(_("Untitled"))) {
            tempfile = true;
            IEditor* editor = m_mgr->FindEditor(secondFile);
            if (!editor) {
                clDEBUG() << "CodeLiteDiff::OnDiff: call to FindEditor() failed" << endl;
                return;
            }
            wxFileName rightFn = SaveEditorToTmpfile(editor);
            if (!rightFn.IsOk()) {
                clDEBUG() << "CodeLiteDiff::OnDiff: call to SaveEditorToTmpfile() failed for secondFile" << endl;
                return;
            }
            secondFile = rightFn.GetFullPath();
        }

        // Check that we're not trying to diff an editor against itself
        // If we are and it's been edited, diff against the unaltered version
        if (m_leftFile.GetFullPath() == secondFile) {
            IEditor* editor = m_mgr->FindEditor(secondFile);
            if (editor && editor->IsEditorModified()) {
                wxFileName rightFn = SaveEditorToTmpfile(editor);
                if (!rightFn.IsOk()) {
                    clDEBUG() << "CodeLiteDiff::OnDiff: call to SaveEditorToTmpfile() failed for secondFile" << endl;
                    return;
                }
                secondFile = rightFn.GetFullPath();

            } else {
                clDEBUG() << "CodeLiteDiff::OnDiff: trying to diff an editor against itself" << endl;
                return;
            }
        }

        clDiffFrame* diffView = new clDiffFrame(EventNotifier::Get()->TopFrame(), m_leftFile, secondFile, tempfile);
        diffView->Show();
    }
}

wxFileName CodeLiteDiff::SaveEditorToTmpfile(IEditor* editor) const
{
    wxString content = editor->GetEditorText();
    if (content.empty()) {
        return wxFileName(); // Nothing to diff
    }

    wxString tpath(wxFileName::GetTempDir());
    tpath << wxFileName::GetPathSeparator() << "CLdiff" << wxFileName::GetPathSeparator();
    wxFileName::Mkdir(tpath, wxS_DIR_DEFAULT, wxPATH_MKDIR_FULL);
    wxFileName tmpFile(wxFileName::CreateTempFileName(tpath + editor->GetFileName().GetName()));
    if (!tmpFile.IsOk()) {
        return wxFileName();
    }

    tmpFile.SetExt(editor->GetFileName().GetExt());
    wxFFile fp(tmpFile.GetFullPath(), "w+b");
    if (fp.IsOpened()) {
        fp.Write(content);
        fp.Close();
    } else {
        return wxFileName();
    }

    return tmpFile;
}

void CodeLiteDiff::OnNewDiffFolder(wxCommandEvent& e)
{
    DiffFoldersFrame* dlg = new DiffFoldersFrame(EventNotifier::Get()->TopFrame());
    dlg->Show();
}
