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(); } }