Dynamic context menu depending on current view. Result export now supports plaintext alongside CSV (untested)

Author: nfinit

src/sqlite-ce-edit/editor.c

@@ -113,6 +113,56 @@ void SyncLineNumScroll(void) {
     UpdateLineNumbers();
 }
 
+static int g_lastMenuMode = 0;  /* Start with Query menu (mode 0) */
+
+static void UpdateContextMenu(int mode) {
+    HMENU hCBMenu, hCtx;
+    
+    /* Only rebuild if mode changed */
+    if (mode == g_lastMenuMode) return;
+    
+    /* Get the actual menu from CommandBar */
+    hCBMenu = CommandBar_GetMenu(g_hwndCB, 0);
+    if (!hCBMenu) return;
+    
+    /* Remove old context menu at position 1 */
+    RemoveMenu(hCBMenu, 1, MF_BYPOSITION);
+    
+    /* Create and insert new context menu */
+    hCtx = CreatePopupMenu();
+    if (mode == 0) {
+        AppendMenuW(hCtx, MF_STRING, IDM_EXECUTE, L"&Execute\tCtrl+Enter");
+        AppendMenuW(hCtx, MF_SEPARATOR, 0, NULL);
+        AppendMenuW(hCtx, MF_STRING, IDM_FIND, L"&Find...\tCtrl+F");
+        AppendMenuW(hCtx, MF_STRING, IDM_FINDNEXT, L"Find &Next\tF3");
+        InsertMenuW(hCBMenu, 1, MF_BYPOSITION | MF_POPUP, (UINT)hCtx, L"&Query");
+    } else if (mode == 1) {
+        AppendMenuW(hCtx, MF_STRING, IDM_EXPORTCSV, L"Export as &CSV...");
+        AppendMenuW(hCtx, MF_STRING, IDM_EXPORTTXT, L"Export as &Text...");
+        AppendMenuW(hCtx, MF_SEPARATOR, 0, NULL);
+        AppendMenuW(hCtx, MF_STRING, IDM_FIND, L"&Find...\tCtrl+F");
+        AppendMenuW(hCtx, MF_STRING, IDM_FINDNEXT, L"Find &Next\tF3");
+        InsertMenuW(hCBMenu, 1, MF_BYPOSITION | MF_POPUP, (UINT)hCtx, L"&Results");
+    } else {
+        AppendMenuW(hCtx, MF_STRING, IDM_REFRESH, L"&Refresh");
+        InsertMenuW(hCBMenu, 1, MF_BYPOSITION | MF_POPUP, (UINT)hCtx, L"&Schema");
+    }
+    
+    g_lastMenuMode = mode;
+    
+    /* Update View menu radio check - get from CommandBar's menu */
+    {
+        HMENU hViewSub = GetSubMenu(hCBMenu, 2);  /* View is at position 2 */
+        if (hViewSub) {
+            CheckMenuRadioItem(hViewSub, IDM_VIEWQUERY, IDM_VIEWSCHEMA,
+                mode == 0 ? IDM_VIEWQUERY : (mode == 1 ? IDM_VIEWRESULT : IDM_VIEWSCHEMA), MF_BYCOMMAND);
+        }
+    }
+    
+    /* Force CommandBar to redraw */
+    CommandBar_DrawMenuBar(g_hwndCB, 0);
+}
+
 void SwitchView(int mode) {
     g_viewMode = mode;
     ShowWindow(g_hwndQuery, mode == 0 ? SW_SHOW : SW_HIDE);
@@ -121,6 +171,8 @@ void SwitchView(int mode) {
     if (g_hwndSchema) ShowWindow(g_hwndSchema, mode == 2 ? SW_SHOW : SW_HIDE);
     /* Disable exec-at-cursor button in non-query views */
     SendMessage(g_hwndCB, TB_ENABLEBUTTON, IDM_EXECATCURSOR, mode == 0);
+    /* Update context menu */
+    UpdateContextMenu(mode);
     if (mode == 0) {
         SetFocus(g_hwndQuery);
         UpdateLineCount();

src/sqlite-ce-edit/fileops.c

@@ -348,6 +348,49 @@ void DoExportCSV(void) {
     if (buf) LocalFree(buf);
 }
 
+/*============================================================================
+** Export Results to Text File
+**============================================================================*/
+
+void DoExportTxt(void) {
+    CE_OPENFILENAME ofn;
+    wchar_t szFile[MAX_PATH] = L"results.txt";
+    HANDLE hFile;
+    DWORD dwLen, dwWritten;
+    wchar_t *wbuf;
+    char *buf;
+    
+    memset(&ofn, 0, sizeof(ofn));
+    ofn.lStructSize = sizeof(ofn);
+    ofn.hwndOwner = g_hwndMain;
+    ofn.lpstrFile = szFile;
+    ofn.nMaxFile = MAX_PATH;
+    ofn.lpstrFilter = L"Text Files (*.txt)\0*.txt\0All Files (*.*)\0*.*\0";
+    ofn.lpstrDefExt = L"txt";
+    ofn.lpstrTitle = L"Export Results";
+    ofn.Flags = OFN_OVERWRITEPROMPT | OFN_PATHMUSTEXIST;
+    
+    if (!GetSaveFileNameW(&ofn)) return;
+    
+    dwLen = GetWindowTextLengthW(g_hwndResult);
+    if (dwLen == 0) return;
+    
+    wbuf = (wchar_t*)LocalAlloc(LMEM_FIXED, (dwLen + 1) * sizeof(wchar_t));
+    buf = (char*)LocalAlloc(LMEM_FIXED, dwLen + 1);
+    if (wbuf && buf) {
+        GetWindowTextW(g_hwndResult, wbuf, dwLen + 1);
+        WideCharToMultiByte(CP_ACP, 0, wbuf, -1, buf, dwLen + 1, NULL, NULL);
+        
+        hFile = CreateFileW(szFile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
+        if (hFile != INVALID_HANDLE_VALUE) {
+            WriteFile(hFile, buf, (DWORD)strlen(buf), &dwWritten, NULL);
+            CloseHandle(hFile);
+        }
+    }
+    if (wbuf) LocalFree(wbuf);
+    if (buf) LocalFree(buf);
+}
+
 /*============================================================================
 ** Export Table to CSV (helper for DoExportDb)
 **============================================================================*/

src/sqlite-ce-edit/globals.h

@@ -96,6 +96,7 @@ BOOL WINAPI GetSaveFileNameW(CE_OPENFILENAME*);
 #define IDM_SAVEQUERY  106
 #define IDM_EXPORTCSV  107
 #define IDM_EXPORTDB   108
+#define IDM_EXPORTTXT  109
 #define IDM_IMPORTCSV  109
 #define IDM_IMPORTCEDB 110
 #define IDM_EXECUTE  201
@@ -119,6 +120,7 @@ BOOL WINAPI GetSaveFileNameW(CE_OPENFILENAME*);
 /* Schema context menu */
 #define IDM_SCHEMA_SELECT 700
 #define IDM_SCHEMA_DROP   701
+#define IDM_REFRESH       702
 
 /* Timer IDs */
 #define IDT_TAPHOLD 1
@@ -223,6 +225,7 @@ void UpdateLineCount(void);
 void UpdateLineNumbers(void);
 void SyncLineNumScroll(void);
 void SwitchView(int mode);
+void BuildMenuBar(int mode);
 void UpdateQueryFont(void);
 void UpdateResultFont(void);
 void CycleFontSize(void);
@@ -269,6 +272,7 @@ void DoFileOpen(void);
 void DoOpenQuery(void);
 void DoSaveQuery(void);
 void DoExportCSV(void);
+void DoExportTxt(void);
 void DoExportDb(void);
 void DoImportCSV(void);
 void DoImportCEDB(void);

src/sqlite-ce-edit/main.c

@@ -248,6 +248,7 @@ static LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lPara
                 case IDM_OPENQUERY: DoOpenQuery(); break;
                 case IDM_SAVEQUERY: DoSaveQuery(); break;
                 case IDM_EXPORTCSV: DoExportCSV(); break;
+                case IDM_EXPORTTXT: DoExportTxt(); break;
                 case IDM_EXPORTDB: DoExportDb(); break;
                 case IDM_IMPORTCSV: DoImportCSV(); break;
                 case IDM_IMPORTCEDB: DoImportCEDB(); break;