import SwiftUI import Charts import AppKit import CryptoKit import UniformTypeIdentifiers // MARK: - Classic Workbench Colours private enum WB { static let toolbar = Color(nsColor: NSColor( calibratedWhite: 0.88, alpha: 1 )) static let sidebar = Color(nsColor: NSColor( calibratedWhite: 0.94, alpha: 1 )) static let border = Color(nsColor: NSColor( calibratedWhite: 0.76, alpha: 1 )) static let header = Color(nsColor: NSColor( calibratedWhite: 0.91, alpha: 1 )) } // MARK: - Main Window struct MainView: View { @EnvironmentObject private var model: DatabaseModel @State private var selectedSQL = "" @State private var resultFilter = "" @State private var resultSortColumn = 0 @State private var resultSortAscending = true @State private var hiddenResultColumns: Set = [] @State private var resultPage = 0 private let resultPageSize = 1_000 @State private var showingEditor = false @State private var showingDatabaseTransfer = false @State private var tableSearchTarget: WorkbenchTableTarget? = nil @State private var tableImportTarget: WorkbenchTableTarget? = nil @State private var schemaEditorTarget: SchemaEditorTarget? = nil @State private var objectEditorTarget: WorkbenchObjectEditorTarget? = nil @State private var showingQueryPlan = false @State private var showingQueryHistory = false @State private var showingSavedQueries = false @State private var showingSaveQuery = false @State private var newSavedQueryName = "" @State private var editing = ConnectionProfile() @State private var editingPassword = "" @State private var expandedSchemas: Set = [] @State private var expandedTableFolders: Set = [] @State private var expandedObjectFolders: Set = [] @State private var schemaFilter = "" @State private var schemaManagement: WorkbenchSchemaManagementTarget? = nil @State private var selectedTableName: String? = nil @State private var outputExpanded = true @State private var navigatorMode = 1 @State private var sqlFocusRequest = 0 @State private var administrationPage: WorkbenchAdminPage? = .dashboard var body: some View { VStack(spacing: 0) { mainToolbar Rectangle() .fill(WB.border) .frame(height: 1) HSplitView { navigator .frame( minWidth: 180, idealWidth: 250, maxWidth: 420 ) workspace .frame(minWidth: 480) } statusBar } .frame(minWidth: 800, minHeight: 520) .preferredColorScheme( UserDefaults.standard.string( forKey: "preferences.appearance" ) == "dark" ? .dark : .light ) .background(Color(nsColor: .windowBackgroundColor)) .onReceive(NotificationCenter.default.publisher(for: .workbenchRevealSQLEditor)) { _ in administrationPage = nil navigatorMode = 1 DispatchQueue.main.async { sqlFocusRequest += 1 } } .task { if model.databases.isEmpty, model.selected != nil { model.loadDatabases() } } .sheet(isPresented: $showingQueryPlan) { WorkbenchExplainView(model: model, statement: selectedSQL.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty ? model.sql : selectedSQL) } .sheet(isPresented: $showingQueryHistory) { SQLHistorySheet(model: model) } .sheet(isPresented: $showingSavedQueries) { SavedSQLQueriesSheet(model: model) } .alert("Save SQL Query", isPresented: $showingSaveQuery) { TextField("Query name", text: $newSavedQueryName) Button("Save") { model.saveCurrentQuery(named: newSavedQueryName) } Button("Cancel", role: .cancel) { } } message: { Text("Save the current SQL in your local query library.") } .sheet(item: $tableSearchTarget) { target in WorkbenchTableSearchSheet(model: model, target: target) { sql in revealSchemaSQL(sql) model.runSQL() } } .sheet(item: $tableImportTarget) { target in WorkbenchCSVImportSheet(model: model, target: target) } .sheet(isPresented: $showingDatabaseTransfer) { WorkbenchDatabaseTransferView(model: model) } .sheet(item: $schemaEditorTarget, onDismiss: { if !model.selectedDatabase.isEmpty { model.loadTables(database: model.selectedDatabase) } }) { target in WorkbenchSchemaEditorView(profile: model.selected, target: target) } .sheet(item: $objectEditorTarget, onDismiss: { if !model.selectedDatabase.isEmpty { model.loadTables(database: model.selectedDatabase) } }) { target in WorkbenchObjectEditorView(profile: model.selected, target: target) } .sheet(item: $schemaManagement, onDismiss: { model.loadDatabases() }) { target in WorkbenchSchemaManagementSheet(model: model, target: target) } .sheet(isPresented: $showingEditor) { ConnectionEditor( profile: $editing, password: $editingPassword ) { model.save(editing, password: editingPassword) showingEditor = false } } } // MARK: - Query execution navigation /// A query launched from the global toolbar must reveal its editor and /// results, even when an Administration page currently owns the workspace. private func executeSQLInEditor() { // Make the editor the active workspace before starting execution. administrationPage = nil navigatorMode = 1 model.runSQL() // The SQLTextView is created only after the workspace changes. Request // focus on the next main-loop turn, after SwiftUI mounts that view. DispatchQueue.main.async { sqlFocusRequest += 1 } } // MARK: - Main Toolbar private var mainToolbar: some View { HStack(spacing: 16) { Image(nsImage: NSImage(named: "WorkbenchBrand") ?? NSImage(contentsOfFile: Bundle.main.path(forResource: "WorkbenchBrand", ofType: "png") ?? "") ?? NSImage()) .resizable() .scaledToFit() .frame(width: 30, height: 30) Text("MySQL Workbench") .font(.system(size: 15, weight: .semibold)) .foregroundStyle(.primary) Rectangle() .fill(WB.border) .frame(width: 1, height: 30) toolbarButton("Dashboard", icon: "chart.xyaxis.line") { navigatorMode = 0 administrationPage = .dashboard } toolbarButton( "New Connection", icon: "plus.rectangle.on.rectangle" ) { editing = ConnectionProfile(name: "New Connection") editingPassword = "" showingEditor = true } toolbarButton( "Import / Export", icon: "arrow.up.arrow.down.square" ) { showingDatabaseTransfer = true } toolbarButton( "Refresh", icon: "arrow.clockwise" ) { model.loadDatabases() } toolbarButton( "Execute", icon: "bolt.fill" ) { executeSQLInEditor() } .keyboardShortcut(.return, modifiers: [.command]) toolbarButton( "Format SQL", icon: "text.alignleft" ) { model.formatSQL() } Spacer() VStack(alignment: .trailing, spacing: 2) { Text(model.selected?.name ?? "No Connection") .font(.system(size: 12, weight: .semibold)) Text(model.selectedDatabase.isEmpty ? "No default schema" : model.selectedDatabase) .font(.system(size: 11)) .foregroundStyle(.secondary) } } .padding(.horizontal, 10) .frame(height: 46) .background(WB.toolbar) } private func toolbarButton( _ title: String, icon: String, action: @escaping () -> Void ) -> some View { Button(action: action) { VStack(spacing: 3) { Image(systemName: icon) .font(.system(size: 19)) .frame(height: 23) Text(title) .font(.system(size: 10)) } .foregroundStyle(.primary) .frame(minWidth: 65) } .buttonStyle(.plain) .help(title) } // MARK: - Navigator private var navigator: some View { VStack(spacing: 0) { Picker("Navigator", selection: $navigatorMode) { Text("Administration").tag(0) Text("Schemas").tag(1) } .pickerStyle(.segmented) .labelsHidden() .padding(8) .onChange(of: navigatorMode) { _, newMode in if newMode == 1 { administrationPage = nil sqlFocusRequest += 1 } } if navigatorMode == 0 { administrationNavigator } else { schemasNavigator } } .background(WB.sidebar) } private var schemasNavigator: some View { VStack(spacing: 0) { HStack { Text("SCHEMAS") .font(.system(size: 11, weight: .bold)) Spacer() Button { model.loadDatabases() } label: { Image(systemName: "arrow.clockwise") } .buttonStyle(.borderless) } .padding(.horizontal, 12) .frame(height: 34) TextField("Filter schemas", text: $schemaFilter) .textFieldStyle(.roundedBorder) .padding(.horizontal, 9) .padding(.bottom, 5) Divider() ScrollView { LazyVStack(alignment: .leading, spacing: 2) { ForEach(model.databases.filter { schemaFilter.isEmpty || $0.localizedCaseInsensitiveContains(schemaFilter) }, id: \.self) { db in schemaNode(db) } } .padding(.vertical, 8) } Divider() sectionHeader("MYSQL CONNECTIONS") ScrollView { VStack(alignment: .leading, spacing: 4) { ForEach(model.profiles) { profile in Button { model.selectedID = profile.id model.loadDatabases() } label: { Label( profile.name, systemImage: "server.rack" ) .frame( maxWidth: .infinity, alignment: .leading ) .padding(6) .background( model.selectedID == profile.id ? Color.blue.opacity(0.14) : Color.clear ) } .buttonStyle(.plain) .contextMenu { Button("Edit Connection") { edit(profile) } } } } .padding(8) } .frame(height: 125) } } private var administrationNavigator: some View { ScrollView { VStack(alignment: .leading, spacing: 4) { administrationSection( "MANAGEMENT", pages: [ .serverStatus, .serverCounters, .clientConnections, .users, .variables, .serverLogs ] ) administrationSection( "PERFORMANCE", pages: [ .dashboard, .performanceReports, .slowQueries, .tableIO, .innodbMetrics, .innodbLocks, .connectionStatistics, .performanceSchema ] ) } .padding(8) } .frame(maxWidth: .infinity, maxHeight: .infinity) } private func administrationSection( _ title: String, pages: [WorkbenchAdminPage] ) -> some View { VStack(alignment: .leading, spacing: 4) { Text(title) .font(.system(size: 11, weight: .bold)) .foregroundStyle(.secondary) .padding(.top, 12) .padding(.bottom, 4) ForEach(pages) { page in Button { administrationPage = page } label: { Label(page.rawValue, systemImage: page.icon) .frame( maxWidth: .infinity, alignment: .leading ) .padding(.horizontal, 8) .padding(.vertical, 7) .background( administrationPage == page ? Color.accentColor.opacity(0.18) : Color.clear ) .clipShape(RoundedRectangle(cornerRadius: 5)) } .buttonStyle(.plain) } } } // MARK: - Compact Schema Tree private func schemaNode(_ db: String) -> some View { VStack(alignment: .leading, spacing: 0) { HStack(spacing: 5) { Image(systemName: expandedSchemas.contains(db) ? "chevron.down" : "chevron.right" ) .font(.system(size: 10)) .foregroundStyle(.secondary) .frame(width: 13) Image(systemName: "cylinder.split.1x2") .foregroundStyle(.blue) .frame(width: 17) Text(db) .font(.system(size: 12, weight: model.selectedDatabase == db ? .semibold : .regular)) .lineLimit(1) Spacer(minLength: 0) } .frame(height: 25) .padding(.leading, 10) .contentShape(Rectangle()) .onTapGesture { toggleSchema(db) } .onTapGesture(count: 2) { model.selectedDatabase = db expandedSchemas.insert(db) expandedTableFolders.insert(db) if model.schemaTables[db] == nil { model.loadTables(database: db) } } .contextMenu { Button("Set as Default Schema") { model.selectedDatabase = db } Button("Filter to This Schema") { schemaFilter = db } Button("Clear Schema Filter") { schemaFilter = "" } .disabled(schemaFilter.isEmpty) Button("Schema Inspector…") { schemaManagement = .init(database: db, action: .inspect) } Divider() Button("Copy Schema Name") { NSPasteboard.general.clearContents() NSPasteboard.general.setString(db, forType: .string) } Button("Send CREATE Statement to SQL Editor") { revealSchemaSQL("SHOW CREATE DATABASE `\(db.replacingOccurrences(of: "`", with: "``"))`;") } Divider() Button("Create Schema…") { schemaManagement = .init(database: nil, action: .create) } Button("Alter Schema…") { schemaManagement = .init(database: db, action: .alter) } Button("Drop Schema…", role: .destructive) { schemaManagement = .init(database: db, action: .drop) } Divider() Button("Create Table…") { schemaEditorTarget = SchemaEditorTarget(database: db, table: nil) } Button("Import CSV into New Table…") { tableImportTarget = .init(database: db, table: nil) } Button("Refresh All") { model.loadDatabases() model.loadTables(database: db) } } if expandedSchemas.contains(db) { schemaChildren(db) } } .frame(maxWidth: .infinity, alignment: .leading) } private func revealSchemaSQL(_ statement: String) { model.sql = statement administrationPage = nil navigatorMode = 1 DispatchQueue.main.async { sqlFocusRequest += 1 } } private func toggleSchema(_ db: String) { if expandedSchemas.contains(db) { expandedSchemas.remove(db) } else { expandedSchemas.insert(db) expandedTableFolders.insert(db) if model.schemaTables[db] == nil { model.loadTables(database: db) } } } private func schemaChildren(_ db: String) -> some View { VStack(alignment: .leading, spacing: 0) { // Tables folder HStack(spacing: 5) { Image(systemName: expandedTableFolders.contains(db) ? "chevron.down" : "chevron.right" ) .font(.system(size: 10)) .foregroundStyle(.secondary) .frame(width: 13) Image(systemName: "folder") .foregroundStyle(.secondary) .frame(width: 17) Text("Tables") .font(.system(size: 12)) Spacer(minLength: 0) } .frame(height: 24) .padding(.leading, 28) .contentShape(Rectangle()) .onTapGesture { if expandedTableFolders.contains(db) { expandedTableFolders.remove(db) } else { expandedTableFolders.insert(db) if model.schemaTables[db] == nil { model.loadTables(database: db) } } } // Tables if expandedTableFolders.contains(db) { if let tables = model.schemaTables[db] { ForEach(tables, id: \.self) { table in tableNode(database: db, table: table) } } else { Button("Load Tables") { model.loadTables(database: db) } .font(.system(size: 11)) .buttonStyle(.borderless) .padding(.leading, 66) .frame(height: 24) } } schemaObjectFolder("Views", database: db, objects: model.schemaViews[db] ?? [], kind: .view) schemaObjectFolder("Stored Procedures", database: db, objects: model.schemaProcedures[db] ?? [], kind: .procedure) schemaObjectFolder("Functions", database: db, objects: model.schemaFunctions[db] ?? [], kind: .function) } .frame(maxWidth: .infinity, alignment: .leading) } private func tableNode(database: String, table: String) -> some View { let identifier = database + "." + table let isSelected = selectedTableName == identifier return HStack(spacing: 5) { Image(systemName: "tablecells") .font(.system(size: 11)) .foregroundStyle( isSelected ? Color.blue : Color.gray ) .frame(width: 17) Text(table) .font(.system(size: 12)) .lineLimit(1) .truncationMode(.middle) Spacer(minLength: 0) } .frame(height: 23) .padding(.leading, 49) .frame(maxWidth: .infinity, alignment: .leading) .background( isSelected ? Color.blue.opacity(0.14) : Color.clear ) .contentShape(Rectangle()) .onTapGesture { selectedTableName = identifier } .onTapGesture(count: 2) { selectedTableName = identifier model.selectTable( database: database, table: table ) } .contextMenu { Button("Select First 100 Rows") { selectedTableName = identifier model.selectTable( database: database, table: table ) } Button("Search Table Data…") { tableSearchTarget = .init(database: database, table: table) } Button("Import CSV Data…") { tableImportTarget = .init(database: database, table: table) } Divider() Button("Edit Table Schema…") { schemaEditorTarget = SchemaEditorTarget(database: database, table: table) } } } private enum SchemaObjectKind { case view, procedure, function } private func schemaObjectFolder(_ title: String, database: String, objects: [String], kind: SchemaObjectKind) -> some View { let key = database + "." + title let expanded = expandedObjectFolders.contains(key) return VStack(alignment: .leading, spacing: 0) { HStack(spacing: 5) { Image(systemName: expanded ? "chevron.down" : "chevron.right") .font(.system(size: 10)).foregroundStyle(.secondary).frame(width: 13) Image(systemName: "folder").foregroundStyle(.secondary).frame(width: 17) Text(title).font(.system(size: 12)) Text("\(objects.count)").font(.caption2).foregroundStyle(.secondary) Spacer(minLength: 0) } .frame(height: 24) .padding(.leading, 28) .contentShape(Rectangle()) .onTapGesture { if expanded { expandedObjectFolders.remove(key) } else { expandedObjectFolders.insert(key) if model.schemaTables[database] == nil { model.loadTables(database: database) } } } if expanded { ForEach(objects, id: \.self) { name in HStack(spacing: 5) { Image(systemName: kind == .view ? "tablecells" : "curlybraces") .foregroundStyle(.secondary).frame(width: 17) Text(name).font(.system(size: 12)).lineLimit(1) Spacer(minLength: 0) } .frame(height: 23) .padding(.leading, 49) .contentShape(Rectangle()) .onTapGesture(count: 2) { openSchemaObject(database: database, name: name, kind: kind) } .contextMenu { Button(kind == .view ? "Select First Rows" : "Show Definition") { openSchemaObject(database: database, name: name, kind: kind) } Button("Edit Definition…") { objectEditorTarget = WorkbenchObjectEditorTarget( database: database, name: name, kind: kind == .view ? .view : (kind == .procedure ? .procedure : .function) ) } } } if objects.isEmpty { Text("No objects (or metadata unavailable)") .font(.caption2).foregroundStyle(.secondary) .padding(.leading, 49).frame(height: 23) } } } } private func openSchemaObject(database: String, name: String, kind: SchemaObjectKind) { let db = database.replacingOccurrences(of: "`", with: "``") let object = name.replacingOccurrences(of: "`", with: "``") model.selectedDatabase = database switch kind { case .view: model.sql = "SELECT * FROM `\(db)`.`\(object)` LIMIT 500;" case .procedure: model.sql = "SHOW CREATE PROCEDURE `\(db)`.`\(object)`;" case .function: model.sql = "SHOW CREATE FUNCTION `\(db)`.`\(object)`;" } administrationPage = nil navigatorMode = 1 model.runSQL() DispatchQueue.main.async { sqlFocusRequest += 1 } } // MARK: - Workspace private var workspace: some View { Group { if navigatorMode == 0, let page = administrationPage { WorkbenchAdministrationView( page: page, profile: model.selected ) { administrationPage = nil navigatorMode = 1 } } else { sqlWorkspace } } } private var sqlWorkspace: some View { VStack(spacing: 0) { queryTabBar Divider() queryToolbar Divider() VSplitView { sqlEditor .frame(minHeight: 150) bottomWorkspace .frame(minHeight: 150) } } } private var queryTabBar: some View { HStack(spacing: 0) { ScrollView(.horizontal) { HStack(spacing: 2) { ForEach(model.queryTabs) { tab in HStack(spacing: 8) { Button { administrationPage = nil navigatorMode = 1 model.activateQuery(tab.id) DispatchQueue.main.async { sqlFocusRequest += 1 } } label: { Label(tab.title, systemImage: "doc.text") .lineLimit(1) } .buttonStyle(.plain) Button { model.closeQuery(tab.id) } label: { Image(systemName: "xmark") .font(.system(size: 9)) } .buttonStyle(.plain) } .font(.system(size: 12)) .padding(.horizontal, 10) .frame(height: 30) .background( model.activeQueryID == tab.id ? Color(nsColor: .textBackgroundColor) : Color.clear ) .clipShape( UnevenRoundedRectangle( topLeadingRadius: 5, topTrailingRadius: 5 ) ) } } } .scrollIndicators(.hidden) Button { model.newQueryTab() } label: { Image(systemName: "plus") } .buttonStyle(.borderless) .padding(.horizontal, 10) .help("New Query Tab") } .background(WB.header) } private var queryToolbar: some View { HStack(spacing: 14) { Button { executeSQLInEditor() } label: { Image(systemName: "bolt.fill") .foregroundStyle(.orange) } .help("Execute SQL") Button { model.runSQL(statement: selectedSQL) } label: { Label("Run Selection", systemImage: "selection.pin.in.out") } .disabled(selectedSQL.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty) .help("Execute selected SQL without modifying the editor") Button { showingQueryPlan = true } label: { Label("Explain", systemImage: "chart.bar.doc.horizontal") } .disabled(model.isRunning || model.selected == nil) .help("Inspect the optimizer plan for the selected SQL or editor contents") Button { model.formatSQL() } label: { Image(systemName: "text.alignleft") } .help("Format SQL") Divider().frame(height: 17) Button { model.openSQLScript() } label: { Image(systemName: "folder") } .help("Open SQL Script…") Button { model.saveSQLScript() } label: { Image(systemName: "square.and.arrow.down") } .help("Save SQL Script") Menu { if model.recentSQLScripts.isEmpty { Text("No Recent Scripts") } else { ForEach(model.recentSQLScripts, id: \.self) { path in Button(URL(fileURLWithPath: path).lastPathComponent) { model.openRecentSQLScript(at: path) } } Divider() Button("Clear Recent Scripts") { model.clearRecentSQLScripts() } } } label: { Image(systemName: "clock.arrow.circlepath") } .help("Recent SQL Scripts") Divider().frame(height: 17) Button { showingQueryHistory = true } label: { Label("History", systemImage: "clock.arrow.circlepath") } .help("Query execution history") Button { showingSavedQueries = true } label: { Label("Saved", systemImage: "bookmark") } .help("Open saved SQL queries") Button { newSavedQueryName = "" showingSaveQuery = true } label: { Image(systemName: "bookmark.fill") } .help("Save current SQL query") Divider().frame(height: 17) Text("Default Schema:") .font(.system(size: 11)) Text(model.selectedDatabase.isEmpty ? "None" : model.selectedDatabase) .font(.system(size: 11, weight: .semibold)) Spacer() Text("SQL Editor") .font(.system(size: 11)) .foregroundStyle(.secondary) } .buttonStyle(.borderless) .padding(.horizontal, 12) .frame(height: 30) .background(WB.header) } private var sqlEditor: some View { VStack(spacing: 0) { if model.status == "Error" && !model.result.message.isEmpty { HStack(alignment: .top, spacing: 8) { Image(systemName: "exclamationmark.triangle.fill") .foregroundStyle(.red) Text(model.result.message) .font(.system(size: 11, design: .monospaced)) .textSelection(.enabled) .frame(maxWidth: .infinity, alignment: .leading) } .padding(8) .background(Color.red.opacity(0.09)) } HStack(spacing: 0) { if UserDefaults.standard.object( forKey: "preferences.showLineNumbers" ) as? Bool ?? true { lineNumbers Rectangle() .fill(WB.border) .frame(width: 1) } Rectangle() .fill(WB.border) .frame(width: 1) SQLTextView(text: $model.sql, selectedSQL: $selectedSQL, focusRequest: sqlFocusRequest, completions: Array(Set(model.schemaTables.values.flatMap { $0 } + model.schemaViews.values.flatMap { $0 } + (model.schemaColumns[model.selectedDatabase] ?? []))).sorted()) } .background(Color(nsColor: .textBackgroundColor)) } } private var lineNumbers: some View { ScrollView { VStack(alignment: .trailing, spacing: 0) { ForEach( 1...max(1, model.sql.components( separatedBy: "\n" ).count), id: \.self ) { number in Text("\(number)") .frame(height: 19) } } .padding(.horizontal, 9) .padding(.top, 8) } .font(.system(size: 12, design: .monospaced)) .foregroundStyle(.secondary) .frame(width: 48) .background(WB.header) .allowsHitTesting(false) } // MARK: - Results and Action Output private var bottomWorkspace: some View { VSplitView { resultsPanel .frame(minHeight: 230) if outputExpanded { actionOutput .frame(minHeight: 90) } } } private var resultsPanel: some View { VStack(spacing: 0) { HStack { Image(systemName: "tablecells") Text("Result Grid") .fontWeight(.semibold) Spacer() if !model.result.columns.isEmpty { TextField("Filter rows", text: $resultFilter) .textFieldStyle(.roundedBorder) .frame(width: 170) Picker("Sort", selection: $resultSortColumn) { ForEach(model.result.columns.indices, id: \.self) { index in Text(model.result.columns[index]).tag(index) } } .frame(width: 155) Button(resultSortAscending ? "Ascending" : "Descending") { resultSortAscending.toggle() } Menu("Columns") { Button("Show All Columns") { hiddenResultColumns.removeAll() } Divider() ForEach(model.result.columns.indices, id: \.self) { index in Button { if hiddenResultColumns.contains(index) { hiddenResultColumns.remove(index) } else { hiddenResultColumns.insert(index) } } label: { Label(model.result.columns[index], systemImage: hiddenResultColumns.contains(index) ? "square" : "checkmark.square") } } } Button("Export CSV") { exportResultCSV() } } Text(model.queryDurationText) .foregroundStyle(.secondary) Text(model.result.message) .foregroundStyle(.secondary) } .font(.system(size: 12)) .padding(.horizontal, 12) .frame(height: 28) .background(WB.header) Divider() if model.result.columns.isEmpty { VStack { Spacer() Text("No query results") .foregroundStyle(.secondary) Spacer() } } else { resultGrid HStack(spacing: 12) { Text("\(displayedRows.count) matching rows · showing \(min(resultPage * resultPageSize + 1, displayedRows.count))–\(min((resultPage + 1) * resultPageSize, displayedRows.count))") .foregroundStyle(.secondary) Spacer() Button("Previous") { resultPage = max(0, resultPage - 1) } .disabled(resultPage == 0) Text("Page \(resultPage + 1) / \(max(1, (displayedRows.count + resultPageSize - 1) / resultPageSize))") Button("Next") { resultPage += 1 } .disabled((resultPage + 1) * resultPageSize >= displayedRows.count) } .font(.system(size: 11)) .padding(.horizontal, 12) .padding(.vertical, 5) .background(WB.header) } } .onChange(of: resultFilter) { _, _ in resultPage = 0 } .onChange(of: resultSortColumn) { _, _ in resultPage = 0 } .onChange(of: resultSortAscending) { _, _ in resultPage = 0 } .onChange(of: model.result.columns) { _, _ in resultPage = 0 hiddenResultColumns.removeAll() } } private var resultGrid: some View { let rows = displayedRows let start = min(resultPage * resultPageSize, rows.count) let end = min(start + resultPageSize, rows.count) return NativeResultGrid(columns: model.result.columns, rows: Array(rows[start.. String { "\"" + value.replacingOccurrences(of: "\"", with: "\"\"") + "\"" } func csvRecord(_ values: [String]) -> Data { Data((values.map(csvField).joined(separator: ",") + "\r\n").utf8) } let rows = displayedRows let tempURL = url.deletingLastPathComponent() .appendingPathComponent(".\(UUID().uuidString).csv.tmp") do { guard FileManager.default.createFile(atPath: tempURL.path, contents: nil) else { throw CocoaError(.fileWriteUnknown) } let handle = try FileHandle(forWritingTo: tempURL) defer { try? handle.close() } try handle.write(contentsOf: Data([0xEF, 0xBB, 0xBF])) try handle.write(contentsOf: csvRecord(model.result.columns)) for row in rows { try handle.write(contentsOf: csvRecord(row)) } try handle.synchronize() if FileManager.default.fileExists(atPath: url.path) { _ = try FileManager.default.replaceItemAt(url, withItemAt: tempURL) } else { try FileManager.default.moveItem(at: tempURL, to: url) } model.status = "Exported \(rows.count) rows to \(url.lastPathComponent)" } catch { try? FileManager.default.removeItem(at: tempURL) model.status = "CSV export failed: \(error.localizedDescription)" } } private var actionOutput: some View { VStack(spacing: 0) { HStack { Image(systemName: "list.bullet.rectangle") Text("Action Output") .fontWeight(.semibold) Spacer() Button { outputExpanded = false } label: { Image(systemName: "chevron.down") } .buttonStyle(.borderless) } .font(.system(size: 12)) .padding(.horizontal, 12) .frame(height: 26) .background(WB.header) Divider() HStack(alignment: .top, spacing: 12) { Image(systemName: model.status == "Error" ? "xmark.circle.fill" : "checkmark.circle.fill" ) .foregroundStyle( model.status == "Error" ? .red : .green ) VStack(alignment: .leading, spacing: 5) { Text(model.status) .fontWeight(.semibold) Text(model.result.message) .foregroundStyle(.secondary) } Spacer() } .font(.system(size: 12)) .padding(12) Spacer() } } // MARK: - Status Bar private var statusBar: some View { HStack(spacing: 10) { Circle() .fill( model.status == "Error" ? Color.red : Color.green ) .frame(width: 8, height: 8) Text(model.status) .lineLimit(1) Spacer() Text(model.selected?.name ?? "Disconnected") Divider() .frame(height: 13) Text(model.selectedDatabase.isEmpty ? "No Schema" : model.selectedDatabase) Divider() .frame(height: 13) Text("UTF-8") } .font(.system(size: 11)) .padding(.horizontal, 12) .frame(height: 22) .background(WB.header) } private func sectionHeader(_ title: String) -> some View { HStack { Text(title) .font(.system(size: 11, weight: .bold)) .foregroundStyle(.secondary) Spacer() } .padding(.horizontal, 12) .frame(height: 30) .background(WB.header) } private func edit(_ profile: ConnectionProfile) { editing = profile editingPassword = PasswordStore.read(profile.id) showingEditor = true } } // MARK: - Phase 17: Query History and Saved Queries private struct SQLHistorySheet: View { @ObservedObject var model: DatabaseModel @Environment(\.dismiss) private var dismiss @State private var filter = "" private var entries: [SQLHistoryEntry] { model.queryHistory.filter { filter.isEmpty || $0.sql.localizedCaseInsensitiveContains(filter) || $0.connectionName.localizedCaseInsensitiveContains(filter) } } var body: some View { VStack(alignment: .leading, spacing: 12) { HStack { Text("Query History").font(.title2).bold() Spacer() Button("Clear History") { model.clearQueryHistory() } .disabled(model.queryHistory.isEmpty) Button("Done") { dismiss() } } TextField("Filter SQL or connection", text: $filter) .textFieldStyle(.roundedBorder) List(entries) { entry in VStack(alignment: .leading, spacing: 5) { HStack { Image(systemName: entry.succeeded ? "checkmark.circle.fill" : "xmark.circle.fill") .foregroundStyle(entry.succeeded ? .green : .red) Text(entry.executedAt, style: .date) Text(entry.executedAt, style: .time) Text(entry.connectionName).foregroundStyle(.secondary) Text(entry.database).foregroundStyle(.secondary) Spacer() Button("Open in New Tab") { model.newQueryTab() model.sql = entry.sql model.saveActiveQuery() dismiss() } } Text(entry.sql).font(.system(.body, design: .monospaced)) .lineLimit(3).textSelection(.enabled) Text(entry.detail).font(.caption).foregroundStyle(.secondary) }.padding(.vertical, 4) } } .padding(20) .frame(minWidth: 740, minHeight: 460) } } private struct SavedSQLQueriesSheet: View { @ObservedObject var model: DatabaseModel @Environment(\.dismiss) private var dismiss @State private var filter = "" var body: some View { VStack(alignment: .leading, spacing: 12) { HStack { Text("Saved SQL Queries").font(.title2).bold() Spacer() Button("Done") { dismiss() } } TextField("Filter saved queries", text: $filter) .textFieldStyle(.roundedBorder) List(model.savedQueries.filter { filter.isEmpty || $0.name.localizedCaseInsensitiveContains(filter) || $0.sql.localizedCaseInsensitiveContains(filter) }) { entry in HStack { VStack(alignment: .leading, spacing: 4) { Text(entry.name).font(.headline) Text(entry.sql).font(.system(.caption, design: .monospaced)) .lineLimit(2).foregroundStyle(.secondary) } Spacer() Button("Open") { model.openSavedQuery(entry) dismiss() } Button(role: .destructive) { model.deleteSavedQuery(entry.id) } label: { Image(systemName: "trash") } }.padding(.vertical, 5) } } .padding(20) .frame(minWidth: 650, minHeight: 420) } } // MARK: - Native SQL Text Editor struct SQLTextView: NSViewRepresentable { @Binding var text: String @Binding var selectedSQL: String var focusRequest: Int = 0 var completions: [String] = [] func makeCoordinator() -> Coordinator { Coordinator(self) } func makeNSView(context: Context) -> NSScrollView { let scroll = NSTextView.scrollableTextView() let editor = scroll.documentView as! NSTextView scroll.appearance = NSAppearance(named: .aqua) editor.appearance = NSAppearance(named: .aqua) editor.backgroundColor = .white editor.textColor = .black scroll.backgroundColor = .white editor.isEditable = true editor.isRichText = false editor.allowsUndo = true editor.isAutomaticQuoteSubstitutionEnabled = false editor.isAutomaticDashSubstitutionEnabled = false editor.font = NSFont.monospacedSystemFont( ofSize: CGFloat( UserDefaults.standard.object( forKey: "preferences.editorFontSize" ) as? Double ?? 13 ), weight: .regular ) editor.textContainerInset = NSSize( width: 10, height: 8 ) editor.delegate = context.coordinator editor.string = text context.coordinator.highlight(editor) // The editor may be mounted after the Schemas navigation transition. DispatchQueue.main.async { [weak editor] in guard let editor, let window = editor.window else { return } window.makeFirstResponder(editor) } return scroll } func updateNSView( _ scroll: NSScrollView, context: Context ) { guard let editor = scroll.documentView as? NSTextView else { return } context.coordinator.parent = self if context.coordinator.lastFocusRequest != focusRequest { context.coordinator.lastFocusRequest = focusRequest DispatchQueue.main.async { [weak editor] in guard let editor, let window = editor.window else { return } window.makeFirstResponder(editor) } } if editor.string != text { let selection = editor.selectedRange() editor.string = text let safeLocation = min( selection.location, (text as NSString).length ) editor.setSelectedRange( NSRange(location: safeLocation, length: 0) ) context.coordinator.highlight(editor) } } final class Coordinator: NSObject, NSTextViewDelegate { var parent: SQLTextView var lastFocusRequest: Int init(_ parent: SQLTextView) { self.parent = parent self.lastFocusRequest = parent.focusRequest } private let completionWords = [ "SELECT", "FROM", "WHERE", "INSERT INTO", "UPDATE", "DELETE FROM", "CREATE TABLE", "ALTER TABLE", "DROP TABLE", "JOIN", "LEFT JOIN", "INNER JOIN", "GROUP BY", "ORDER BY", "HAVING", "LIMIT", "DISTINCT", "COUNT", "SUM", "AVG", "MIN", "MAX", "AS", "AND", "OR", "NOT", "NULL", "IS NULL", "LIKE", "IN", "BETWEEN", "EXISTS", "UNION", "WITH", "CASE", "WHEN", "THEN", "ELSE", "END", "SHOW TABLES", "SHOW DATABASES", "DESCRIBE", "EXPLAIN", "PRIMARY KEY", "FOREIGN KEY", "INDEX", "VALUES" ] func textView(_ textView: NSTextView, completions words: [String], forPartialWordRange charRange: NSRange, indexOfSelectedItem index: UnsafeMutablePointer?) -> [String] { let text = textView.string as NSString guard charRange.location != NSNotFound, NSMaxRange(charRange) <= text.length else { return [] } let prefix = text.substring(with: charRange) guard !prefix.isEmpty else { return [] } return Array(Set(completionWords + parent.completions)) .filter { $0.range(of: prefix, options: [.anchored, .caseInsensitive]) != nil } .sorted() } func textViewDidChangeSelection(_ notification: Notification) { guard let editor = notification.object as? NSTextView else { return } let range = editor.selectedRange() let value = editor.string as NSString let selection = range.length > 0 && NSMaxRange(range) <= value.length ? value.substring(with: range) : "" if parent.selectedSQL != selection { parent.selectedSQL = selection } } func textDidChange(_ notification: Notification) { guard let editor = notification.object as? NSTextView else { return } parent.text = editor.string highlight(editor) } func highlight(_ editor: NSTextView) { guard let storage = editor.textStorage else { return } let text = editor.string as NSString let range = NSRange( location: 0, length: text.length ) storage.beginEditing() storage.setAttributes([ .foregroundColor: NSColor.textColor, .font: NSFont.monospacedSystemFont( ofSize: CGFloat( UserDefaults.standard.object( forKey: "preferences.editorFontSize" ) as? Double ?? 13 ), weight: .regular ) ], range: range) let pattern = """ \\b(SELECT|FROM|WHERE|INSERT|UPDATE|DELETE|CREATE|TABLE|DATABASE|SHOW|USE|JOIN|LEFT|RIGHT|INNER|OUTER|ON|AS|AND|OR|ORDER|GROUP|BY|LIMIT|VALUES|SET|NULL|NOT|IN|LIKE|DISTINCT|ALTER|DROP|DESCRIBE|EXPLAIN)\\b """ if let regex = try? NSRegularExpression( pattern: pattern, options: [.caseInsensitive] ) { for match in regex.matches( in: editor.string, range: range ) { storage.addAttribute( .foregroundColor, value: NSColor.systemBlue, range: match.range ) } } storage.endEditing() } } } // MARK: - Connection Editor // MARK: - Phase 23: Locally stored EXPLAIN snapshots private struct WorkbenchPlanSnapshot: Identifiable, Codable { var id = UUID() var capturedAt = Date() var connectionName: String var database: String var statement: String var format: String var columnNames: [String] var resultRows: [[String]] var elapsedSeconds: Double var label: String { "\(capturedAt.formatted(date: .abbreviated, time: .shortened)) · \(format) · \(database)" } } private enum WorkbenchPlanArchive { private static let key = "WorkbenchReplacement.explainSnapshots.v1" static func load() -> [WorkbenchPlanSnapshot] { guard let data = UserDefaults.standard.data(forKey: key), let snapshots = try? JSONDecoder().decode([WorkbenchPlanSnapshot].self, from: data) else { return [] } return snapshots } static func save(_ snapshots: [WorkbenchPlanSnapshot]) { // Cap both snapshot count and size so EXPLAIN JSON cannot grow preferences without bound. let trimmed = Array(snapshots.prefix(40)) if let data = try? JSONEncoder().encode(trimmed), data.count <= 2_000_000 { UserDefaults.standard.set(data, forKey: key) } } } // MARK: - Phase 21: Native MySQL query plan inspector private struct WorkbenchExplainView: View { @Environment(\.dismiss) private var dismiss @ObservedObject var model: DatabaseModel let statement: String @State private var plan = QueryResult() @State private var errorText: String? @State private var isLoading = false @State private var useJSON = false @State private var duration: TimeInterval? @State private var snapshots: [WorkbenchPlanSnapshot] = WorkbenchPlanArchive.load() @State private var baselineID: UUID? @State private var showHistory = false private var matchingSnapshots: [WorkbenchPlanSnapshot] { snapshots.filter { $0.statement == normalizedStatement && $0.database == (model.selectedDatabase.isEmpty ? (model.selected?.database ?? "") : model.selectedDatabase) && $0.connectionName == (model.selected?.name ?? "") } } private var baseline: WorkbenchPlanSnapshot? { matchingSnapshots.first { $0.id == baselineID } } private var comparisonText: String? { guard let baseline, !plan.columns.isEmpty else { return nil } guard baseline.format == (useJSON ? "JSON" : "Table") else { return "Baseline format differs from the current plan. Select a matching format." } if useJSON { return baseline.resultRows == plan.rows ? "JSON plan unchanged from baseline." : "JSON plan differs from baseline. Inspect the plan text for details." } func estimatedRows(_ columns: [String], _ rows: [[String]]) -> Int? { guard let index = columns.firstIndex(where: { $0.lowercased() == "rows" }) else { return nil } let values = rows.compactMap { index < $0.count ? Int($0[index]) : nil } return values.isEmpty ? nil : values.reduce(0, +) } let oldEstimate = estimatedRows(baseline.columnNames, baseline.resultRows) let newEstimate = estimatedRows(plan.columns, plan.rows) let changed = baseline.columnNames != plan.columns || baseline.resultRows != plan.rows var message = changed ? "Plan changed since baseline." : "Plan unchanged from baseline." if let oldEstimate, let newEstimate { message += " Estimated rows across steps: \(oldEstimate) → \(newEstimate)." } return message + " Estimates are not measured query timings." } // Phase 24: Compare tabular plan steps by position and table, not only // by aggregate row estimates. EXPLAIN estimates are not runtime metrics. private var stepComparison: [String] { guard let baseline, !useJSON, baseline.format == "Table", !plan.columns.isEmpty else { return [] } func value(_ columns: [String], _ row: [String], _ field: String) -> String { guard let index = columns.firstIndex(where: { $0.caseInsensitiveCompare(field) == .orderedSame }), index < row.count else { return "—" } return row[index] } var details: [String] = [] let count = max(baseline.resultRows.count, plan.rows.count) for index in 0.. String { guard let index = names.firstIndex(of: name), index < row.count else { return "" } return row[index] } return plan.rows.compactMap { row -> String? in let table = value(row, "table") let access = value(row, "type").uppercased() let selected = value(row, "key") let possible = value(row, "possible_keys") let label = table.isEmpty ? "Query step" : "Table \(table)" if access == "ALL" && possible.uppercased() != "NULL" && !possible.isEmpty { return "\(label): full scan despite candidate indexes [\(possible)]. Review predicate selectivity and statistics." } if access == "ALL" && (possible.isEmpty || possible.uppercased() == "NULL") { return "\(label): full scan with no candidate index listed. Inspect WHERE/JOIN predicates before considering an index." } if selected.isEmpty || selected.uppercased() == "NULL" { return "\(label): optimizer did not select an index. This may be optimal for small tables." } return nil } } private var diagnosticReport: String { let sections = [ "SQL:", normalizedStatement, "", "Plan observations:", planInsights.isEmpty ? "None" : planInsights.joined(separator: "\n"), "", "Index analysis:", indexDiagnostics.isEmpty ? "None (or JSON format selected)" : indexDiagnostics.joined(separator: "\n"), "", "Baseline comparison:", comparisonText ?? "No baseline selected", "", "Changed plan steps:", stepComparison.isEmpty ? "None" : stepComparison.joined(separator: "\n"), "", "These are optimizer estimates, not measured query performance." ] return sections.joined(separator: "\n") } private var normalizedStatement: String { var value = statement.trimmingCharacters(in: .whitespacesAndNewlines) if value.hasSuffix(";") { value.removeLast() } return value.trimmingCharacters(in: .whitespacesAndNewlines) } // Advisory analysis of the tabular EXPLAIN result. These are optimizer // estimates, not measured execution statistics or automatic index DDL. private var planInsights: [String] { guard !useJSON, !plan.columns.isEmpty else { return [] } let headings = plan.columns.map { $0.lowercased() } func field(_ row: [String], _ name: String) -> String { guard let index = headings.firstIndex(of: name), index < row.count else { return "" } return row[index] } var messages: [String] = [] for row in plan.rows { let table = field(row, "table") let label = table.isEmpty ? "A query step" : "Table \(table)" let access = field(row, "type").uppercased() let key = field(row, "key") let estimatedRows = field(row, "rows") let extra = field(row, "extra").lowercased() if access == "ALL" { messages.append("\(label): full table scan (type=ALL). Check selectivity, predicates and available indexes before considering a new index.") } else if access == "INDEX" { messages.append("\(label): full index scan (type=index). This is not necessarily a problem for small tables or covering indexes.") } if key.isEmpty || key.uppercased() == "NULL" { messages.append("\(label): no index selected for this step. An index is not always beneficial.") } if extra.contains("using temporary") { messages.append("\(label): temporary table may be used for grouping, sorting or other operations.") } if extra.contains("using filesort") { messages.append("\(label): additional sort operation indicated (Using filesort).") } if !estimatedRows.isEmpty, estimatedRows.uppercased() != "NULL" { messages.append("\(label): optimizer estimates \(estimatedRows) rows examined for this step.") } } return messages } private var copyablePlan: String { if useJSON { return plan.rows.first?.first ?? "" } return ([plan.columns.joined(separator: "\t")] + plan.rows.map { $0.joined(separator: "\t") }) .joined(separator: "\n") } private var eligible: Bool { // Deliberately restrict this inspector to SELECT queries. EXPLAIN for // UPDATE/DELETE is supported by MySQL but not offered here to keep // the first diagnostic release conservative. let upper = normalizedStatement.uppercased() return upper.hasPrefix("SELECT ") || upper.hasPrefix("SELECT\n") || upper.hasPrefix("WITH ") || upper.hasPrefix("WITH\n") } var body: some View { VStack(alignment: .leading, spacing: 12) { HStack { Label("Query Execution Plan", systemImage: "chart.bar.doc.horizontal") .font(.title2.bold()) Spacer() Button("Close") { dismiss() } } Text("Uses MySQL EXPLAIN to inspect the optimizer's estimated plan without executing the SELECT statement itself.") .font(.callout) .foregroundStyle(.secondary) Text(normalizedStatement) .font(.system(.callout, design: .monospaced)) .lineLimit(4) .textSelection(.enabled) .frame(maxWidth: .infinity, alignment: .leading) .padding(8) .background(Color(nsColor: .controlBackgroundColor)) HStack { Picker("Plan format", selection: $useJSON) { Text("Table").tag(false) Text("JSON").tag(true) } .pickerStyle(.segmented) .frame(width: 240) Button("Run EXPLAIN") { Task { await loadPlan() } } .disabled(isLoading || !eligible || model.selected == nil) Button("Save Snapshot") { saveSnapshot() } .disabled(plan.columns.isEmpty || isLoading) Button("History") { showHistory.toggle() } if let baseline { Text("Comparing with \(baseline.capturedAt.formatted(date: .abbreviated, time: .shortened))") .font(.caption) .foregroundStyle(.secondary) Button("Clear Baseline") { baselineID = nil } } Button("Copy Plan") { NSPasteboard.general.clearContents() NSPasteboard.general.setString(copyablePlan, forType: .string) } .disabled(plan.columns.isEmpty) if isLoading { ProgressView().controlSize(.small) } if let duration { Text(String(format: "%.3f s", duration)) .foregroundStyle(.secondary) } Spacer() } if !eligible { ContentUnavailableView("SELECT query required", systemImage: "exclamationmark.triangle", description: Text("Select a SELECT or WITH query in the SQL editor before opening EXPLAIN.")) } else if let errorText { ScrollView { Text(errorText) .foregroundStyle(.red) .textSelection(.enabled) .frame(maxWidth: .infinity, alignment: .leading) } } else if useJSON, let json = plan.rows.first?.first { ScrollView([.vertical, .horizontal]) { Text(json) .font(.system(.body, design: .monospaced)) .textSelection(.enabled) .frame(maxWidth: .infinity, alignment: .leading) } } else if !plan.columns.isEmpty { NativeResultGrid(columns: plan.columns, rows: plan.rows) } else { ContentUnavailableView("No plan loaded", systemImage: "chart.bar.doc.horizontal") } if showHistory { DisclosureGroup("Saved plans for this query (\(matchingSnapshots.count))", isExpanded: $showHistory) { if matchingSnapshots.isEmpty { Text("No saved plans for this connection, schema and SQL statement.") .foregroundStyle(.secondary) } else { ScrollView { VStack(alignment: .leading, spacing: 6) { ForEach(matchingSnapshots) { snapshot in HStack { Text(snapshot.label) .font(.callout) Spacer() Button("Compare") { baselineID = snapshot.id } Button("Delete", role: .destructive) { snapshots.removeAll { $0.id == snapshot.id } if baselineID == snapshot.id { baselineID = nil } WorkbenchPlanArchive.save(snapshots) } } } } } .frame(maxHeight: 130) } } } if let comparisonText { Label(comparisonText, systemImage: "arrow.left.arrow.right") .font(.callout) .textSelection(.enabled) } if !plan.columns.isEmpty { HStack { Label("Index analysis", systemImage: "magnifyingglass") .font(.headline) Spacer() Button("Copy Diagnostics") { NSPasteboard.general.clearContents() NSPasteboard.general.setString(diagnosticReport, forType: .string) } } if !indexDiagnostics.isEmpty { ScrollView { VStack(alignment: .leading, spacing: 6) { ForEach(Array(indexDiagnostics.enumerated()), id: \.offset) { _, note in Text(note) .frame(maxWidth: .infinity, alignment: .leading) .textSelection(.enabled) } } } .frame(maxHeight: 100) } else { Text("No additional index warnings for this plan format.") .font(.caption) .foregroundStyle(.secondary) } } if !stepComparison.isEmpty { DisclosureGroup("Changed plan steps (\(stepComparison.count))") { ScrollView { VStack(alignment: .leading, spacing: 6) { ForEach(Array(stepComparison.enumerated()), id: \.offset) { _, change in Text(change) .frame(maxWidth: .infinity, alignment: .leading) .textSelection(.enabled) } } } .frame(maxHeight: 130) } } if !planInsights.isEmpty { DisclosureGroup("Plan observations (\(planInsights.count))") { ScrollView { VStack(alignment: .leading, spacing: 6) { ForEach(Array(planInsights.enumerated()), id: \.offset) { _, observation in Label(observation, systemImage: "info.circle") .frame(maxWidth: .infinity, alignment: .leading) } } .font(.callout) .padding(.vertical, 6) } .frame(maxHeight: 160) } } Text("EXPLAIN shows estimates, not actual query execution times. Observations are heuristics, not index recommendations. Confirm with representative data before changing production indexes.") .font(.caption) .foregroundStyle(.secondary) } .padding(16) .frame(minWidth: 760, minHeight: 490) .onChange(of: useJSON) { _, _ in plan = QueryResult() errorText = nil Task { await loadPlan() } } .task { await loadPlan() } } @MainActor private func saveSnapshot() { guard let profile = model.selected, !plan.columns.isEmpty else { return } let database = model.selectedDatabase.isEmpty ? profile.database : model.selectedDatabase let snapshot = WorkbenchPlanSnapshot( connectionName: profile.name, database: database, statement: normalizedStatement, format: useJSON ? "JSON" : "Table", columnNames: plan.columns, resultRows: plan.rows, elapsedSeconds: duration ?? 0 ) snapshots.insert(snapshot, at: 0) WorkbenchPlanArchive.save(snapshots) showHistory = true } @MainActor private func loadPlan() async { guard eligible, let profile = model.selected, !isLoading else { return } isLoading = true errorText = nil plan = QueryResult() let started = Date() defer { duration = Date().timeIntervalSince(started) isLoading = false } let database = model.selectedDatabase.isEmpty ? profile.database : model.selectedDatabase let query = (useJSON ? "EXPLAIN FORMAT=JSON " : "EXPLAIN ") + normalizedStatement do { plan = try await NativeMySQLRunner.run( profile: profile, database: database, password: PasswordStore.read(profile.id), sql: query ) } catch { errorText = "EXPLAIN failed: " + error.localizedDescription } } } struct ConnectionEditor: View { @Binding var profile: ConnectionProfile @Binding var password: String let save: () -> Void @Environment(\.dismiss) private var dismiss var body: some View { VStack(alignment: .leading, spacing: 14) { Text("Connection Settings") .font(.title2.bold()) Form { TextField( "Connection Name", text: $profile.name ) TextField( "Hostname", text: $profile.host ) TextField( "Port", value: $profile.port, format: .number ) TextField( "Username", text: $profile.username ) SecureField( "Password (macOS Keychain)", text: $password ) TextField( "Default Schema", text: $profile.database ) } HStack { Spacer() Button("Cancel") { dismiss() } Button("Save") { save() } .buttonStyle(.borderedProminent) } } .padding(24) .frame(width: 520) } } // MARK: - High Performance Native Result Grid struct NativeResultGrid: NSViewRepresentable { let columns: [String] let rows: [[String]] var hiddenColumnIndices: Set = [] func makeCoordinator() -> Coordinator { Coordinator() } func makeNSView(context: Context) -> NSScrollView { let scroll = NSScrollView() scroll.hasVerticalScroller = true scroll.hasHorizontalScroller = true scroll.autohidesScrollers = true scroll.drawsBackground = true scroll.backgroundColor = .white let table = NSTableView() table.headerView = NSTableHeaderView() table.usesAlternatingRowBackgroundColors = UserDefaults.standard.object( forKey: "preferences.alternatingRows" ) as? Bool ?? true table.rowHeight = CGFloat( UserDefaults.standard.object( forKey: "preferences.resultRowHeight" ) as? Double ?? 24 ) table.intercellSpacing = NSSize(width: 1, height: 1) table.allowsMultipleSelection = true table.allowsColumnResizing = true table.allowsColumnReordering = true table.columnAutoresizingStyle = .noColumnAutoresizing table.backgroundColor = .white context.coordinator.columns = columns context.coordinator.rows = rows table.delegate = context.coordinator table.dataSource = context.coordinator configureColumns(table, hiddenIndices: hiddenColumnIndices) scroll.documentView = table table.reloadData() return scroll } func updateNSView( _ scroll: NSScrollView, context: Context ) { guard let table = scroll.documentView as? NSTableView else { return } let oldColumns = context.coordinator.columns context.coordinator.columns = columns context.coordinator.rows = rows if oldColumns != columns || Set(table.tableColumns.compactMap { column -> Int? in let identifier = column.identifier.rawValue return identifier.hasPrefix("column_") ? Int(identifier.dropFirst(7)) : nil }) != Set(columns.indices).subtracting(hiddenColumnIndices) { configureColumns(table, hiddenIndices: hiddenColumnIndices) } table.reloadData() } private func configureColumns(_ table: NSTableView, hiddenIndices: Set) { for column in Array(table.tableColumns) { table.removeTableColumn(column) } for (index, name) in columns.enumerated() where !hiddenIndices.contains(index) { let identifier = NSUserInterfaceItemIdentifier( "column_\(index)" ) let column = NSTableColumn(identifier: identifier) column.title = name column.width = CGFloat( UserDefaults.standard.object( forKey: "preferences.resultColumnWidth" ) as? Double ?? 170 ) column.minWidth = 70 column.maxWidth = 700 column.resizingMask = .userResizingMask table.addTableColumn(column) } } final class Coordinator: NSObject, NSTableViewDataSource, NSTableViewDelegate { var columns: [String] = [] var rows: [[String]] = [] func numberOfRows(in tableView: NSTableView) -> Int { rows.count } func tableView( _ tableView: NSTableView, viewFor tableColumn: NSTableColumn?, row: Int ) -> NSView? { guard let tableColumn, row >= 0, row < rows.count else { return nil } let identifier = tableColumn.identifier.rawValue guard identifier.hasPrefix("column_"), let columnIndex = Int(identifier.dropFirst(7)), columnIndex < columns.count else { return nil } let cellIdentifier = NSUserInterfaceItemIdentifier("ResultCell") let cell: NSTableCellView if let reused = tableView.makeView( withIdentifier: cellIdentifier, owner: nil ) as? NSTableCellView { cell = reused } else { cell = NSTableCellView() cell.identifier = cellIdentifier let field = NSTextField(labelWithString: "") field.font = NSFont.monospacedSystemFont( ofSize: 12, weight: .regular ) field.lineBreakMode = .byTruncatingTail field.translatesAutoresizingMaskIntoConstraints = false cell.addSubview(field) cell.textField = field NSLayoutConstraint.activate([ field.leadingAnchor.constraint( equalTo: cell.leadingAnchor, constant: 7 ), field.trailingAnchor.constraint( equalTo: cell.trailingAnchor, constant: -7 ), field.centerYAnchor.constraint( equalTo: cell.centerYAnchor ) ]) } let values = rows[row] cell.textField?.stringValue = columnIndex < values.count ? values[columnIndex] : "" return cell } } } // MARK: - MySQL Administration enum WorkbenchAdminPage: String, CaseIterable, Identifiable { case serverStatus = "Server Status" case serverCounters = "Server Counters" case clientConnections = "Client Connections" case users = "Users and Privileges" case variables = "Status and System Variables" case dashboard = "Dashboard" case performanceReports = "Performance Reports" case performanceSchema = "Performance Schema Setup" case serverLogs = "Server Error Log" case slowQueries = "Slow Query Analysis" case tableIO = "Table I/O Hotspots" case innodbMetrics = "InnoDB Metrics" case innodbLocks = "InnoDB Lock Waits" case connectionStatistics = "Connection Statistics" var id: String { rawValue } var icon: String { switch self { case .slowQueries: return "tortoise" case .tableIO: return "externaldrive" case .innodbMetrics: return "internaldrive" case .innodbLocks: return "lock.shield" case .connectionStatistics: return "network" case .serverLogs: return "text.alignleft" case .serverStatus: return "server.rack" case .serverCounters: return "chart.xyaxis.line" case .clientConnections: return "person.2" case .users: return "person.crop.circle.badge.checkmark" case .variables: return "slider.horizontal.3" case .dashboard: return "chart.xyaxis.line" case .performanceReports: return "chart.bar.xaxis" case .performanceSchema: return "gauge.with.dots.needle.67percent" } } var statement: String { switch self { case .slowQueries: return """ SELECT DIGEST_TEXT, COUNT_STAR, SUM_TIMER_WAIT, AVG_TIMER_WAIT, MAX_TIMER_WAIT, SUM_ROWS_EXAMINED, SUM_ROWS_SENT, SUM_NO_INDEX_USED, FIRST_SEEN, LAST_SEEN FROM performance_schema.events_statements_summary_by_digest WHERE DIGEST_TEXT IS NOT NULL ORDER BY SUM_TIMER_WAIT DESC LIMIT 100 """ case .tableIO: return """ SELECT OBJECT_SCHEMA, OBJECT_NAME, COUNT_READ, COUNT_WRITE, SUM_TIMER_READ, SUM_TIMER_WRITE FROM performance_schema.table_io_waits_summary_by_table WHERE OBJECT_SCHEMA IS NOT NULL ORDER BY SUM_TIMER_WAIT DESC LIMIT 100 """ case .innodbMetrics: return """ SHOW GLOBAL STATUS WHERE Variable_name IN ( 'Innodb_buffer_pool_read_requests', 'Innodb_buffer_pool_reads', 'Innodb_buffer_pool_pages_total', 'Innodb_buffer_pool_pages_free', 'Innodb_buffer_pool_pages_dirty', 'Innodb_data_reads', 'Innodb_data_writes', 'Innodb_log_waits', 'Innodb_row_lock_waits', 'Innodb_row_lock_time', 'Innodb_rows_read', 'Innodb_rows_inserted', 'Innodb_rows_updated', 'Innodb_rows_deleted' ) """ case .innodbLocks: return """ SELECT REQUESTING_ENGINE_TRANSACTION_ID, BLOCKING_ENGINE_TRANSACTION_ID, REQUESTING_ENGINE_LOCK_ID, BLOCKING_ENGINE_LOCK_ID FROM performance_schema.data_lock_waits LIMIT 100 """ case .connectionStatistics: return """ SHOW GLOBAL STATUS WHERE Variable_name IN ( 'Uptime', 'Connections', 'Max_used_connections', 'Threads_connected', 'Threads_running', 'Threads_cached', 'Threads_created', 'Aborted_clients', 'Aborted_connects', 'Connection_errors_max_connections', 'Slow_queries', 'Questions', 'Queries', 'Created_tmp_disk_tables', 'Select_full_join', 'Sort_merge_passes' ) """ case .serverLogs: return "SELECT LOGGED, PRIO, ERROR_CODE, SUBSYSTEM, DATA FROM performance_schema.error_log ORDER BY LOGGED DESC LIMIT 500" case .serverCounters: return "SHOW GLOBAL STATUS" case .serverStatus: return """ SHOW GLOBAL STATUS WHERE Variable_name IN ( 'Uptime', 'Connections', 'Max_used_connections', 'Threads_connected', 'Threads_running', 'Threads_cached', 'Queries', 'Questions', 'Slow_queries', 'Bytes_received', 'Bytes_sent', 'Aborted_clients', 'Aborted_connects', 'Innodb_buffer_pool_read_requests', 'Innodb_buffer_pool_reads', 'Innodb_data_reads', 'Innodb_data_writes', 'Created_tmp_tables', 'Created_tmp_disk_tables' ) """ case .clientConnections: return "SHOW FULL PROCESSLIST" case .users: return """ SELECT User, Host, plugin, account_locked FROM mysql.user ORDER BY User, Host """ case .variables: return "SHOW GLOBAL VARIABLES" case .dashboard: return """ SHOW GLOBAL STATUS WHERE Variable_name IN ( 'Uptime', 'Connections', 'Threads_connected', 'Threads_running', 'Questions', 'Queries', 'Bytes_received', 'Bytes_sent', 'Innodb_buffer_pool_reads', 'Innodb_buffer_pool_read_requests', 'Slow_queries' ) """ case .performanceReports: return """ SELECT * FROM sys.statement_analysis LIMIT 100 """ case .performanceSchema: return """ SELECT NAME, ENABLED, TIMED FROM performance_schema.setup_instruments LIMIT 250 """ } } } private enum WorkbenchProcessAction: String { case query = "Stop query for" case connection = "Disconnect" } struct WorkbenchAdministrationView: View { let page: WorkbenchAdminPage let profile: ConnectionProfile? let close: () -> Void @State private var result = QueryResult() @State private var loading = false @State private var errorMessage: String? @State private var lastUpdated: Date? @State private var searchText = "" @State private var autoRefresh = true @State private var selectedAccount: String? = nil @State private var grants: [String] = [] @State private var grantsError: String? = nil @State private var showingGrants = false @State private var showingAccountManager = false @State private var selectedProcessID: String? = nil @State private var pendingProcessAction: WorkbenchProcessAction? = nil @State private var processActionError: String? = nil @State private var processActionStatus: String? = nil @State private var terminatingProcess = false // Phase 25/26: bounded, in-session samples. No credentials or SQL persisted. @State private var diagnosticSamples: [DiagnosticSample] = [] @State private var showingTrends = false private var supportsTrends: Bool { page == .slowQueries || page == .connectionStatistics || page == .innodbMetrics || page == .dashboard || page == .serverStatus || page == .serverCounters } private var diagnosticGuidance: String? { switch page { case .slowQueries: return "Aggregated statement digests since Performance Schema statistics were last reset. Times are in picoseconds; this is not the server slow-query log. SQL text may be truncated." case .tableIO: return "Cumulative Performance Schema table I/O counters since last reset. Times are in picoseconds; refresh to observe changes." case .innodbMetrics: return "Cumulative InnoDB status counters; calculate rates from measurements at two different times." case .innodbLocks: return "Current InnoDB lock-wait relationships only. An empty result means no waits were visible at the time of the query." case .connectionStatistics: return "Server-wide counters since startup. Connections is cumulative, whereas Threads_connected is the current count." default: return nil } } private var filteredRows: [[String]] { guard !searchText.isEmpty else { return result.rows } return result.rows.filter { row in row.contains { $0.localizedCaseInsensitiveContains(searchText) } } } var body: some View { VStack(spacing: 0) { header Divider() if loading && result.columns.isEmpty { Spacer() ProgressView("Loading \(page.rawValue)…") Spacer() } else if let errorMessage { ContentUnavailableView { Label( "Unable to Load \(page.rawValue)", systemImage: "exclamationmark.triangle" ) } description: { Text(errorMessage) } actions: { Button("Retry") { Task { await refresh() } } } .frame(maxWidth: .infinity, maxHeight: .infinity) } else if result.columns.isEmpty { ContentUnavailableView( "No Data", systemImage: "tablecells", description: Text( "The server did not return any columns." ) ) .frame(maxWidth: .infinity, maxHeight: .infinity) } else { resultContent .layoutPriority(1) } if let guidance = diagnosticGuidance { Text(guidance) .font(.caption) .foregroundStyle(.secondary) .frame(maxWidth: .infinity, alignment: .leading) .padding(.horizontal, 16) .padding(.vertical, 8) } if supportsTrends && !diagnosticSamples.isEmpty && page != .dashboard && page != .serverStatus { HStack(spacing: 12) { Label("\(diagnosticSamples.count) samples", systemImage: "chart.xyaxis.line") .foregroundStyle(.secondary) Button("View Trends…") { showingTrends = true } Button("Clear Samples") { diagnosticSamples.removeAll() } Spacer() Text("Samples are kept in memory for this screen only") .font(.caption).foregroundStyle(.secondary) } .padding(.horizontal, 16) .padding(.vertical, 6) } if page == .clientConnections { processControls } if page == .users && !result.rows.isEmpty { HStack { Picker("Account", selection: $selectedAccount) { Text("Select account").tag(String?.none) ForEach(result.rows.indices, id: \.self) { i in let row = result.rows[i] if row.count >= 2 { Text("\(row[0])@\(row[1])") .tag(Optional("\(row[0])\u{001f}\(row[1])")) } } } .frame(maxWidth: 320) Button("Show Grants") { Task { await showGrants() } } .disabled(selectedAccount == nil) Button("Manage Accounts…") { autoRefresh = false showingAccountManager = true } .disabled(profile == nil) Spacer() Text("Changes require confirmation") .foregroundStyle(.secondary) } .padding(.horizontal, 16) .padding(.vertical, 8) } Divider() footer } .frame(maxWidth: .infinity, maxHeight: .infinity) .task(id: "\(page.id)-\(profile?.id.uuidString ?? "")") { await refresh() } .task(id: "\(page.id)-\(autoRefresh)-\(profile?.id.uuidString ?? "")") { // Gather a second snapshot promptly on first display so rate graphs // draw a line without waiting for the full five-second interval. if autoRefresh && supportsTrends { try? await Task.sleep(for: .seconds(1.5)) if !Task.isCancelled && autoRefresh { if loading { try? await Task.sleep(for: .seconds(1)) } if !Task.isCancelled && autoRefresh { await refresh() } } } while autoRefresh && !Task.isCancelled { try? await Task.sleep(for: .seconds(5)) if Task.isCancelled || !autoRefresh { break } await refresh() } } .confirmationDialog( "Terminate MySQL session?", isPresented: Binding( get: { pendingProcessAction != nil }, set: { if !$0 { pendingProcessAction = nil } } ), titleVisibility: .visible ) { if let action = pendingProcessAction, let id = selectedProcessID { Button("\(action.rawValue) session \(id)", role: .destructive) { pendingProcessAction = nil Task { await terminateProcess(id: id, action: action) } } } Button("Cancel", role: .cancel) { pendingProcessAction = nil } } message: { Text("This operation affects a live MySQL session. Terminating a connection can interrupt transactions and roll back uncommitted work. Verify the session ID before proceeding.") } .sheet(isPresented: $showingAccountManager, onDismiss: { Task { await refresh() } }) { if let profile { WorkbenchAccountManagerView( profile: profile, selectedAccount: selectedAccount ) } } .sheet(isPresented: $showingTrends) { WorkbenchDiagnosticTrendsView(page: page, samples: diagnosticSamples) } .sheet(isPresented: $showingGrants) { VStack(alignment: .leading, spacing: 12) { Text("Grants: \(selectedAccount ?? "")") .font(.title2.bold()) if let grantsError { Text(grantsError).foregroundStyle(.red) } else { ScrollView { Text(grants.joined(separator: "\n\n")) .font(.system(.body, design: .monospaced)) .textSelection(.enabled) .frame(maxWidth: .infinity, alignment: .leading) } } Button("Close") { showingGrants = false } } .padding(20) .frame(minWidth: 580, minHeight: 300) } } private var processControls: some View { VStack(alignment: .leading, spacing: 6) { HStack(spacing: 12) { Picker("Session", selection: $selectedProcessID) { Text("Select a session").tag(String?.none) ForEach(processIDs, id: \.self) { id in Text("ID \(id)").tag(Optional(id)) } } .frame(maxWidth: 250) Button("Stop Query") { pendingProcessAction = .query } .disabled(selectedProcessID == nil || terminatingProcess) Button("Disconnect Session") { pendingProcessAction = .connection } .disabled(selectedProcessID == nil || terminatingProcess) Spacer() if terminatingProcess { ProgressView().controlSize(.small) } } if let processActionError { Text(processActionError).foregroundStyle(.red).textSelection(.enabled) } if let processActionStatus { Text(processActionStatus).foregroundStyle(.secondary) } Text("Requires PROCESS privilege to see other users' sessions and appropriate privileges to terminate them.") .font(.caption).foregroundStyle(.secondary) } .padding(.horizontal, 16) .padding(.vertical, 8) } private var processIDs: [String] { guard let index = result.columns.firstIndex(where: { $0.caseInsensitiveCompare("Id") == .orderedSame }) else { return [] } return result.rows.compactMap { row in guard row.indices.contains(index), let id = UInt64(row[index]), id > 0 else { return nil } return String(id) } } @MainActor private func terminateProcess(id: String, action: WorkbenchProcessAction) async { guard let profile, let numericID = UInt64(id), numericID > 0 else { return } terminatingProcess = true processActionError = nil processActionStatus = nil defer { terminatingProcess = false } do { let own = try await NativeMySQLRunner.run( profile: profile, database: "", password: PasswordStore.read(profile.id), sql: "SELECT CONNECTION_ID()" ) if own.rows.first?.first == id { processActionError = "Cannot terminate this application's own MySQL session." return } let sql = "KILL \(action == .query ? "QUERY" : "CONNECTION") \(numericID)" _ = try await NativeMySQLRunner.run( profile: profile, database: "", password: PasswordStore.read(profile.id), sql: sql ) processActionStatus = "\(action.rawValue) requested for session \(id)." selectedProcessID = nil await refresh() } catch { processActionError = error.localizedDescription } } private var header: some View { HStack(spacing: 14) { Image(systemName: page.icon) .font(.system(size: 27)) .foregroundStyle(.blue) .frame(width: 40) VStack(alignment: .leading, spacing: 3) { Text(profile?.name ?? "No Connection") .font(.caption) .foregroundStyle(.secondary) Text(page.rawValue) .font(.system(size: 22, weight: .semibold)) } Spacer() if page != .dashboard { TextField("Filter results", text: $searchText) .textFieldStyle(.roundedBorder) .frame(width: 175) } Toggle("Auto-refresh", isOn: $autoRefresh) .toggleStyle(.checkbox) .help("Refresh every five seconds") Button { Task { await refresh() } } label: { Label("Refresh", systemImage: "arrow.clockwise") } .disabled(loading) Button { close() } label: { Label("SQL Editor", systemImage: "chevron.left.forwardslash.chevron.right") } } .padding(18) } // The live charts occupy a fixed region ABOVE an independently scrolling // counter table. They cannot disappear when the user scrolls the rows. @ViewBuilder private var resultContent: some View { switch page { case .serverStatus, .dashboard: // Phase 51: one outer scroll container owns the entire dashboard. // The former fixed-height top chart pane consumed the available // window height and left the overview pane impossible to reach. ScrollView(.vertical) { VStack(alignment: .leading, spacing: 18) { WorkbenchInlinePerformanceDashboard( samples: diagnosticSamples, chartHeight: 155 ) .accessibilityIdentifier("live-mysql-line-graphs-phase51") Divider() Label("Server overview · Live graphs", systemImage: "chart.xyaxis.line") .font(.title3.bold()) WorkbenchServerOverview(rows: filteredRows, samples: diagnosticSamples) if page == .serverStatus { Divider() HStack { Text("Server status metrics").font(.headline) Spacer() Text("\(filteredRows.count) metrics") .font(.caption).foregroundStyle(.secondary) } // The grid owns its own bounded viewport; the outer // page can still scroll to every section below charts. WorkbenchColouredCounterTable( columns: ["Metric", "Value"], rows: filteredRows ) .frame(height: 320) } } .padding(18) .frame(maxWidth: .infinity, alignment: .leading) } .frame(maxWidth: .infinity, maxHeight: .infinity) .accessibilityIdentifier("scrollable-dashboard-phase51") case .serverCounters, .connectionStatistics, .innodbMetrics: WorkbenchLiveMonitorScreen( page: page, samples: diagnosticSamples, rows: filteredRows, allRowCount: result.rows.count ) .frame(maxWidth: .infinity, maxHeight: .infinity) default: // Status and System Variables is deliberately a distinct variable // inspector, never replaced by the monitoring screen. NativeResultGrid(columns: result.columns, rows: filteredRows) .frame(maxWidth: .infinity, maxHeight: .infinity) } } private var footer: some View { HStack { if let lastUpdated { Text( "Updated \(lastUpdated.formatted(date: .omitted, time: .standard))" ) } else { Text("Not loaded") } Spacer() Text("\(result.rows.count) rows") } .font(.caption) .foregroundStyle(.secondary) .padding(.horizontal, 16) .frame(height: 30) } @MainActor private func showGrants() async { guard let profile, let selectedAccount else { return } let components = selectedAccount.split(separator: "\u{001f}", omittingEmptySubsequences: false) guard components.count == 2 else { return } func quoted(_ text: String) -> String { "'" + text.replacingOccurrences(of: "'", with: "''") + "'" } grants = [] grantsError = nil showingGrants = true do { let sql = "SHOW GRANTS FOR \(quoted(String(components[0])))@\(quoted(String(components[1])))" let response = try await NativeMySQLRunner.run( profile: profile, database: "", password: PasswordStore.read(profile.id), sql: sql ) grants = response.rows.compactMap { $0.first } } catch { grantsError = error.localizedDescription } } @MainActor private func refresh() async { guard let profile else { errorMessage = "Select a MySQL connection first." return } guard !loading else { return } loading = true errorMessage = nil defer { loading = false } do { let response = try await NativeMySQLRunner.run( profile: profile, database: "", password: PasswordStore.read(profile.id), sql: page.statement ) result = response if supportsTrends { diagnosticSamples.append(DiagnosticSample(at: Date(), columns: response.columns, rows: response.rows)) if diagnosticSamples.count > 60 { diagnosticSamples.removeFirst(diagnosticSamples.count - 60) } } if page == .clientConnections, let selectedProcessID, !processIDs.contains(selectedProcessID) { self.selectedProcessID = nil } lastUpdated = Date() } catch { errorMessage = error.localizedDescription } } } // MARK: - Built-in Database Import / Export struct WorkbenchDatabaseTransferView: View { @ObservedObject var model: DatabaseModel @Environment(\.dismiss) private var dismiss @State private var selectedMode = 0 @State private var database = "" @State private var isWorking = false @State private var progress = 0.0 @State private var transferredRows: Int64 = 0 @State private var currentTableRows: Int64 = 0 @State private var currentTableExpected: Int64 = 0 @State private var status = "Ready" @State private var operationTask: Task? @State private var showingImportConfirmation = false @State private var pendingImportURL: URL? @State private var availableTables: [String] = [] @State private var selectedTables: Set = [] @State private var exportStructure = true @State private var exportData = true @State private var includeDrop = false @State private var includeViews = true @State private var includeTriggers = false @State private var includeRoutines = false @State private var availableViews: [String] = [] @State private var selectedViews: Set = [] @State private var loadingTables = false @State private var availableDatabases: [String] = [] @State private var loadingDatabases = false @State private var lastBackupURL: URL? @State private var lastVerificationReport = "" private var selectedProfile: ConnectionProfile? { model.selected } var body: some View { VStack(spacing: 0) { header Divider() Picker("Operation", selection: $selectedMode) { Text("Data Export").tag(0) Text("Data Import / Restore").tag(1) } .pickerStyle(.segmented) .padding(20) ScrollView { VStack(alignment: .leading, spacing: 16) { GroupBox("MySQL Connection") { VStack(alignment: .leading, spacing: 10) { LabeledContent("Connection") { Text(selectedProfile?.name ?? "Not selected") } Picker("Database", selection: $database) { Text("Select Database").tag("") ForEach(availableDatabases, id: \.self) { name in Text(name).tag(name) } } HStack { if loadingDatabases { ProgressView().controlSize(.small) } Button("Refresh Databases") { Task { await loadTransferDatabases() } } .disabled(loadingDatabases || selectedProfile == nil) } } } if selectedMode == 0 { GroupBox("Export Options") { VStack(alignment: .leading, spacing: 10) { Toggle("Table structure", isOn: $exportStructure) Toggle("Table data", isOn: $exportData) Toggle("Add DROP TABLE IF EXISTS", isOn: $includeDrop) .disabled(!exportStructure) Toggle("Include views (definitions)", isOn: $includeViews) .disabled(!exportStructure) Toggle("Include triggers", isOn: $includeTriggers) .disabled(!exportStructure) Toggle("Include stored procedures and functions", isOn: $includeRoutines) .disabled(!exportStructure) HStack { Text("Tables (\(selectedTables.count) selected)") Spacer() Button("All") { selectedTables = Set(availableTables) } Button("None") { selectedTables.removeAll() } Button("Refresh") { Task { await loadTables() } } .disabled(loadingTables) } if loadingTables { ProgressView("Loading tables…") } if availableTables.isEmpty && !loadingTables { Text("No tables found. Select a database or refresh the list.") .foregroundStyle(.secondary) } ForEach(availableTables, id: \.self) { table in Toggle(table, isOn: Binding( get: { selectedTables.contains(table) }, set: { checked in if checked { selectedTables.insert(table) } else { selectedTables.remove(table) } } )) } if includeViews && exportStructure { HStack { Text("Views (\(selectedViews.count) selected)") Spacer() Button("All") { selectedViews = Set(availableViews) } Button("None") { selectedViews.removeAll() } } ForEach(availableViews, id: \.self) { view in Toggle(view, isOn: Binding( get: { selectedViews.contains(view) }, set: { checked in if checked { selectedViews.insert(view) } else { selectedViews.remove(view) } } )) } } Text("InnoDB data is exported using a consistent snapshot and a single streamed query per table. Tables without unique indexes are supported.") .font(.caption).foregroundStyle(.secondary) } } } else { GroupBox("Import Options") { VStack(alignment: .leading, spacing: 10) { Text("Import SQL statements into the selected database.") .foregroundStyle(.secondary) Text("Supports MySQL DELIMITER directives, triggers and stored routines. Import is not transactional; use a verified backup before restoring.") .font(.caption) .foregroundStyle(.secondary) } } } GroupBox("Progress") { VStack(alignment: .leading, spacing: 10) { ProgressView(value: progress) .frame(maxWidth: .infinity) Text(status) .font(.caption) .foregroundStyle(.secondary) .textSelection(.enabled) if !lastVerificationReport.isEmpty { Text(lastVerificationReport) .font(.system(.caption, design: .monospaced)) .textSelection(.enabled) .frame(maxWidth: .infinity, alignment: .leading) } if isWorking && selectedMode == 0 && currentTableExpected > 0 { ProgressView(value: Double(currentTableRows), total: Double(currentTableExpected)) Text("Current table: \(currentTableRows.formatted()) / \(currentTableExpected.formatted()) rows · Total exported: \(transferredRows.formatted())") .font(.caption.monospacedDigit()) .foregroundStyle(.secondary) } } } } } .frame(maxWidth: .infinity, alignment: .leading) .padding(20) Divider() HStack { Button("Close") { operationTask?.cancel() dismiss() } Spacer() if !isWorking { Button("Verify Backup…") { chooseBackupToVerify() } .help("Verify an SQL file against its SHA-256 checksum sidecar") Button("Verify Restored Database…") { chooseBackupToValidate() } .disabled(selectedProfile == nil || database.isEmpty) .help("Compare restored table row counts and object names with the export manifest") } if isWorking { Button("Cancel") { operationTask?.cancel() status = "Cancelling after current statement…" } } else { Button( selectedMode == 0 ? "Export SQL…" : "Import SQL…" ) { if selectedMode == 0 { chooseExportDestination() } else { chooseImportSource() } } .buttonStyle(.borderedProminent) .disabled( selectedProfile == nil || database.isEmpty || (selectedMode == 0 && ((!exportStructure && !exportData) || (selectedTables.isEmpty && !(exportStructure && includeViews && !selectedViews.isEmpty)))) ) } } .padding(16) } .frame(minWidth: 650, idealWidth: 780, minHeight: 500) .onAppear { if database.isEmpty { database = model.selectedDatabase.isEmpty ? (model.databases.first ?? "") : model.selectedDatabase } } .task(id: selectedProfile?.id) { await loadTransferDatabases() } .task(id: database) { availableTables = [] selectedTables = [] availableViews = [] selectedViews = [] if !database.isEmpty { await loadTables() } } .confirmationDialog( "Import SQL into \(database)?", isPresented: $showingImportConfirmation ) { Button("Import SQL") { guard let url = pendingImportURL else { return } operationTask = Task { await importSQL(from: url) } } Button("Cancel", role: .cancel) { pendingImportURL = nil } } message: { Text( "SQL files can modify or delete existing database objects and data. Ensure you have a backup before continuing." ) } } private var header: some View { HStack(spacing: 12) { Image(systemName: "externaldrive.badge.timemachine") .font(.system(size: 28)) .foregroundStyle(.blue) VStack(alignment: .leading, spacing: 4) { Text("Database Import / Export") .font(.system(size: 22, weight: .semibold)) Text("Built-in native MySQL database tools") .font(.caption) .foregroundStyle(.secondary) } Spacer() } .padding(20) } private func chooseExportDestination() { let panel = NSSavePanel() panel.title = "Export MySQL Database" panel.nameFieldStringValue = "\(database).sql" panel.canCreateDirectories = true guard panel.runModal() == .OK, let url = panel.url else { return } operationTask = Task { await exportSQL(to: url) } } private func chooseBackupToVerify() { let panel = NSOpenPanel() panel.title = "Verify SQL Backup" panel.canChooseDirectories = false panel.canChooseFiles = true guard panel.runModal() == .OK, let url = panel.url else { return } operationTask = Task { isWorking = true progress = 0 defer { isWorking = false; operationTask = nil } do { try await Task.detached(priority: .utility) { try WorkbenchBackupIntegrity.verify(url) }.value progress = 1 status = "Verified SHA-256: \(url.lastPathComponent)" } catch { status = "Verification failed: \(error.localizedDescription)" } } } private func chooseBackupToValidate() { let panel = NSOpenPanel() panel.title = "Select Exported SQL Backup to Validate" panel.canChooseDirectories = false panel.canChooseFiles = true guard panel.runModal() == .OK, let url = panel.url else { return } operationTask = Task { await validateRestoredDatabase(from: url) } } @MainActor private func validateRestoredDatabase(from url: URL) async { isWorking = true progress = 0 lastVerificationReport = "" defer { isWorking = false; operationTask = nil } do { let manifest = try WorkbenchBackupManifest.read(for: url) guard manifest.database == database else { throw WorkbenchTransferError.message( "Manifest is for database \(manifest.database), but \(database) is selected." ) } if WorkbenchBackupIntegrity.hasSidecar(for: url) { status = "Verifying SQL checksum…" try await Task.detached(priority: .utility) { try WorkbenchBackupIntegrity.verify(url) }.value } var failures: [String] = [] var checked = 0 let total = manifest.tables.count + manifest.views.count + manifest.triggers.count + manifest.routines.count for table in manifest.tables { try Task.checkCancellation() status = "Validating table \(table.name)" do { let rows = try await execute("SELECT COUNT(*) FROM \(quoteIdentifier(database)).\(quoteIdentifier(table.name))") guard let actual = rows.rows.first?.first.flatMap(Int64.init) else { failures.append("\(table.name): count unavailable") continue } if let expected = table.rows, actual != expected { failures.append("\(table.name): expected \(expected), found \(actual) rows") } } catch { failures.append("\(table.name): \(error.localizedDescription)") } checked += 1 progress = Double(checked) / Double(max(total, 1)) } for view in manifest.views { try Task.checkCancellation() status = "Validating view \(view)" do { _ = try await execute("SHOW CREATE VIEW \(quoteIdentifier(database)).\(quoteIdentifier(view))") } catch { failures.append("View \(view): \(error.localizedDescription)") } checked += 1 progress = Double(checked) / Double(max(total, 1)) } for trigger in manifest.triggers { try Task.checkCancellation() status = "Validating trigger \(trigger)" do { _ = try await execute("SHOW CREATE TRIGGER \(quoteIdentifier(database)).\(quoteIdentifier(trigger))") } catch { failures.append("Trigger \(trigger): \(error.localizedDescription)") } checked += 1 progress = Double(checked) / Double(max(total, 1)) } for routine in manifest.routines { try Task.checkCancellation() status = "Validating \(routine.kind) \(routine.name)" do { _ = try await execute("SHOW CREATE \(routine.kind) \(quoteIdentifier(database)).\(quoteIdentifier(routine.name))") } catch { failures.append("\(routine.kind) \(routine.name): \(error.localizedDescription)") } checked += 1 progress = Double(checked) / Double(max(total, 1)) } progress = 1 lastVerificationReport = failures.isEmpty ? "PASS: \(checked) objects checked. Counts and object existence match the manifest." : "FAILED: \(failures.count) mismatches:\n" + failures.joined(separator: "\n") status = failures.isEmpty ? "Restore validation passed" : "Restore validation found \(failures.count) problems" } catch is CancellationError { status = "Validation cancelled" } catch { status = "Validation failed: \(error.localizedDescription)" } } private func chooseImportSource() { let panel = NSOpenPanel() panel.title = "Import MySQL SQL File" panel.allowedContentTypes = [.plainText] panel.allowsOtherFileTypes = true panel.canChooseDirectories = false panel.canChooseFiles = true guard panel.runModal() == .OK, let url = panel.url else { return } pendingImportURL = url showingImportConfirmation = true } // Export operations use a distinct authenticated session so their snapshot // cannot be affected by SQL editor queries or administration refreshes. @State private var exportSession: ConnectionProfile? @State private var exportPassword = "" @MainActor private func execute(_ statement: String) async throws -> QueryResult { guard let profile = selectedProfile else { throw WorkbenchTransferError.message( "No MySQL connection selected" ) } return try await NativeMySQLRunner.run( profile: exportSession ?? profile, database: statement == "SHOW DATABASES" ? "" : database, password: exportSession == nil ? PasswordStore.read(profile.id) : exportPassword, sql: statement ) } private func quoteIdentifier(_ identifier: String) -> String { "`" + identifier.replacingOccurrences( of: "`", with: "``" ) + "`" } @MainActor private func loadTransferDatabases() async { guard selectedProfile != nil else { availableDatabases = [] database = "" status = "Select a MySQL connection first" return } loadingDatabases = true defer { loadingDatabases = false } do { // Do not depend on the schema navigator having been opened first. let response = try await execute("SHOW DATABASES") let names = response.rows.compactMap { $0.first } availableDatabases = names if !names.contains(database) { let preferred = model.selectedDatabase database = names.contains(preferred) ? preferred : (names.first ?? "") } status = names.isEmpty ? "No accessible databases" : "Ready" } catch { availableDatabases = [] database = "" status = "Cannot load databases: \(error.localizedDescription)" } } @MainActor private func loadTables() async { guard !database.isEmpty else { availableTables = []; selectedTables = [] availableViews = []; selectedViews = [] return } loadingTables = true defer { loadingTables = false } do { let rows = try await execute("SHOW FULL TABLES FROM \(quoteIdentifier(database))") availableTables = rows.rows.compactMap { row in row.count > 1 && row[1].uppercased() == "BASE TABLE" ? row[0] : nil } selectedTables = Set(availableTables) availableViews = rows.rows.compactMap { row in row.count > 1 && row[1].uppercased() == "VIEW" ? row[0] : nil } selectedViews = Set(availableViews) } catch { status = "Cannot list tables: \(error.localizedDescription)" availableTables = [] selectedTables = [] availableViews = [] selectedViews = [] } } // The server constructs SQL literals, so NULL, quotes and binary data are // preserved rather than reconstructed from QueryResult's String values. @MainActor private func exportSQL(to url: URL) async { guard exportStructure || exportData else { return } let views = exportStructure && includeViews ? availableViews.filter { selectedViews.contains($0) } : [] guard !selectedTables.isEmpty || !views.isEmpty || (exportStructure && (includeTriggers || includeRoutines)) else { return } guard let activeProfile = selectedProfile else { status = "Export failed: No connection selected" return } // Clone the connection profile with a fresh identity; the session pool // then creates a separate socket, independent of the editor session. var dedicatedProfile = activeProfile dedicatedProfile.id = UUID() exportPassword = PasswordStore.read(activeProfile.id) exportSession = dedicatedProfile isWorking = true progress = 0 transferredRows = 0 lastVerificationReport = "" var manifestTables: [WorkbenchBackupManifest.Table] = [] var manifestRoutines: [WorkbenchBackupManifest.Routine] = [] var manifestTriggers: [String] = [] currentTableRows = 0 currentTableExpected = 0 let tables = availableTables.filter { selectedTables.contains($0) } let tempURL = url.deletingLastPathComponent().appendingPathComponent(".\(UUID().uuidString).sql.tmp") var handle: FileHandle? defer { try? handle?.close() try? FileManager.default.removeItem(at: tempURL) NativeMySQLRunner.disconnect(profileID: dedicatedProfile.id) exportSession = nil exportPassword = "" isWorking = false operationTask = nil } do { // A consistent InnoDB snapshot requires every SELECT to execute on // this SAME connection. MyISAM/MEMORY cannot participate. for table in tables where exportData { let engine = try await execute(""" SELECT ENGINE FROM information_schema.TABLES WHERE TABLE_SCHEMA = \(sqlLiteral(database)) AND TABLE_NAME = \(sqlLiteral(table)) AND TABLE_TYPE = 'BASE TABLE' """) guard let name = engine.rows.first?.first, name.uppercased() == "INNODB" else { throw WorkbenchTransferError.message( "Consistent data export requires InnoDB. Table \(table) uses \(engine.rows.first?.first ?? "an unknown engine"). Export its structure only." ) } } _ = try await execute("SET SESSION TRANSACTION ISOLATION LEVEL REPEATABLE READ") _ = try await execute("START TRANSACTION WITH CONSISTENT SNAPSHOT, READ ONLY") guard FileManager.default.createFile(atPath: tempURL.path, contents: nil) else { throw WorkbenchTransferError.message("Cannot create temporary export file") } handle = try FileHandle(forWritingTo: tempURL) func write(_ value: String) throws { try handle?.write(contentsOf: Data(value.utf8)) } try write("-- WorkbenchReplacement native export\n-- Database: \(database)\n-- Generated: \(Date())\nSET NAMES utf8mb4;\n\n") let db = quoteIdentifier(database) for (index, table) in tables.enumerated() { try Task.checkCancellation() let qualified = "\(db).\(quoteIdentifier(table))" status = "Exporting \(table) (\(index + 1)/\(tables.count))" try write("\n-- Table \(qualified)\n") if exportStructure { let definition = try await execute("SHOW CREATE TABLE \(qualified)") guard let row = definition.rows.first, row.count > 1 else { throw WorkbenchTransferError.message("Missing CREATE TABLE for \(table)") } if includeDrop { try write("DROP TABLE IF EXISTS \(qualified);\n") } try write(row[1] + ";\n") } if exportData { // Column metadata drives binary-safe SQL literal formatting. let metadata = try await execute(""" SELECT COLUMN_NAME, DATA_TYPE FROM information_schema.COLUMNS WHERE TABLE_SCHEMA = \(sqlLiteral(database)) AND TABLE_NAME = \(sqlLiteral(table)) ORDER BY ORDINAL_POSITION """) guard !metadata.rows.isEmpty else { throw WorkbenchTransferError.message("No column metadata for \(table)") } let names = metadata.rows.map { quoteIdentifier($0[0]) }.joined(separator: ", ") let expressions = metadata.rows.map { row -> String in let col = quoteIdentifier(row[0]) let kind = row[1].lowercased() // HEX avoids decoding arbitrary bytes as UTF-8. if ["binary", "varbinary", "tinyblob", "blob", "mediumblob", "longblob", "bit"].contains(kind) { return "IF(\(col) IS NULL, 'NULL', CONCAT('0x', HEX(\(col))))" } return "IF(\(col) IS NULL, 'NULL', QUOTE(\(col)))" }.joined(separator: ", ") // A single streamed SELECT keeps every row in the same // InnoDB read view and works without a primary/unique key. // No OFFSET, no 100,000-row cap, no full result buffering. let indexes = try await execute(""" SELECT INDEX_NAME, COLUMN_NAME, SEQ_IN_INDEX FROM information_schema.STATISTICS WHERE TABLE_SCHEMA = \(sqlLiteral(database)) AND TABLE_NAME = \(sqlLiteral(table)) AND NON_UNIQUE = 0 ORDER BY CASE WHEN INDEX_NAME = 'PRIMARY' THEN 0 ELSE 1 END, INDEX_NAME, SEQ_IN_INDEX """) let nullable = try await execute(""" SELECT COLUMN_NAME FROM information_schema.COLUMNS WHERE TABLE_SCHEMA = \(sqlLiteral(database)) AND TABLE_NAME = \(sqlLiteral(table)) AND IS_NULLABLE = 'YES' """) let nullableColumns = Set(nullable.rows.compactMap { $0.first }) var grouped: [String: [String]] = [:] var indexOrder: [String] = [] for indexRow in indexes.rows where indexRow.count >= 2 { let name = indexRow[0] if grouped[name] == nil { grouped[name] = [] indexOrder.append(name) } grouped[name, default: []].append(indexRow[1]) } let usableIndex = indexOrder.first { name in guard let columns = grouped[name], !columns.isEmpty else { return false } return columns.allSatisfy { !nullableColumns.contains($0) } } let orderBy = usableIndex.flatMap { grouped[$0] }? .map(quoteIdentifier).joined(separator: ", ") let countResult = try await execute("SELECT COUNT(*) FROM \(qualified)") guard let countText = countResult.rows.first?.first, let expectedCount = Int64(countText) else { throw WorkbenchTransferError.message("Cannot count rows in \(table)") } currentTableRows = 0 currentTableExpected = expectedCount try write("-- Streamed snapshot: \(expectedCount) rows; order: \(usableIndex ?? "server scan")\n") let select = "SELECT CONCAT('(', CONCAT_WS(',', \(expressions)), ')') FROM \(qualified)" + (orderBy.map { " ORDER BY " + $0 } ?? "") let outputHandle = handle! let prefix = Data("INSERT INTO \(qualified) (\(names)) VALUES ".utf8) let suffix = Data(";\n".utf8) let completedBeforeTable = transferredRows let completedTables = index let totalUnits = max(tables.count + views.count, 1) var streamedInTable: Int64 = 0 let written = try await NativeMySQLRunner.stream( profile: dedicatedProfile, database: database, password: exportPassword, sql: select ) { literal in try outputHandle.write(contentsOf: prefix) try outputHandle.write(contentsOf: Data(literal.utf8)) try outputHandle.write(contentsOf: suffix) streamedInTable += 1 if streamedInTable % 500 == 0 || streamedInTable == expectedCount { let completed = streamedInTable DispatchQueue.main.async { currentTableRows = completed transferredRows = completedBeforeTable + completed progress = (Double(completedTables) + Double(completed) / Double(max(expectedCount, 1))) / Double(totalUnits) status = "Exporting \(table): \(completed.formatted()) / \(expectedCount.formatted()) rows" } } } currentTableRows = written transferredRows = completedBeforeTable + written guard written == expectedCount else { throw WorkbenchTransferError.message( "Table \(table): expected \(expectedCount) rows, streamed \(written). Backup discarded." ) } manifestTables.append(.init(name: table, rows: written)) status = "Exported \(table): \(written) rows" } if !exportData { manifestTables.append(.init(name: table, rows: nil)) } currentTableExpected = 0 currentTableRows = 0 progress = Double(index + 1) / Double(max(tables.count + views.count, 1)) } // Views are exported after base tables so dependencies exist on restore. // The server's SHOW CREATE VIEW response may contain a DEFINER clause; // importing under a different MySQL account can require extra privileges. for (index, view) in views.enumerated() { try Task.checkCancellation() let qualified = "\(db).\(quoteIdentifier(view))" status = "Exporting view \(view)" let definition = try await execute("SHOW CREATE VIEW \(qualified)") guard let row = definition.rows.first, row.count > 1 else { throw WorkbenchTransferError.message("Missing CREATE VIEW for \(view)") } try write("\n-- View \(qualified)\n") if includeDrop { try write("DROP VIEW IF EXISTS \(qualified);\n") } try write(row[1] + ";\n") progress = Double(tables.count + index + 1) / Double(max(tables.count + views.count, 1)) } // Routines and triggers use client-side delimiters. The importer // strips these directives and sends each definition as one query. if exportStructure && includeRoutines { let routines = try await execute(""" SELECT ROUTINE_NAME, ROUTINE_TYPE FROM information_schema.ROUTINES WHERE ROUTINE_SCHEMA = \(sqlLiteral(database)) ORDER BY ROUTINE_TYPE, ROUTINE_NAME """) for routine in routines.rows { try Task.checkCancellation() guard routine.count >= 2 else { continue } let name = routine[0] let kind = routine[1].uppercased() guard kind == "PROCEDURE" || kind == "FUNCTION" else { continue } status = "Exporting \(kind.lowercased()) \(name)" let qualified = "\(db).\(quoteIdentifier(name))" let definition = try await execute("SHOW CREATE \(kind) \(qualified)") // SHOW CREATE PROCEDURE/FUNCTION: Name, sql_mode, Create ..., ... guard let row = definition.rows.first, row.count > 2, !row[2].isEmpty else { throw WorkbenchTransferError.message("Missing CREATE \(kind) for \(name)") } try write("\n-- \(kind) \(qualified)\n") try write("DELIMITER $$\n") if includeDrop { try write("DROP \(kind) IF EXISTS \(qualified)$$\n") } try write(row[2] + "$$\nDELIMITER ;\n") manifestRoutines.append(.init(name: name, kind: kind)) } } if exportStructure && includeTriggers { let triggers = try await execute(""" SELECT TRIGGER_NAME FROM information_schema.TRIGGERS WHERE TRIGGER_SCHEMA = \(sqlLiteral(database)) ORDER BY TRIGGER_NAME """) for trigger in triggers.rows { try Task.checkCancellation() guard let name = trigger.first else { continue } status = "Exporting trigger \(name)" let qualified = "\(db).\(quoteIdentifier(name))" let definition = try await execute("SHOW CREATE TRIGGER \(qualified)") // SHOW CREATE TRIGGER: Trigger, sql_mode, SQL Original Statement, ... guard let row = definition.rows.first, row.count > 2, !row[2].isEmpty else { throw WorkbenchTransferError.message("Missing CREATE TRIGGER for \(name)") } try write("\n-- Trigger \(qualified)\n") try write("DELIMITER $$\n") if includeDrop { try write("DROP TRIGGER IF EXISTS \(qualified)$$\n") } try write(row[2] + "$$\nDELIMITER ;\n") manifestTriggers.append(name) } } try Task.checkCancellation() // Release the read view before finalizing the export file. _ = try await execute("COMMIT") try handle?.close() handle = nil if FileManager.default.fileExists(atPath: url.path) { _ = try FileManager.default.replaceItemAt(url, withItemAt: tempURL) } else { try FileManager.default.moveItem(at: tempURL, to: url) } // The SQL file is complete before creating its checksum sidecar. // Sidecar failure is reported; it does not delete a completed export. do { try WorkbenchBackupIntegrity.createSidecar(for: url) let manifest = WorkbenchBackupManifest( database: database, tables: manifestTables, views: views, triggers: manifestTriggers, routines: manifestRoutines ) try manifest.write(for: url) lastBackupURL = url progress = 1 status = "Export complete and SHA-256 verified: \(tables.count) tables, \(views.count) views, \(transferredRows.formatted()) rows" } catch { progress = 1 status = "SQL export complete, but checksum creation failed: \(error.localizedDescription)" } } catch is CancellationError { status = "Export cancelled; no final backup was written" } catch { status = "Export failed: \(error.localizedDescription)" } } private func sqlLiteral(_ value: String) -> String { "'" + value.replacingOccurrences(of: "\\", with: "\\\\") .replacingOccurrences(of: "'", with: "''") + "'" } @MainActor private func importSQL(from url: URL) async { isWorking = true progress = 0 status = "Reading SQL file…" defer { isWorking = false operationTask = nil } do { // Refuse a corrupted backup when a checksum sidecar is present. // Third-party SQL files without a sidecar remain importable. if WorkbenchBackupIntegrity.hasSidecar(for: url) { status = "Verifying SQL backup checksum…" try await Task.detached(priority: .utility) { try WorkbenchBackupIntegrity.verify(url) }.value } try Task.checkCancellation() let fileSize = (try FileManager.default.attributesOfItem(atPath: url.path)[.size] as? NSNumber)?.int64Value ?? 0 var statementNumber = 0 var lastLine = 1 let reader = try WorkbenchStreamingSQLReader(url: url) defer { reader.close() } while let entry = try reader.nextStatement() { try Task.checkCancellation() statementNumber += 1 lastLine = entry.line status = "Importing statement \(statementNumber) · line \(lastLine)" do { _ = try await execute(entry.sql) } catch { throw WorkbenchTransferError.message( "Statement \(statementNumber), line \(lastLine): \(error.localizedDescription)" ) } if statementNumber % 10 == 0 || reader.bytesRead >= fileSize { progress = fileSize > 0 ? min(1, Double(reader.bytesRead) / Double(fileSize)) : 1 } } progress = 1 status = "Import complete: \(statementNumber) statements. Use Verify Restored Database to compare with the export manifest." lastBackupURL = url model.loadDatabases() } catch is CancellationError { status = "Import cancelled; previously executed statements may remain applied" } catch { status = "Import failed: \(error.localizedDescription)" } } } enum WorkbenchTransferError: LocalizedError { case message(String) var errorDescription: String? { switch self { case .message(let message): return message } } } // MARK: - SQL Import Parser // Supports MySQL DELIMITER directives, quoted text, executable /*! */ // comments, and routine bodies. Parsing does not execute SQL. enum WorkbenchSQLImportParser { static func parse(_ sql: String) throws -> [String] { let chars = Array(sql) var statements: [String] = [] var current = "" var delimiter = ";" var quote: Character? = nil var lineComment = false var blockComment = false var escaped = false var lineStart = true var i = 0 func matches(_ value: String, at index: Int) -> Bool { let pattern = Array(value) guard index + pattern.count <= chars.count else { return false } return Array(chars[index..<(index + pattern.count)]) == pattern } func emit() { let value = current.trimmingCharacters(in: .whitespacesAndNewlines) if !value.isEmpty { statements.append(value) } current = "" } while i < chars.count { let ch = chars[i] let next: Character? = i + 1 < chars.count ? chars[i + 1] : nil // DELIMITER is a client-side directive, never sent to MySQL. if lineStart && quote == nil && !lineComment && !blockComment { var j = i while j < chars.count && (chars[j] == " " || chars[j] == "\t") { j += 1 } if j + 9 <= chars.count && String(chars[j.. URL { URL(fileURLWithPath: sqlURL.path + ".sha256") } static func hasSidecar(for sqlURL: URL) -> Bool { FileManager.default.fileExists(atPath: sidecar(for: sqlURL).path) } static func digest(_ url: URL) throws -> String { let file = try FileHandle(forReadingFrom: url) defer { try? file.close() } var hasher = SHA256() while true { let chunk = try file.read(upToCount: 1024 * 1024) ?? Data() if chunk.isEmpty { break } hasher.update(data: chunk) } return hasher.finalize().map { String(format: "%02x", $0) }.joined() } static func createSidecar(for sqlURL: URL) throws { let value = try digest(sqlURL) // Standard sha256sum-compatible format, with filename rather than path. let line = value + " " + sqlURL.lastPathComponent + "\n" try line.write(to: sidecar(for: sqlURL), atomically: true, encoding: .utf8) } static func verify(_ sqlURL: URL) throws { let checksumURL = sidecar(for: sqlURL) guard FileManager.default.fileExists(atPath: checksumURL.path) else { throw WorkbenchTransferError.message("No checksum file found: \(checksumURL.lastPathComponent)") } let contents = try String(contentsOf: checksumURL, encoding: .utf8) .trimmingCharacters(in: .whitespacesAndNewlines) let expected = String(contents.prefix(64)).lowercased() guard expected.count == 64, expected.utf8.allSatisfy({ (48...57).contains($0) || (97...102).contains($0) }) else { throw WorkbenchTransferError.message("Invalid SHA-256 checksum file") } let actual = try digest(sqlURL) guard actual == expected else { throw WorkbenchTransferError.message("SHA-256 mismatch: backup contents have changed") } } } // MARK: - Phase 11: bounded-memory SQL dump reader // Keeps only the current SQL statement in memory. A single very large // INSERT still requires enough memory to hold that statement. final class WorkbenchStreamingSQLReader { struct Entry { let sql: String; let line: Int } private let handle: FileHandle private var bytes = Data() private var reachedEOF = false private(set) var bytesRead: Int64 = 0 private var lineNumber = 0 private var statementStart = 1 private var statement = "" private var delimiter = ";" private var quote: Character? private var escaped = false private var blockComment = false private var lineComment = false private var pending: [Entry] = [] init(url: URL) throws { handle = try FileHandle(forReadingFrom: url) } func close() { try? handle.close() } func nextStatement() throws -> Entry? { while pending.isEmpty { guard let line = try readLine() else { if quote != nil || blockComment { throw WorkbenchTransferError.message("Unterminated quote or comment near line \(lineNumber)") } let tail = statement.trimmingCharacters(in: .whitespacesAndNewlines) statement = "" if !tail.isEmpty { return Entry(sql: tail, line: statementStart) } return nil } lineNumber += 1 try consume(line) } return pending.removeFirst() } private func readLine() throws -> String? { while true { if let end = bytes.firstIndex(of: 10) { let data = bytes.prefix(through: end) bytes.removeSubrange(...end) guard let line = String(data: data, encoding: .utf8) else { throw WorkbenchTransferError.message("Invalid UTF-8 at line \(lineNumber + 1)") } return line } if reachedEOF { guard !bytes.isEmpty else { return nil } let data = bytes bytes.removeAll(keepingCapacity: true) guard let line = String(data: data, encoding: .utf8) else { throw WorkbenchTransferError.message("Invalid UTF-8 at end of SQL file") } return line } let chunk = try handle.read(upToCount: 64 * 1024) ?? Data() bytesRead += Int64(chunk.count) if chunk.isEmpty { reachedEOF = true } else { bytes.append(chunk) } } } private func consume(_ line: String) throws { let trimmed = line.trimmingCharacters(in: .whitespacesAndNewlines) if quote == nil && !blockComment && statement.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty && trimmed.uppercased().hasPrefix("DELIMITER ") { let token = String(trimmed.dropFirst(10)).trimmingCharacters(in: .whitespaces) guard !token.isEmpty && !token.contains(where: { $0.isWhitespace }) else { throw WorkbenchTransferError.message("Invalid DELIMITER at line \(lineNumber)") } delimiter = token return } let chars = Array(line) let delim = Array(delimiter) var i = 0 lineComment = false while i < chars.count { let ch = chars[i] let next: Character? = i + 1 < chars.count ? chars[i + 1] : nil if lineComment { break } if blockComment { statement.append(ch) if ch == "*" && next == "/" { statement.append("/"); i += 2; blockComment = false } else { i += 1 } continue } if let q = quote { statement.append(ch) if escaped { escaped = false } else if ch == "\\" && q != "`" { escaped = true } else if ch == q { if next == q { statement.append(ch); i += 1 } else { quote = nil } } i += 1 continue } if ch == "'" || ch == "\"" || ch == "`" { quote = ch; statement.append(ch); i += 1; continue } if ch == "#" || (ch == "-" && next == "-" && (i + 2 >= chars.count || chars[i + 2].isWhitespace)) { lineComment = true; break } if ch == "/" && next == "*" { statement.append("/*"); blockComment = true; i += 2; continue } if i + delim.count <= chars.count && Array(chars[i..<(i + delim.count)]) == delim { let sql = statement.trimmingCharacters(in: .whitespacesAndNewlines) if !sql.isEmpty { pending.append(Entry(sql: sql, line: statementStart)) } statement = ""; i += delim.count; statementStart = lineNumber; continue } if statement.isEmpty && !ch.isWhitespace { statementStart = lineNumber } statement.append(ch) i += 1 } if !statement.isEmpty { statement.append("\n") } } } // MARK: - Phase 12 restore validation manifest // This is a post-restore comparison, not a proof of byte-for-byte data equality. struct WorkbenchBackupManifest: Codable { struct Table: Codable { let name: String let rows: Int64? // nil when the export contains no table data } struct Routine: Codable { let name: String let kind: String } let version: Int let database: String let tables: [Table] let views: [String] let triggers: [String] let routines: [Routine] init(database: String, tables: [Table], views: [String], triggers: [String], routines: [Routine]) { self.version = 1 self.database = database self.tables = tables self.views = views self.triggers = triggers self.routines = routines } static func sidecarURL(for sql: URL) -> URL { URL(fileURLWithPath: sql.path + ".manifest.json") } func write(for sql: URL) throws { let encoder = JSONEncoder() encoder.outputFormatting = [.prettyPrinted, .sortedKeys] try encoder.encode(self).write(to: Self.sidecarURL(for: sql), options: .atomic) } static func read(for sql: URL) throws -> Self { let url = sidecarURL(for: sql) guard FileManager.default.fileExists(atPath: url.path) else { throw WorkbenchTransferError.message("No .manifest.json sidecar found. Export a backup with Phase 12 first.") } let manifest = try JSONDecoder().decode(Self.self, from: Data(contentsOf: url)) guard manifest.version == 1 else { throw WorkbenchTransferError.message("Unsupported backup manifest version \(manifest.version)") } return manifest } } // MARK: - Phase 14: Explicitly confirmed MySQL account administration private enum WorkbenchAccountAction: String, CaseIterable, Identifiable { case create = "Create Account" case password = "Change Password" case lock = "Lock Account" case unlock = "Unlock Account" case grant = "Grant Privileges" case revoke = "Revoke Privileges" var id: String { rawValue } } private struct WorkbenchAccountManagerView: View { let profile: ConnectionProfile let selectedAccount: String? @Environment(\.dismiss) private var dismiss @State private var action: WorkbenchAccountAction = .create @State private var username = "" @State private var hostname = "%" @State private var password = "" @State private var databaseName = "" @State private var privilege = "SELECT" @State private var previewSQL = "" @State private var confirm = false @State private var running = false @State private var status: String? @State private var failure: String? private let privileges = ["SELECT", "INSERT", "UPDATE", "DELETE", "CREATE", "ALTER", "DROP", "INDEX", "REFERENCES", "EXECUTE", "CREATE VIEW", "SHOW VIEW", "TRIGGER"] var body: some View { VStack(alignment: .leading, spacing: 16) { Text("Manage MySQL Accounts").font(.title2.bold()) Text("Account changes run on \(profile.name). Review the SQL before applying it.") .foregroundStyle(.secondary) Form { Picker("Operation", selection: $action) { ForEach(WorkbenchAccountAction.allCases) { option in Text(option.rawValue).tag(option) } } TextField("Username", text: $username) .textContentType(.username) TextField("Host", text: $hostname) if action == .create || action == .password { SecureField("Password", text: $password) } if action == .grant || action == .revoke { TextField("Database", text: $databaseName) Picker("Privilege", selection: $privilege) { ForEach(privileges, id: \.self) { item in Text(item).tag(item) } } } } .formStyle(.grouped) if let failure { Text(failure).foregroundStyle(.red).textSelection(.enabled) } if let status { Text(status).foregroundStyle(.green) } HStack { Button("Close") { dismiss() } Spacer() Button("Review SQL…") { review() } .disabled(running) .keyboardShortcut(.defaultAction) } } .padding(20) .frame(width: 590) .onAppear { if let selectedAccount { let parts = selectedAccount.split(separator: "\u{001f}", omittingEmptySubsequences: false) if parts.count == 2 { username = String(parts[0]) hostname = String(parts[1]) } } } .alert("Confirm \(action.rawValue)", isPresented: $confirm) { Button("Cancel", role: .cancel) { previewSQL = "" } Button("Apply Change", role: .destructive) { Task { await execute() } } } message: { // Passwords are never shown in the SQL preview. Text("Run this operation against \(profile.name)?\n\n\(previewSQL)") } } // Strict account and database validation keeps identifiers out of SQL syntax. private func sql() throws -> (command: String, preview: String) { let allowed = CharacterSet(charactersIn: "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789_.-%") guard !username.isEmpty, username.unicodeScalars.allSatisfy({ allowed.contains($0) }), !hostname.isEmpty, hostname.unicodeScalars.allSatisfy({ allowed.contains($0) }) else { throw AccountError.invalid("Username/host may contain only letters, digits, _, ., -, and %.") } let account = "'\(username)'@'\(hostname)'" switch action { case .create, .password: guard !password.isEmpty else { throw AccountError.invalid("Enter a password.") } // MySQL's default SQL mode interprets backslashes as escapes. // We check SQL mode before execution to avoid changing password meaning. let literal = password.replacingOccurrences(of: "\\", with: "\\\\") .replacingOccurrences(of: "'", with: "''") let prefix = action == .create ? "CREATE USER" : "ALTER USER" return ("\(prefix) \(account) IDENTIFIED BY '\(literal)'", "\(prefix) \(account) IDENTIFIED BY '••••••••'") case .lock: return ("ALTER USER \(account) ACCOUNT LOCK", "ALTER USER \(account) ACCOUNT LOCK") case .unlock: return ("ALTER USER \(account) ACCOUNT UNLOCK", "ALTER USER \(account) ACCOUNT UNLOCK") case .grant, .revoke: guard !databaseName.isEmpty, databaseName.unicodeScalars.allSatisfy({ allowed.contains($0) }), privileges.contains(privilege) else { throw AccountError.invalid("Enter a valid database name and privilege.") } let db = "`\(databaseName)`.*" let command = action == .grant ? "GRANT \(privilege) ON \(db) TO \(account)" : "REVOKE \(privilege) ON \(db) FROM \(account)" return (command, command) } } private func review() { failure = nil status = nil do { previewSQL = try sql().preview confirm = true } catch { failure = error.localizedDescription } } @MainActor private func execute() async { running = true defer { running = false; password = "" } do { let command = try sql().command let credentials = PasswordStore.read(profile.id) if action == .create || action == .password { let mode = try await NativeMySQLRunner.run( profile: profile, database: "", password: credentials, sql: "SELECT @@SESSION.sql_mode" ) let flags = mode.rows.first?.first ?? "" if flags.split(separator: ",").contains(where: { $0 == "NO_BACKSLASH_ESCAPES" }) { throw AccountError.invalid("Password changes are disabled for NO_BACKSLASH_ESCAPES SQL mode.") } } _ = try await NativeMySQLRunner.run( profile: profile, database: "", password: credentials, sql: command ) status = "\(action.rawValue) completed successfully." failure = nil } catch { failure = error.localizedDescription status = nil } } private enum AccountError: LocalizedError { case invalid(String) var errorDescription: String? { if case let .invalid(message) = self { return message } return nil } } } // MARK: - Phase 28: Front-screen performance charts private struct WorkbenchInlinePerformanceDashboard: View { let samples: [DiagnosticSample] var chartHeight: CGFloat = 205 private struct Reading: Identifiable { let id: Int let time: Date let value: Double } private func counter(_ sample: DiagnosticSample, _ name: String) -> Double? { guard let row = sample.rows.first(where: { $0.count > 1 && $0[0].caseInsensitiveCompare(name) == .orderedSame }) else { return nil } return Double(row[1]) } private func points(_ name: String, rate: Bool) -> [Reading] { samples.indices.compactMap { i in guard let current = counter(samples[i], name) else { return nil } if !rate { return Reading(id: i, time: samples[i].at, value: current) } guard i > 0, let previous = counter(samples[i - 1], name) else { return nil } let interval = samples[i].at.timeIntervalSince(samples[i - 1].at) guard interval > 0, current >= previous else { return nil } return Reading(id: i, time: samples[i].at, value: (current - previous) / interval) } } private func panel(_ title: String, metric: String, rate: Bool, tint: Color) -> some View { let readings = points(metric, rate: rate) return VStack(alignment: .leading, spacing: 10) { HStack { Text(title).font(.headline) Spacer() Circle().fill(tint).frame(width: 9, height: 9) } if let last = readings.last { Text(last.value.formatted(.number.precision(.fractionLength(1))) + (rate ? " /sec" : "")) .font(.system(size: 15, weight: .semibold, design: .rounded)) .foregroundStyle(tint) } else { Text("Collecting samples…") .font(.caption).foregroundStyle(.secondary) } Chart(readings) { reading in AreaMark(x: .value("Time", reading.time), y: .value("Value", reading.value)) .foregroundStyle(tint.opacity(0.14)) LineMark(x: .value("Time", reading.time), y: .value("Value", reading.value)) .foregroundStyle(tint) .lineStyle(StrokeStyle(lineWidth: 2.5)) PointMark(x: .value("Time", reading.time), y: .value("Value", reading.value)) .foregroundStyle(tint) } .chartYAxis { AxisMarks(position: .leading) } .frame(height: chartHeight) .accessibilityLabel("Live time-series graph for \(title)") } .padding(14) .frame(maxWidth: .infinity, alignment: .leading) .background(Color(nsColor: .controlBackgroundColor)) .clipShape(RoundedRectangle(cornerRadius: 12)) } var body: some View { VStack(alignment: .leading, spacing: 12) { HStack { Label("Live MySQL Line Graphs", systemImage: "waveform.path.ecg") .font(.title2.bold()) Spacer() Text("\(samples.count) samples · refreshes every 5 seconds") .font(.caption).foregroundStyle(.secondary) } LazyVGrid(columns: [GridItem(.adaptive(minimum: 360), spacing: 16)], spacing: 16) { panel("Queries per second", metric: "Queries", rate: true, tint: .blue) panel("Active connections", metric: "Threads_connected", rate: false, tint: .teal) panel("Running threads", metric: "Threads_running", rate: false, tint: .orange) panel("Slow queries per second", metric: "Slow_queries", rate: true, tint: .red) panel("Bytes received per second", metric: "Bytes_received", rate: true, tint: .purple) panel("Bytes sent per second", metric: "Bytes_sent", rate: true, tint: .green) } Text("Rates are calculated from consecutive MySQL status snapshots. Counters reset after server restarts; graphs show up to 60 in-session samples.") .font(.caption).foregroundStyle(.secondary) } } } // MARK: - Phase 34: Responsive server-counter workspace // MARK: - Phase 39: pinned, visible monitoring graphs // The charts and counters are siblings, not nested in one long ScrollView. // Single administration monitoring view. The chart region is deliberately outside // the counter ScrollView, so scrolling hundreds of counters cannot hide charts. private struct WorkbenchServerOverview: View { let rows: [[String]] let samples: [DiagnosticSample] private let names = [ "Uptime", "Threads_connected", "Threads_running", "Queries", "Slow_queries", "Bytes_received", "Bytes_sent", "Connections" ] private struct OverviewPoint: Identifiable { let id: Int let time: Date let value: Double } private func points(for name: String) -> [OverviewPoint] { samples.enumerated().compactMap { index, sample in guard let row = sample.rows.first(where: { $0.count > 1 && $0[0].caseInsensitiveCompare(name) == .orderedSame }), let value = Double(row[1]), value.isFinite else { return nil } return OverviewPoint(id: index, time: sample.at, value: value) } } var body: some View { LazyVGrid(columns: [GridItem(.adaptive(minimum: 260), spacing: 14)], spacing: 14) { ForEach(names, id: \.self) { name in let raw = rows.first(where: { $0.count >= 2 && $0[0].caseInsensitiveCompare(name) == .orderedSame })?.last ?? "—" let tint = WorkbenchCounterCharts.tint(for: name) let series = points(for: name) VStack(alignment: .leading, spacing: 10) { HStack(alignment: .firstTextBaseline) { Text(name.replacingOccurrences(of: "_", with: " ")) .font(.subheadline.weight(.semibold)) Spacer(minLength: 8) Text(WorkbenchCounterCharts.displayValue(raw, for: name)) .font(.headline.monospacedDigit()) .foregroundStyle(tint) } Chart(series) { point in AreaMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(tint.opacity(0.12)) LineMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(tint) .lineStyle(StrokeStyle(lineWidth: 2.5)) PointMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(tint) } .chartYAxis { AxisMarks(position: .leading) } .frame(height: 135) .overlay { if series.isEmpty { Text("Waiting for live samples…") .font(.caption) .foregroundStyle(.secondary) } } } .padding(14) .frame(maxWidth: .infinity, alignment: .leading) .background(Color(nsColor: .controlBackgroundColor), in: RoundedRectangle(cornerRadius: 12)) .overlay(RoundedRectangle(cornerRadius: 12).stroke(tint.opacity(0.20))) } } } } private struct WorkbenchLiveMonitorScreen: View { let page: WorkbenchAdminPage let samples: [DiagnosticSample] let rows: [[String]] let allRowCount: Int @State private var categorySelection = "All" @State private var chartGroup = "Activity" private let categories = ["All", "Connections", "Queries", "InnoDB", "Network", "Errors", "Replication", "Other"] private var pageTitle: String { switch page { case .serverStatus: return "Server Status" case .serverCounters: return "Server Counters" case .connectionStatistics: return "Connection Statistics" case .innodbMetrics: return "InnoDB Metrics" default: return "Server Monitoring" } } private func category(for name: String) -> String { let n = name.lowercased() if n.hasPrefix("innodb_") { return "InnoDB" } if n.contains("replica") || n.contains("slave") || n.contains("binlog") || n.contains("relay_log") { return "Replication" } if n.contains("abort") || n.contains("error") || n.contains("failed") || n.contains("denied") { return "Errors" } if n.contains("bytes_") || n.contains("network") { return "Network" } if n.contains("connection") || n.contains("thread") || n.contains("client") { return "Connections" } if n.contains("quer") || n.contains("question") || n.hasPrefix("com_") || n.contains("select") { return "Queries" } return "Other" } private var visibleRows: [[String]] { categorySelection == "All" ? rows : rows.filter { $0.count > 1 && category(for: $0[0]) == categorySelection } } private var metrics: [String] { if page == .connectionStatistics { return ["Threads_connected", "Threads_running", "Connections", "Aborted_connects"] } if page == .innodbMetrics { return ["Innodb_buffer_pool_read_requests", "Innodb_buffer_pool_reads", "Innodb_data_reads", "Innodb_data_writes"] } switch chartGroup { case "Network": return ["Bytes_received", "Bytes_sent", "Connections", "Aborted_connects"] case "InnoDB": return ["Innodb_buffer_pool_read_requests", "Innodb_buffer_pool_reads", "Innodb_data_reads", "Innodb_data_writes"] default: return ["Queries", "Threads_connected", "Threads_running", "Slow_queries"] } } private func value(_ metric: String, in sample: DiagnosticSample) -> Double? { guard let row = sample.rows.first(where: { $0.count > 1 && $0[0].caseInsensitiveCompare(metric) == .orderedSame }) else { return nil } return Double(row[1]) } private struct Reading: Identifiable { let id: Int let time: Date let value: Double } private func readings(for metric: String) -> [Reading] { let gauge = metric == "Threads_connected" || metric == "Threads_running" return samples.indices.compactMap { index in guard let current = value(metric, in: samples[index]), current.isFinite else { return nil } if gauge { return Reading(id: index, time: samples[index].at, value: current) } guard index > 0, let previous = value(metric, in: samples[index - 1]), current >= previous else { return nil } let interval = samples[index].at.timeIntervalSince(samples[index - 1].at) guard interval > 0 else { return nil } return Reading(id: index, time: samples[index].at, value: (current - previous) / interval) } } private func chart(_ metric: String) -> some View { let points = readings(for: metric) let tint = WorkbenchCounterCharts.tint(for: metric) let gauge = metric == "Threads_connected" || metric == "Threads_running" return VStack(alignment: .leading, spacing: 6) { HStack(alignment: .firstTextBaseline) { Text(metric.replacingOccurrences(of: "_", with: " ")) .font(.subheadline.weight(.semibold)).lineLimit(1) Spacer(minLength: 8) Text(points.last.map { $0.value.formatted(.number.precision(.fractionLength(1))) } ?? "—") .font(.headline.monospacedDigit()).foregroundStyle(tint) Text(gauge ? "current" : "/sec") .font(.caption2).foregroundStyle(.secondary) } Chart(points) { point in AreaMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(tint.opacity(0.16)) LineMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(tint).lineStyle(StrokeStyle(lineWidth: 2.5)) } .chartYAxis { AxisMarks(position: .leading) } .frame(maxWidth: .infinity) .frame(height: 90) .overlay { if points.isEmpty { Text("Waiting for samples…") .font(.caption).foregroundStyle(.secondary) } } } .padding(9) .frame(maxWidth: .infinity, alignment: .leading) .background(tint.opacity(0.065), in: RoundedRectangle(cornerRadius: 12)) .overlay(RoundedRectangle(cornerRadius: 12).stroke(tint.opacity(0.24))) } private var statusSummary: some View { let names = ["Uptime", "Threads_connected", "Threads_running", "Queries", "Slow_queries", "Max_used_connections"] return HStack(spacing: 8) { ForEach(names, id: \.self) { name in let raw = rows.first(where: { $0.count > 1 && $0[0].caseInsensitiveCompare(name) == .orderedSame })?.last ?? "—" VStack(alignment: .leading, spacing: 4) { Text(name.replacingOccurrences(of: "_", with: " ")) .font(.caption2).foregroundStyle(.secondary).lineLimit(1) Text(WorkbenchCounterCharts.displayValue(raw, for: name)) .font(.headline.monospacedDigit()).lineLimit(1).minimumScaleFactor(0.7) } .frame(maxWidth: .infinity, minHeight: 45, alignment: .leading) .padding(9) .background(Color(nsColor: .controlBackgroundColor), in: RoundedRectangle(cornerRadius: 9)) } } } // Fixed-height chart cards live OUTSIDE the independently scrolling table. // Avoid GeometryReader and nested graph ScrollViews: they previously allowed // the graph viewport to collapse when the parent administration view laid out. var body: some View { VStack(alignment: .leading, spacing: 10) { HStack(spacing: 10) { Label(pageTitle + " · LIVE GRAPHS", systemImage: "chart.xyaxis.line") .font(.title3.bold()) Spacer(minLength: 4) Text("\(samples.count) samples · 5s refresh") .font(.caption) .foregroundStyle(.secondary) } if page == .serverStatus || page == .serverCounters { Picker("Graphs", selection: $chartGroup) { Text("Activity").tag("Activity") Text("Network").tag("Network") Text("InnoDB").tag("InnoDB") } .pickerStyle(.segmented) .frame(maxWidth: 350) } // Exactly two rows of two graphs. No graph-region scrolling and no // dependence on the size proposed by a surrounding GeometryReader. LazyVGrid( columns: [GridItem(.flexible(minimum: 0), spacing: 12), GridItem(.flexible(minimum: 0), spacing: 12)], spacing: 12 ) { ForEach(metrics, id: \.self) { metric in chart(metric) .frame(height: 148) .frame(maxWidth: .infinity) } } .frame(maxWidth: .infinity) .accessibilityIdentifier("monitor-graphs-visible") Divider() if page == .serverStatus { Text("Server overview").font(.headline) ScrollView(.vertical) { WorkbenchServerOverview(rows: rows, samples: samples) .frame(maxWidth: .infinity, alignment: .leading) } .frame(maxWidth: .infinity, maxHeight: .infinity) } else { HStack { Text(page == .connectionStatistics ? "Connection statistics" : page == .innodbMetrics ? "InnoDB statistics" : "All MySQL server counters") .font(.headline) Spacer(minLength: 6) if page == .serverCounters { Picker("Category", selection: $categorySelection) { ForEach(categories, id: \.self) { Text($0).tag($0) } } .frame(width: 155) } Text("\(visibleRows.count) counters") .font(.caption).foregroundStyle(.secondary) } WorkbenchColouredCounterTable(columns: ["Metric", "Value"], rows: visibleRows) .frame(maxWidth: .infinity, maxHeight: .infinity) } } .padding(16) .frame(maxWidth: .infinity, maxHeight: .infinity, alignment: .topLeading) } } private struct WorkbenchServerCountersScreen: View { let samples: [DiagnosticSample] let rows: [[String]] @State private var category = "All" private let categories = ["All", "Connections", "Queries", "InnoDB", "Network", "Errors", "Replication", "Other"] private func group(_ name: String) -> String { let n = name.lowercased() if n.hasPrefix("innodb_") { return "InnoDB" } if n.contains("replica") || n.contains("slave") || n.contains("binlog") || n.contains("relay_log") { return "Replication" } if n.contains("abort") || n.contains("error") || n.contains("failed") || n.contains("denied") { return "Errors" } if n.contains("bytes_") || n.contains("network") { return "Network" } if n.contains("connection") || n.contains("thread") || n.contains("client") { return "Connections" } if n.contains("quer") || n.contains("question") || n.hasPrefix("com_") || n.contains("select") || n.contains("sort") || n.contains("tmp_") { return "Queries" } return "Other" } private func value(_ name: String) -> String { guard let row = rows.first(where: { $0.count >= 2 && $0[0].caseInsensitiveCompare(name) == .orderedSame }) else { return "—" } return WorkbenchCounterCharts.displayValue(row[1], for: name) } private func overview(_ title: String, metric: String, tint: Color) -> some View { VStack(alignment: .leading, spacing: 6) { Text(title).font(.caption.weight(.medium)).foregroundStyle(.secondary) Text(value(metric)).font(.system(size: 23, weight: .bold, design: .rounded)) .foregroundStyle(tint).lineLimit(1).minimumScaleFactor(0.7) } .frame(maxWidth: .infinity, minHeight: 66, alignment: .leading) .padding(12) .background(tint.opacity(0.08), in: RoundedRectangle(cornerRadius: 10)) } private var visibleRows: [[String]] { category == "All" ? rows : rows.filter { $0.count >= 2 && group($0[0]) == category } } var body: some View { ScrollView(.vertical) { VStack(alignment: .leading, spacing: 20) { // Graphs are the primary content, never hidden below summary cards. WorkbenchCounterCharts(page: .serverCounters, samples: samples) .frame(maxWidth: .infinity, alignment: .leading) Divider() HStack { Label("All MySQL server counters", systemImage: "list.bullet.rectangle") .font(.title3.bold()) Spacer() Text("\(visibleRows.count) of \(rows.count) counters") .font(.caption).foregroundStyle(.secondary) } Picker("Counter category", selection: $category) { ForEach(categories, id: \.self) { Text($0).tag($0) } } .pickerStyle(.segmented) WorkbenchColouredCounterTable(columns: ["Variable_name", "Value"], rows: visibleRows) .frame(height: 450) Text("Colours identify counter families, not server health. Error totals are cumulative; nonzero values do not necessarily indicate a current fault.") .font(.caption).foregroundStyle(.secondary) } .padding(20) .frame(maxWidth: .infinity, alignment: .leading) } } } // MARK: - Phase 29: Inline coloured counter graphs // MARK: - Phase 33: pinned server graphs and complete coloured statistics private struct WorkbenchPinnedCounterGraphs: View { let samples: [DiagnosticSample] private let metrics = ["Queries", "Questions", "Threads_connected", "Threads_running", "Slow_queries", "Bytes_received", "Bytes_sent", "Innodb_buffer_pool_reads", "Innodb_data_reads"] private func sampleValue(_ sample: DiagnosticSample, metric: String) -> Double? { guard let row = sample.rows.first(where: { $0.count > 1 && $0[0].caseInsensitiveCompare(metric) == .orderedSame }) else { return nil } return Double(row[1]) } private func values(for metric: String) -> [CounterGraphPoint] { samples.indices.compactMap { index in guard let current = sampleValue(samples[index], metric: metric) else { return nil } if metric.hasPrefix("Threads_") { return CounterGraphPoint(time: samples[index].at, value: current) } guard index > 0, let prior = sampleValue(samples[index - 1], metric: metric), current >= prior else { return nil } let seconds = samples[index].at.timeIntervalSince(samples[index - 1].at) guard seconds > 0 else { return nil } return CounterGraphPoint(time: samples[index].at, value: (current - prior) / seconds) } } var body: some View { ScrollView(.horizontal) { HStack(spacing: 12) { ForEach(metrics, id: \.self) { metric in let points = values(for: metric) let colour = WorkbenchCounterCharts.tint(for: metric) VStack(alignment: .leading, spacing: 8) { Text(metric.replacingOccurrences(of: "_", with: " ")) .font(.subheadline.bold()) HStack(alignment: .firstTextBaseline) { Text(points.last.map { $0.value.formatted(.number.precision(.fractionLength(1))) } ?? "—") .font(.title2.bold()).foregroundStyle(colour) Text(metric.hasPrefix("Threads_") ? "current" : "/sec") .font(.caption).foregroundStyle(.secondary) } Chart(points) { point in AreaMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(colour.opacity(0.18)) LineMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(colour) .lineStyle(StrokeStyle(lineWidth: 2.5)) } .chartYAxis { AxisMarks(position: .leading) } .frame(height: 105) if points.isEmpty { Text("Waiting for samples") .font(.caption2).foregroundStyle(.secondary) } } .padding(12) .frame(width: 290, height: 200, alignment: .topLeading) .background(colour.opacity(0.09)) .clipShape(RoundedRectangle(cornerRadius: 12)) } } .padding(.horizontal, 2) } } } private struct CounterGraphPoint: Identifiable { let id = UUID() let time: Date let value: Double } private struct WorkbenchColouredCounterTable: View { let columns: [String] let rows: [[String]] var body: some View { VStack(spacing: 0) { HStack { Text("Variable name").frame(maxWidth: .infinity, alignment: .leading) Text("Value").frame(width: 210, alignment: .trailing) } .font(.subheadline.bold()) .padding(.horizontal, 12).padding(.vertical, 10) .background(Color.secondary.opacity(0.09)) Divider() ScrollView { LazyVStack(spacing: 0) { ForEach(rows.indices, id: \.self) { index in let row = rows[index] if row.count >= 2 { let colour = WorkbenchCounterCharts.tint(for: row[0]) HStack(spacing: 10) { RoundedRectangle(cornerRadius: 2) .fill(colour).frame(width: 4, height: 22) Text(row[0]) .font(.system(.body, design: .monospaced)) .lineLimit(1) .help(row[0]) Spacer(minLength: 8) Text(WorkbenchCounterCharts.displayValue(row[1], for: row[0])) .font(.system(.body, design: .monospaced).weight(.semibold)) .foregroundStyle(colour) .lineLimit(1) .help(row[1]) } .padding(.horizontal, 12).padding(.vertical, 7) .background(index.isMultiple(of: 2) ? Color.secondary.opacity(0.045) : Color.clear) Divider().opacity(0.35) } } } } } .clipShape(RoundedRectangle(cornerRadius: 10)) .overlay(RoundedRectangle(cornerRadius: 10).stroke(Color.secondary.opacity(0.18))) } } private struct WorkbenchCounterCharts: View { let page: WorkbenchAdminPage let samples: [DiagnosticSample] private struct Point: Identifiable { let id: Int let time: Date let value: Double } static func tint(for name: String) -> Color { let key = name.lowercased() if key.contains("abort") || key.contains("error") || key.contains("slow") || key.contains("wait") { return .red } if key.contains("thread") || key.contains("connection") { return .teal } if key.contains("innodb") || key.contains("buffer") { return .purple } if key.contains("bytes") || key.contains("network") { return .green } if key.contains("disk") || key.contains("tmp") { return .orange } return .blue } static func displayValue(_ raw: String, for name: String) -> String { guard let number = Double(raw), number.isFinite else { return raw } let key = name.lowercased() if key == "uptime" { let seconds = max(0, Int(number)) let days = seconds / 86400 let hours = (seconds % 86400) / 3600 let minutes = (seconds % 3600) / 60 return days > 0 ? "\(days)d \(hours)h \(minutes)m" : "\(hours)h \(minutes)m" } if key.contains("bytes") { return ByteCountFormatter.string(fromByteCount: Int64(number), countStyle: .binary) } return number.formatted(.number.grouping(.automatic).precision(.fractionLength(0...2))) } private var metrics: [String] { switch page { case .innodbMetrics: return ["Innodb_buffer_pool_reads", "Innodb_buffer_pool_read_requests", "Innodb_data_reads", "Innodb_data_writes", "Innodb_rows_read", "Innodb_row_lock_waits"] case .connectionStatistics: return ["Threads_connected", "Threads_running", "Connections", "Queries", "Slow_queries", "Aborted_connects"] case .serverCounters: return ["Queries", "Questions", "Threads_connected", "Threads_running", "Slow_queries", "Bytes_received", "Bytes_sent", "Innodb_buffer_pool_reads", "Innodb_data_reads"] case .variables: return ["Queries", "Threads_connected", "Threads_running", "Connections", "Slow_queries", "Bytes_sent"] default: return ["Queries", "Threads_connected", "Threads_running", "Slow_queries"] } } private func value(_ sample: DiagnosticSample, _ metric: String) -> Double? { guard let row = sample.rows.first(where: { $0.count >= 2 && $0[0].caseInsensitiveCompare(metric) == .orderedSame }) else { return nil } return Double(row[1]) } private func points(_ metric: String, rate: Bool) -> [Point] { samples.indices.compactMap { i in guard let current = value(samples[i], metric), current.isFinite else { return nil } if !rate { return Point(id: i, time: samples[i].at, value: current) } guard i > 0, let previous = value(samples[i - 1], metric), current >= previous else { return nil } let elapsed = samples[i].at.timeIntervalSince(samples[i - 1].at) guard elapsed > 0 else { return nil } return Point(id: i, time: samples[i].at, value: (current - previous) / elapsed) } } private func graph(_ metric: String) -> some View { let instantaneous = metric.hasPrefix("Threads_") let values = points(metric, rate: !instantaneous) let tint = Self.tint(for: metric) return VStack(alignment: .leading, spacing: 8) { Text(metric.replacingOccurrences(of: "_", with: " ")).font(.subheadline.weight(.semibold)) .lineLimit(1).minimumScaleFactor(0.8) HStack(alignment: .firstTextBaseline) { Text(values.last.map { $0.value.formatted(.number.precision(.fractionLength(1))) } ?? "—") .font(.system(size: 23, weight: .bold, design: .rounded)) .foregroundStyle(tint) Text(instantaneous ? "current" : "/sec") .font(.caption).foregroundStyle(.secondary) } Chart(values) { point in AreaMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(tint.opacity(0.13)) LineMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(tint).lineStyle(StrokeStyle(lineWidth: 2.5)) } .chartYAxis { AxisMarks(position: .leading) } .frame(height: 205) if values.isEmpty { Text("Waiting for samples").font(.caption2).foregroundStyle(.secondary) } } .padding(16) .frame(maxWidth: .infinity, alignment: .leading) .background(Color(nsColor: .controlBackgroundColor), in: RoundedRectangle(cornerRadius: 12)) .overlay(RoundedRectangle(cornerRadius: 12).stroke(tint.opacity(0.35), lineWidth: 1)) } var body: some View { VStack(alignment: .leading, spacing: 12) { HStack { Label(page == .serverCounters ? "Server Counters · Live Graphs" : "Live counter graphs", systemImage: "chart.xyaxis.line") .font(.title2.bold()) Spacer() Text("\(samples.count) samples · 5-second refresh") .font(.caption).foregroundStyle(.secondary) } LazyVGrid(columns: [GridItem(.adaptive(minimum: 285), spacing: 14)], spacing: 12) { ForEach(metrics, id: \.self) { metric in graph(metric) } } Text("Live MySQL data • refreshes every five seconds. Rates appear after two successful samples; thread counts show current values.") .font(.caption).foregroundStyle(.secondary) } } } // MARK: - Phase 25/26: Live performance samples and counter rates private struct DiagnosticSample: Identifiable { let id = UUID() let at: Date let columns: [String] let rows: [[String]] } private struct WorkbenchDiagnosticTrendsView: View { let page: WorkbenchAdminPage let samples: [DiagnosticSample] @Environment(\.dismiss) private var dismiss private func metricMap(_ sample: DiagnosticSample) -> [String: Double] { var values: [String: Double] = [:] if page == .slowQueries { guard let name = sample.columns.firstIndex(of: "DIGEST_TEXT"), let count = sample.columns.firstIndex(of: "COUNT_STAR") else { return values } for row in sample.rows where row.indices.contains(name) && row.indices.contains(count) { if let value = Double(row[count]) { values[row[name]] = value } } } else { for row in sample.rows where row.count >= 2 { if let value = Double(row[1]) { values[row[0]] = value } } } return values } private var comparisons: [(String, Double, Double, Double?)] { guard let latest = samples.last else { return [] } let current = metricMap(latest) let prior = samples.count > 1 ? metricMap(samples[samples.count - 2]) : [:] let seconds = samples.count > 1 ? latest.at.timeIntervalSince(samples[samples.count - 2].at) : 0 return current.keys.sorted().map { key in let value = current[key] ?? 0 let before = prior[key] ?? value let rate: Double? = seconds > 0 && prior[key] != nil && value >= before ? (value - before) / seconds : nil return (key, value, before, rate) } } @State private var selectedMetric = "" private var metricNames: [String] { guard let last = samples.last else { return [] } return Array(metricMap(last).keys).sorted() } private var activeMetric: String { metricNames.contains(selectedMetric) ? selectedMetric : (metricNames.first ?? "") } private struct ChartPoint: Identifiable { let id: Int let time: Date let value: Double } private var chartPoints: [ChartPoint] { samples.enumerated().compactMap { index, sample in guard let value = metricMap(sample)[activeMetric], value.isFinite else { return nil } return ChartPoint(id: index, time: sample.at, value: value) } } private var chartTint: Color { let name = activeMetric.lowercased() if name.contains("abort") || name.contains("error") || name.contains("lock_wait") { return .red } if name.contains("slow") || name.contains("temporary") { return .orange } if name.contains("thread") || name.contains("connection") { return .teal } return .blue } var body: some View { VStack(alignment: .leading, spacing: 12) { HStack { Text("\(page.rawValue) — Trends").font(.title2.bold()) Spacer() Button("Close") { dismiss() } } Text("\(samples.count) samples. Rates compare the two latest successful refreshes; cumulative counters can reset when MySQL restarts or statistics are cleared.") .font(.caption).foregroundStyle(.secondary) if !metricNames.isEmpty { HStack { Picker("Chart metric", selection: $selectedMetric) { ForEach(metricNames, id: \.self) { name in Text(name).tag(name) } } .frame(maxWidth: 480) Spacer() Label("Live samples", systemImage: "chart.xyaxis.line") .foregroundStyle(chartTint) } .onAppear { if selectedMetric.isEmpty { selectedMetric = metricNames.first ?? "" } } Chart(chartPoints) { point in AreaMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(chartTint.opacity(0.13)) .interpolationMethod(.linear) LineMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(chartTint) .lineStyle(StrokeStyle(lineWidth: 2.5)) .interpolationMethod(.linear) PointMark(x: .value("Time", point.time), y: .value("Value", point.value)) .foregroundStyle(chartTint) } .chartYAxis { AxisMarks(position: .leading) } .frame(height: 205) } if samples.count < 2 { ContentUnavailableView("More samples needed", systemImage: "chart.xyaxis.line", description: Text("Refresh this screen twice, or enable auto-refresh.")) } else { ScrollView { LazyVStack(alignment: .leading, spacing: 8) { HStack { Text("Metric").frame(maxWidth: .infinity, alignment: .leading) Text("Current").frame(width: 110, alignment: .trailing) Text("Previous").frame(width: 110, alignment: .trailing) Text("Change/sec").frame(width: 110, alignment: .trailing) }.font(.headline) Divider() ForEach(comparisons.indices, id: \.self) { index in let item = comparisons[index] HStack { Text(item.0).lineLimit(2).frame(maxWidth: .infinity, alignment: .leading) Text(item.1.formatted()).frame(width: 110, alignment: .trailing) Text(item.2.formatted()).frame(width: 110, alignment: .trailing) Text(item.3.map { $0.formatted(.number.precision(.fractionLength(2))) } ?? "—") .foregroundStyle(item.3 == nil ? Color.secondary : (item.3! > 0 ? Color.teal : Color.secondary)) .frame(width: 110, alignment: .trailing) }.font(.system(.caption, design: .monospaced)) Divider() } } } } Text("Query digest counts are grouped by SQL fingerprint. This is a sampled comparison, not a durable monitoring database.") .font(.caption).foregroundStyle(.secondary) } .padding(20) .frame(minWidth: 820, minHeight: 480) } } // MARK: - Phase 27: Native Schema Inspector and SQL DDL Editor private struct SchemaEditorTarget: Identifiable { let id = UUID() let database: String let table: String? } private struct WorkbenchSchemaEditorView: View { let profile: ConnectionProfile? let target: SchemaEditorTarget @Environment(\.dismiss) private var dismiss @State private var columns = QueryResult() @State private var indexes = QueryResult() @State private var originalDDL = "" @State private var proposedSQL = "" @State private var busy = false @State private var message: String? @State private var confirmExecution = false @State private var successful = false private func identifier(_ value: String) -> String { "`" + value.replacingOccurrences(of: "`", with: "``") + "`" } private var qualifiedName: String { identifier(target.database) + "." + identifier(target.table ?? "new_table") } private func query(_ sql: String) async throws -> QueryResult { guard let profile else { throw NSError(domain: "WorkbenchSchemaEditor", code: 1, userInfo: [NSLocalizedDescriptionKey: "Connect to MySQL first."]) } return try await NativeMySQLRunner.run( profile: profile, database: target.database, password: PasswordStore.read(profile.id), sql: sql ) } var body: some View { VStack(alignment: .leading, spacing: 12) { HStack { Label(target.table == nil ? "Create Table" : "Table Schema Editor", systemImage: "tablecells.badge.ellipsis") .font(.title2.bold()) Spacer() Text(target.database).foregroundStyle(.secondary) Button("Close") { dismiss() } } if let message { Label(message, systemImage: successful ? "checkmark.circle" : "exclamationmark.triangle") .foregroundStyle(successful ? .green : .red) .textSelection(.enabled) } if target.table != nil { TabView { schemaGrid(columns).tabItem { Text("Columns") } schemaGrid(indexes).tabItem { Text("Indexes") } ScrollView { Text(originalDDL).font(.system(.body, design: .monospaced)) .textSelection(.enabled).frame(maxWidth: .infinity, alignment: .leading) .padding() }.tabItem { Text("Original DDL") } } .frame(minHeight: 180) } HStack { Text("SQL schema changes").font(.headline) Spacer() Text("Review before execution").font(.caption).foregroundStyle(.orange) } TextEditor(text: $proposedSQL) .font(.system(.body, design: .monospaced)) .frame(minHeight: 160) .border(Color.secondary.opacity(0.3)) HStack { Text("DDL may implicitly commit and cannot reliably be rolled back.") .font(.caption).foregroundStyle(.secondary) Spacer() Button("Reload") { Task { await load() } }.disabled(busy) Button("Execute DDL…") { confirmExecution = true } .disabled(busy || proposedSQL.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty) .buttonStyle(.borderedProminent) } } .padding(20) .frame(minWidth: 820, minHeight: 620) .task { await load() } .confirmationDialog("Execute schema modification?", isPresented: $confirmExecution, titleVisibility: .visible) { Button("Execute SQL", role: .destructive) { Task { await executeDDL() } } Button("Cancel", role: .cancel) { } } message: { Text("This SQL changes the database schema and may implicitly commit. Verify the statement and take a backup first.") } } private func schemaGrid(_ data: QueryResult) -> some View { VStack(alignment: .leading, spacing: 6) { if data.columns.isEmpty { Text("No metadata loaded").foregroundStyle(.secondary) } else { ScrollView([.horizontal, .vertical]) { Grid(alignment: .leading, horizontalSpacing: 18, verticalSpacing: 7) { GridRow { ForEach(data.columns, id: \.self) { name in Text(name).fontWeight(.bold) } } Divider() ForEach(data.rows.indices, id: \.self) { index in GridRow { ForEach(data.rows[index].indices, id: \.self) { column in Text(data.rows[index][column]) .font(.system(.caption, design: .monospaced)) .textSelection(.enabled) } } } }.padding(8) } } }.frame(maxWidth: .infinity, maxHeight: .infinity) } @MainActor private func load() async { guard target.table != nil else { proposedSQL = "CREATE TABLE " + qualifiedName + " (\n id BIGINT NOT NULL AUTO_INCREMENT,\n PRIMARY KEY (id)\n) ENGINE=InnoDB;" return } busy = true defer { busy = false } do { columns = try await query("SHOW FULL COLUMNS FROM " + qualifiedName) indexes = try await query("SHOW INDEX FROM " + qualifiedName) let ddl = try await query("SHOW CREATE TABLE " + qualifiedName) originalDDL = ddl.rows.first?.last ?? "" if proposedSQL.isEmpty { proposedSQL = "-- Enter ALTER TABLE statements here.\nALTER TABLE " + qualifiedName + "\n ADD COLUMN new_column VARCHAR(255) NULL;" } message = nil } catch { message = error.localizedDescription successful = false } } @MainActor private func executeDDL() async { let sql = proposedSQL.trimmingCharacters(in: .whitespacesAndNewlines) // Single-statement editor: avoid silently executing partial multi-statement input. let statement = sql.hasSuffix(";") ? String(sql.dropLast()).trimmingCharacters(in: .whitespacesAndNewlines) : sql guard !statement.isEmpty else { return } busy = true defer { busy = false } do { _ = try await query(statement) successful = true message = "Schema statement executed successfully." if target.table != nil { await load() } } catch { successful = false message = error.localizedDescription } } } // MARK: - View and Stored Routine Definition Editor private struct WorkbenchObjectEditorTarget: Identifiable { enum Kind: String { case view = "VIEW", procedure = "PROCEDURE", function = "FUNCTION" } let database: String let name: String let kind: Kind var id: String { database + "." + kind.rawValue + "." + name } } private struct WorkbenchObjectEditorView: View { let profile: ConnectionProfile? let target: WorkbenchObjectEditorTarget @Environment(\.dismiss) private var dismiss @State private var definition = "" @State private var originalDefinition = "" @State private var message: String? @State private var isError = false @State private var busy = false @State private var confirmSave = false private func quote(_ value: String) -> String { "`" + value.replacingOccurrences(of: "`", with: "``") + "`" } private var qualified: String { quote(target.database) + "." + quote(target.name) } private func query(_ sql: String) async throws -> QueryResult { guard let profile else { throw NSError(domain: "WorkbenchObjectEditor", code: 1, userInfo: [NSLocalizedDescriptionKey: "Connect to MySQL first."]) } return try await NativeMySQLRunner.run( profile: profile, database: target.database, password: PasswordStore.read(profile.id), sql: sql ) } var body: some View { VStack(alignment: .leading, spacing: 12) { HStack { Label("Edit " + target.kind.rawValue.capitalized, systemImage: "curlybraces") .font(.title2.bold()) Text(qualified).font(.subheadline.monospaced()).foregroundStyle(.secondary) Spacer() Button("Close") { dismiss() } } if let message { Text(message) .foregroundStyle(isError ? .red : .green) .textSelection(.enabled) } Text("Edit the complete CREATE statement. The original definition is kept in memory until this window closes.") .font(.caption).foregroundStyle(.secondary) TextEditor(text: $definition) .font(.system(.body, design: .monospaced)) .frame(minHeight: 350) .border(Color.secondary.opacity(0.3)) if target.kind != .view { Text("MySQL does not support CREATE OR REPLACE for routines. Saving drops and recreates the routine. This is not atomic: grants, DEFINER and concurrent callers may be affected. Back up the routine first.") .font(.caption).foregroundStyle(.orange) } HStack { Button("Reload Original") { definition = originalDefinition }.disabled(busy) Spacer() ProgressView().controlSize(.small).opacity(busy ? 1 : 0) Button("Save Definition…") { confirmSave = true } .disabled(busy || definition.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty || definition == originalDefinition) .buttonStyle(.borderedProminent) } } .padding(20) .frame(minWidth: 850, minHeight: 560) .task { await load() } .confirmationDialog("Apply database definition?", isPresented: $confirmSave, titleVisibility: .visible) { Button(target.kind == .view ? "Replace View" : "Drop and Recreate " + target.kind.rawValue.capitalized, role: .destructive) { Task { await save() } } Button("Cancel", role: .cancel) { } } message: { Text(target.kind == .view ? "This will replace the view definition on the MySQL server." : "This drops the existing routine before recreating it. MySQL cannot roll back this operation.") } } @MainActor private func load() async { busy = true defer { busy = false } do { let result = try await query("SHOW CREATE " + target.kind.rawValue + " " + qualified) guard let row = result.rows.first, let index = result.columns.firstIndex(where: { $0.lowercased().hasPrefix("create ") }), row.indices.contains(index) else { throw NSError(domain: "WorkbenchObjectEditor", code: 2, userInfo: [NSLocalizedDescriptionKey: "MySQL did not return the CREATE definition. Check SHOW VIEW or routine privileges."]) } originalDefinition = row[index] definition = originalDefinition message = nil } catch { isError = true message = error.localizedDescription } } private func normalized(_ sql: String) -> String { let trimmed = sql.trimmingCharacters(in: .whitespacesAndNewlines) return trimmed.hasSuffix(";") ? String(trimmed.dropLast()) : trimmed } @MainActor private func save() async { busy = true defer { busy = false } let sql = normalized(definition) guard sql.range(of: #"^CREATE\s+"#, options: [.regularExpression, .caseInsensitive]) != nil else { isError = true message = "The definition must start with CREATE." return } let expectedKind = target.kind.rawValue guard sql.range(of: "\\b" + expectedKind + "\\b", options: [.regularExpression, .caseInsensitive]) != nil else { isError = true message = "Expected a CREATE " + expectedKind + " statement." return } do { if target.kind == .view { // SHOW CREATE VIEW may include ALGORITHM / DEFINER / SQL SECURITY. // Insert OR REPLACE immediately after CREATE, retaining those clauses. let replacement = sql.replacingOccurrences( of: #"^CREATE\s+(?:OR\s+REPLACE\s+)?"#, with: "CREATE OR REPLACE ", options: [.regularExpression, .caseInsensitive] ) _ = try await query(replacement) } else { // MySQL has no CREATE OR REPLACE PROCEDURE/FUNCTION. _ = try await query("DROP " + target.kind.rawValue + " " + qualified) do { _ = try await query(sql) } catch { // Best effort restoration of the original definition. do { _ = try await query(normalized(originalDefinition)) throw NSError(domain: "WorkbenchObjectEditor", code: 3, userInfo: [NSLocalizedDescriptionKey: "Updated routine failed: \(error.localizedDescription). The original definition was restored; verify privileges and grants."]) } catch let restoreError { if (restoreError as NSError).code == 3 && (restoreError as NSError).domain == "WorkbenchObjectEditor" { throw restoreError } throw NSError(domain: "WorkbenchObjectEditor", code: 4, userInfo: [NSLocalizedDescriptionKey: "Updated routine failed: \(error.localizedDescription). Restoration also failed: \(restoreError.localizedDescription). Restore from your backup immediately."]) } } } originalDefinition = sql definition = sql isError = false message = "Definition saved successfully." } catch { isError = true message = error.localizedDescription } } } // MARK: - Phase 57: Schema management private struct WorkbenchSchemaManagementTarget: Identifiable { enum Action { case create, alter, drop, inspect } let id = UUID() let database: String? let action: Action } private struct WorkbenchSchemaManagementSheet: View { @Environment(\.dismiss) private var dismiss @ObservedObject var model: DatabaseModel let target: WorkbenchSchemaManagementTarget @State private var name = "" @State private var charset = "utf8mb4" @State private var collation = "utf8mb4_0900_ai_ci" @State private var confirmation = "" @State private var submitted = false private var escapedName: String { name.replacingOccurrences(of: "`", with: "``") } private var title: String { switch target.action { case .create: return "Create Schema" case .alter: return "Alter Schema" case .drop: return "Drop Schema" case .inspect: return "Schema Inspector" } } var body: some View { VStack(alignment: .leading, spacing: 16) { Text(title).font(.title2.bold()) if target.action == .inspect { LabeledContent("Schema", value: name) Text("Inspect the schema definition and available metadata using the SQL editor.") .foregroundStyle(.secondary) Button("Show CREATE DATABASE") { model.sql = "SHOW CREATE DATABASE `\(escapedName)`;" dismiss() } } else { TextField("Schema name", text: $name) .disabled(target.action != .create) if target.action != .drop { TextField("Character set", text: $charset) TextField("Collation", text: $collation) Text("Character set and collation must be valid identifiers supported by the server.") .font(.caption).foregroundStyle(.secondary) } else { Text("This permanently deletes the schema and ALL its tables, views, routines and data. This cannot be undone.") .foregroundStyle(.red) TextField("Type the schema name to confirm", text: $confirmation) } if submitted { Text("Statement submitted. Check the SQL result/status before continuing.") .font(.caption).foregroundStyle(.secondary) } } HStack { Spacer() Button("Close") { dismiss() } if target.action != .inspect { Button(target.action == .drop ? "Drop Schema" : "Apply") { apply() } .disabled(!valid || submitted) .keyboardShortcut(.defaultAction) } } } .padding(24) .frame(width: 480) .onAppear { name = target.database ?? "" } } private var valid: Bool { guard !name.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty else { return false } if target.action == .drop { return confirmation == name } if target.action == .inspect { return true } let allowed = CharacterSet.alphanumerics.union(CharacterSet(charactersIn: "_")) return !charset.isEmpty && !collation.isEmpty && charset.unicodeScalars.allSatisfy(allowed.contains) && collation.unicodeScalars.allSatisfy(allowed.contains) } private func apply() { guard valid else { return } let sql: String switch target.action { case .create: sql = "CREATE DATABASE `\(escapedName)` CHARACTER SET \(charset) COLLATE \(collation);" case .alter: sql = "ALTER DATABASE `\(escapedName)` CHARACTER SET \(charset) COLLATE \(collation);" case .drop: sql = "DROP DATABASE `\(escapedName)`;" case .inspect: return } model.runSQL(statement: sql) submitted = true } } // MARK: - Phase 59: Table data search and CSV import private struct WorkbenchTableTarget: Identifiable { let id = UUID() let database: String let table: String? } private enum WorkbenchSQLValue { static func identifier(_ name: String) -> String { "`" + name.replacingOccurrences(of: "`", with: "``") + "`" } static func literal(_ value: String) -> String { // MySQL backslash escaping; use the connection's SQL mode for interpretation. let escaped = value.replacingOccurrences(of: "\\", with: "\\\\") .replacingOccurrences(of: "'", with: "''") .replacingOccurrences(of: "\0", with: "\\0") .replacingOccurrences(of: "\n", with: "\\n") .replacingOccurrences(of: "\r", with: "\\r") return "'" + escaped + "'" } } private struct WorkbenchTableSearchSheet: View { @Environment(\.dismiss) private var dismiss @ObservedObject var model: DatabaseModel let target: WorkbenchTableTarget let execute: (String) -> Void @State private var columns: [String] = [] @State private var column = "" @State private var term = "" @State private var mode = "Contains" @State private var limit = 500 @State private var error = "" var body: some View { VStack(alignment: .leading, spacing: 16) { Text("Search Table Data").font(.title2.bold()) Text("\(target.database).\(target.table ?? "")").foregroundStyle(.secondary) if !error.isEmpty { Text(error).foregroundStyle(.red) } Picker("Column", selection: $column) { ForEach(columns, id: \.self) { Text($0).tag($0) } } Picker("Condition", selection: $mode) { ForEach(["Contains", "Equals", "Starts with", "Ends with"], id: \.self) { Text($0).tag($0) } } TextField("Search value", text: $term) Stepper("Maximum rows: \(limit)", value: $limit, in: 1...10000, step: 100) HStack { Spacer() Button("Cancel") { dismiss() } Button("Search") { let db = WorkbenchSQLValue.identifier(target.database) let tbl = WorkbenchSQLValue.identifier(target.table ?? "") let expression: String switch mode { case "Equals": expression = WorkbenchSQLValue.literal(term) case "Starts with": expression = WorkbenchSQLValue.literal(term + "%") case "Ends with": expression = WorkbenchSQLValue.literal("%" + term) default: expression = WorkbenchSQLValue.literal("%" + term + "%") } let op = mode == "Equals" ? "=" : "LIKE" execute("SELECT * FROM \(db).\(tbl) WHERE \(WorkbenchSQLValue.identifier(column)) \(op) \(expression) LIMIT \(limit);") dismiss() } .disabled(column.isEmpty) .keyboardShortcut(.defaultAction) } } .padding(24) .frame(width: 520) .task { guard let profile = model.selected, let table = target.table else { return } do { let sql = "SHOW COLUMNS FROM \(WorkbenchSQLValue.identifier(target.database)).\(WorkbenchSQLValue.identifier(table));" let result = try await NativeMySQLRunner.run(profile: profile, database: target.database, password: PasswordStore.read(profile.id), sql: sql) columns = result.rows.compactMap { $0.first } column = columns.first ?? "" } catch { error = error.localizedDescription } } } } private struct WorkbenchCSVImportSheet: View { @Environment(\.dismiss) private var dismiss @ObservedObject var model: DatabaseModel let target: WorkbenchTableTarget @State private var fileURL: URL? @State private var headers: [String] = [] @State private var records: [[String]] = [] @State private var selectedColumns: [String] = [] @State private var availableColumns: [String] = [] @State private var newTableName = "imported_data" @State private var firstRowHeaders = true @State private var spatialColumns: Set = [] @State private var busy = false @State private var progress = 0 @State private var error = "" @State private var completed = false var body: some View { VStack(alignment: .leading, spacing: 12) { Text("Table Data Import Wizard").font(.title2.bold()) Text("Schema: \(target.database)").foregroundStyle(.secondary) if target.table == nil { TextField("New table name", text: $newTableName) } else { Text("Destination table: \(target.table!)") } HStack { Button("Choose CSV File…") { chooseFile() } Text(fileURL?.lastPathComponent ?? "No file selected") .lineLimit(1).foregroundStyle(.secondary) } Toggle("First row contains column names", isOn: $firstRowHeaders) .onChange(of: firstRowHeaders) { _, _ in prepareMapping() } if !records.isEmpty { Text("\(dataRows.count) data rows · \(headers.count) columns").font(.caption) ScrollView { VStack(alignment: .leading, spacing: 8) { ForEach(headers.indices, id: \.self) { index in HStack { Text(headers[index]).frame(width: 150, alignment: .leading).lineLimit(1) if target.table == nil { TextField("Destination column", text: $selectedColumns[index]) .frame(width: 190) } else { Picker("Destination", selection: $selectedColumns[index]) { Text("Skip").tag("") ForEach(availableColumns, id: \.self) { Text($0).tag($0) } }.frame(width: 220) } Toggle("WKT geometry", isOn: Binding( get: { spatialColumns.contains(index) }, set: { if $0 { spatialColumns.insert(index) } else { spatialColumns.remove(index) } } )).toggleStyle(.checkbox) } } Divider() Text("Preview (first 5 rows)").font(.headline) ForEach(Array(dataRows.prefix(5).enumerated()), id: \.offset) { _, row in Text(row.joined(separator: " | ")) .font(.system(.caption, design: .monospaced)).lineLimit(2) } } }.frame(maxHeight: 350) } if !error.isEmpty { Text(error).foregroundStyle(.red).textSelection(.enabled) } if busy { ProgressView(value: Double(progress), total: Double(max(1, dataRows.count))) } if completed { Label("Imported \(progress) rows", systemImage: "checkmark.circle.fill").foregroundStyle(.green) } HStack { Spacer() Button("Close") { dismiss() }.disabled(busy) Button("Import \(dataRows.count) Rows") { Task { await importRows() } } .disabled(busy || records.isEmpty || dataRows.isEmpty || (target.table == nil && newTableName.isEmpty)) .keyboardShortcut(.defaultAction) } } .padding(22) .frame(width: 740, height: 660) .task { await loadColumns() } } private var dataRows: [[String]] { firstRowHeaders ? Array(records.dropFirst()) : records } private func chooseFile() { let panel = NSOpenPanel() panel.allowedContentTypes = [.commaSeparatedText, .plainText] panel.allowsMultipleSelection = false guard panel.runModal() == .OK, let url = panel.url else { return } do { let data = try Data(contentsOf: url) guard let text = String(data: data, encoding: .utf8) else { error = "CSV must be UTF-8 encoded."; return } let parsed = try WorkbenchCSVParser.parse(text) guard !parsed.isEmpty else { error = "CSV file is empty."; return } fileURL = url records = parsed error = "" progress = 0 completed = false prepareMapping() } catch { error = error.localizedDescription } } private func prepareMapping() { let width = records.map(\.count).max() ?? 0 headers = (0.. String in let value = index < row.count ? row[index] : "" let literal = WorkbenchSQLValue.literal(value) return spatialColumns.contains(index) ? "ST_GeomFromText(\(literal))" : literal }.joined(separator: ", ") _ = try await NativeMySQLRunner.run(profile: profile, database: target.database, password: password, sql: "INSERT INTO \(qualified) (\(names)) VALUES (\(values));") progress += 1 } completed = true model.loadTables(database: target.database) } catch { error = "Stopped after \(progress) rows: \(error.localizedDescription). Previously inserted rows may remain." } busy = false } } private enum WorkbenchCSVParser { enum CSVError: LocalizedError { case unterminatedQuote var errorDescription: String? { "CSV contains an unterminated quoted field." } } static func parse(_ text: String) throws -> [[String]] { var rows: [[String]] = [] var row: [String] = [] var field = "" var quoted = false let chars = Array(text) var i = 0 while i < chars.count { let c = chars[i] if quoted { if c == "\"" { if i + 1 < chars.count && chars[i + 1] == "\"" { field.append("\""); i += 1 } else { quoted = false } } else { field.append(c) } } else if c == "\"" && field.isEmpty { quoted = true } else if c == "," { row.append(field); field = "" } else if c == "\n" { row.append(field.trimmingCharacters(in: .init(charactersIn: "\r"))) rows.append(row); row = []; field = "" } else { field.append(c) } i += 1 } if quoted { throw CSVError.unterminatedQuote } if !row.isEmpty || !field.isEmpty { row.append(field); rows.append(row) } return rows } }