#include <wx/artprov.h>

#include <wx/button.h>

#include <wx/notebook.h>

#include <wx/textctrl.h>

#include <wx/regex.h>

#include <globals.h>

#include <logmanager.h>

#include <manager.h>

#include "ShellCtrlBase.h"

// The global instance of the shell registry

ShellRegistry& GlobalShellRegistry()

{

static ShellRegistry* theRegistry = new ShellRegistry();

return *theRegistry;

}

// Unique IDs for Timer and Shell Manager messages

int ID_SHELLPOLLTIMER=wxNewId();

int ID_SHELLMGR=wxNewId();

bool ShellRegistry::Register(const wxString &name, fnCreate create, fnFree free) //register/deregister are called by the plugin registrant instance

{

Manager::Get()->GetLogManager()->Log(wxString::Format(_("Python Interpreter Plugin: Registering shell type %s"),name.c_str()));

std::map<wxString, ShellRegInfo>::iterator it;

if(m_reginfo.find(name)!=m_reginfo.end())

return false;

ShellRegInfo sri;

sri.create=create;

sri.free=free;

m_reginfo[name]=sri;

return true;

}

bool ShellRegistry::Deregister(const wxString &name)

{

std::map<wxString, ShellRegInfo>::iterator it

=m_reginfo.find(name);

if(it==m_reginfo.end())

return false;

m_reginfo.erase(it);

return true;

}

ShellCtrlBase *ShellRegistry::CreateControl(const wxString &type,wxWindow* parent, int id, const wxString &windowname, ShellManager *shellmgr)

{

std::map<wxString, ShellRegInfo>::iterator it

=m_reginfo.find(type);

if(it==m_reginfo.end())

return NULL;

return it->second.create(parent, id, windowname, shellmgr);

}

void ShellRegistry::FreeControl(ShellCtrlBase */*sh*/) //TODO: Don't think this is necessary?

{

// std::map<wxString, ShellRegInfo>::iterator it

// =m_reginfo.find(type);

// if(it!=m_reginfo.end())

// it.second->free(); //TODO: Can't compile

}

//IMPLEMENT_DYNAMIC_CLASS(ShellCtrlBase, wxPanel)

ShellCtrlBase::ShellCtrlBase(wxWindow* parent, int id, const wxString &name, ShellManager *shellmgr)

: wxPanel(parent, id)

{

m_parent=parent;

m_name=name;

m_id=id;

m_shellmgr=shellmgr;

}

DEFINE_EVENT_TYPE(wxEVT_SHELL_ADD_CLICKED)

////////////////////////////////////// ShellManager /////////////////////////////////////////////

BEGIN_EVENT_TABLE(ShellManager, wxPanel)

EVT_CHAR(ShellManager::OnUserInput)

EVT_TIMER(ID_SHELLPOLLTIMER, ShellManager::OnPollandSyncOutput)

EVT_AUINOTEBOOK_PAGE_CLOSE(ID_SHELLMGR, ShellManager::OnPageClosing)

EVT_AUINOTEBOOK_PAGE_CHANGING(ID_SHELLMGR, ShellManager::OnPageChanging)

END_EVENT_TABLE()

void ShellManager::OnPageClosing(wxAuiNotebookEvent& event)

{

ShellCtrlBase* sh = GetPage(event.GetSelection());

// LOGSTREAM << wxString::Format(_T("OnPageClosing(): ed=%p, title=%s\n"), eb, eb ? eb->GetTitle().c_str() : _T(""));

if (!QueryClose(sh))

event.Veto();

m_nb->SetSelection(0);

// event.Skip(); // allow others to process it too

}

void ShellManager::OnPageChanging(wxAuiNotebookEvent& event)

{

if(event.GetSelection() == int(m_nb->GetPageCount()-1))

{

wxCommandEvent pe(wxEVT_SHELL_ADD_CLICKED,0);

AddPendingEvent(pe);

event.Veto();

return;

}

event.Skip();

}

bool ShellManager::QueryClose(ShellCtrlBase* sh)

{

if(!sh)

return true;

if(!sh->IsDead())

{

wxString msg(_("Process \"")+sh->GetName()+_("\" is still running...\nDo you want to kill it?"));

switch (cbMessageBox(msg, _("Kill process?"), wxICON_QUESTION | wxYES_NO))

{

case wxID_YES:

sh->KillProcess();

return false;

case wxID_NO:

return false;

default:

break;

}

}

return true;

}

