void DbViewerPanel::OnPopupClick(wxCommandEvent& evt)
{
if(!m_selectedID.IsOk()) return;
try {
if(evt.GetId() == XRCID("IDR_DBVIEWER_ADD_DATABASE")) {
if(m_pEditedConnection) {
// TODO:LANG:
wxString dbName = wxGetTextFromUser(_("Database name"), _("Add database"));
if(!dbName.IsEmpty()) {
DatabaseLayerPtr pDbLayer = m_pEditedConnection->GetDbAdapter()->GetDatabaseLayer(wxT(""));
wxString sql = m_pEditedConnection->GetDbAdapter()->GetCreateDatabaseSql(dbName);
if(!sql.empty()) {
pDbLayer->RunQuery(sql);
pDbLayer->Close();
// TODO:LANG:
wxMessageBox(_("Database created successfully"));
RefreshDbView();
} else {
// TODO:LANG:
wxMessageDialog dlg(
this, _("Can't create new db in this database engine!"), _("Error"), wxOK | wxICON_ERROR);
dlg.ShowModal();
}
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_DROP_DATABASE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Database* pDb = (Database*)wxDynamicCast(data->GetData(), Database);
if(pDb) {
wxString dropSQL = pDb->GetDbAdapter()->GetDropDatabaseSql(pDb);
if(!dropSQL.IsEmpty()) {
// TODO:LANG:
wxMessageDialog dlg(this,
wxString::Format(_("Remove database '%s'?"), pDb->GetName().c_str()),
_("Drop database"),
wxYES_NO);
if(dlg.ShowModal() == wxID_YES) {
DatabaseLayerPtr pDbLayer = pDb->GetDbAdapter()->GetDatabaseLayer(wxT(""));
pDbLayer->RunQuery(dropSQL);
pDbLayer->Close();
// TODO:LANG:
wxMessageBox(_("Database dropped successfully"));
RefreshDbView();
}
}
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_ERD_TABLE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Table* pTab = (Table*)wxDynamicCast(data->GetData(), Table);
if(pTab) {
wxString pagename;
pagename = CreatePanelName(pTab, DbViewerPanel::Erd);
ErdPanel* erdpanel =
new ErdPanel(m_pNotebook, pTab->GetDbAdapter()->Clone(), m_pConnections, (Table*)pTab->Clone());
AddEditorPage(erdpanel, pagename);
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_ERD_DB")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Database* pDb = (Database*)wxDynamicCast(data->GetData(), Database);
if(pDb) {
wxString pagename;
pagename = CreatePanelName(pDb, DbViewerPanel::Erd);
ErdPanel* erdpanel = new ErdPanel(
m_pNotebook, pDb->GetDbAdapter()->Clone(), m_pConnections, (Database*)pDb->Clone());
AddEditorPage(erdpanel, pagename);
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_CLASS_DB")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Database* pDb = (Database*)wxDynamicCast(data->GetData(), Database);
if(pDb) {
pDb = (Database*)pDb->Clone();
// NOTE: the refresh functions must be here for propper code generation (they translate views into
// tables)
pDb->RefreshChildren(true);
pDb->RefreshChildrenDetails();
ClassGenerateDialog dlg(m_mgr->GetTheApp()->GetTopWindow(), pDb->GetDbAdapter(), pDb, m_mgr);
dlg.ShowModal();
delete pDb;
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_CLASS_TABLE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Table* pTab = (Table*)wxDynamicCast(data->GetData(), Table);
if(pTab) {
ClassGenerateDialog dlg(
m_mgr->GetTheApp()->GetTopWindow(), pTab->GetDbAdapter(), (Table*)pTab->Clone(), m_mgr);
dlg.ShowModal();
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_DROP_TABLE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Table* pTab = (Table*)wxDynamicCast(data->GetData(), Table);
if(pTab) {
wxMessageDialog dlg(this,
wxString::Format(_("Remove table '%s'?"), pTab->GetName().c_str()),
_("Drop table"),
wxYES_NO);
if(dlg.ShowModal() == wxID_YES) {
DatabaseLayerPtr pDbLayer = pTab->GetDbAdapter()->GetDatabaseLayer(pTab->GetParentName());
pDbLayer->RunQuery(pTab->GetDbAdapter()->GetDropTableSql(pTab));
pDbLayer->Close();
wxMessageBox(_("Table dropped successfully"));
RefreshDbView();
}
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_DROP_VIEW")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
View* pView = (View*)wxDynamicCast(data->GetData(), View);
if(pView) {
wxMessageDialog dlg(this,
wxString::Format(_("Remove view '%s'?"), pView->GetName().c_str()),
_("Drop view"),
wxYES_NO);
if(dlg.ShowModal() == wxID_YES) {
DatabaseLayerPtr pDbLayer = pView->GetDbAdapter()->GetDatabaseLayer(pView->GetParentName());
pDbLayer->RunQuery(pView->GetDbAdapter()->GetDropViewSql(pView));
pDbLayer->Close();
wxMessageBox(_("View dropped successfully"));
RefreshDbView();
}
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_SQL_TABLE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Table* pTab = (Table*)wxDynamicCast(data->GetData(), Table);
if(pTab) {
#ifdef __WXMSW__
clWindowUpdateLocker locker(m_mgr->GetEditorPaneNotebook());
#endif
wxString pagename = CreatePanelName(pTab, DbViewerPanel::Sql);
if(!DoSelectPage(pagename)) {
SQLCommandPanel* sqlpage = new SQLCommandPanel(
m_pNotebook, pTab->GetDbAdapter()->Clone(), pTab->GetParentName(), pTab->GetName());
AddEditorPage(sqlpage, pagename);
}
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_SQL_VIEW")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
View* pView = (View*)wxDynamicCast(data->GetData(), View);
if(pView) {
#ifdef __WXMSW__
clWindowUpdateLocker locker(m_mgr->GetEditorPaneNotebook());
#endif
wxString pagename = CreatePanelName(pView, DbViewerPanel::Sql);
if(!DoSelectPage(pagename)) {
SQLCommandPanel* sqlpage = new SQLCommandPanel(
m_pNotebook, pView->GetDbAdapter()->Clone(), pView->GetParentName(), pView->GetName());
AddEditorPage(sqlpage, pagename);
}
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_SQL_DATABASE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Database* pDb = (Database*)wxDynamicCast(data->GetData(), Database);
if(pDb) {
wxString pagename = CreatePanelName(pDb, DbViewerPanel::Sql);
if(!DoSelectPage(pagename)) {
SQLCommandPanel* sqlpage =
new SQLCommandPanel(m_pNotebook, pDb->GetDbAdapter()->Clone(), pDb->GetName(), wxT(""));
#ifndef __WXMSW__
sqlpage->Show();
#endif
AddEditorPage(sqlpage, pagename);
}
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_IMPORT_DATABASE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Database* pDb = (Database*)wxDynamicCast(data->GetData(), Database);
if(pDb) {
// TODO:LANG:
wxFileDialog dlg(this,
_("Import database from SQL file ..."),
wxGetCwd(),
wxT(""),
wxT("SQL Files (*.sql)|*.sql"),
wxFD_OPEN | wxFD_FILE_MUST_EXIST);
if(dlg.ShowModal() == wxID_OK) {
ImportDb(dlg.GetPath(), pDb);
}
}
}
RefreshDbView();
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_DUMP_DATABASE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Database* pDb = (Database*)wxDynamicCast(data->GetData(), Database);
if(pDb) {
// TODO:LANG:
wxFileDialog dlg(this,
_("Dump data into file ..."),
wxT(""),
pDb->GetName() + wxT(".sql"),
wxT("SQL files (*.sql)|*.sql"),
wxFD_SAVE | wxFD_OVERWRITE_PROMPT);
if(dlg.ShowModal() == wxID_OK) {
DumpClass* dump = new DumpClass(pDb->GetDbAdapter(), pDb, dlg.GetPath());
dump->DumpData();
wxMessageBox(_("Data was saved to ") + dlg.GetPath());
}
}
}
} else if(evt.GetId() == XRCID("IDR_DBVIEWER_EXPORT_DATABASE")) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(m_selectedID);
if(data) {
Database* pDb = (Database*)wxDynamicCast(data->GetData(), Database);
if(pDb) {
wxFileDialog dlg(this,
_("Export database..."),
wxGetCwd(),
wxT(""),
wxT("SQL Files (*.sql)|*.sql"),
wxFD_SAVE | wxFD_OVERWRITE_PROMPT);
if(dlg.ShowModal() == wxID_OK) {
// CreateStructure
wxString retStr = wxT("-- SQL script created by wxDbExplorer\n\n ");
SerializableList::compatibility_iterator tabNode = pDb->GetFirstChildNode();
while(tabNode) {
Table* tab = wxDynamicCast(tabNode->GetData(), Table);
if(tab) {
retStr.append(pDb->GetDbAdapter()->GetCreateTableSql(tab, true));
}
tabNode = tabNode->GetNext();
}
tabNode = pDb->GetFirstChildNode();
while(tabNode) {
View* view = wxDynamicCast(tabNode->GetData(), View);
if(view) {
retStr.append(pDb->GetDbAdapter()->GetCreateViewSql(view, true));
}
tabNode = tabNode->GetNext();
}
tabNode = pDb->GetFirstChildNode();
while(tabNode) {
Table* tab = wxDynamicCast(tabNode->GetData(), Table);
if(tab) {
retStr.append(pDb->GetDbAdapter()->GetAlterTableConstraintSql(tab));
}
tabNode = tabNode->GetNext();
}
DumpClass dump(pDb->GetDbAdapter(), pDb, dlg.GetPath());
dump.DumpData();
wxTextFile file(dlg.GetPath());
if(!file.Exists()) file.Create();
file.Open();
if(file.IsOpened()) {
file.InsertLine(retStr, 0);
file.Write(wxTextFileType_None, wxConvUTF8);
file.Close();
}
wxMessageBox(
wxString::Format(_("The database has been exported to '%s'."), dlg.GetPath().GetData()),
_("wxDbExplorer"));
}
}
}
} else {
wxMessageBox(_("Sorry, requested feature isn't implemented yet. "), _("Sorry"));
}
} catch(DatabaseLayerException& e) {
wxString errorMessage = wxString::Format(_("Error (%d): %s"), e.GetErrorCode(), e.GetErrorMessage().c_str());
wxMessageDialog dlg(this, errorMessage, _("DB Error"), wxOK | wxCENTER | wxICON_ERROR);
dlg.ShowModal();
} catch(...) {
wxMessageDialog dlg(this, _("Unknown error."), _("DB Error"), wxOK | wxCENTER | wxICON_ERROR);
dlg.ShowModal();
}
}
void DbViewerPanel::RefreshDbView()
{
// Refresh all connections
wxTreeItemId root = m_treeDatabases->GetRootItem();
if(root.IsOk()) {
wxTreeItemIdValue cookie;
wxTreeItemId child = m_treeDatabases->GetFirstChild(root, cookie);
while(child.IsOk()) {
DbItem* data = (DbItem*)m_treeDatabases->GetItemData(child);
if(data) {
DbConnection* pCon = wxDynamicCast(data->GetData(), DbConnection);
if(pCon) pCon->RefreshChildren();
Database* pDb = wxDynamicCast(data->GetData(), Database);
if(pDb) pDb->RefreshChildren(false);
Table* pTab = wxDynamicCast(data->GetData(), Table);
if(pTab) pTab->RefreshChildren();
}
child = m_treeDatabases->GetNextChild(root, cookie);
}
}
// clear items from tree
m_treeDatabases->DeleteAllItems();
// create imageList for icons
wxImageList* pImageList = new wxImageList(16, 16, true, 3);
pImageList->Add(m_mgr->GetStdIcons()->LoadBitmap(wxT("toolbars/16/standard/file_open"))); // folder icon
pImageList->Add(m_mgr->GetStdIcons()->LoadBitmap(wxT("db-explorer/16/table"))); // table icon
pImageList->Add(m_mgr->GetStdIcons()->LoadBitmap(wxT("toolbars/16/search/find"))); // view icon
pImageList->Add(m_mgr->GetStdIcons()->LoadBitmap(wxT("db-explorer/16/database"))); // database
pImageList->Add(m_mgr->GetStdIcons()->LoadBitmap(wxT("db-explorer/16/column"))); // column
m_treeDatabases->AssignImageList(pImageList);
wxTreeItemId totalRootID = m_treeDatabases->AddRoot(wxString::Format(wxT("Databases")), -1);
// ---------------- load connections ----------------------------
SerializableList::compatibility_iterator connectionNode = m_pConnections->GetFirstChildNode();
while(connectionNode) {
DbConnection* pDbCon = (DbConnection*)wxDynamicCast(connectionNode->GetData(), DbConnection);
if(pDbCon) {
wxTreeItemId rootID =
m_treeDatabases->AppendItem(totalRootID,
wxString::Format(wxT("Databases (%s)"), pDbCon->GetServerName().c_str()),
3,
3,
new DbItem(pDbCon));
// ----------------------- load databases -------------------------------
SerializableList::compatibility_iterator dbNode = pDbCon->GetFirstChildNode();
while(dbNode) {
Database* pDatabase = wxDynamicCast(dbNode->GetData(), Database);
if(pDatabase) {
wxTreeItemId dbID =
m_treeDatabases->AppendItem(rootID, pDatabase->GetName(), 3, 3, new DbItem(pDatabase));
m_treeDatabases->Expand(rootID);
wxTreeItemId idFolder = m_treeDatabases->AppendItem(dbID, wxT("Tables"), 0, 0, NULL);
// m_treeDatabases->Expand(dbID);
// ----------------------------- load tables ----------------------------------
SerializableList::compatibility_iterator tabNode = pDatabase->GetFirstChildNode();
while(tabNode) {
Table* pTable = wxDynamicCast(tabNode->GetData(), Table);
if(pTable) {
// wxTreeItemId tabID = m_treeDatabases->AppendItem(idFolder,pTable->getName(),1,-1,new
// DbItem(NULL,pTable)); //NULL);
wxTreeItemId tabID = m_treeDatabases->AppendItem(idFolder,
pTable->GetName(),
1,
1,
new DbItem(pTable)); // NULL);
///////////////////////////////////////////////////////////
// Add the columns
///////////////////////////////////////////////////////////
SerializableList::compatibility_iterator columnNode = pTable->GetFirstChildNode();
while(columnNode) {
Column* col = wxDynamicCast(columnNode->GetData(), Column);
if(col) {
m_treeDatabases->AppendItem(
tabID, col->FormatName().c_str(), 4, 4, new DbItem(col));
}
columnNode = columnNode->GetNext();
}
}
tabNode = tabNode->GetNext();
}
// ----------------------------------------------------------------------------
idFolder = m_treeDatabases->AppendItem(dbID, wxT("Views"), 0, 0, NULL);
// m_treeDatabases->Expand(dbID);
// ----------------------------- load views ----------------------------------
tabNode = pDatabase->GetFirstChildNode();
while(tabNode) {
View* pView = wxDynamicCast(tabNode->GetData(), View);
if(pView) {
// wxTreeItemId tabID = m_treeDatabases->AppendItem(idFolder,pTable->getName(),1,-1,new
// DbItem(NULL,pTable)); //NULL);
m_treeDatabases->AppendItem(idFolder, pView->GetName(), 2, 2, new DbItem(pView)); // NULL);
}
tabNode = tabNode->GetNext();
}
// ----------------------------------------------------------------------------
}
dbNode = dbNode->GetNext();
}
// -----------------------------------------------------------------------
}
connectionNode = connectionNode->GetNext();
}
}