/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2023 KiCad Developers, see AUTHORS.TXT for contributors.
 *
 * 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 3
 * of the License, or (at your option) any later version.
 *
 * This program 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 this program; if not, you may find one here:
 * http://www.gnu.org/licenses/gpl-3.0.html
 * or you may search the http://www.gnu.org website for the version 3 license,
 * or you may write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */


#include "dialog_git_switch.h"

#include <wx/button.h>
#include <wx/checkbox.h>
#include <wx/listctrl.h>
#include <wx/event.h>
#include <wx/sizer.h>
#include <wx/timer.h>
#include <wx/stattext.h>
#include <wx/textctrl.h>

#include <git2.h>


DIALOG_GIT_SWITCH::DIALOG_GIT_SWITCH( wxWindow* aParent, git_repository* aRepository ) :
        DIALOG_SHIM( aParent, wxID_ANY, _( "Git Branch Switch" ), wxDefaultPosition, wxDefaultSize,
                     wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER ),
        m_timer( this ), m_repository( aRepository )
{
    wxBoxSizer* sizer = new wxBoxSizer( wxVERTICAL );

    // Add explanation text
    wxStaticText* explanationText =
            new wxStaticText( this, wxID_ANY, _( "Select or enter a branch name:" ) );
    sizer->Add( explanationText, 0, wxALL, 10 );

    // Add branch list with three columns
    m_branchList = new wxListView( this, wxID_ANY, wxDefaultPosition, wxDefaultSize,
                                   wxLC_REPORT | wxLC_SINGLE_SEL );
    m_branchList->InsertColumn( 0, _( "Branch" ) );
    m_branchList->InsertColumn( 1, _( "Last Commit" ) );
    m_branchList->InsertColumn( 2, _( "Last Updated" ) );
    sizer->Add( m_branchList, 1, wxALL | wxEXPAND, 10 );

    // Add branch name text box
    m_branchNameText = new wxTextCtrl( this, wxID_ANY, wxEmptyString, wxDefaultPosition,
                                       wxDefaultSize, wxTE_PROCESS_ENTER );
    sizer->Add( m_branchNameText, 0, wxALL | wxEXPAND, 10 );

    // Add buttons
    wxStdDialogButtonSizer* buttonSizer = new wxStdDialogButtonSizer();
    m_switchButton = new wxButton( this, wxID_OK, _( "Switch" ) );
    buttonSizer->AddButton( m_switchButton );
    m_switchButton->Disable();
    wxButton* cancelButton = new wxButton( this, wxID_CANCEL, _( "Cancel" ) );
    buttonSizer->AddButton( cancelButton );
    buttonSizer->Realize();
    sizer->Add( buttonSizer, 0, wxALIGN_RIGHT | wxALL, 10 );

    // Bind events
    Bind( wxEVT_LIST_ITEM_SELECTED, &DIALOG_GIT_SWITCH::OnBranchListSelection, this, m_branchList->GetId() );
    Bind( wxEVT_LIST_ITEM_ACTIVATED, &DIALOG_GIT_SWITCH::OnBranchListDClick, this, m_branchList->GetId() );
    Bind( wxEVT_BUTTON, &DIALOG_GIT_SWITCH::OnSwitchButton, this, m_switchButton->GetId() );
    Bind( wxEVT_BUTTON, &DIALOG_GIT_SWITCH::OnCancelButton, this, cancelButton->GetId() );
    Bind( wxEVT_TEXT, &DIALOG_GIT_SWITCH::OnTextChanged, this, m_branchNameText->GetId() );
    Bind( wxEVT_TIMER, &DIALOG_GIT_SWITCH::OnTimer, this, m_timer.GetId() );

    // Populate branch list
    PopulateBranchList();

    // Set sizer for the dialog
    SetSizerAndFit( sizer );

    finishDialogSettings();

    m_existingBranch = false;
}

DIALOG_GIT_SWITCH::~DIALOG_GIT_SWITCH()
{
    StopTimer();
    Unbind( wxEVT_TIMER, &DIALOG_GIT_SWITCH::OnTimer, this, m_timer.GetId() );
}

void DIALOG_GIT_SWITCH::PopulateBranchList()
{
    m_branchList->DeleteAllItems();

    // Get the branches
    GetBranches();

    // Populate the list
    for( auto& [ name, data ] : m_branches )
    {
        wxDateTime lastUpdated( data.lastUpdated );
        wxString   lastUpdatedString = lastUpdated.Format();

        long itemIndex = m_branchList->InsertItem( m_branchList->GetItemCount(), name );
        m_branchList->SetItem( itemIndex, 1, data.commitString );
        m_branchList->SetItem( itemIndex, 2, lastUpdatedString );
    }

    m_branchList->SetColumnWidth( 0, wxLIST_AUTOSIZE );
    m_branchList->SetColumnWidth( 1, wxLIST_AUTOSIZE );
    m_branchList->SetColumnWidth( 2, wxLIST_AUTOSIZE );

}