long ShellManager::LaunchProcess(const wxString &processcmd, const wxString &name, const wxString &type, const wxArrayString &options)

{

int id=wxNewId();

ShellCtrlBase *shell=GlobalShellRegistry().CreateControl(type,this,id,name,this);

if(!shell)

{

cbMessageBox(wxString::Format(_("Console type %s not found in registry."),type.c_str()));

return -1;

}

long procid=shell->LaunchProcess(processcmd,options);

if(procid>0)

{

if(!m_synctimer.IsRunning())

m_synctimer.Start(100);

}

else

{

cbMessageBox(_("process launch failed."));

delete shell; //TODO: GlobalShellRegistry.FreeControl() ???

return -1;

}

m_nb->InsertPage(m_nb->GetPageCount()-1,shell,name);

m_nb->SetSelection(m_nb->GetPageCount()-2);

// shell->Show();

return procid;

}

ShellCtrlBase *ShellManager::GetPage(size_t i)

{

return (ShellCtrlBase*)m_nb->GetPage(i);

}

ShellCtrlBase *ShellManager::GetPage(const wxString &name)

{

for(unsigned int i=0;i<m_nb->GetPageCount()-1;i++)

{

ShellCtrlBase *sh=GetPage(i);

if(name==sh->GetName())

return sh;

}

return NULL;

}

// Forecefully kill the process

void ShellManager::KillProcess(int /*id*/)

{

}

void ShellManager::KillWindow(int /*id*/)

{

}

void ShellManager::RemoveDeadPages()

{

unsigned int i=0;

while(i<m_nb->GetPageCount()-1)

{

ShellCtrlBase *shell=GetPage(i);

if(shell->IsDead())

m_nb->DeletePage(i);

else

i++;

}

}

size_t ShellManager::GetTermNum(ShellCtrlBase *term)

{

for(unsigned int i=0;i<m_nb->GetPageCount()-1;i++)

{

ShellCtrlBase *shell=GetPage(i);

if(shell==term)

return i;

}

return m_nb->GetPageCount();

}

int ShellManager::NumAlive()

{

int count=0;

for(unsigned int i=0;i<m_nb->GetPageCount()-1;i++)

count+=!GetPage(i)->IsDead();

return count;

}

void ShellManager::OnShellTerminate(ShellCtrlBase *term)

{

size_t i=GetTermNum(term);

if(i<m_nb->GetPageCount()-1)

m_nb->SetPageText(i,_("[DONE]")+m_nb->GetPageText(i));

if(NumAlive()==0)

m_synctimer.Stop();

}

void ShellManager::OnPollandSyncOutput(wxTimerEvent& /*te*/)

{

for(unsigned int i=0;i<m_nb->GetPageCount()-1;i++)

{

GetPage(i)->SyncOutput();

}

}

void ShellManager::OnUserInput(wxKeyEvent& /*ke*/)

{ //TODO: This shouldn't be necessary as individual pages will have the focus

// ShellCtrlBase *sh=(ShellCtrlBase*)m_nb->GetCurrentPage();

// sh->OnUserInput(ke);

}

ShellManager::ShellManager(wxWindow* parent)

: wxPanel(parent, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTAB_TRAVERSAL/* | wxCLIP_CHILDREN*/)

{

m_synctimer.SetOwner(this, ID_SHELLPOLLTIMER);

wxBoxSizer* bs = new wxBoxSizer(wxVERTICAL);

m_nb = new wxAuiNotebook(this, ID_SHELLMGR, wxDefaultPosition, wxDefaultSize, wxAUI_NB_SCROLL_BUTTONS|wxAUI_NB_CLOSE_ON_ACTIVE_TAB);

bs->Add(m_nb, 1, wxEXPAND | wxALL);

wxBitmap bm = wxArtProvider::GetBitmap(wxART_NEW);

wxButton *wnd = new wxButton(this, wxID_ANY);

m_nb->AddPage(wnd,_(""),false,bm);

SetAutoLayout(TRUE);

SetSizer(bs);

}

ShellManager::~ShellManager()

{

//dtor

//All of the subwindows owned by this panel will be destroyed automatically

}