void DIALOG_GIT_SWITCH::OnBranchListDClick( wxListEvent& aEvent )
{
    int selection = aEvent.GetIndex();

    if( selection != wxNOT_FOUND )
    {
        wxString branchName = m_branchList->GetItemText( selection );
        m_branchNameText->SetValue( branchName );

        if( branchName != m_currentBranch )
            EndModal( wxID_OK );
    }
}


void DIALOG_GIT_SWITCH::OnBranchListSelection( wxListEvent& aEvent )
{
    int selection = aEvent.GetIndex();

    if( selection != wxNOT_FOUND )
    {
        wxString branchName = m_branchList->GetItemText( selection );
        m_branchNameText->SetValue( branchName );
        m_switchButton->SetLabel( _( "Switch" ) );
        m_switchButton->Enable( branchName != m_currentBranch );
    }
    else
    {
        // Deselect all elements in the list
        for( int ii = 0; ii < m_branchList->GetItemCount(); ++ii )
            m_branchList->SetItemState( ii, 0, 0 );
    }
}

void DIALOG_GIT_SWITCH::OnSwitchButton(wxCommandEvent& aEvent)
{
    wxString branchName = m_branchNameText->GetValue();

    // Check if the branch name exists
    bool branchExists = m_branches.count(branchName);

    if (branchExists)
    {
        EndModal(wxID_OK); // Return Switch code
    }
    else
    {
        EndModal(wxID_ADD); // Return Add code
    }
}


void DIALOG_GIT_SWITCH::OnCancelButton(wxCommandEvent& aEvent)
{
    EndModal(wxID_CANCEL); // Return Cancel code
}


wxString DIALOG_GIT_SWITCH::GetBranchName() const
{
    return m_branchNameText->GetValue();
}


void DIALOG_GIT_SWITCH::StartTimer()
{
    m_timer.Start( 500, true );
}


void DIALOG_GIT_SWITCH::StopTimer()
{
    m_timer.Stop();
}


void DIALOG_GIT_SWITCH::OnTimer( wxTimerEvent& aEvt )
{
    wxString branchName = m_branchNameText->GetValue();

    if( branchName == m_lastEnteredText )
        return;

    m_lastEnteredText = branchName;

    // Check if the branch name exists
    bool branchExists = m_branches.count( branchName );

    if( branchExists )
    {
        m_switchButton->SetLabel( _( "Switch" ) );
        m_switchButton->Enable( branchName != m_currentBranch );
    }
    else
    {
        m_switchButton->SetLabel( _( "Add" ) );
        m_switchButton->Enable();
    }
}


void DIALOG_GIT_SWITCH::OnTextChanged( wxCommandEvent& aEvt )
{
    StartTimer();
}


void DIALOG_GIT_SWITCH::GetBranches()
{
    // Clear the branch list
    m_branches.clear();

    git_branch_iterator* branchIterator = nullptr;
    git_branch_t         branchType;

    // Get Current Branch
    git_reference* currentBranchReference = nullptr;
    git_repository_head( &currentBranchReference, m_repository );

    // Get the current branch name
    if( currentBranchReference )
    {
        m_currentBranch = git_reference_shorthand( currentBranchReference );
        git_reference_free( currentBranchReference );
    }

    // Initialize branch iterator
    git_branch_iterator_new( &branchIterator, m_repository, GIT_BRANCH_ALL );

    // Iterate over local branches
    git_reference* branchReference = nullptr;
    while( git_branch_next( &branchReference, &branchType, branchIterator ) == 0 )
    {
        // Get the branch OID
        const git_oid* branchOid = git_reference_target( branchReference );

        // Skip this branch if it doesn't have an OID
        if( !branchOid )
        {
            git_reference_free( branchReference );
            continue;
        }

        git_commit* commit = nullptr;

        if( git_commit_lookup( &commit, m_repository, branchOid ) )
        {
            // Skip this branch if it doesn't have a commit
            git_reference_free( branchReference );
            continue;
        }

        // Retrieve commit details
        BranchData branchData;
        branchData.commitString = git_commit_message( commit );
        branchData.lastUpdated = static_cast<time_t>( git_commit_time( commit ) );
        branchData.isRemote = branchType == GIT_BRANCH_REMOTE;

        m_branches[git_reference_shorthand( branchReference )] = branchData;

        git_commit_free( commit );
        git_reference_free( branchReference );
    }

    git_branch_iterator_free( branchIterator );
}