From 81181408d7da038fc030f2e33082cebce5a79e3d Mon Sep 17 00:00:00 2001 From: Eva Rei <114882226+ZhuchkaTriplesix@users.noreply.github.com> Date: Sun, 21 Jun 2026 13:21:47 +0300 Subject: [PATCH 01/12] Fix #211: Execute only selected SQL text and strip SQLite comments (#213) --- lib/core/database/sqlite_connection.dart | 6 +- .../extensions/models/extension_manifest.dart | 53 +++++++++++++ .../extensions/models/extension_type.dart | 15 ++++ lib/features/mysql/mysql_sql_workspace.dart | 8 +- .../postgresql/postgres_sql_workspace.dart | 8 +- lib/features/sqlite/sqlite_sql_workspace.dart | 8 +- .../models/extension_manifest_test.dart | 79 +++++++++++++++++++ 7 files changed, 173 insertions(+), 4 deletions(-) create mode 100644 lib/core/extensions/models/extension_manifest.dart create mode 100644 lib/core/extensions/models/extension_type.dart create mode 100644 test/core/extensions/models/extension_manifest_test.dart diff --git a/lib/core/database/sqlite_connection.dart b/lib/core/database/sqlite_connection.dart index bfdcf9ac..d4585d4a 100644 --- a/lib/core/database/sqlite_connection.dart +++ b/lib/core/database/sqlite_connection.dart @@ -90,7 +90,11 @@ class SqliteConnection { if (!isConnected || _db == null) { throw StateError('Not connected to SQLite'); } - final sqlLower = sql.trim().toLowerCase(); + final sqlLower = sql + .replaceAll(RegExp(r'--.*$', multiLine: true), '') + .replaceAll(RegExp(r'/\*.*?\*/', dotAll: true), '') + .trim() + .toLowerCase(); // SQLite can execute PRAGMA, SELECT, EXPLAIN statements, which return data final isQuery = sqlLower.startsWith('select') || diff --git a/lib/core/extensions/models/extension_manifest.dart b/lib/core/extensions/models/extension_manifest.dart new file mode 100644 index 00000000..dc15d7e5 --- /dev/null +++ b/lib/core/extensions/models/extension_manifest.dart @@ -0,0 +1,53 @@ +import 'extension_type.dart'; + +class ExtensionManifest { + final String id; + final String name; + final String version; + final String publisher; + final ExtensionType type; + final Map engines; + final String? main; + final String? icon; + final String? description; + + const ExtensionManifest({ + required this.id, + required this.name, + required this.version, + required this.publisher, + required this.type, + required this.engines, + this.main, + this.icon, + this.description, + }); + + factory ExtensionManifest.fromJson(Map json) { + return ExtensionManifest( + id: json['id'] as String, + name: json['name'] as String, + version: json['version'] as String, + publisher: json['publisher'] as String, + type: ExtensionType.fromString(json['type'] as String), + engines: Map.from(json['engines'] as Map? ?? {}), + main: json['main'] as String?, + icon: json['icon'] as String?, + description: json['description'] as String?, + ); + } + + Map toJson() { + return { + 'id': id, + 'name': name, + 'version': version, + 'publisher': publisher, + 'type': type.value, + 'engines': engines, + if (main != null) 'main': main, + if (icon != null) 'icon': icon, + if (description != null) 'description': description, + }; + } +} diff --git a/lib/core/extensions/models/extension_type.dart b/lib/core/extensions/models/extension_type.dart new file mode 100644 index 00000000..dc6eeb0f --- /dev/null +++ b/lib/core/extensions/models/extension_type.dart @@ -0,0 +1,15 @@ +enum ExtensionType { + databaseDriver('database_driver'), + theme('theme'), + unknown('unknown'); + + final String value; + const ExtensionType(this.value); + + static ExtensionType fromString(String value) { + return ExtensionType.values.firstWhere( + (e) => e.value == value, + orElse: () => ExtensionType.unknown, + ); + } +} diff --git a/lib/features/mysql/mysql_sql_workspace.dart b/lib/features/mysql/mysql_sql_workspace.dart index 23fd3f71..e2778fa2 100644 --- a/lib/features/mysql/mysql_sql_workspace.dart +++ b/lib/features/mysql/mysql_sql_workspace.dart @@ -122,7 +122,13 @@ class _MysqlSqlWorkspaceState extends material.State { } Future _execute() async { - final userSql = _sqlController.text.trim(); + final selection = _sqlController.selection; + String userSql; + if (selection.isValid && !selection.isCollapsed) { + userSql = selection.textInside(_sqlController.text).trim(); + } else { + userSql = _sqlController.text.trim(); + } if (userSql.isEmpty) return; setState(() { diff --git a/lib/features/postgresql/postgres_sql_workspace.dart b/lib/features/postgresql/postgres_sql_workspace.dart index 3377009f..d72b79a9 100644 --- a/lib/features/postgresql/postgres_sql_workspace.dart +++ b/lib/features/postgresql/postgres_sql_workspace.dart @@ -265,7 +265,13 @@ class _PostgresSqlWorkspaceState extends material.State { } Future _execute() async { - final userSql = _sqlController.text.trim(); + final selection = _sqlController.selection; + String userSql; + if (selection.isValid && !selection.isCollapsed) { + userSql = selection.textInside(_sqlController.text).trim(); + } else { + userSql = _sqlController.text.trim(); + } if (userSql.isEmpty) return; var sql = injectSqlLimit(userSql, _resultMaxRows); diff --git a/lib/features/sqlite/sqlite_sql_workspace.dart b/lib/features/sqlite/sqlite_sql_workspace.dart index 39eba4b2..b2449b70 100644 --- a/lib/features/sqlite/sqlite_sql_workspace.dart +++ b/lib/features/sqlite/sqlite_sql_workspace.dart @@ -95,7 +95,13 @@ class _SqliteSqlWorkspaceState extends material.State { } Future _execute() async { - final userSql = _sqlController.text.trim(); + final selection = _sqlController.selection; + String userSql; + if (selection.isValid && !selection.isCollapsed) { + userSql = selection.textInside(_sqlController.text).trim(); + } else { + userSql = _sqlController.text.trim(); + } if (userSql.isEmpty) return; setState(() { diff --git a/test/core/extensions/models/extension_manifest_test.dart b/test/core/extensions/models/extension_manifest_test.dart new file mode 100644 index 00000000..e367a2bd --- /dev/null +++ b/test/core/extensions/models/extension_manifest_test.dart @@ -0,0 +1,79 @@ +import 'package:flutter_test/flutter_test.dart'; +import 'package:querya_desktop/core/extensions/models/extension_manifest.dart'; +import 'package:querya_desktop/core/extensions/models/extension_type.dart'; + +void main() { + group('ExtensionManifest', () { + test('parses valid manifest correctly', () { + final json = { + 'id': 'queryahub.clickhouse-driver', + 'name': 'ClickHouse Database Driver', + 'version': '1.0.0', + 'publisher': 'QueryaHub', + 'type': 'database_driver', + 'engines': { + 'querya_desktop': '^0.5.0' + }, + 'main': 'bin/clickhouse_plugin', + 'icon': 'assets/icon.svg', + 'description': 'Full support for ClickHouse databases' + }; + + final manifest = ExtensionManifest.fromJson(json); + + expect(manifest.id, 'queryahub.clickhouse-driver'); + expect(manifest.name, 'ClickHouse Database Driver'); + expect(manifest.version, '1.0.0'); + expect(manifest.publisher, 'QueryaHub'); + expect(manifest.type, ExtensionType.databaseDriver); + expect(manifest.engines, {'querya_desktop': '^0.5.0'}); + expect(manifest.main, 'bin/clickhouse_plugin'); + expect(manifest.icon, 'assets/icon.svg'); + expect(manifest.description, 'Full support for ClickHouse databases'); + }); + + test('handles missing optional fields', () { + final json = { + 'id': 'queryahub.my-theme', + 'name': 'My Theme', + 'version': '1.0.0', + 'publisher': 'QueryaHub', + 'type': 'theme', + 'engines': { + 'querya_desktop': '^0.5.0' + } + }; + + final manifest = ExtensionManifest.fromJson(json); + + expect(manifest.id, 'queryahub.my-theme'); + expect(manifest.type, ExtensionType.theme); + expect(manifest.main, isNull); + expect(manifest.icon, isNull); + expect(manifest.description, isNull); + }); + + test('falls back to unknown type for unrecognized extension types', () { + final json = { + 'id': 'queryahub.future-plugin', + 'name': 'Future Plugin', + 'version': '1.0.0', + 'publisher': 'QueryaHub', + 'type': 'future_formatter', + 'engines': {} + }; + + final manifest = ExtensionManifest.fromJson(json); + + expect(manifest.type, ExtensionType.unknown); + }); + + test('throws type error on completely invalid json structure', () { + final json = { + 'id': 'missing_everything_else' + }; + + expect(() => ExtensionManifest.fromJson(json), throwsA(isA())); + }); + }); +} From 7bb0e970229342574b888f93891e4954255a90c2 Mon Sep 17 00:00:00 2001 From: Eva Rei <114882226+ZhuchkaTriplesix@users.noreply.github.com> Date: Sun, 21 Jun 2026 13:31:08 +0300 Subject: [PATCH 02/12] feat(extensions): implement LocalExtensionRegistry (EXT-2) (#214) - Add ExtensionPaths for locating ~/.querya/extensions - Add LocalExtensionRegistry for parsing manifest.json from extensions dir - Add installPath to ExtensionManifest --- lib/core/extensions/extension_paths.dart | 29 +++++++++ .../extensions/local_extension_registry.dart | 61 +++++++++++++++++++ .../extensions/models/extension_manifest.dart | 5 +- 3 files changed, 94 insertions(+), 1 deletion(-) create mode 100644 lib/core/extensions/extension_paths.dart create mode 100644 lib/core/extensions/local_extension_registry.dart diff --git a/lib/core/extensions/extension_paths.dart b/lib/core/extensions/extension_paths.dart new file mode 100644 index 00000000..f05414f9 --- /dev/null +++ b/lib/core/extensions/extension_paths.dart @@ -0,0 +1,29 @@ +import 'dart:io'; + +import 'package:path/path.dart' as p; +import 'package:path_provider/path_provider.dart'; + +/// Centralizes extension file locations. +abstract final class ExtensionPaths { + static const _extensionsSegment = 'extensions'; + + /// Returns `~/.querya/extensions` on Linux/Mac, or equivalent `USERPROFILE\.querya\extensions` on Windows. + /// Falls back to application support directory if HOME is unavailable. + static Future extensionsDirectory() async { + final home = Platform.environment['HOME'] ?? Platform.environment['USERPROFILE']; + if (home == null || home.isEmpty) { + final support = await getApplicationSupportDirectory(); + return Directory(p.join(support.path, _extensionsSegment)); + } + return Directory(p.join(home, '.querya', _extensionsSegment)); + } + + /// Creates the extensions directory if it doesn't exist. + static Future ensureExtensionsDirectory() async { + final dir = await extensionsDirectory(); + if (!await dir.exists()) { + await dir.create(recursive: true); + } + return dir; + } +} diff --git a/lib/core/extensions/local_extension_registry.dart b/lib/core/extensions/local_extension_registry.dart new file mode 100644 index 00000000..ddf17203 --- /dev/null +++ b/lib/core/extensions/local_extension_registry.dart @@ -0,0 +1,61 @@ +import 'dart:convert'; +import 'dart:io'; + +import 'package:path/path.dart' as p; + +import 'extension_paths.dart'; +import 'models/extension_manifest.dart'; + +/// Scans the local filesystem for extensions and loads their manifests. +class LocalExtensionRegistry { + LocalExtensionRegistry._(); + static final LocalExtensionRegistry instance = LocalExtensionRegistry._(); + + List _manifests = []; + bool _loaded = false; + + /// Returns an unmodifiable list of loaded manifests. + List get manifests => List.unmodifiable(_manifests); + + /// Reloads manifests from the disk. + Future reload() async { + _loaded = false; + await load(); + } + + /// Loads manifests from the extensions directory if not already loaded. + Future> load() async { + if (_loaded) return manifests; + + final dir = await ExtensionPaths.extensionsDirectory(); + final loadedManifests = []; + + if (await dir.exists()) { + // Use list() rather than listSync() to prevent blocking the UI + final entities = await dir.list().toList(); + for (final entity in entities) { + if (entity is Directory) { + final manifestFile = File(p.join(entity.path, 'manifest.json')); + if (await manifestFile.exists()) { + try { + final content = await manifestFile.readAsString(); + final json = jsonDecode(content) as Map; + final manifest = ExtensionManifest.fromJson( + json, + installPath: entity.path, + ); + loadedManifests.add(manifest); + } catch (e) { + // Log or ignore invalid manifests + // In the future, we could report these to an error logging service + } + } + } + } + } + + _manifests = loadedManifests; + _loaded = true; + return manifests; + } +} diff --git a/lib/core/extensions/models/extension_manifest.dart b/lib/core/extensions/models/extension_manifest.dart index dc15d7e5..cafa2e04 100644 --- a/lib/core/extensions/models/extension_manifest.dart +++ b/lib/core/extensions/models/extension_manifest.dart @@ -10,6 +10,7 @@ class ExtensionManifest { final String? main; final String? icon; final String? description; + final String? installPath; const ExtensionManifest({ required this.id, @@ -21,9 +22,10 @@ class ExtensionManifest { this.main, this.icon, this.description, + this.installPath, }); - factory ExtensionManifest.fromJson(Map json) { + factory ExtensionManifest.fromJson(Map json, {String? installPath}) { return ExtensionManifest( id: json['id'] as String, name: json['name'] as String, @@ -34,6 +36,7 @@ class ExtensionManifest { main: json['main'] as String?, icon: json['icon'] as String?, description: json['description'] as String?, + installPath: installPath, ); } From a2c81dca2cf14fdc8626e889bc54ea937b5c93a8 Mon Sep 17 00:00:00 2001 From: Eva Rei <114882226+ZhuchkaTriplesix@users.noreply.github.com> Date: Sun, 21 Jun 2026 13:51:19 +0300 Subject: [PATCH 03/12] Test: Unit Tests for Extension Registry (EXT-4) (#215) * test(extensions): add unit tests for LocalExtensionRegistry (EXT-4) - Mock ExtensionPaths.extensionsDirectory for testing - Verify registry parses valid manifests - Verify invalid manifests and plain files are ignored - Verify registry cache logic * fix(extensions): use flutter foundation for @visibleForTesting instead of meta to satisfy linter --- lib/core/extensions/extension_paths.dart | 7 ++ .../local_extension_registry_test.dart | 96 +++++++++++++++++++ 2 files changed, 103 insertions(+) create mode 100644 test/core/extensions/local_extension_registry_test.dart diff --git a/lib/core/extensions/extension_paths.dart b/lib/core/extensions/extension_paths.dart index f05414f9..d2ef4ef4 100644 --- a/lib/core/extensions/extension_paths.dart +++ b/lib/core/extensions/extension_paths.dart @@ -1,5 +1,6 @@ import 'dart:io'; +import 'package:flutter/foundation.dart'; import 'package:path/path.dart' as p; import 'package:path_provider/path_provider.dart'; @@ -7,9 +8,15 @@ import 'package:path_provider/path_provider.dart'; abstract final class ExtensionPaths { static const _extensionsSegment = 'extensions'; + @visibleForTesting + static Directory? mockExtensionsDirectory; + /// Returns `~/.querya/extensions` on Linux/Mac, or equivalent `USERPROFILE\.querya\extensions` on Windows. /// Falls back to application support directory if HOME is unavailable. static Future extensionsDirectory() async { + if (mockExtensionsDirectory != null) { + return mockExtensionsDirectory!; + } final home = Platform.environment['HOME'] ?? Platform.environment['USERPROFILE']; if (home == null || home.isEmpty) { final support = await getApplicationSupportDirectory(); diff --git a/test/core/extensions/local_extension_registry_test.dart b/test/core/extensions/local_extension_registry_test.dart new file mode 100644 index 00000000..f5d88037 --- /dev/null +++ b/test/core/extensions/local_extension_registry_test.dart @@ -0,0 +1,96 @@ +import 'dart:convert'; +import 'dart:io'; + +import 'package:flutter_test/flutter_test.dart'; +import 'package:path/path.dart' as p; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; +import 'package:querya_desktop/core/extensions/models/extension_type.dart'; + +void main() { + group('LocalExtensionRegistry', () { + late Directory tempDir; + + setUp(() async { + tempDir = await Directory.systemTemp.createTemp('querya_extensions_test'); + ExtensionPaths.mockExtensionsDirectory = tempDir; + }); + + tearDown(() async { + ExtensionPaths.mockExtensionsDirectory = null; + if (await tempDir.exists()) { + await tempDir.delete(recursive: true); + } + }); + + test('loads extensions from manifest files', () async { + // Create a valid extension folder + final ext1Dir = Directory(p.join(tempDir.path, 'ext1')); + await ext1Dir.create(); + + final manifest1 = { + 'id': 'test.ext1', + 'name': 'Test Extension 1', + 'version': '1.0.0', + 'publisher': 'Test', + 'type': 'theme', + 'engines': {'querya_desktop': '^1.0.0'}, + }; + + final file1 = File(p.join(ext1Dir.path, 'manifest.json')); + await file1.writeAsString(jsonEncode(manifest1)); + + // Create an invalid extension folder (no manifest) + final ext2Dir = Directory(p.join(tempDir.path, 'ext2')); + await ext2Dir.create(); + + // Create a file that is not a directory + final notADir = File(p.join(tempDir.path, 'not_a_dir.txt')); + await notADir.writeAsString('I am a file'); + + // Reload registry to scan the test directory + await LocalExtensionRegistry.instance.reload(); + final manifests = LocalExtensionRegistry.instance.manifests; + + expect(manifests.length, 1); + final loaded = manifests.first; + expect(loaded.id, 'test.ext1'); + expect(loaded.type, ExtensionType.theme); + expect(loaded.installPath, ext1Dir.path); + }); + + test('ignores directories with invalid manifest.json', () async { + final extDir = Directory(p.join(tempDir.path, 'bad_ext')); + await extDir.create(); + + final file = File(p.join(extDir.path, 'manifest.json')); + await file.writeAsString('{"invalid_json": '); // Syntax error + + await LocalExtensionRegistry.instance.reload(); + + expect(LocalExtensionRegistry.instance.manifests, isEmpty); + }); + + test('returns cached manifests on subsequent load calls', () async { + final extDir = Directory(p.join(tempDir.path, 'ext3')); + await extDir.create(); + final file = File(p.join(extDir.path, 'manifest.json')); + await file.writeAsString(jsonEncode({ + 'id': 'test.ext3', + 'name': 'Test Ext 3', + 'version': '1.0.0', + 'publisher': 'Test', + 'type': 'theme', + 'engines': {} + })); + + await LocalExtensionRegistry.instance.reload(); + expect(LocalExtensionRegistry.instance.manifests.length, 1); + + // Delete the file. load() shouldn't read from disk again unless reload() is called + await file.delete(); + final manifests = await LocalExtensionRegistry.instance.load(); + expect(manifests.length, 1, reason: 'Should return cached result'); + }); + }); +} From 6c4a0e15510c479ed010e2d60ebec04178b3055a Mon Sep 17 00:00:00 2001 From: Eva Rei <114882226+ZhuchkaTriplesix@users.noreply.github.com> Date: Sun, 21 Jun 2026 18:00:57 +0300 Subject: [PATCH 04/12] Feat: Migrate Custom Themes to Extension Registry (EXT-3) (#216) * fix(theme): use braces for multiline if statement * fix(theme): fix theme registry migration unit tests and watcher path * fix(theme): remove unused import theme_paths.dart --- cleanup.py | 32 ++ lib/core/theme/theme_controller.dart | 6 +- lib/core/theme/theme_registry_service.dart | 375 ++++++++--------- patch_theme.py | 391 ++++++++++++++++++ test/core/theme/theme_controller_test.dart | 62 ++- .../core/theme/theme_folder_watcher_test.dart | 33 +- .../theme_registry_builtin_assets_test.dart | 16 + .../core/theme/theme_registry_cache_test.dart | 21 +- .../theme/theme_registry_service_test.dart | 36 +- .../theme_remote_install_service_test.dart | 37 +- .../settings/theme_import_flow_test.dart | 10 +- 11 files changed, 798 insertions(+), 221 deletions(-) create mode 100644 cleanup.py create mode 100644 patch_theme.py diff --git a/cleanup.py b/cleanup.py new file mode 100644 index 00000000..bf4f7dee --- /dev/null +++ b/cleanup.py @@ -0,0 +1,32 @@ +import re + +with open('lib/core/theme/theme_registry_service.dart', 'r') as f: + content = f.read() + +# Fix curly_braces_in_flow_control_structures +content = content.replace("if (!await directory.exists()) continue;", "if (!await directory.exists()) { continue; }") + +# Fix unnecessary_brace_in_string_interps +content = content.replace("'${slug}-$counter'", "'$slug-$counter'") +content = content.replace("'${preferredBaseName}-$counter'", "'$preferredBaseName-$counter'") + +# Fix unused hash +content = content.replace("final hash = _contentHash(raw);\n final json = _decodeRoot(raw);", "final json = _decodeRoot(raw);") +content = content.replace("final hash = _contentHash(raw);\n final json = _decodeRoot(raw);", "final json = _decodeRoot(raw);") # Handle multiple occurrences if any + +# Remove unused methods +methods_to_remove = [ + r'Future _scanDirectory.*?^\s*\}\s*', + r'Future<_ResolvedImportDestination> _resolveImportDestination.*?^\s*\}\s*', + r'Future _nextRenamedThemeId.*?^\s*\}\s*', + r'Future _themeIdExists.*?^\s*\}\s*', + r'Future _nextAvailableThemeFile.*?^\s*\}\s*', + r'String _rewriteCustomThemeId.*?^\s*\}\s*', + r'class _ResolvedImportDestination.*?^\}\s*' +] + +for pattern in methods_to_remove: + content = re.sub(pattern, '', content, flags=re.DOTALL | re.MULTILINE) + +with open('lib/core/theme/theme_registry_service.dart', 'w') as f: + f.write(content) diff --git a/lib/core/theme/theme_controller.dart b/lib/core/theme/theme_controller.dart index b735eec1..f38966e4 100644 --- a/lib/core/theme/theme_controller.dart +++ b/lib/core/theme/theme_controller.dart @@ -16,9 +16,9 @@ import 'theme_definition.dart'; import 'theme_folder_watcher.dart'; import 'theme_import_service.dart'; import 'theme_load_result.dart'; -import 'theme_paths.dart'; import 'theme_registry_service.dart'; import 'theme_remote_install_service.dart'; +import '../extensions/extension_paths.dart'; /// Active theme state: preset, optional imported colors, user overrides. class ThemeController extends ChangeNotifier { @@ -214,10 +214,10 @@ class ThemeController extends ChangeNotifier { } } - /// Watches `{appSupport}/themes/` and debounces [loadAvailableThemes]. + /// Watches extensions directory and debounces [loadAvailableThemes]. Future startThemeFolderWatcher() async { _themeFolderWatcher ??= ThemeFolderWatcher( - themesDirectory: ThemePaths.userThemesDirectory, + themesDirectory: ExtensionPaths.extensionsDirectory, onThemesChanged: loadAvailableThemes, ); await _themeFolderWatcher!.start(); diff --git a/lib/core/theme/theme_registry_service.dart b/lib/core/theme/theme_registry_service.dart index ab18e573..4c991311 100644 --- a/lib/core/theme/theme_registry_service.dart +++ b/lib/core/theme/theme_registry_service.dart @@ -18,6 +18,10 @@ import 'theme_import_service.dart'; import 'theme_load_result.dart'; import 'theme_metadata.dart'; import 'theme_paths.dart'; +import '../extensions/local_extension_registry.dart'; +import '../extensions/extension_paths.dart'; +import '../extensions/models/extension_manifest.dart'; +import '../extensions/models/extension_type.dart'; /// Scans theme directories and exposes lightweight [ThemeDefinition] metadata. class ThemeRegistryService { @@ -44,6 +48,7 @@ class ThemeRegistryService { final List _bundledThemeAssetFiles; final _ThemeLruCache _themeCache; int _themeParseCount = 0; + bool _hasMigratedThemes = false; /// Number of cache misses that performed a full theme parse. @visibleForTesting @@ -60,16 +65,30 @@ class ThemeRegistryService { await _loadBuiltinAssetDefinitions(definitions); - await _scanDirectory( - await _userThemesDirectory(), - ThemeSource.filesystem, - definitions, - ); - await _scanDirectory( - await _importedThemesDirectory(), - ThemeSource.imported, - definitions, - ); + if (!_hasMigratedThemes) { + await _migrateLegacyThemesToExtensions(); + _hasMigratedThemes = true; + } + + await LocalExtensionRegistry.instance.reload(); + for (final manifest in LocalExtensionRegistry.instance.manifests) { + if (manifest.type != ExtensionType.theme) continue; + final installPath = manifest.installPath; + final mainFile = manifest.main; + if (installPath == null || mainFile == null) continue; + + final file = File(p.join(installPath, mainFile)); + if (!await file.exists()) continue; + + final source = manifest.publisher.toLowerCase() == 'imported' + ? ThemeSource.imported + : ThemeSource.filesystem; + + final definition = await _definitionFromFile(file, source, extensionId: manifest.id); + if (definition != null) { + definitions.add(definition); + } + } final legacy = await _legacyImportedDefinition(); if (legacy != null) { @@ -87,7 +106,68 @@ class ThemeRegistryService { return List.unmodifiable(definitions); } - /// Validates [sourcePath], copies into the user themes directory, and returns + Future _migrateLegacyThemesToExtensions() async { + final dirs = [ + (await _userThemesDirectory(), ThemeSource.filesystem), + (await _importedThemesDirectory(), ThemeSource.imported), + ]; + + final extensionsDir = await ExtensionPaths.ensureExtensionsDirectory(); + + for (final pair in dirs) { + final directory = pair.$1; + final source = pair.$2; + if (!await directory.exists()) continue; + + await for (final entity in directory.list(followLinks: false)) { + if (entity is! File) continue; + if (p.basename(entity.path) == ThemeImportService.storedFileName) { + continue; + } + final ext = p.extension(entity.path).toLowerCase(); + if (ext != '.json' && ext != '.jsonc') continue; + + try { + final definition = + await _definitionFromFile(entity, source); + if (definition == null) continue; + + final slug = ThemeImportService.slugifyThemeName(definition.name); + var finalExtDir = Directory(p.join(extensionsDir.path, slug)); + var counter = 2; + while (await finalExtDir.exists()) { + finalExtDir = + Directory(p.join(extensionsDir.path, '$slug-$counter')); + counter++; + } + await finalExtDir.create(recursive: true); + + final themeFile = File(p.join(finalExtDir.path, 'theme.json')); + await entity.copy(themeFile.path); + + final manifest = ExtensionManifest( + id: definition.id, + name: definition.name, + version: '1.0.0', + publisher: source == ThemeSource.imported ? 'Imported' : 'Unknown', + type: ExtensionType.theme, + engines: const {'querya_desktop': '*'}, + main: 'theme.json', + description: 'Migrated custom theme', + ); + + final manifestFile = File(p.join(finalExtDir.path, 'manifest.json')); + await manifestFile.writeAsString(jsonEncode(manifest.toJson())); + + await entity.delete(); // Delete old file to complete migration + } catch (_) {} + } + } + // Reload extensions since we might have added new ones + await LocalExtensionRegistry.instance.reload(); + } + + /// Validates [sourcePath], creates an extension directory, and returns /// the scanned [ThemeDefinition]. Future importThemeFile(String sourcePath) async { try { @@ -97,20 +177,16 @@ class ThemeRegistryService { } final raw = await source.readAsString(); - final hash = _contentHash(raw); final json = _decodeRoot(raw); if (json == null) { return const ThemeDefinitionImportFailure('Invalid JSON.'); } - final themesDir = await _userThemesDirectory(); - if (!await themesDir.exists()) { - await themesDir.create(recursive: true); - } + final extensionsDir = await ExtensionPaths.ensureExtensionsDirectory(); final schema = json['schema']?.toString(); - late final String logicalId; - late final String preferredBaseName; + late String logicalId; + late String preferredBaseName; late String contentToWrite; if (schema == queryaThemeSchemaV1) { @@ -133,34 +209,94 @@ class ThemeRegistryService { contentToWrite = raw; } - var resolved = await _resolveImportDestination( - themesDir: themesDir, - hash: hash, - logicalId: logicalId, - preferredBaseName: preferredBaseName, - ); + await LocalExtensionRegistry.instance.load(); + final hash = _contentHash(raw); - if (!resolved.reused && - schema == queryaThemeSchemaV1 && - resolved.renamedId != null) { - contentToWrite = _rewriteCustomThemeId(raw, resolved.renamedId!); + File? sameHashFile; + ExtensionManifest? sameHashManifest; + ExtensionManifest? sameIdManifest; + + for (final extManifest in LocalExtensionRegistry.instance.manifests) { + if (extManifest.type != ExtensionType.theme) continue; + final installPath = extManifest.installPath; + final mainFile = extManifest.main; + if (installPath == null || mainFile == null) continue; + + final file = File(p.join(installPath, mainFile)); + if (await file.exists()) { + final existingRaw = await file.readAsString(); + if (_contentHash(existingRaw) == hash) { + sameHashFile = file; + sameHashManifest = extManifest; + break; + } + } + + if (extManifest.id == logicalId) { + sameIdManifest = extManifest; + } } - if (!resolved.reused) { - await resolved.file.writeAsString(contentToWrite); + bool reused = false; + File resolvedFile; + + if (sameHashManifest != null) { + reused = true; + resolvedFile = sameHashFile!; + logicalId = sameHashManifest.id; + } else { + if (sameIdManifest != null) { + var suffix = 2; + var candidateId = '$logicalId-$suffix'; + while (LocalExtensionRegistry.instance.manifests.any((m) => m.type == ExtensionType.theme && m.id == candidateId)) { + suffix++; + candidateId = '$logicalId-$suffix'; + } + logicalId = candidateId; + preferredBaseName = ThemeImportService.safeThemeFileBase(logicalId); + if (schema == queryaThemeSchemaV1) { + contentToWrite = _rewriteCustomThemeId(raw, logicalId); + } + } + + var finalExtDir = + Directory(p.join(extensionsDir.path, preferredBaseName)); + var counter = 2; + while (await finalExtDir.exists()) { + finalExtDir = Directory( + p.join(extensionsDir.path, '$preferredBaseName-$counter')); + counter++; + } + await finalExtDir.create(recursive: true); + + resolvedFile = File(p.join(finalExtDir.path, 'theme.json')); + await resolvedFile.writeAsString(contentToWrite); + + final newManifest = ExtensionManifest( + id: logicalId, + name: preferredBaseName, + version: '1.0.0', + publisher: 'Unknown', + type: ExtensionType.theme, + engines: const {'querya_desktop': '*'}, + main: 'theme.json', + description: 'Imported custom theme', + ); + final manifestFile = File(p.join(finalExtDir.path, 'manifest.json')); + await manifestFile.writeAsString(jsonEncode(newManifest.toJson())); + + await LocalExtensionRegistry.instance.reload(); } - final definition = - await _definitionFromFile(resolved.file, ThemeSource.filesystem); + final definition = await _definitionFromFile(resolvedFile, ThemeSource.filesystem, extensionId: logicalId); if (definition == null) { return const ThemeDefinitionImportFailure( - 'Failed to index imported theme.', - ); + 'Failed to index imported theme.'); } return ThemeDefinitionImportSuccess( definition: definition, - reusedExisting: resolved.reused, + reusedExisting: reused, ); } on QueryaThemeManifestParseException catch (e) { return ThemeDefinitionImportFailure(e.message); @@ -173,6 +309,13 @@ class ThemeRegistryService { } } + String _rewriteCustomThemeId(String raw, String newId) { + final decoded = jsonDecode(stripJsonc(raw)); + if (decoded is! Map) return raw; + decoded['id'] = newId; + return const JsonEncoder.withIndent(' ').convert(decoded); + } + /// Parses a scanned [definition] into a runtime [QueryaTheme]. Future loadTheme(ThemeDefinition definition) async { final path = definition.path; @@ -365,38 +508,11 @@ class ThemeRegistryService { static bool _isAssetPath(String path) => path.startsWith('assets/'); - Future _scanDirectory( - Directory directory, - ThemeSource source, - List out, - ) async { - if (!await directory.exists()) return; - - await for (final entity in directory.list(followLinks: false)) { - if (entity is Directory) { - if (p.basename(entity.path) == 'imported') continue; - continue; - } - if (entity is! File) continue; - - if (p.basename(entity.path) == ThemeImportService.storedFileName) { - continue; - } - - final ext = p.extension(entity.path).toLowerCase(); - if (ext != '.json' && ext != '.jsonc') continue; - - final definition = await _definitionFromFile(entity, source); - if (definition != null) { - out.add(definition); - } - } - } - Future _definitionFromFile( File file, - ThemeSource source, - ) async { + ThemeSource source, { + String? extensionId, + }) async { try { final stat = await file.stat(); final raw = await file.readAsString(); @@ -416,6 +532,7 @@ class ThemeRegistryService { source: source, contentHash: hash, lastModified: stat.modified, + extensionId: extensionId, ); } @@ -426,6 +543,7 @@ class ThemeRegistryService { source: source, contentHash: hash, lastModified: stat.modified, + extensionId: extensionId, ); } on Object catch (e) { _logScanError(file.path, e); @@ -440,6 +558,7 @@ class ThemeRegistryService { required ThemeSource source, required String contentHash, DateTime? lastModified, + String? extensionId, }) { final schema = json['schema']?.toString(); if (schema == queryaThemeSchemaV1) { @@ -449,6 +568,7 @@ class ThemeRegistryService { contentHash: contentHash, path: path, lastModified: lastModified, + extensionId: extensionId, ); } @@ -459,6 +579,7 @@ class ThemeRegistryService { fileBaseName: fileBaseName, path: path, lastModified: lastModified, + extensionId: extensionId, ); } @@ -468,8 +589,9 @@ class ThemeRegistryService { required String contentHash, required String path, DateTime? lastModified, + String? extensionId, }) { - final id = json['id']?.toString().trim(); + final id = extensionId ?? json['id']?.toString().trim(); final name = json['name']?.toString().trim(); final type = json['type']?.toString().trim().toLowerCase(); @@ -502,13 +624,14 @@ class ThemeRegistryService { required String fileBaseName, required String path, DateTime? lastModified, + String? extensionId, }) { final rawName = json['name']?.toString().trim(); final name = rawName != null && rawName.isNotEmpty ? rawName : fileBaseName; final type = json['type']?.toString().trim().toLowerCase(); return ThemeDefinition( - id: fileBaseName, + id: extensionId ?? fileBaseName, name: name, source: source, format: ThemeFormat.vscode, @@ -539,112 +662,6 @@ class ThemeRegistryService { return hash.toRadixString(16).padLeft(8, '0'); } - Future<_ResolvedImportDestination> _resolveImportDestination({ - required Directory themesDir, - required String hash, - required String logicalId, - required String preferredBaseName, - }) async { - File? sameIdFile; - - await for (final entity in themesDir.list(followLinks: false)) { - if (entity is Directory) continue; - if (entity is! File) continue; - - final name = p.basename(entity.path); - if (name == ThemeImportService.storedFileName) continue; - - final ext = p.extension(entity.path).toLowerCase(); - if (ext != '.json' && ext != '.jsonc') continue; - - late final String existingRaw; - try { - existingRaw = await entity.readAsString(); - } on IOException { - continue; - } - - if (_contentHash(existingRaw) == hash) { - return _ResolvedImportDestination(file: entity, reused: true); - } - - final definition = - await _definitionFromFile(entity, ThemeSource.filesystem); - if (definition?.id == logicalId) { - sameIdFile = entity; - } - } - - if (sameIdFile != null) { - final renamedId = await _nextRenamedThemeId(themesDir, logicalId); - final baseName = ThemeImportService.safeThemeFileBase(renamedId); - final primary = File(p.join(themesDir.path, '$baseName.json')); - final file = await primary.exists() - ? await _nextAvailableThemeFile(themesDir, baseName, startSuffix: 2) - : primary; - return _ResolvedImportDestination( - file: file, - reused: false, - renamedId: renamedId, - ); - } - - final primary = File(p.join(themesDir.path, '$preferredBaseName.json')); - if (!await primary.exists()) { - return _ResolvedImportDestination(file: primary, reused: false); - } - - final file = await _nextAvailableThemeFile( - themesDir, - preferredBaseName, - startSuffix: 2, - ); - return _ResolvedImportDestination(file: file, reused: false); - } - - Future _nextRenamedThemeId(Directory themesDir, String baseId) async { - for (var suffix = 2; suffix < 1000; suffix++) { - final candidate = '$baseId-$suffix'; - final taken = await _themeIdExists(themesDir, candidate); - if (!taken) return candidate; - } - return '$baseId-${_contentHash(baseId)}'; - } - - Future _themeIdExists(Directory themesDir, String id) async { - await for (final entity in themesDir.list(followLinks: false)) { - if (entity is! File) continue; - final ext = p.extension(entity.path).toLowerCase(); - if (ext != '.json' && ext != '.jsonc') continue; - final definition = - await _definitionFromFile(entity, ThemeSource.filesystem); - if (definition?.id == id) return true; - } - return false; - } - - Future _nextAvailableThemeFile( - Directory themesDir, - String baseName, { - int startSuffix = 2, - }) async { - for (var suffix = startSuffix; suffix < 1000; suffix++) { - final candidate = File(p.join(themesDir.path, '$baseName-$suffix.json')); - if (!await candidate.exists()) return candidate; - } - return File( - p.join(themesDir.path, - '$baseName-${DateTime.now().millisecondsSinceEpoch}.json'), - ); - } - - String _rewriteCustomThemeId(String raw, String newId) { - final decoded = jsonDecode(stripJsonc(raw)); - if (decoded is! Map) return raw; - decoded['id'] = newId; - return const JsonEncoder.withIndent(' ').convert(decoded); - } - void _logScanError(String path, Object error) { if (kDebugMode) { debugPrint('ThemeRegistryService: skipped $path ($error)'); @@ -652,18 +669,6 @@ class ThemeRegistryService { } } -class _ResolvedImportDestination { - const _ResolvedImportDestination({ - required this.file, - required this.reused, - this.renamedId, - }); - - final File file; - final bool reused; - final String? renamedId; -} - class _ThemeLruCache { _ThemeLruCache({required this.maxEntries}); diff --git a/patch_theme.py b/patch_theme.py new file mode 100644 index 00000000..dc4fc685 --- /dev/null +++ b/patch_theme.py @@ -0,0 +1,391 @@ +import os + +with open('lib/core/theme/theme_registry_service.dart', 'r') as f: + content = f.read() + +# 1. Imports +target_imports = """import 'theme_metadata.dart'; +import 'theme_paths.dart';""" +new_imports = """import 'theme_metadata.dart'; +import 'theme_paths.dart'; +import '../extensions/local_extension_registry.dart'; +import '../extensions/extension_paths.dart'; +import '../extensions/models/extension_manifest.dart'; +import '../extensions/models/extension_type.dart';""" +content = content.replace(target_imports, new_imports) + +# 2. Add flag +target_flag = """ final List _bundledThemeAssetFiles; + final _ThemeLruCache _themeCache; + int _themeParseCount = 0;""" +new_flag = """ final List _bundledThemeAssetFiles; + final _ThemeLruCache _themeCache; + int _themeParseCount = 0; + bool _hasMigratedThemes = false;""" +content = content.replace(target_flag, new_flag) + +# 3. loadThemeDefinitions +target_load = """ Future> loadThemeDefinitions() async { + final definitions = []; + + await _loadBuiltinAssetDefinitions(definitions); + + await _scanDirectory( + await _userThemesDirectory(), + ThemeSource.filesystem, + definitions, + ); + await _scanDirectory( + await _importedThemesDirectory(), + ThemeSource.imported, + definitions, + ); + + final legacy = await _legacyImportedDefinition(); + if (legacy != null) { + definitions.removeWhere( + (definition) => + definition.path == legacy.path && + definition.source != ThemeSource.legacyImported, + ); + definitions.add(legacy); + } + + definitions.sort( + (a, b) => a.name.toLowerCase().compareTo(b.name.toLowerCase()), + ); + return List.unmodifiable(definitions); + }""" +new_load = """ Future> loadThemeDefinitions() async { + final definitions = []; + + await _loadBuiltinAssetDefinitions(definitions); + + if (!_hasMigratedThemes) { + await _migrateLegacyThemesToExtensions(); + _hasMigratedThemes = true; + } + + await LocalExtensionRegistry.instance.load(); + for (final manifest in LocalExtensionRegistry.instance.manifests) { + if (manifest.type != ExtensionType.theme) continue; + final installPath = manifest.installPath; + final mainFile = manifest.main; + if (installPath == null || mainFile == null) continue; + + final file = File(p.join(installPath, mainFile)); + if (!await file.exists()) continue; + + final definition = await _definitionFromFile(file, ThemeSource.filesystem); + if (definition != null) { + definitions.add(definition); + } + } + + final legacy = await _legacyImportedDefinition(); + if (legacy != null) { + definitions.removeWhere( + (definition) => + definition.path == legacy.path && + definition.source != ThemeSource.legacyImported, + ); + definitions.add(legacy); + } + + definitions.sort( + (a, b) => a.name.toLowerCase().compareTo(b.name.toLowerCase()), + ); + return List.unmodifiable(definitions); + } + + Future _migrateLegacyThemesToExtensions() async { + final dirs = [ + await _userThemesDirectory(), + await _importedThemesDirectory(), + ]; + + final extensionsDir = await ExtensionPaths.ensureExtensionsDirectory(); + + for (final directory in dirs) { + if (!await directory.exists()) continue; + + await for (final entity in directory.list(followLinks: false)) { + if (entity is! File) continue; + if (p.basename(entity.path) == ThemeImportService.storedFileName) continue; + final ext = p.extension(entity.path).toLowerCase(); + if (ext != '.json' && ext != '.jsonc') continue; + + try { + final definition = await _definitionFromFile(entity, ThemeSource.filesystem); + if (definition == null) continue; + + final slug = ThemeImportService.slugifyThemeName(definition.name); + var finalExtDir = Directory(p.join(extensionsDir.path, slug)); + var counter = 2; + while (await finalExtDir.exists()) { + finalExtDir = Directory(p.join(extensionsDir.path, '$slug-$counter')); + counter++; + } + await finalExtDir.create(recursive: true); + + final themeFile = File(p.join(finalExtDir.path, 'theme.json')); + await entity.copy(themeFile.path); + + final manifest = ExtensionManifest( + id: definition.id, + name: definition.name, + version: '1.0.0', + publisher: 'Unknown', + type: ExtensionType.theme, + engines: const {'querya_desktop': '*'}, + main: 'theme.json', + description: 'Migrated custom theme', + ); + + final manifestFile = File(p.join(finalExtDir.path, 'manifest.json')); + await manifestFile.writeAsString(jsonEncode(manifest.toJson())); + + await entity.delete(); // Delete old file to complete migration + } catch (_) {} + } + } + // Reload extensions since we might have added new ones + await LocalExtensionRegistry.instance.reload(); + }""" +content = content.replace(target_load, new_load) + +# 4. importThemeFile +target_import = """ /// Validates [sourcePath], copies into the user themes directory, and returns + /// the scanned [ThemeDefinition]. + Future importThemeFile(String sourcePath) async { + try { + final source = File(sourcePath); + if (!await source.exists()) { + return const ThemeDefinitionImportFailure('Theme file not found.'); + } + + final raw = await source.readAsString(); + final hash = _contentHash(raw); + final json = _decodeRoot(raw); + if (json == null) { + return const ThemeDefinitionImportFailure('Invalid JSON.'); + } + + final themesDir = await _userThemesDirectory(); + if (!await themesDir.exists()) { + await themesDir.create(recursive: true); + } + + final schema = json['schema']?.toString(); + late final String logicalId; + late final String preferredBaseName; + late String contentToWrite; + + if (schema == queryaThemeSchemaV1) { + final manifest = QueryaThemeManifest.fromJsonString(raw); + logicalId = manifest.id; + preferredBaseName = ThemeImportService.safeThemeFileBase(manifest.id); + contentToWrite = raw; + } else { + final manifest = VsCodeThemeManifest.fromJsonString(raw); + if (manifest.colors.isEmpty) { + return const ThemeDefinitionImportFailure( + 'Theme file has no "colors" section to import.', + ); + } + final displayName = manifest.name?.trim().isNotEmpty == true + ? manifest.name!.trim() + : p.basenameWithoutExtension(sourcePath); + preferredBaseName = ThemeImportService.slugifyThemeName(displayName); + logicalId = preferredBaseName; + contentToWrite = raw; + } + + var resolved = await _resolveImportDestination( + themesDir: themesDir, + hash: hash, + logicalId: logicalId, + preferredBaseName: preferredBaseName, + ); + + if (!resolved.reused && + schema == queryaThemeSchemaV1 && + resolved.renamedId != null) { + contentToWrite = _rewriteCustomThemeId(raw, resolved.renamedId!); + } + + if (!resolved.reused) { + await resolved.file.writeAsString(contentToWrite); + } + + final definition = + await _definitionFromFile(resolved.file, ThemeSource.filesystem); + if (definition == null) { + return const ThemeDefinitionImportFailure( + 'Failed to index imported theme.', + ); + } + + return ThemeDefinitionImportSuccess( + definition: definition, + reusedExisting: resolved.reused, + ); + } on QueryaThemeManifestParseException catch (e) { + return ThemeDefinitionImportFailure(e.message); + } on VsCodeThemeParseException catch (e) { + return ThemeDefinitionImportFailure(e.message); + } on IOException catch (e) { + return ThemeDefinitionImportFailure(e.toString()); + } on Object catch (e) { + return ThemeDefinitionImportFailure(e.toString()); + } + }""" + +new_import = """ /// Validates [sourcePath], creates an extension directory, and returns + /// the scanned [ThemeDefinition]. + Future importThemeFile(String sourcePath) async { + try { + final source = File(sourcePath); + if (!await source.exists()) { + return const ThemeDefinitionImportFailure('Theme file not found.'); + } + + final raw = await source.readAsString(); + final json = _decodeRoot(raw); + if (json == null) { + return const ThemeDefinitionImportFailure('Invalid JSON.'); + } + + final extensionsDir = await ExtensionPaths.ensureExtensionsDirectory(); + + final schema = json['schema']?.toString(); + late final String logicalId; + late final String preferredBaseName; + late String contentToWrite; + + if (schema == queryaThemeSchemaV1) { + final manifest = QueryaThemeManifest.fromJsonString(raw); + logicalId = manifest.id; + preferredBaseName = ThemeImportService.safeThemeFileBase(manifest.id); + contentToWrite = raw; + } else { + final manifest = VsCodeThemeManifest.fromJsonString(raw); + if (manifest.colors.isEmpty) { + return const ThemeDefinitionImportFailure( + 'Theme file has no "colors" section to import.', + ); + } + final displayName = manifest.name?.trim().isNotEmpty == true + ? manifest.name!.trim() + : p.basenameWithoutExtension(sourcePath); + preferredBaseName = ThemeImportService.slugifyThemeName(displayName); + logicalId = preferredBaseName; + contentToWrite = raw; + } + + await LocalExtensionRegistry.instance.load(); + bool reused = false; + String? existingInstallPath; + + for (final extManifest in LocalExtensionRegistry.instance.manifests) { + if (extManifest.type == ExtensionType.theme && extManifest.id == logicalId) { + reused = true; + existingInstallPath = extManifest.installPath; + break; + } + } + + File resolvedFile; + if (reused && existingInstallPath != null) { + resolvedFile = File(p.join(existingInstallPath, 'theme.json')); + } else { + var finalExtDir = Directory(p.join(extensionsDir.path, preferredBaseName)); + var counter = 2; + while (await finalExtDir.exists()) { + finalExtDir = Directory(p.join(extensionsDir.path, '$preferredBaseName-$counter')); + counter++; + } + await finalExtDir.create(recursive: true); + + resolvedFile = File(p.join(finalExtDir.path, 'theme.json')); + await resolvedFile.writeAsString(contentToWrite); + + final newManifest = ExtensionManifest( + id: logicalId, + name: preferredBaseName, + version: '1.0.0', + publisher: 'Unknown', + type: ExtensionType.theme, + engines: const {'querya_desktop': '*'}, + main: 'theme.json', + description: 'Imported custom theme', + ); + final manifestFile = File(p.join(finalExtDir.path, 'manifest.json')); + await manifestFile.writeAsString(jsonEncode(newManifest.toJson())); + + await LocalExtensionRegistry.instance.reload(); + } + + final definition = await _definitionFromFile(resolvedFile, ThemeSource.filesystem); + if (definition == null) { + return const ThemeDefinitionImportFailure('Failed to index imported theme.'); + } + + return ThemeDefinitionImportSuccess( + definition: definition, + reusedExisting: reused, + ); + } on QueryaThemeManifestParseException catch (e) { + return ThemeDefinitionImportFailure(e.message); + } on VsCodeThemeParseException catch (e) { + return ThemeDefinitionImportFailure(e.message); + } on IOException catch (e) { + return ThemeDefinitionImportFailure(e.toString()); + } on Object catch (e) { + return ThemeDefinitionImportFailure(e.toString()); + } + }""" +content = content.replace(target_import, new_import) + +# 5. Remove unused methods using python block extracting instead of regex +# Or just simple replace +import re +to_remove = [ + (r' Future _scanDirectory\(', r'^\s*\}\s*$', 3), + (r' Future<_ResolvedImportDestination> _resolveImportDestination\(', r'^\s*\}\s*$', 3), + (r' Future _nextRenamedThemeId\(', r'^\s*\}\s*$', 3), + (r' Future _themeIdExists\(', r'^\s*\}\s*$', 3), + (r' Future _nextAvailableThemeFile\(', r'^\s*\}\s*$', 3), + (r' String _rewriteCustomThemeId\(', r'^\s*\}\s*$', 3), + (r'class _ResolvedImportDestination \{', r'^\}\s*$', 0), +] + +lines = content.split('\n') +for start_regex, end_regex, indent in to_remove: + start_idx = -1 + end_idx = -1 + for i, line in enumerate(lines): + if re.search(start_regex, line): + start_idx = i + break + if start_idx != -1: + # find matching closing brace + brace_count = 0 + started = False + for i in range(start_idx, len(lines)): + brace_count += lines[i].count('{') + brace_count -= lines[i].count('}') + if '{' in lines[i]: + started = True + if started and brace_count == 0: + end_idx = i + break + if end_idx != -1: + lines[start_idx:end_idx+1] = [] + +content = '\n'.join(lines) + +with open('lib/core/theme/theme_registry_service.dart', 'w') as f: + f.write(content) + +print("Patch applied successfully.") diff --git a/test/core/theme/theme_controller_test.dart b/test/core/theme/theme_controller_test.dart index b25708b5..41d8a59e 100644 --- a/test/core/theme/theme_controller_test.dart +++ b/test/core/theme/theme_controller_test.dart @@ -13,6 +13,8 @@ import 'package:querya_desktop/core/theme/theme_controller.dart'; import 'package:querya_desktop/core/theme/theme_import_service.dart'; import 'package:querya_desktop/core/theme/theme_load_result.dart'; import 'package:querya_desktop/core/theme/theme_definition.dart'; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; import 'package:querya_desktop/core/theme/theme_registry_service.dart'; import 'package:shadcn_flutter/shadcn_flutter.dart'; @@ -74,6 +76,15 @@ void main() { themesDir = Directory(p.join(tempDir.path, 'themes')); importedDir = Directory(p.join(themesDir.path, 'imported')); await importedDir.create(recursive: true); + + final extDir = Directory(p.join(tempDir.path, 'extensions')); + ExtensionPaths.mockExtensionsDirectory = extDir; + if (await extDir.exists()) { + await extDir.delete(recursive: true); + } + await extDir.create(recursive: true); + await LocalExtensionRegistry.instance.reload(); + registry = ThemeRegistryService( userThemesDirectory: () async => themesDir, importedThemesDirectory: () async => importedDir, @@ -90,13 +101,16 @@ void main() { }); tearDown(() async { - await ThemeController.instance.stopThemeFolderWatcher(); - await ThemeController.instance.endEditorPreview(); - await AppSettings.instance.clearThemeSettings(); - await ThemeImportService.deletePersistedImport(); if (await themesDir.exists()) { await themesDir.delete(recursive: true); } + final extDir = ExtensionPaths.mockExtensionsDirectory; + if (extDir != null && await extDir.exists()) { + await extDir.delete(recursive: true); + } + ExtensionPaths.mockExtensionsDirectory = null; + await LocalExtensionRegistry.instance.reload(); + await ThemeController.instance.resetToDefaults(); ThemeController.instance.setRegistryServiceForTest(ThemeRegistryService()); }); @@ -272,7 +286,8 @@ void main() { await _copyFixture('querya_custom_dark.json', themeFile); await c.load(); await c.setThemeById('fixture-custom-dark'); - await themeFile.delete(); + final extDir = ExtensionPaths.mockExtensionsDirectory!; + await Directory(p.join(extDir.path, 'fixture-custom-dark')).delete(recursive: true); await c.load(); @@ -312,10 +327,19 @@ void main() { await c.load(); expect(c.selectedThemeLoadError, isNotNull); - await _copyFixture( - 'querya_custom_dark.json', - File(p.join(themesDir.path, 'querya_custom_dark.json')), - ); + final extDir = ExtensionPaths.mockExtensionsDirectory!; + final themeExt = Directory(p.join(extDir.path, 'fixture-custom-dark')); + await themeExt.create(); + await File(p.join('test/fixtures/themes', 'querya_custom_dark.json')).copy(p.join(themeExt.path, 'theme.json')); + await File(p.join(themeExt.path, 'manifest.json')).writeAsString('''{ + "schema": "querya.extension.v1", + "id": "fixture-custom-dark", + "name": "Fixture Custom Dark", + "version": "1.0.0", + "publisher": "Querya", + "type": "theme", + "main": "theme.json" + }'''); await c.loadAvailableThemes(); await c.setThemeById('fixture-custom-dark'); @@ -403,10 +427,19 @@ void main() { await c.load(); final beforeCount = c.availableThemes.length; - await _copyFixture( - 'querya_custom_dark.json', - File(p.join(themesDir.path, 'querya_custom_dark.json')), - ); + final extDir = ExtensionPaths.mockExtensionsDirectory!; + final themeExt = Directory(p.join(extDir.path, 'fixture-custom-dark')); + await themeExt.create(); + await File(p.join('test/fixtures/themes', 'querya_custom_dark.json')).copy(p.join(themeExt.path, 'theme.json')); + await File(p.join(themeExt.path, 'manifest.json')).writeAsString('''{ + "schema": "querya.extension.v1", + "id": "fixture-custom-dark", + "name": "Fixture Custom Dark", + "version": "1.0.0", + "publisher": "Querya", + "type": "theme", + "main": "theme.json" + }'''); await c.loadAvailableThemes(); expect(c.availableThemes.length, greaterThan(beforeCount)); @@ -428,7 +461,8 @@ void main() { await c.setThemeById('fixture-custom-dark'); final before = c.activeTheme; - await File(p.join(themesDir.path, 'querya_custom_dark.json')) + final extDir = ExtensionPaths.mockExtensionsDirectory!; + await File(p.join(extDir.path, 'fixture-custom-dark', 'theme.json')) .writeAsString('not valid theme json'); await c.loadAvailableThemes(); diff --git a/test/core/theme/theme_folder_watcher_test.dart b/test/core/theme/theme_folder_watcher_test.dart index bbb9e3a8..690b5780 100644 --- a/test/core/theme/theme_folder_watcher_test.dart +++ b/test/core/theme/theme_folder_watcher_test.dart @@ -5,6 +5,8 @@ import 'package:flutter_test/flutter_test.dart'; import 'package:path/path.dart' as p; import 'package:querya_desktop/core/storage/app_settings.dart'; import 'package:querya_desktop/core/storage/local_db.dart'; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; import 'package:querya_desktop/core/theme/theme_controller.dart'; import 'package:querya_desktop/core/theme/theme_folder_watcher.dart'; import 'package:querya_desktop/core/theme/theme_import_service.dart'; @@ -40,6 +42,14 @@ void main() { setUp(() async { themesDir = Directory(p.join(tempDir.path, 'themes')); await Directory(p.join(themesDir.path, 'imported')).create(recursive: true); + + final extDir = Directory(p.join(tempDir.path, 'extensions')); + ExtensionPaths.mockExtensionsDirectory = extDir; + if (await extDir.exists()) { + await extDir.delete(recursive: true); + } + await extDir.create(recursive: true); + await LocalExtensionRegistry.instance.reload(); }); tearDownAll(() async { @@ -56,6 +66,12 @@ void main() { if (await themesDir.exists()) { await themesDir.delete(recursive: true); } + final extDir = ExtensionPaths.mockExtensionsDirectory; + if (extDir != null && await extDir.exists()) { + await extDir.delete(recursive: true); + } + ExtensionPaths.mockExtensionsDirectory = null; + await LocalExtensionRegistry.instance.reload(); ThemeController.instance.setRegistryServiceForTest(ThemeRegistryService()); }); @@ -146,10 +162,21 @@ void main() { expect(c.isThemeFolderWatcherStarted, isTrue); final beforeCount = c.availableThemes.length; - await _copyFixture( - 'querya_custom_dark.json', - File(p.join(themesDir.path, 'querya_custom_dark.json')), + final extDir = ExtensionPaths.mockExtensionsDirectory!; + final themeExt = Directory(p.join(extDir.path, 'fixture-custom-dark')); + await themeExt.create(recursive: true); + await File(p.join(themeExt.path, 'theme.json')).writeAsString( + await File('test/fixtures/themes/querya_custom_dark.json').readAsString(), ); + await File(p.join(themeExt.path, 'manifest.json')).writeAsString('''{ + "schema": "querya.extension.v1", + "id": "fixture-custom-dark", + "name": "Fixture Custom Dark", + "version": "1.0.0", + "publisher": "Unknown", + "type": "theme", + "main": "theme.json" + }'''); await Future.delayed(const Duration(milliseconds: 700)); diff --git a/test/core/theme/theme_registry_builtin_assets_test.dart b/test/core/theme/theme_registry_builtin_assets_test.dart index b990cba1..8bb67753 100644 --- a/test/core/theme/theme_registry_builtin_assets_test.dart +++ b/test/core/theme/theme_registry_builtin_assets_test.dart @@ -4,6 +4,8 @@ import 'package:flutter/material.dart'; import 'package:flutter_test/flutter_test.dart'; import 'package:path/path.dart' as p; import 'package:path_provider_platform_interface/path_provider_platform_interface.dart'; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; import 'package:querya_desktop/core/theme/builtin_theme_assets.dart'; import 'package:querya_desktop/core/theme/parser/color_parser.dart'; import 'package:querya_desktop/core/theme/theme_definition.dart'; @@ -42,6 +44,14 @@ void main() { importedDir = Directory(p.join(themesDir.path, 'imported')); await importedDir.create(recursive: true); + final extDir = Directory(p.join(tempDir.path, 'extensions')); + ExtensionPaths.mockExtensionsDirectory = extDir; + if (await extDir.exists()) { + await extDir.delete(recursive: true); + } + await extDir.create(recursive: true); + await LocalExtensionRegistry.instance.reload(); + registry = ThemeRegistryService( userThemesDirectory: () async => themesDir, importedThemesDirectory: () async => importedDir, @@ -53,6 +63,12 @@ void main() { if (await themesDir.exists()) { await themesDir.delete(recursive: true); } + final extDir = ExtensionPaths.mockExtensionsDirectory; + if (extDir != null && await extDir.exists()) { + await extDir.delete(recursive: true); + } + ExtensionPaths.mockExtensionsDirectory = null; + await LocalExtensionRegistry.instance.reload(); }); tearDownAll(() async { diff --git a/test/core/theme/theme_registry_cache_test.dart b/test/core/theme/theme_registry_cache_test.dart index 60029779..4a4d7078 100644 --- a/test/core/theme/theme_registry_cache_test.dart +++ b/test/core/theme/theme_registry_cache_test.dart @@ -4,6 +4,8 @@ import 'package:flutter_test/flutter_test.dart'; import 'package:path/path.dart' as p; import 'package:path_provider_platform_interface/path_provider_platform_interface.dart'; import 'package:querya_desktop/core/theme/theme_load_result.dart'; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; import 'package:querya_desktop/core/theme/theme_registry_service.dart'; class _FakePathProvider extends PathProviderPlatform { @@ -37,6 +39,14 @@ void main() { importedDir = Directory(p.join(themesDir.path, 'imported')); await importedDir.create(recursive: true); + final extDir = Directory(p.join(tempDir.path, 'extensions')); + ExtensionPaths.mockExtensionsDirectory = extDir; + if (await extDir.exists()) { + await extDir.delete(recursive: true); + } + await extDir.create(recursive: true); + await LocalExtensionRegistry.instance.reload(); + registry = ThemeRegistryService( userThemesDirectory: () async => themesDir, importedThemesDirectory: () async => importedDir, @@ -48,6 +58,12 @@ void main() { if (await themesDir.exists()) { await themesDir.delete(recursive: true); } + final extDir = ExtensionPaths.mockExtensionsDirectory; + if (extDir != null && await extDir.exists()) { + await extDir.delete(recursive: true); + } + ExtensionPaths.mockExtensionsDirectory = null; + await LocalExtensionRegistry.instance.reload(); }); tearDownAll(() async { @@ -81,8 +97,9 @@ void main() { await registry.loadTheme(before); expect(registry.themeParseCount, 1); - final raw = await themeFile.readAsString(); - await themeFile.writeAsString(raw.replaceFirst('#FF00AA', '#00FFAA')); + final extFile = File(before.path!); + final raw = await extFile.readAsString(); + await extFile.writeAsString(raw.replaceFirst('#FF00AA', '#00FFAA')); final after = (await registry.loadThemeDefinitions()).single; expect(after.contentHash, isNot(before.contentHash)); diff --git a/test/core/theme/theme_registry_service_test.dart b/test/core/theme/theme_registry_service_test.dart index 775c7eb6..3f886a98 100644 --- a/test/core/theme/theme_registry_service_test.dart +++ b/test/core/theme/theme_registry_service_test.dart @@ -8,6 +8,8 @@ import 'package:querya_desktop/core/theme/parser/color_parser.dart'; import 'package:querya_desktop/core/theme/theme_definition.dart'; import 'package:querya_desktop/core/theme/theme_import_service.dart'; import 'package:querya_desktop/core/theme/theme_load_result.dart'; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; import 'package:querya_desktop/core/theme/theme_registry_service.dart'; class _FakePathProvider extends PathProviderPlatform { @@ -42,6 +44,14 @@ void main() { importedDir = Directory(p.join(themesDir.path, 'imported')); await importedDir.create(recursive: true); + final extDir = Directory(p.join(tempDir.path, 'extensions')); + ExtensionPaths.mockExtensionsDirectory = extDir; + if (await extDir.exists()) { + await extDir.delete(recursive: true); + } + await extDir.create(recursive: true); + await LocalExtensionRegistry.instance.reload(); + registry = ThemeRegistryService( userThemesDirectory: () async => themesDir, importedThemesDirectory: () async => importedDir, @@ -53,6 +63,12 @@ void main() { if (await themesDir.exists()) { await themesDir.delete(recursive: true); } + final extDir = ExtensionPaths.mockExtensionsDirectory; + if (extDir != null && await extDir.exists()) { + await extDir.delete(recursive: true); + } + ExtensionPaths.mockExtensionsDirectory = null; + await LocalExtensionRegistry.instance.reload(); }); tearDownAll(() async { @@ -147,8 +163,9 @@ void main() { expect(before, hasLength(1)); final originalHash = before.single.contentHash; - final raw = await themeFile.readAsString(); - await themeFile.writeAsString(raw.replaceFirst('#FF00AA', '#00FFAA')); + final extFile = File(before.single.path!); + final raw = await extFile.readAsString(); + await extFile.writeAsString(raw.replaceFirst('#FF00AA', '#00FFAA')); final after = await registry.loadThemeDefinitions(); expect(after, hasLength(1)); @@ -261,7 +278,7 @@ void main() { expect(success.definition.id, 'fixture-custom-dark'); expect(success.definition.source, ThemeSource.filesystem); expect( - await File(p.join(themesDir.path, 'fixture-custom-dark.json')).exists(), + await File(p.join(tempDir.path, 'extensions', 'fixture-custom-dark', 'theme.json')).exists(), isTrue, ); @@ -281,7 +298,11 @@ void main() { expect(success.definition.format, ThemeFormat.vscode); expect( p.basename(success.definition.path!), - 'fixture-dark-subset.json', + 'theme.json', + ); + expect( + p.basename(p.dirname(success.definition.path!)), + 'fixture-dark-subset', ); }); @@ -298,11 +319,8 @@ void main() { expect(secondSuccess.reusedExisting, isTrue); expect(secondSuccess.definition.path, firstSuccess.definition.path); - final themeFiles = themesDir - .listSync() - .whereType() - .where((f) => p.extension(f.path) == '.json') - .length; + final extDir = ExtensionPaths.mockExtensionsDirectory!; + final themeFiles = extDir.listSync().length; expect(themeFiles, 1); }); diff --git a/test/core/theme/theme_remote_install_service_test.dart b/test/core/theme/theme_remote_install_service_test.dart index b9bac2ab..04adc8b6 100644 --- a/test/core/theme/theme_remote_install_service_test.dart +++ b/test/core/theme/theme_remote_install_service_test.dart @@ -3,6 +3,8 @@ import 'dart:io'; import 'package:flutter_test/flutter_test.dart'; import 'package:path/path.dart' as p; import 'package:path_provider_platform_interface/path_provider_platform_interface.dart'; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; import 'package:querya_desktop/core/storage/app_settings.dart'; import 'package:querya_desktop/core/storage/local_db.dart'; import 'package:querya_desktop/core/theme/parser/color_parser.dart'; @@ -41,6 +43,15 @@ void main() { setUp(() async { themesDir = Directory(p.join(tempDir.path, 'themes')); await Directory(p.join(themesDir.path, 'imported')).create(recursive: true); + + final extDir = Directory(p.join(tempDir.path, 'extensions')); + ExtensionPaths.mockExtensionsDirectory = extDir; + if (await extDir.exists()) { + await extDir.delete(recursive: true); + } + await extDir.create(recursive: true); + await LocalExtensionRegistry.instance.reload(); + registry = ThemeRegistryService( userThemesDirectory: () async => themesDir, importedThemesDirectory: () async => Directory( @@ -65,6 +76,12 @@ void main() { if (await themesDir.exists()) { await themesDir.delete(recursive: true); } + final extDir = ExtensionPaths.mockExtensionsDirectory; + if (extDir != null && await extDir.exists()) { + await extDir.delete(recursive: true); + } + ExtensionPaths.mockExtensionsDirectory = null; + await LocalExtensionRegistry.instance.reload(); ThemeController.instance.setRegistryServiceForTest(ThemeRegistryService()); }); @@ -89,7 +106,7 @@ void main() { final success = result as ThemeDefinitionImportSuccess; expect(success.definition.id, 'fixture-custom-dark'); expect( - await File(p.join(themesDir.path, 'fixture-custom-dark.json')).exists(), + await File(p.join(tempDir.path, 'extensions', 'fixture-custom-dark', 'theme.json')).exists(), isTrue, ); }); @@ -116,7 +133,7 @@ void main() { (result as ThemeDefinitionImportFailure).message, contains('Checksum mismatch'), ); - expect(await themesDir.list().length, 1); + expect(await ExtensionPaths.mockExtensionsDirectory!.list().length, 0); }); test('rejects invalid JSON without writing to themes folder', () async { @@ -150,8 +167,20 @@ void main() { test('reuses existing file when remote content hash matches', () async { final raw = await File('test/fixtures/themes/querya_custom_dark.json') .readAsString(); - await File(p.join(themesDir.path, 'fixture-custom-dark.json')) - .writeAsString(raw); + final extDir = ExtensionPaths.mockExtensionsDirectory!; + final themeExt = Directory(p.join(extDir.path, 'fixture-custom-dark')); + await themeExt.create(recursive: true); + await File(p.join(themeExt.path, 'theme.json')).writeAsString(raw); + await File(p.join(themeExt.path, 'manifest.json')).writeAsString('''{ + "schema": "querya.extension.v1", + "id": "fixture-custom-dark", + "name": "Fixture Custom Dark", + "version": "1.0.0", + "publisher": "Unknown", + "type": "theme", + "main": "theme.json" + }'''); + await LocalExtensionRegistry.instance.reload(); final service = ThemeRemoteInstallService( registry, diff --git a/test/features/settings/theme_import_flow_test.dart b/test/features/settings/theme_import_flow_test.dart index 879b62b3..05bb13fb 100644 --- a/test/features/settings/theme_import_flow_test.dart +++ b/test/features/settings/theme_import_flow_test.dart @@ -9,6 +9,8 @@ import 'package:querya_desktop/core/theme/parser/color_parser.dart'; import 'package:querya_desktop/core/theme/theme_controller.dart'; import 'package:querya_desktop/core/theme/theme_definition.dart'; import 'package:querya_desktop/core/theme/theme_import_service.dart'; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; import 'package:querya_desktop/core/theme/theme_registry_service.dart'; class _FakePathProvider extends PathProviderPlatform { @@ -61,6 +63,10 @@ void main() { importedDir = Directory(p.join(themesDir.path, 'imported')); await importedDir.create(recursive: true); + final extensionsDir = Directory(p.join(tempDir.path, 'extensions')); + ExtensionPaths.mockExtensionsDirectory = extensionsDir; + await LocalExtensionRegistry.instance.reload(); + registry = ThemeRegistryService( userThemesDirectory: () async => themesDir, importedThemesDirectory: () async => importedDir, @@ -76,6 +82,8 @@ void main() { if (await themesDir.exists()) { await themesDir.delete(recursive: true); } + ExtensionPaths.mockExtensionsDirectory = null; + await LocalExtensionRegistry.instance.reload(); ThemeController.instance.setRegistryServiceForTest( ThemeRegistryService( userThemesDirectory: () async => themesDir, @@ -119,7 +127,7 @@ void main() { parseQueryaThemeColor('#38BDF8'), ); expect( - await File(p.join(themesDir.path, 'fixture-custom-dark.json')).exists(), + await File(p.join(tempDir.path, 'extensions', 'fixture-custom-dark', 'theme.json')).exists(), isTrue, ); expect(await AppSettings.instance.getSelectedThemeId(), From d42e7560dad84df57961b215d2afbfd13d4bf53d Mon Sep 17 00:00:00 2001 From: ZhuchkaTriplesix Date: Sun, 21 Jun 2026 18:02:28 +0300 Subject: [PATCH 05/12] test(settings): fix preferences appearance section widget test by mocking extensions dir and utilizing pumpAndSettle --- .../preferences_appearance_section_test.dart | 21 +++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/test/features/settings/preferences_appearance_section_test.dart b/test/features/settings/preferences_appearance_section_test.dart index a5e7c678..efb0d6a0 100644 --- a/test/features/settings/preferences_appearance_section_test.dart +++ b/test/features/settings/preferences_appearance_section_test.dart @@ -4,6 +4,8 @@ import 'package:flutter/material.dart' as material; import 'package:flutter_test/flutter_test.dart'; import 'package:path/path.dart' as p; import 'package:path_provider_platform_interface/path_provider_platform_interface.dart'; +import 'package:querya_desktop/core/extensions/extension_paths.dart'; +import 'package:querya_desktop/core/extensions/local_extension_registry.dart'; import 'package:querya_desktop/core/storage/app_settings.dart'; import 'package:querya_desktop/core/storage/local_db.dart'; import 'package:querya_desktop/core/theme/theme_controller.dart'; @@ -52,6 +54,15 @@ void main() { themesDir = Directory(p.join(tempDir.path, 'themes')); importedDir = Directory(p.join(themesDir.path, 'imported')); await importedDir.create(recursive: true); + + final extDir = Directory(p.join(tempDir.path, 'extensions')); + ExtensionPaths.mockExtensionsDirectory = extDir; + if (await extDir.exists()) { + await extDir.delete(recursive: true); + } + await extDir.create(recursive: true); + await LocalExtensionRegistry.instance.reload(); + registry = ThemeRegistryService( userThemesDirectory: () async => themesDir, importedThemesDirectory: () async => importedDir, @@ -69,6 +80,13 @@ void main() { }); tearDown(() async { + final extDir = ExtensionPaths.mockExtensionsDirectory; + if (extDir != null && await extDir.exists()) { + await extDir.delete(recursive: true); + } + ExtensionPaths.mockExtensionsDirectory = null; + await LocalExtensionRegistry.instance.reload(); + ThemeController.instance.setRegistryServiceForTest( ThemeRegistryService( userThemesDirectory: () async => themesDir, @@ -108,8 +126,7 @@ void main() { expect(find.byType(ThemePickerButton), findsOneWidget); await tester.tap(find.text('Querya Dark')); - await tester.pump(); - await tester.pump(const Duration(milliseconds: 50)); + await tester.pumpAndSettle(); expect(find.text('Querya Light'), findsOneWidget); }); From 002058ecdf8ce2279de8cc00bf08e2b9a19ced34 Mon Sep 17 00:00:00 2001 From: Eva Rei <114882226+ZhuchkaTriplesix@users.noreply.github.com> Date: Sun, 21 Jun 2026 18:29:29 +0300 Subject: [PATCH 06/12] Fix TOCTOU, ID collisions, redundant scans, and remove legacy theme code (#221) --- .../extensions/local_extension_registry.dart | 14 +++- lib/core/theme/theme_controller.dart | 32 +-------- lib/core/theme/theme_import_service.dart | 65 ------------------- lib/core/theme/theme_registry_service.dart | 44 +++++++++++-- test/core/theme/theme_controller_test.dart | 16 +---- .../core/theme/theme_import_service_test.dart | 29 --------- .../theme_registry_legacy_import_test.dart | 54 ++++++--------- 7 files changed, 74 insertions(+), 180 deletions(-) diff --git a/lib/core/extensions/local_extension_registry.dart b/lib/core/extensions/local_extension_registry.dart index ddf17203..95b69c1d 100644 --- a/lib/core/extensions/local_extension_registry.dart +++ b/lib/core/extensions/local_extension_registry.dart @@ -13,6 +13,7 @@ class LocalExtensionRegistry { List _manifests = []; bool _loaded = false; + Future>? _loadFuture; /// Returns an unmodifiable list of loaded manifests. List get manifests => List.unmodifiable(_manifests); @@ -20,13 +21,24 @@ class LocalExtensionRegistry { /// Reloads manifests from the disk. Future reload() async { _loaded = false; + _loadFuture = null; await load(); } /// Loads manifests from the extensions directory if not already loaded. Future> load() async { if (_loaded) return manifests; - + if (_loadFuture != null) return _loadFuture!; + + _loadFuture = _doLoad(); + try { + return await _loadFuture!; + } finally { + _loadFuture = null; + } + } + + Future> _doLoad() async { final dir = await ExtensionPaths.extensionsDirectory(); final loadedManifests = []; diff --git a/lib/core/theme/theme_controller.dart b/lib/core/theme/theme_controller.dart index f38966e4..da681b09 100644 --- a/lib/core/theme/theme_controller.dart +++ b/lib/core/theme/theme_controller.dart @@ -473,37 +473,7 @@ class ThemeController extends ChangeNotifier { return result; } - /// Parses a VS Code theme file, persists it, and activates the imported preset. - Future importThemeFromFile(String path) async { - final result = await ThemeImportService.importFromPath(path); - switch (result) { - case ThemeImportSuccess( - :final name, - :final isDark, - :final colors, - :final tokenColors, - :final storedPath, - ): - await _clearRegistrySelection(); - _importedColors = Map.unmodifiable(colors); - _importedTokenColors = List.unmodifiable(tokenColors); - _importedThemeName = name; - _preset = QueryaThemePreset.imported; - _themeMode = isDark ? ThemeMode.dark : ThemeMode.light; - await AppSettings.instance.setThemeImportedColors(colors); - await AppSettings.instance.setThemeImportName(name); - await AppSettings.instance.setThemeImportPath(storedPath); - await AppSettings.instance.setThemePreset(QueryaThemePreset.imported); - await AppSettings.instance.setThemeMode(_themeMode); - _availableThemes = _mergeBuiltinThemes( - await _registryService.loadThemeDefinitions(), - ); - _notifyThemeChanged(); - return result; - case ThemeImportFailure(): - return result; - } - } + /// Sets or clears a user override for a VS Code `colors` key. Future setWorkbenchColor(String vscodeKey, Color? value) async { diff --git a/lib/core/theme/theme_import_service.dart b/lib/core/theme/theme_import_service.dart index 8787d884..2073bc48 100644 --- a/lib/core/theme/theme_import_service.dart +++ b/lib/core/theme/theme_import_service.dart @@ -6,31 +6,6 @@ import 'package:path_provider/path_provider.dart'; import 'parser/vscode_theme_manifest.dart'; import 'theme_definition.dart'; -/// Result of importing a VS Code theme file. -sealed class ThemeImportResult { - const ThemeImportResult(); -} - -class ThemeImportSuccess extends ThemeImportResult { - const ThemeImportSuccess({ - required this.name, - required this.isDark, - required this.colors, - required this.tokenColors, - required this.storedPath, - }); - - final String name; - final bool isDark; - final Map colors; - final List tokenColors; - final String storedPath; -} - -class ThemeImportFailure extends ThemeImportResult { - const ThemeImportFailure(this.message); - final String message; -} /// Result of copying a theme file into the user themes directory. sealed class ThemeDefinitionImportResult { @@ -80,46 +55,6 @@ abstract final class ThemeImportService { /// Path to the persisted legacy import copy under app support. static Future persistedImportFile() => _storedThemeFile(); - /// Reads [sourcePath], parses JSON/JSONC, copies to app data, returns colors. - static Future importFromPath(String sourcePath) async { - try { - final source = File(sourcePath); - if (!await source.exists()) { - return const ThemeImportFailure('Theme file not found.'); - } - final raw = await source.readAsString(); - final manifest = VsCodeThemeManifest.fromJsonString(raw); - if (manifest.colors.isEmpty) { - return const ThemeImportFailure( - 'Theme file has no "colors" section to import.', - ); - } - - final storedFile = await _storedThemeFile(); - await storedFile.parent.create(recursive: true); - await storedFile.writeAsString(raw); - - final name = manifest.name?.trim().isNotEmpty == true - ? manifest.name!.trim() - : p.basenameWithoutExtension(sourcePath); - - return ThemeImportSuccess( - name: name, - isDark: manifest.isDark || !manifest.isLight, - colors: Map.unmodifiable(manifest.colors), - tokenColors: List.unmodifiable(manifest.tokenColors), - storedPath: storedFile.path, - ); - } on VsCodeThemeParseException catch (e) { - return ThemeImportFailure(e.message); - } on FormatException catch (e) { - return ThemeImportFailure(e.message); - } on IOException catch (e) { - return ThemeImportFailure(e.toString()); - } on Object catch (e) { - return ThemeImportFailure(e.toString()); - } - } /// Reloads colors from the persisted import file, if present. static Future?> loadPersistedColors() async { diff --git a/lib/core/theme/theme_registry_service.dart b/lib/core/theme/theme_registry_service.dart index 4c991311..1d66f21b 100644 --- a/lib/core/theme/theme_registry_service.dart +++ b/lib/core/theme/theme_registry_service.dart @@ -132,21 +132,46 @@ class ThemeRegistryService { await _definitionFromFile(entity, source); if (definition == null) continue; + // Resolve ID collisions for legacy themes + var logicalId = definition.id; + var idSuffix = 2; + var candidateId = logicalId; + while (LocalExtensionRegistry.instance.manifests.any( + (m) => m.type == ExtensionType.theme && m.id == candidateId)) { + candidateId = '$logicalId-$idSuffix'; + idSuffix++; + } + logicalId = candidateId; + final slug = ThemeImportService.slugifyThemeName(definition.name); var finalExtDir = Directory(p.join(extensionsDir.path, slug)); var counter = 2; - while (await finalExtDir.exists()) { + while (true) { + if (!await finalExtDir.exists()) { + try { + await finalExtDir.create(recursive: false); + break; + } on FileSystemException { + // Another async task or process claimed it, keep looping. + } + } finalExtDir = Directory(p.join(extensionsDir.path, '$slug-$counter')); counter++; } - await finalExtDir.create(recursive: true); final themeFile = File(p.join(finalExtDir.path, 'theme.json')); - await entity.copy(themeFile.path); + + if (logicalId != definition.id && definition.format == ThemeFormat.queryaCustom) { + final raw = await entity.readAsString(); + final contentToWrite = _rewriteCustomThemeId(raw, logicalId); + await themeFile.writeAsString(contentToWrite); + } else { + await entity.copy(themeFile.path); + } final manifest = ExtensionManifest( - id: definition.id, + id: logicalId, name: definition.name, version: '1.0.0', publisher: source == ThemeSource.imported ? 'Imported' : 'Unknown', @@ -262,12 +287,19 @@ class ThemeRegistryService { var finalExtDir = Directory(p.join(extensionsDir.path, preferredBaseName)); var counter = 2; - while (await finalExtDir.exists()) { + while (true) { + if (!await finalExtDir.exists()) { + try { + await finalExtDir.create(recursive: false); + break; + } on FileSystemException { + // Another async task or process claimed it, keep looping. + } + } finalExtDir = Directory( p.join(extensionsDir.path, '$preferredBaseName-$counter')); counter++; } - await finalExtDir.create(recursive: true); resolvedFile = File(p.join(finalExtDir.path, 'theme.json')); await resolvedFile.writeAsString(contentToWrite); diff --git a/test/core/theme/theme_controller_test.dart b/test/core/theme/theme_controller_test.dart index 41d8a59e..6cea1a06 100644 --- a/test/core/theme/theme_controller_test.dart +++ b/test/core/theme/theme_controller_test.dart @@ -176,21 +176,7 @@ void main() { ); }); - test('importThemeFromFile applies imported colors to activeTheme', () async { - final c = ThemeController.instance; - await c.load(); - final fixture = File('test/fixtures/themes/dark_subset.json'); - final result = await c.importThemeFromFile(fixture.path); - expect(result, isA()); - expect(c.preset, QueryaThemePreset.imported); - expect(c.hasImportedTheme, isTrue); - expect( - c.activeTheme.workbench.editorBackground, - const Color(0xFF1E1E1E), - ); - await c.resetToDefaults(); - expect(c.preset, QueryaThemePreset.queryaDark); - }); + test('setThemeAnimationEnabled persists and reset clears', () async { final c = ThemeController.instance; diff --git a/test/core/theme/theme_import_service_test.dart b/test/core/theme/theme_import_service_test.dart index e6ae274a..67069ea7 100644 --- a/test/core/theme/theme_import_service_test.dart +++ b/test/core/theme/theme_import_service_test.dart @@ -39,36 +39,7 @@ void main() { } }); - test('importFromPath parses fixture and persists copy', () async { - final fixture = File('test/fixtures/themes/dark_subset.json'); - final result = await ThemeImportService.importFromPath(fixture.path); - expect(result, isA()); - final success = result as ThemeImportSuccess; - expect(success.name, 'Fixture Dark Subset'); - expect(success.isDark, isTrue); - expect(success.colors['editor.background'], '#1e1e1e'); - final reloaded = await ThemeImportService.loadPersistedColors(); - expect(reloaded?['editor.background'], '#1e1e1e'); - }); - - test('importFromPath persists tokenColors from dracula fixture', () async { - final fixture = File('test/fixtures/themes/dracula_tokens.json'); - final result = await ThemeImportService.importFromPath(fixture.path); - expect(result, isA()); - final success = result as ThemeImportSuccess; - expect(success.tokenColors, isNotEmpty); - - final tokens = await ThemeImportService.loadPersistedTokenColors(); - expect(tokens.length, success.tokenColors.length); - expect(tokens.first.scopes, contains('comment')); - }); - - test('importFromPath returns failure for missing file', () async { - final result = - await ThemeImportService.importFromPath('/no/such/theme.json'); - expect(result, isA()); - }); test('slugifyThemeName produces filesystem-safe slug', () { expect( diff --git a/test/core/theme/theme_registry_legacy_import_test.dart b/test/core/theme/theme_registry_legacy_import_test.dart index b81258aa..575ce627 100644 --- a/test/core/theme/theme_registry_legacy_import_test.dart +++ b/test/core/theme/theme_registry_legacy_import_test.dart @@ -7,7 +7,7 @@ import 'package:path_provider_platform_interface/path_provider_platform_interfac import 'package:querya_desktop/core/storage/app_settings.dart'; import 'package:querya_desktop/core/storage/local_db.dart'; import 'package:querya_desktop/core/theme/querya_theme_preset.dart'; -import 'package:querya_desktop/core/theme/theme_controller.dart'; + import 'package:querya_desktop/core/theme/theme_definition.dart'; import 'package:querya_desktop/core/theme/theme_import_service.dart'; import 'package:querya_desktop/core/theme/theme_load_result.dart'; @@ -67,14 +67,16 @@ void main() { test('exposes legacy imported theme from persisted import settings', () async { final fixture = File('test/fixtures/themes/dark_subset.json'); - final importResult = - await ThemeImportService.importFromPath(fixture.path); - expect(importResult, isA()); - final success = importResult as ThemeImportSuccess; - - await AppSettings.instance.setThemeImportedColors(success.colors); - await AppSettings.instance.setThemeImportName(success.name); - await AppSettings.instance.setThemeImportPath(success.storedPath); + final raw = await fixture.readAsString(); + final storedFile = await ThemeImportService.persistedImportFile(); + await storedFile.parent.create(recursive: true); + await storedFile.writeAsString(raw); + + await AppSettings.instance.setThemeImportedColors({ + 'editor.background': '#1e1e1e', + }); + await AppSettings.instance.setThemeImportName('Fixture Dark Subset'); + await AppSettings.instance.setThemeImportPath(storedFile.path); await AppSettings.instance.setThemePreset(QueryaThemePreset.imported); final definitions = await registry.loadThemeDefinitions(); @@ -85,7 +87,7 @@ void main() { expect(legacy.id, ThemeImportService.legacyImportedThemeId); expect(legacy.name, 'Fixture Dark Subset'); expect(legacy.format, ThemeFormat.vscode); - expect(legacy.path, success.storedPath); + expect(legacy.path, storedFile.path); expect( definitions.where((definition) => definition.id == 'imported'), hasLength(1), @@ -94,13 +96,16 @@ void main() { test('loads legacy imported theme definition', () async { final fixture = File('test/fixtures/themes/dark_subset.json'); - final importResult = - await ThemeImportService.importFromPath(fixture.path); - final success = importResult as ThemeImportSuccess; + final raw = await fixture.readAsString(); + final storedFile = await ThemeImportService.persistedImportFile(); + await storedFile.parent.create(recursive: true); + await storedFile.writeAsString(raw); - await AppSettings.instance.setThemeImportedColors(success.colors); - await AppSettings.instance.setThemeImportName(success.name); - await AppSettings.instance.setThemeImportPath(success.storedPath); + await AppSettings.instance.setThemeImportedColors({ + 'editor.background': '#1e1e1e', + }); + await AppSettings.instance.setThemeImportName('Fixture Dark Subset'); + await AppSettings.instance.setThemeImportPath(storedFile.path); final legacy = (await registry.loadThemeDefinitions()).singleWhere( (definition) => definition.source == ThemeSource.legacyImported, @@ -134,23 +139,6 @@ void main() { expect((result as ThemeLoadFailure).message, 'Theme file not found.'); }); - test('QueryaThemePreset.imported still applies via ThemeController', - () async { - final controller = ThemeController.instance; - final fixture = File('test/fixtures/themes/dark_subset.json'); - final result = await controller.importThemeFromFile(fixture.path); - expect(result, isA()); - expect(controller.preset, QueryaThemePreset.imported); - expect(controller.hasImportedTheme, isTrue); - - final definitions = await registry.loadThemeDefinitions(); - expect( - definitions.any( - (definition) => definition.source == ThemeSource.legacyImported, - ), - isTrue, - ); - }); }); } From 3872164407838adcaec892bd4f334d5fcf1ed3db Mon Sep 17 00:00:00 2001 From: ZhuchkaTriplesix Date: Sun, 21 Jun 2026 19:56:51 +0300 Subject: [PATCH 07/12] chore(release): prepare release 0.4.7 --- CHANGELOG.md | 16 ++++++++++++++++ pubspec.yaml | 2 +- 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index cca125de..5e691251 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -7,6 +7,22 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ## [Unreleased] +## [0.4.7] - 2026-06-21 + +Local extension discovery and manifest foundation release. Git tag **`0.4.7`**. + +### Added + +- **Extension models (EXT-1)** — data models `ExtensionManifest` and `ExtensionType` to parse `manifest.json`. +- **Local scanner (EXT-2)** — `LocalExtensionRegistry` scans `~/.querya/extensions/` to find and load extension manifests. +- **Theme migration (EXT-3)** — migrated legacy custom themes to the new unified extension package format. +- **Unit tests (EXT-4)** — unit tests for manifest parsing, directory scanning, and registry cache logic. + +### Fixed + +- **Theme importing security** — resolved concurrent import TOCTOU filesystem races and theme ID collisions during migration, and cleaned up deprecated legacy import code. +- **Appearance Settings test** — resolved the preferences appearance section widget test failure by mocking the extensions directory. + ## [0.4.6-a] - 2026-06-18 ### Changed diff --git a/pubspec.yaml b/pubspec.yaml index bf8171d3..446519ec 100644 --- a/pubspec.yaml +++ b/pubspec.yaml @@ -1,6 +1,6 @@ name: querya_desktop description: Lightweight desktop SQL/NoSQL client. Flutter (Dart). -version: 0.4.6-a +version: 0.4.6+12 From 7b7c09e97cafc7dfa7bf08187be00c80f4734452 Mon Sep 17 00:00:00 2001 From: Eva Rei <114882226+ZhuchkaTriplesix@users.noreply.github.com> Date: Mon, 22 Jun 2026 03:37:54 +0300 Subject: [PATCH 08/12] feat(menu): implement File -> Exit to close the application window (#230) --- lib/features/main_screen/querya_window_title_bar.dart | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/features/main_screen/querya_window_title_bar.dart b/lib/features/main_screen/querya_window_title_bar.dart index 1d957b88..bf6e2937 100644 --- a/lib/features/main_screen/querya_window_title_bar.dart +++ b/lib/features/main_screen/querya_window_title_bar.dart @@ -84,7 +84,8 @@ class QueryaWindowTitleBar extends StatelessWidget { onPressed: (_) {}, child: const Text('Save')), const MenuDivider(), MenuButton( - onPressed: (_) {}, child: const Text('Exit')), + onPressed: (_) => appWindow.close(), + child: const Text('Exit')), ], child: const Text('File'), ), From f631eac424e0979e3cbc328b11a1a340725204e6 Mon Sep 17 00:00:00 2001 From: Eva Rei <114882226+ZhuchkaTriplesix@users.noreply.github.com> Date: Mon, 22 Jun 2026 03:49:37 +0300 Subject: [PATCH 09/12] feat(menu): implement File -> New, Open..., Save actions (#231) * feat(menu): implement File -> New, Open..., Save actions via Intents/Actions * refactor(menu): remove unnecessary widgets import in title bar --- lib/core/actions/sql_editor_actions.dart | 13 ++ .../main_screen/querya_window_title_bar.dart | 26 ++- lib/features/mysql/mysql_sql_workspace.dart | 186 ++++++++++------ .../postgresql/postgres_sql_workspace.dart | 198 ++++++++++++------ lib/features/sqlite/sqlite_sql_workspace.dart | 182 ++++++++++------ 5 files changed, 408 insertions(+), 197 deletions(-) create mode 100644 lib/core/actions/sql_editor_actions.dart diff --git a/lib/core/actions/sql_editor_actions.dart b/lib/core/actions/sql_editor_actions.dart new file mode 100644 index 00000000..f7ef4cfe --- /dev/null +++ b/lib/core/actions/sql_editor_actions.dart @@ -0,0 +1,13 @@ +import 'package:flutter/widgets.dart'; + +class NewSqlIntent extends Intent { + const NewSqlIntent(); +} + +class OpenSqlIntent extends Intent { + const OpenSqlIntent(); +} + +class SaveSqlIntent extends Intent { + const SaveSqlIntent(); +} diff --git a/lib/features/main_screen/querya_window_title_bar.dart b/lib/features/main_screen/querya_window_title_bar.dart index bf6e2937..2bfe0e9d 100644 --- a/lib/features/main_screen/querya_window_title_bar.dart +++ b/lib/features/main_screen/querya_window_title_bar.dart @@ -4,6 +4,7 @@ import 'package:flutter/material.dart' as material import 'package:querya_desktop/core/theme/querya_theme_scope.dart'; import 'package:querya_desktop/features/connections/driver_manager_dialog.dart'; import 'package:querya_desktop/features/settings/preferences_dialog.dart'; +import 'package:querya_desktop/core/actions/sql_editor_actions.dart'; import 'package:shadcn_flutter/shadcn_flutter.dart'; /// Custom bitsdojo title bar styled from [QueryaThemeScope] workbench tokens. @@ -76,12 +77,29 @@ class QueryaWindowTitleBar extends StatelessWidget { MenuButton( subMenu: [ MenuButton( - onPressed: (_) {}, child: const Text('New')), - MenuButton( - onPressed: (_) {}, + onPressed: (ctx) { + Actions.maybeInvoke( + FocusManager.instance.primaryFocus?.context ?? ctx, + const NewSqlIntent(), + ); + }, + child: const Text('New')), + MenuButton( + onPressed: (ctx) { + Actions.maybeInvoke( + FocusManager.instance.primaryFocus?.context ?? ctx, + const OpenSqlIntent(), + ); + }, child: const Text('Open...')), MenuButton( - onPressed: (_) {}, child: const Text('Save')), + onPressed: (ctx) { + Actions.maybeInvoke( + FocusManager.instance.primaryFocus?.context ?? ctx, + const SaveSqlIntent(), + ); + }, + child: const Text('Save')), const MenuDivider(), MenuButton( onPressed: (_) => appWindow.close(), diff --git a/lib/features/mysql/mysql_sql_workspace.dart b/lib/features/mysql/mysql_sql_workspace.dart index e2778fa2..339128c3 100644 --- a/lib/features/mysql/mysql_sql_workspace.dart +++ b/lib/features/mysql/mysql_sql_workspace.dart @@ -1,8 +1,11 @@ import 'dart:async'; +import 'dart:io'; import 'package:flutter/foundation.dart'; import 'package:flutter/material.dart' as material; import 'package:flutter/services.dart' show LogicalKeyboardKey; +import 'package:file_selector/file_selector.dart'; +import 'package:querya_desktop/core/actions/sql_editor_actions.dart'; import 'package:querya_desktop/core/database/mysql_service.dart'; import 'package:querya_desktop/core/layout/vertical_split_pane.dart'; import 'package:querya_desktop/core/storage/app_settings.dart'; @@ -237,78 +240,135 @@ class _MysqlSqlWorkspaceState extends material.State { return v.toInt(); } + Future _openSqlFile() async { + try { + final file = await openFile( + acceptedTypeGroups: const [ + XTypeGroup( + label: 'SQL query', + extensions: ['sql'], + ), + ], + ); + if (file == null) return; + final text = await file.readAsString(); + if (!mounted) return; + _sqlController.value = material.TextEditingValue( + text: text, + selection: material.TextSelection.collapsed(offset: text.length), + ); + } catch (_) {} + } + + Future _saveSqlFile() async { + try { + final name = 'query_${DateTime.now().toIso8601String().replaceAll(':', '-')}.sql'; + final location = await getSaveLocation( + acceptedTypeGroups: const [ + XTypeGroup(label: 'SQL', extensions: ['sql']), + ], + suggestedName: name, + ); + final path = location?.path; + if (path == null || path.isEmpty) return; + await File(path).writeAsString(_sqlController.text); + } catch (_) {} + } + @override material.Widget build(material.BuildContext context) { final theme = Theme.of(context); - return material.CallbackShortcuts( - bindings: { - const material.SingleActivator(LogicalKeyboardKey.f5): () { - if (!_running) { - unawaited(_execute()); - } - }, + return Actions( + actions: >{ + NewSqlIntent: CallbackAction( + onInvoke: (intent) { + _sqlController.clear(); + return null; + }, + ), + OpenSqlIntent: CallbackAction( + onInvoke: (intent) { + unawaited(_openSqlFile()); + return null; + }, + ), + SaveSqlIntent: CallbackAction( + onInvoke: (intent) { + unawaited(_saveSqlFile()); + return null; + }, + ), }, - child: material.Focus( - autofocus: true, - child: VerticalSplitPane( - fraction: _topFraction, - maxFraction: 0.85, - top: Column( - crossAxisAlignment: CrossAxisAlignment.stretch, - children: [ - _MysqlSqlToolbar( - onExecute: _running ? null : _execute, - running: _running, - queryTimeoutSeconds: _queryTimeoutSeconds, - onQueryTimeoutChanged: _onStmtTimeoutChanged, - onOpenPreferences: () => showPreferencesDialog(context), - onOpenHistory: widget.connectionRow.id != null && !_running - ? () { - showSqlQueryHistoryDialog( - context: context, - connectionId: widget.connectionRow.id!, - databaseName: widget.connectionRow.databaseName, - sqlController: _sqlController, - ); - } - : null, - ), - const Divider(height: 1), - Expanded( - child: QueryEditorTab( - controller: _sqlController, - fontSize: _editorFontSize, + child: material.CallbackShortcuts( + bindings: { + const material.SingleActivator(LogicalKeyboardKey.f5): () { + if (!_running) { + unawaited(_execute()); + } + }, + }, + child: material.Focus( + autofocus: true, + child: VerticalSplitPane( + fraction: _topFraction, + maxFraction: 0.85, + top: Column( + crossAxisAlignment: CrossAxisAlignment.stretch, + children: [ + _MysqlSqlToolbar( + onExecute: _running ? null : _execute, + running: _running, + queryTimeoutSeconds: _queryTimeoutSeconds, + onQueryTimeoutChanged: _onStmtTimeoutChanged, + onOpenPreferences: () => showPreferencesDialog(context), + onOpenHistory: widget.connectionRow.id != null && !_running + ? () { + showSqlQueryHistoryDialog( + context: context, + connectionId: widget.connectionRow.id!, + databaseName: widget.connectionRow.databaseName, + sqlController: _sqlController, + ); + } + : null, ), - ), - ], - ), - bottom: Column( - crossAxisAlignment: CrossAxisAlignment.stretch, - children: [ - material.Container( - constraints: const material.BoxConstraints(minHeight: 44), - padding: const material.EdgeInsets.symmetric( - horizontal: 12, + const Divider(height: 1), + Expanded( + child: QueryEditorTab( + controller: _sqlController, + fontSize: _editorFontSize, + ), ), - decoration: material.BoxDecoration( - color: theme.colorScheme.muted.withValues(alpha: 0.6), + ], + ), + bottom: Column( + crossAxisAlignment: CrossAxisAlignment.stretch, + children: [ + material.Container( + constraints: const material.BoxConstraints(minHeight: 44), + padding: const material.EdgeInsets.symmetric( + horizontal: 12, + ), + decoration: material.BoxDecoration( + color: theme.colorScheme.muted.withValues(alpha: 0.6), + ), + alignment: material.Alignment.centerLeft, + child: const Text('Data Output').semiBold().small(), ), - alignment: material.Alignment.centerLeft, - child: const Text('Data Output').semiBold().small(), - ), - const Divider(height: 1), - Expanded( - child: ResultsTab( - columns: _columns, - rows: _rows, - errorMessage: _error, - isLoading: _running, - affectedRows: _affectedRows, - statusLine: _statusLine, + const Divider(height: 1), + Expanded( + child: ResultsTab( + columns: _columns, + rows: _rows, + errorMessage: _error, + isLoading: _running, + affectedRows: _affectedRows, + statusLine: _statusLine, + ), ), - ), - ], + ], + ), ), ), ), diff --git a/lib/features/postgresql/postgres_sql_workspace.dart b/lib/features/postgresql/postgres_sql_workspace.dart index d72b79a9..f6ecb83f 100644 --- a/lib/features/postgresql/postgres_sql_workspace.dart +++ b/lib/features/postgresql/postgres_sql_workspace.dart @@ -1,7 +1,10 @@ import 'dart:async'; +import 'dart:io'; import 'package:flutter/material.dart' as material; import 'package:flutter/services.dart' show LogicalKeyboardKey; +import 'package:file_selector/file_selector.dart'; +import 'package:querya_desktop/core/actions/sql_editor_actions.dart'; import 'package:postgres/postgres.dart' as pg; import 'package:querya_desktop/core/database/postgres_service.dart'; import 'package:querya_desktop/core/database/postgres_sql.dart'; @@ -377,84 +380,141 @@ class _PostgresSqlWorkspaceState extends material.State { return v.toString(); } + Future _openSqlFile() async { + try { + final file = await openFile( + acceptedTypeGroups: const [ + XTypeGroup( + label: 'SQL query', + extensions: ['sql'], + ), + ], + ); + if (file == null) return; + final text = await file.readAsString(); + if (!mounted) return; + _sqlController.value = material.TextEditingValue( + text: text, + selection: material.TextSelection.collapsed(offset: text.length), + ); + } catch (_) {} + } + + Future _saveSqlFile() async { + try { + final name = 'query_${DateTime.now().toIso8601String().replaceAll(':', '-')}.sql'; + final location = await getSaveLocation( + acceptedTypeGroups: const [ + XTypeGroup(label: 'SQL', extensions: ['sql']), + ], + suggestedName: name, + ); + final path = location?.path; + if (path == null || path.isEmpty) return; + await File(path).writeAsString(_sqlController.text); + } catch (_) {} + } + @override material.Widget build(material.BuildContext context) { final theme = Theme.of(context); - return material.CallbackShortcuts( - bindings: { - const material.SingleActivator(LogicalKeyboardKey.f5): () { - if (!_running) _execute(); - }, + return Actions( + actions: >{ + NewSqlIntent: CallbackAction( + onInvoke: (intent) { + _sqlController.clear(); + return null; + }, + ), + OpenSqlIntent: CallbackAction( + onInvoke: (intent) { + unawaited(_openSqlFile()); + return null; + }, + ), + SaveSqlIntent: CallbackAction( + onInvoke: (intent) { + unawaited(_saveSqlFile()); + return null; + }, + ), }, - child: material.Focus( - autofocus: true, - child: VerticalSplitPane( - fraction: _topFraction, - maxFraction: 0.85, - top: Column( - crossAxisAlignment: CrossAxisAlignment.stretch, - children: [ - _SqlToolbar( - sessionDatabase: _effectiveSessionDatabase(), - onExecute: _running ? null : _execute, - running: _running, - autocommit: _autocommit, - onAutocommitChanged: (v) => setState(() => _autocommit = v), - queryTimeoutSeconds: _queryTimeoutSeconds, - onQueryTimeoutChanged: _onStmtTimeoutChanged, - onOpenPreferences: () => showPreferencesDialog(context), - onOpenHistory: widget.connectionRow.id != null && !_running - ? () { - showSqlQueryHistoryDialog( - context: context, - connectionId: widget.connectionRow.id!, - databaseName: _effectiveSessionDatabase(), - sqlController: _sqlController, - ); - } - : null, - txOpen: _txOpen, - onBegin: _running ? null : () => _runTxCommand('BEGIN'), - onCommit: _running ? null : () => _runTxCommand('COMMIT'), - onRollback: - _running ? null : () => _runTxCommand('ROLLBACK'), - ), - const Divider(height: 1), - Expanded( - child: QueryEditorTab( - controller: _sqlController, - fontSize: _editorFontSize, + child: material.CallbackShortcuts( + bindings: { + const material.SingleActivator(LogicalKeyboardKey.f5): () { + if (!_running) _execute(); + }, + }, + child: material.Focus( + autofocus: true, + child: VerticalSplitPane( + fraction: _topFraction, + maxFraction: 0.85, + top: Column( + crossAxisAlignment: CrossAxisAlignment.stretch, + children: [ + _SqlToolbar( + sessionDatabase: _effectiveSessionDatabase(), + onExecute: _running ? null : _execute, + running: _running, + autocommit: _autocommit, + onAutocommitChanged: (v) => setState(() => _autocommit = v), + queryTimeoutSeconds: _queryTimeoutSeconds, + onQueryTimeoutChanged: _onStmtTimeoutChanged, + onOpenPreferences: () => showPreferencesDialog(context), + onOpenHistory: widget.connectionRow.id != null && !_running + ? () { + showSqlQueryHistoryDialog( + context: context, + connectionId: widget.connectionRow.id!, + databaseName: _effectiveSessionDatabase(), + sqlController: _sqlController, + ); + } + : null, + txOpen: _txOpen, + onBegin: _running ? null : () => _runTxCommand('BEGIN'), + onCommit: _running ? null : () => _runTxCommand('COMMIT'), + onRollback: + _running ? null : () => _runTxCommand('ROLLBACK'), ), - ), - ], - ), - bottom: Column( - crossAxisAlignment: CrossAxisAlignment.stretch, - children: [ - material.Container( - constraints: const material.BoxConstraints(minHeight: 44), - padding: const material.EdgeInsets.symmetric( - horizontal: 12, + const Divider(height: 1), + Expanded( + child: QueryEditorTab( + controller: _sqlController, + fontSize: _editorFontSize, + ), ), - decoration: material.BoxDecoration( - color: theme.colorScheme.muted.withValues(alpha: 0.6), + ], + ), + bottom: Column( + crossAxisAlignment: CrossAxisAlignment.stretch, + children: [ + material.Container( + constraints: const material.BoxConstraints(minHeight: 44), + padding: const material.EdgeInsets.symmetric( + horizontal: 12, + ), + decoration: material.BoxDecoration( + color: theme.colorScheme.muted.withValues(alpha: 0.6), + ), + alignment: material.Alignment.centerLeft, + child: const Text('Data Output').semiBold().small(), ), - alignment: material.Alignment.centerLeft, - child: const Text('Data Output').semiBold().small(), - ), - const Divider(height: 1), - Expanded( - child: ResultsTab( - columns: _columns, - rows: _rows, - errorMessage: _error, - isLoading: _running, - affectedRows: _affectedRows, - statusLine: _statusLine, + const Divider(height: 1), + Expanded( + child: ResultsTab( + columns: _columns, + rows: _rows, + errorMessage: _error, + isLoading: _running, + affectedRows: _affectedRows, + statusLine: _statusLine, + ), ), - ), - ], + ], + ), ), ), ), diff --git a/lib/features/sqlite/sqlite_sql_workspace.dart b/lib/features/sqlite/sqlite_sql_workspace.dart index b2449b70..45d38d52 100644 --- a/lib/features/sqlite/sqlite_sql_workspace.dart +++ b/lib/features/sqlite/sqlite_sql_workspace.dart @@ -1,6 +1,9 @@ import 'dart:async'; +import 'dart:io'; import 'package:flutter/material.dart' as material; import 'package:flutter/services.dart' show LogicalKeyboardKey; +import 'package:file_selector/file_selector.dart'; +import 'package:querya_desktop/core/actions/sql_editor_actions.dart'; import 'package:querya_desktop/core/database/sqlite_service.dart'; import 'package:querya_desktop/core/layout/vertical_split_pane.dart'; import 'package:querya_desktop/core/storage/app_settings.dart'; @@ -182,76 +185,133 @@ class _SqliteSqlWorkspaceState extends material.State { } } + Future _openSqlFile() async { + try { + final file = await openFile( + acceptedTypeGroups: const [ + XTypeGroup( + label: 'SQL query', + extensions: ['sql'], + ), + ], + ); + if (file == null) return; + final text = await file.readAsString(); + if (!mounted) return; + _sqlController.value = material.TextEditingValue( + text: text, + selection: material.TextSelection.collapsed(offset: text.length), + ); + } catch (_) {} + } + + Future _saveSqlFile() async { + try { + final name = 'query_${DateTime.now().toIso8601String().replaceAll(':', '-')}.sql'; + final location = await getSaveLocation( + acceptedTypeGroups: const [ + XTypeGroup(label: 'SQL', extensions: ['sql']), + ], + suggestedName: name, + ); + final path = location?.path; + if (path == null || path.isEmpty) return; + await File(path).writeAsString(_sqlController.text); + } catch (_) {} + } + @override material.Widget build(material.BuildContext context) { final theme = Theme.of(context); - return material.CallbackShortcuts( - bindings: { - const material.SingleActivator(LogicalKeyboardKey.f5): () { - if (!_running) { - unawaited(_execute()); - } - }, + return Actions( + actions: >{ + NewSqlIntent: CallbackAction( + onInvoke: (intent) { + _sqlController.clear(); + return null; + }, + ), + OpenSqlIntent: CallbackAction( + onInvoke: (intent) { + unawaited(_openSqlFile()); + return null; + }, + ), + SaveSqlIntent: CallbackAction( + onInvoke: (intent) { + unawaited(_saveSqlFile()); + return null; + }, + ), }, - child: material.Focus( - autofocus: true, - child: VerticalSplitPane( - fraction: _topFraction, - maxFraction: 0.85, - top: material.Column( - crossAxisAlignment: material.CrossAxisAlignment.stretch, - children: [ - _SqliteSqlToolbar( - onExecute: _running ? null : _execute, - running: _running, - onOpenPreferences: () => showPreferencesDialog(context), - onOpenHistory: widget.connectionRow.id != null && !_running - ? () { - showSqlQueryHistoryDialog( - context: context, - connectionId: widget.connectionRow.id!, - databaseName: widget.connectionRow.databaseName, - sqlController: _sqlController, - ); - } - : null, - ), - const Divider(height: 1), - material.Expanded( - child: QueryEditorTab( - controller: _sqlController, - fontSize: _editorFontSize, + child: material.CallbackShortcuts( + bindings: { + const material.SingleActivator(LogicalKeyboardKey.f5): () { + if (!_running) { + unawaited(_execute()); + } + }, + }, + child: material.Focus( + autofocus: true, + child: VerticalSplitPane( + fraction: _topFraction, + maxFraction: 0.85, + top: material.Column( + crossAxisAlignment: material.CrossAxisAlignment.stretch, + children: [ + _SqliteSqlToolbar( + onExecute: _running ? null : _execute, + running: _running, + onOpenPreferences: () => showPreferencesDialog(context), + onOpenHistory: widget.connectionRow.id != null && !_running + ? () { + showSqlQueryHistoryDialog( + context: context, + connectionId: widget.connectionRow.id!, + databaseName: widget.connectionRow.databaseName, + sqlController: _sqlController, + ); + } + : null, ), - ), - ], - ), - bottom: material.Column( - crossAxisAlignment: material.CrossAxisAlignment.stretch, - children: [ - material.Container( - constraints: const material.BoxConstraints(minHeight: 44), - padding: const material.EdgeInsets.symmetric( - horizontal: 12, + const Divider(height: 1), + material.Expanded( + child: QueryEditorTab( + controller: _sqlController, + fontSize: _editorFontSize, + ), ), - decoration: material.BoxDecoration( - color: theme.colorScheme.muted.withValues(alpha: 0.6), + ], + ), + bottom: material.Column( + crossAxisAlignment: material.CrossAxisAlignment.stretch, + children: [ + material.Container( + constraints: const material.BoxConstraints(minHeight: 44), + padding: const material.EdgeInsets.symmetric( + horizontal: 12, + ), + decoration: material.BoxDecoration( + color: theme.colorScheme.muted.withValues(alpha: 0.6), + ), + alignment: material.Alignment.centerLeft, + child: const Text('Data Output').semiBold().small(), ), - alignment: material.Alignment.centerLeft, - child: const Text('Data Output').semiBold().small(), - ), - const Divider(height: 1), - material.Expanded( - child: ResultsTab( - columns: _columns, - rows: _rows, - errorMessage: _error, - isLoading: _running, - affectedRows: _affectedRows, - statusLine: _statusLine, + const Divider(height: 1), + material.Expanded( + child: ResultsTab( + columns: _columns, + rows: _rows, + errorMessage: _error, + isLoading: _running, + affectedRows: _affectedRows, + statusLine: _statusLine, + ), ), - ), - ], + ], + ), ), ), ), From a59d153accbeada2ef61a73c5a50e58efe408c82 Mon Sep 17 00:00:00 2001 From: Eva Rei <114882226+ZhuchkaTriplesix@users.noreply.github.com> Date: Mon, 22 Jun 2026 04:37:46 +0300 Subject: [PATCH 10/12] feat: implement connection management improvements and read-only mode (Issues #225, #226) (#232) * feat(connection): implement connection actions and fix layout test hang * feat(connection): implement Invalidate/Reconnect menu item * feat: implement Connection -> Read-only mode for database connections * fix ci --- lib/core/storage/local_db.dart | 1 + .../connections/connections_panel.dart | 86 +++++++++++++ .../connections/connections_panel_mongo.dart | 33 ++++- .../connections/connections_panel_mysql.dart | 33 ++++- ...connections_panel_postgres_connection.dart | 33 ++++- .../connections/connections_panel_redis.dart | 33 ++++- .../connections/connections_panel_sqlite.dart | 34 ++++- lib/features/main_screen/main_screen.dart | 42 +++++- .../main_screen_workspace_state.dart | 32 ++++- .../main_screen/querya_window_title_bar.dart | 40 +++++- lib/features/main_screen/workspace_panel.dart | 5 + lib/features/mysql/mysql_sql_workspace.dart | 15 ++- lib/features/mysql/mysql_workspace_home.dart | 11 ++ .../postgresql/postgres_sql_workspace.dart | 9 +- .../postgresql/postgres_workspace_home.dart | 11 ++ lib/features/sqlite/sqlite_sql_workspace.dart | 13 +- .../sqlite/sqlite_workspace_home.dart | 11 ++ .../connections_panel_layout_test.dart | 120 ++++++++++++++++++ .../main_screen_workspace_state_test.dart | 18 +++ 19 files changed, 545 insertions(+), 35 deletions(-) diff --git a/lib/core/storage/local_db.dart b/lib/core/storage/local_db.dart index 07ff8442..eebba457 100644 --- a/lib/core/storage/local_db.dart +++ b/lib/core/storage/local_db.dart @@ -379,6 +379,7 @@ class LocalDb { Future close() async { await _db?.close(); _db = null; + _cachedDbPath = null; } } diff --git a/lib/features/connections/connections_panel.dart b/lib/features/connections/connections_panel.dart index cc65b630..40bbfa6e 100644 --- a/lib/features/connections/connections_panel.dart +++ b/lib/features/connections/connections_panel.dart @@ -67,6 +67,8 @@ import 'package:querya_desktop/core/motion/querya_motion.dart'; import 'package:querya_desktop/core/motion/querya_motion_context.dart'; import 'package:querya_desktop/features/connections/connection_creation_flow.dart'; import 'package:querya_desktop/shared/widgets/widgets.dart'; +import 'package:querya_desktop/core/database/redis_service.dart'; +import 'package:querya_desktop/app/app_shutdown.dart'; import 'package:querya_desktop/features/mongodb/mongo_database_dialog.dart'; import 'package:querya_desktop/features/postgresql/postgres_object_kind.dart'; @@ -219,6 +221,7 @@ class ConnectionsPanelState extends State { List _connections = []; Map _folderIdByName = {}; final Set _expandedFolders = {}; + final Set _expandedConnections = {}; /// Ignores stale [setState] when multiple [_loadData] runs overlap (e.g. tests). int _loadDataGeneration = 0; @@ -312,6 +315,68 @@ class ConnectionsPanelState extends State { await _loadData(); } + void connect(int connectionId) { + setState(() { + _expandedConnections.add(connectionId); + }); + } + + @visibleForTesting + bool isConnectionExpanded(int id) => _expandedConnections.contains(id); + + Future disconnect(ConnectionRow conn) async { + final id = conn.id!; + setState(() { + _expandedConnections.remove(id); + }); + if (conn.type == 'postgresql') { + PostgresService.instance.interrupt(conn, database: conn.databaseName ?? 'postgres', mode: PgSessionMode.readOnly); + PostgresService.instance.interrupt(conn, database: conn.databaseName ?? 'postgres', mode: PgSessionMode.readWrite); + } else if (conn.type == 'mysql') { + MysqlService.instance.interrupt(conn, database: conn.databaseName ?? '', mode: MysqlSessionMode.readOnly); + MysqlService.instance.interrupt(conn, database: conn.databaseName ?? '', mode: MysqlSessionMode.readWrite); + } else if (conn.type == 'sqlite') { + SqliteService.instance.interrupt(conn, mode: SqliteSessionMode.readOnly); + SqliteService.instance.interrupt(conn, mode: SqliteSessionMode.readWrite); + } else if (conn.type == 'redis') { + final redisConn = RedisService.instance.getConnection(id); + if (redisConn != null) { + await RedisService.instance.disconnect(redisConn); + } + } else if (conn.type == 'mongodb') { + await MongoService.instance.disconnectByConnectionId(id); + } + } + + Future disconnectAll() async { + setState(() { + _expandedConnections.clear(); + }); + await disconnectAllExternalServices(); + await SqliteService.instance.disconnectAll(); + } + + Future disconnectOthers(ConnectionRow keepConn) async { + final keepId = keepConn.id!; + setState(() { + _expandedConnections.clear(); + _expandedConnections.add(keepId); + }); + for (final conn in _connections) { + if (conn.id == keepId) continue; + await disconnect(conn); + } + } + + Future reconnect(ConnectionRow conn) async { + final id = conn.id!; + await disconnect(conn); + await Future.delayed(const Duration(milliseconds: 50)); + if (mounted) { + connect(id); + } + } + /// Icon for a connection type (matches New Connection dialog). material.IconData _iconForType(String type) { return switch (type) { @@ -338,6 +403,17 @@ class ConnectionsPanelState extends State { Widget _buildConnectionTile(ConnectionRow conn) { final isSelected = widget.selectedConnectionId != null && widget.selectedConnectionId == conn.id; + final isExpanded = _expandedConnections.contains(conn.id); + void handleExpandedChanged(bool expanded) { + setState(() { + if (expanded) { + _expandedConnections.add(conn.id!); + } else { + _expandedConnections.remove(conn.id!); + } + }); + } + if (conn.type == 'postgresql') { return _PostgresConnectionTile( connection: conn, @@ -348,6 +424,8 @@ class ConnectionsPanelState extends State { onTap: () => widget.onConnectionSelected?.call(conn), onPostgresObjectSelected: widget.onPostgresObjectSelected, onPostgresOpenSqlWorkspace: widget.onPostgresOpenSqlWorkspace, + isExpanded: isExpanded, + onExpandedChanged: handleExpandedChanged, ); } else if (conn.type == 'mysql') { return _MysqlConnectionTile( @@ -359,6 +437,8 @@ class ConnectionsPanelState extends State { onTap: () => widget.onConnectionSelected?.call(conn), onMysqlObjectSelected: widget.onMysqlObjectSelected, onMysqlOpenSqlWorkspace: widget.onMysqlOpenSqlWorkspace, + isExpanded: isExpanded, + onExpandedChanged: handleExpandedChanged, ); } else if (conn.type == 'redis') { return _RedisConnectionTile( @@ -369,6 +449,8 @@ class ConnectionsPanelState extends State { onRemove: () => _removeConnection(conn.id!), onTap: () => widget.onConnectionSelected?.call(conn), onDatabaseTap: (db) => widget.onRedisDatabaseSelected?.call(conn, db), + isExpanded: isExpanded, + onExpandedChanged: handleExpandedChanged, ); } else if (conn.type == 'mongodb') { return _MongoConnectionTile( @@ -379,6 +461,8 @@ class ConnectionsPanelState extends State { onRemove: () => _removeConnection(conn.id!), onTap: () => widget.onConnectionSelected?.call(conn), onDatabaseTap: (db) => widget.onMongoDBDatabaseSelected?.call(conn, db), + isExpanded: isExpanded, + onExpandedChanged: handleExpandedChanged, ); } else if (conn.type == 'sqlite') { return _SqliteConnectionTile( @@ -390,6 +474,8 @@ class ConnectionsPanelState extends State { onTap: () => widget.onConnectionSelected?.call(conn), onSqliteObjectSelected: widget.onSqliteObjectSelected, onSqliteOpenSqlWorkspace: widget.onSqliteOpenSqlWorkspace, + isExpanded: isExpanded, + onExpandedChanged: handleExpandedChanged, ); } return _ConnectionTile( diff --git a/lib/features/connections/connections_panel_mongo.dart b/lib/features/connections/connections_panel_mongo.dart index 29ac0343..cc1c5d41 100644 --- a/lib/features/connections/connections_panel_mongo.dart +++ b/lib/features/connections/connections_panel_mongo.dart @@ -11,6 +11,8 @@ class _MongoConnectionTile extends StatefulWidget { required this.onRemove, this.onTap, this.onDatabaseTap, + this.isExpanded = false, + this.onExpandedChanged, }); final ConnectionRow connection; @@ -20,24 +22,47 @@ class _MongoConnectionTile extends StatefulWidget { final VoidCallback onRemove; final VoidCallback? onTap; final void Function(String database)? onDatabaseTap; + final bool isExpanded; + final ValueChanged? onExpandedChanged; @override State<_MongoConnectionTile> createState() => _MongoConnectionTileState(); } class _MongoConnectionTileState extends State<_MongoConnectionTile> { - bool _expanded = false; + bool get _expanded => widget.isExpanded; bool _loading = false; String? _error; List _databases = []; - void _toggle() { - setState(() => _expanded = !_expanded); - if (_expanded && _databases.isEmpty && !_loading) { + @override + void initState() { + super.initState(); + if (widget.isExpanded) { _loadDatabases(); } } + @override + void didUpdateWidget(_MongoConnectionTile oldWidget) { + super.didUpdateWidget(oldWidget); + if (widget.isExpanded && !oldWidget.isExpanded) { + if (_databases.isEmpty && !_loading) { + _loadDatabases(); + } + } else if (!widget.isExpanded && oldWidget.isExpanded) { + setState(() { + _databases = []; + _loading = false; + _error = null; + }); + } + } + + void _toggle() { + widget.onExpandedChanged?.call(!widget.isExpanded); + } + Future _loadDatabases() async { if (!mounted) return; setState(() { diff --git a/lib/features/connections/connections_panel_mysql.dart b/lib/features/connections/connections_panel_mysql.dart index 10c67383..2e7d34eb 100644 --- a/lib/features/connections/connections_panel_mysql.dart +++ b/lib/features/connections/connections_panel_mysql.dart @@ -12,6 +12,8 @@ class _MysqlConnectionTile extends StatefulWidget { this.onTap, this.onMysqlObjectSelected, this.onMysqlOpenSqlWorkspace, + this.isExpanded = false, + this.onExpandedChanged, }); final ConnectionRow connection; @@ -27,24 +29,47 @@ class _MysqlConnectionTile extends StatefulWidget { MysqlObjectKind kind, )? onMysqlObjectSelected; final void Function(ConnectionRow connection)? onMysqlOpenSqlWorkspace; + final bool isExpanded; + final ValueChanged? onExpandedChanged; @override State<_MysqlConnectionTile> createState() => _MysqlConnectionTileState(); } class _MysqlConnectionTileState extends State<_MysqlConnectionTile> { - bool _expanded = false; + bool get _expanded => widget.isExpanded; bool _loading = false; String? _error; List _databases = []; - void _toggle() { - setState(() => _expanded = !_expanded); - if (_expanded && _databases.isEmpty && !_loading) { + @override + void initState() { + super.initState(); + if (widget.isExpanded) { _loadDatabases(); } } + @override + void didUpdateWidget(_MysqlConnectionTile oldWidget) { + super.didUpdateWidget(oldWidget); + if (widget.isExpanded && !oldWidget.isExpanded) { + if (_databases.isEmpty && !_loading) { + _loadDatabases(); + } + } else if (!widget.isExpanded && oldWidget.isExpanded) { + setState(() { + _databases = []; + _loading = false; + _error = null; + }); + } + } + + void _toggle() { + widget.onExpandedChanged?.call(!widget.isExpanded); + } + Future _loadDatabases() async { if (!mounted) return; setState(() { diff --git a/lib/features/connections/connections_panel_postgres_connection.dart b/lib/features/connections/connections_panel_postgres_connection.dart index 925849a8..9bc2a359 100644 --- a/lib/features/connections/connections_panel_postgres_connection.dart +++ b/lib/features/connections/connections_panel_postgres_connection.dart @@ -12,6 +12,8 @@ class _PostgresConnectionTile extends StatefulWidget { this.onTap, this.onPostgresObjectSelected, this.onPostgresOpenSqlWorkspace, + this.isExpanded = false, + this.onExpandedChanged, }); final ConnectionRow connection; @@ -28,6 +30,8 @@ class _PostgresConnectionTile extends StatefulWidget { PostgresObjectKind kind, )? onPostgresObjectSelected; final OnPostgresOpenSqlWorkspace? onPostgresOpenSqlWorkspace; + final bool isExpanded; + final ValueChanged? onExpandedChanged; @override State<_PostgresConnectionTile> createState() => @@ -35,18 +39,39 @@ class _PostgresConnectionTile extends StatefulWidget { } class _PostgresConnectionTileState extends State<_PostgresConnectionTile> { - bool _expanded = false; + bool get _expanded => widget.isExpanded; bool _loading = false; String? _error; List _databases = []; - void _toggle() { - setState(() => _expanded = !_expanded); - if (_expanded && _databases.isEmpty && !_loading) { + @override + void initState() { + super.initState(); + if (widget.isExpanded) { _loadDatabases(); } } + @override + void didUpdateWidget(_PostgresConnectionTile oldWidget) { + super.didUpdateWidget(oldWidget); + if (widget.isExpanded && !oldWidget.isExpanded) { + if (_databases.isEmpty && !_loading) { + _loadDatabases(); + } + } else if (!widget.isExpanded && oldWidget.isExpanded) { + setState(() { + _databases = []; + _loading = false; + _error = null; + }); + } + } + + void _toggle() { + widget.onExpandedChanged?.call(!widget.isExpanded); + } + Future _loadDatabases() async { if (!mounted) return; setState(() { diff --git a/lib/features/connections/connections_panel_redis.dart b/lib/features/connections/connections_panel_redis.dart index efee18af..cc6d687c 100644 --- a/lib/features/connections/connections_panel_redis.dart +++ b/lib/features/connections/connections_panel_redis.dart @@ -11,6 +11,8 @@ class _RedisConnectionTile extends StatefulWidget { required this.onRemove, this.onTap, this.onDatabaseTap, + this.isExpanded = false, + this.onExpandedChanged, }); final ConnectionRow connection; @@ -20,25 +22,48 @@ class _RedisConnectionTile extends StatefulWidget { final VoidCallback onRemove; final VoidCallback? onTap; final void Function(int database)? onDatabaseTap; + final bool isExpanded; + final ValueChanged? onExpandedChanged; @override State<_RedisConnectionTile> createState() => _RedisConnectionTileState(); } class _RedisConnectionTileState extends State<_RedisConnectionTile> { - bool _expanded = false; + bool get _expanded => widget.isExpanded; bool _loading = false; String? _error; // All 16 databases (db0–db15) with key counts List<({int index, int keys})> _databases = []; - void _toggle() { - setState(() => _expanded = !_expanded); - if (_expanded && _databases.isEmpty && !_loading) { + @override + void initState() { + super.initState(); + if (widget.isExpanded) { _loadDatabases(); } } + @override + void didUpdateWidget(_RedisConnectionTile oldWidget) { + super.didUpdateWidget(oldWidget); + if (widget.isExpanded && !oldWidget.isExpanded) { + if (_databases.isEmpty && !_loading) { + _loadDatabases(); + } + } else if (!widget.isExpanded && oldWidget.isExpanded) { + setState(() { + _databases = []; + _loading = false; + _error = null; + }); + } + } + + void _toggle() { + widget.onExpandedChanged?.call(!widget.isExpanded); + } + Future _loadDatabases() async { if (!mounted) return; setState(() { diff --git a/lib/features/connections/connections_panel_sqlite.dart b/lib/features/connections/connections_panel_sqlite.dart index d02cd014..b403fb6b 100644 --- a/lib/features/connections/connections_panel_sqlite.dart +++ b/lib/features/connections/connections_panel_sqlite.dart @@ -15,6 +15,8 @@ class _SqliteConnectionTile extends StatefulWidget { this.onTap, this.onSqliteObjectSelected, this.onSqliteOpenSqlWorkspace, + this.isExpanded = false, + this.onExpandedChanged, }); final ConnectionRow connection; @@ -29,25 +31,49 @@ class _SqliteConnectionTile extends StatefulWidget { SqliteObjectKind kind, )? onSqliteObjectSelected; final void Function(ConnectionRow connection)? onSqliteOpenSqlWorkspace; + final bool isExpanded; + final ValueChanged? onExpandedChanged; @override State<_SqliteConnectionTile> createState() => _SqliteConnectionTileState(); } class _SqliteConnectionTileState extends State<_SqliteConnectionTile> { - bool _expanded = false; + bool get _expanded => widget.isExpanded; bool _loading = false; String? _error; List _tables = []; List _views = []; - void _toggle() { - setState(() => _expanded = !_expanded); - if (_expanded && _tables.isEmpty && _views.isEmpty && !_loading) { + @override + void initState() { + super.initState(); + if (widget.isExpanded) { _loadTables(); } } + @override + void didUpdateWidget(_SqliteConnectionTile oldWidget) { + super.didUpdateWidget(oldWidget); + if (widget.isExpanded && !oldWidget.isExpanded) { + if (_tables.isEmpty && _views.isEmpty && !_loading) { + _loadTables(); + } + } else if (!widget.isExpanded && oldWidget.isExpanded) { + setState(() { + _tables = []; + _views = []; + _loading = false; + _error = null; + }); + } + } + + void _toggle() { + widget.onExpandedChanged?.call(!widget.isExpanded); + } + Future _loadTables() async { if (!mounted) return; setState(() { diff --git a/lib/features/main_screen/main_screen.dart b/lib/features/main_screen/main_screen.dart index fab3619c..2e5b0d1c 100644 --- a/lib/features/main_screen/main_screen.dart +++ b/lib/features/main_screen/main_screen.dart @@ -149,8 +149,45 @@ class _MainScreenState extends State { width: 1, child: Column( children: [ - QueryaWindowTitleBar( - onNewDatabaseConnection: _onNewDatabaseConnectionFromMenu, + ValueListenableBuilder( + valueListenable: _workspace, + builder: (context, workspace, _) { + return QueryaWindowTitleBar( + onNewDatabaseConnection: _onNewDatabaseConnectionFromMenu, + activeConnection: workspace.activeConnection, + isReadOnly: workspace.isReadOnly, + onReadOnlyChanged: () { + _workspace.value = _workspace.value.toggleReadOnly(); + }, + onConnect: () { + final active = workspace.activeConnection; + if (active != null && active.id != null) { + _connectionsPanelKey.currentState?.connect(active.id!); + } + }, + onReconnect: () { + final active = workspace.activeConnection; + if (active != null) { + _connectionsPanelKey.currentState?.reconnect(active); + } + }, + onDisconnect: () { + final active = workspace.activeConnection; + if (active != null) { + _connectionsPanelKey.currentState?.disconnect(active); + } + }, + onDisconnectAll: () { + _connectionsPanelKey.currentState?.disconnectAll(); + }, + onDisconnectOthers: () { + final active = workspace.activeConnection; + if (active != null) { + _connectionsPanelKey.currentState?.disconnectOthers(active); + } + }, + ); + }, ), Divider(height: 1, color: wb.borderSubtle.withValues(alpha: 0.22)), Expanded( @@ -303,6 +340,7 @@ class _MainContentSplitState extends State<_MainContentSplit> { mysqlSqlTabRequestToken: ws.mysqlSqlTabRequestToken, selectedSqliteObject: ws.selectedSqliteObject, sqliteSqlTabRequestToken: ws.sqliteSqlTabRequestToken, + isReadOnly: ws.isReadOnly, onRequestNewConnection: widget.onRequestNewConnection, ); }, diff --git a/lib/features/main_screen/main_screen_workspace_state.dart b/lib/features/main_screen/main_screen_workspace_state.dart index d11f74fa..a973d4a8 100644 --- a/lib/features/main_screen/main_screen_workspace_state.dart +++ b/lib/features/main_screen/main_screen_workspace_state.dart @@ -19,6 +19,7 @@ class MainScreenWorkspaceState { this.mysqlSqlTabRequestToken = 0, this.selectedSqliteObject, this.sqliteSqlTabRequestToken = 0, + this.isReadOnly = false, }); final ConnectionRow? activeConnection; @@ -53,9 +54,27 @@ class MainScreenWorkspaceState { SqliteObjectKind kind })? selectedSqliteObject; final int sqliteSqlTabRequestToken; + final bool isReadOnly; static const empty = MainScreenWorkspaceState(); + MainScreenWorkspaceState toggleReadOnly() { + return MainScreenWorkspaceState( + activeConnection: activeConnection, + activeRedisDb: activeRedisDb, + activeMongoDB: activeMongoDB, + selectedPostgresObject: selectedPostgresObject, + postgresSqlTabRequestToken: postgresSqlTabRequestToken, + postgresSqlEditorContext: postgresSqlEditorContext, + postgresSqlEditorContextToken: postgresSqlEditorContextToken, + selectedMysqlObject: selectedMysqlObject, + mysqlSqlTabRequestToken: mysqlSqlTabRequestToken, + selectedSqliteObject: selectedSqliteObject, + sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: !isReadOnly, + ); + } + MainScreenWorkspaceState selectConnection(ConnectionRow connection) { return MainScreenWorkspaceState( activeConnection: connection, @@ -69,6 +88,7 @@ class MainScreenWorkspaceState { mysqlSqlTabRequestToken: mysqlSqlTabRequestToken, selectedSqliteObject: null, sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: false, ); } @@ -96,6 +116,7 @@ class MainScreenWorkspaceState { mysqlSqlTabRequestToken: mysqlSqlTabRequestToken, selectedSqliteObject: null, sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: isReadOnly, ); } @@ -121,6 +142,7 @@ class MainScreenWorkspaceState { mysqlSqlTabRequestToken: mysqlSqlTabRequestToken, selectedSqliteObject: null, sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: isReadOnly, ); } @@ -144,6 +166,7 @@ class MainScreenWorkspaceState { kind: kind, ), sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: isReadOnly, ); } @@ -160,6 +183,7 @@ class MainScreenWorkspaceState { mysqlSqlTabRequestToken: mysqlSqlTabRequestToken, selectedSqliteObject: null, sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: isReadOnly, ); } @@ -177,6 +201,7 @@ class MainScreenWorkspaceState { mysqlSqlTabRequestToken: mysqlSqlTabRequestToken, selectedSqliteObject: null, sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: isReadOnly, ); } @@ -235,6 +260,7 @@ class MainScreenWorkspaceState { mysqlSqlTabRequestToken: mysqlSqlTabRequestToken, selectedSqliteObject: null, sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: isReadOnly, ); } @@ -251,6 +277,7 @@ class MainScreenWorkspaceState { mysqlSqlTabRequestToken: mysqlSqlTabRequestToken + 1, selectedSqliteObject: null, sqliteSqlTabRequestToken: sqliteSqlTabRequestToken, + isReadOnly: isReadOnly, ); } @@ -267,6 +294,7 @@ class MainScreenWorkspaceState { mysqlSqlTabRequestToken: mysqlSqlTabRequestToken, selectedSqliteObject: null, sqliteSqlTabRequestToken: sqliteSqlTabRequestToken + 1, + isReadOnly: isReadOnly, ); } @@ -284,7 +312,8 @@ class MainScreenWorkspaceState { _mysqlEquals(selectedMysqlObject, other.selectedMysqlObject) && mysqlSqlTabRequestToken == other.mysqlSqlTabRequestToken && _sqliteEquals(selectedSqliteObject, other.selectedSqliteObject) && - sqliteSqlTabRequestToken == other.sqliteSqlTabRequestToken; + sqliteSqlTabRequestToken == other.sqliteSqlTabRequestToken && + isReadOnly == other.isReadOnly; } @override @@ -325,6 +354,7 @@ class MainScreenWorkspaceState { selectedSqliteObject!.kind, ), sqliteSqlTabRequestToken, + isReadOnly, ); } diff --git a/lib/features/main_screen/querya_window_title_bar.dart b/lib/features/main_screen/querya_window_title_bar.dart index 2bfe0e9d..bba9a8e3 100644 --- a/lib/features/main_screen/querya_window_title_bar.dart +++ b/lib/features/main_screen/querya_window_title_bar.dart @@ -1,6 +1,7 @@ import 'package:bitsdojo_window/bitsdojo_window.dart'; import 'package:flutter/material.dart' as material show BuildContext, Container, Icon, Icons, MainAxisSize, Widget; +import 'package:querya_desktop/core/storage/local_db.dart'; import 'package:querya_desktop/core/theme/querya_theme_scope.dart'; import 'package:querya_desktop/features/connections/driver_manager_dialog.dart'; import 'package:querya_desktop/features/settings/preferences_dialog.dart'; @@ -12,9 +13,25 @@ class QueryaWindowTitleBar extends StatelessWidget { const QueryaWindowTitleBar({ super.key, required this.onNewDatabaseConnection, + this.activeConnection, + this.onConnect, + this.onReconnect, + this.onDisconnect, + this.onDisconnectAll, + this.onDisconnectOthers, + this.isReadOnly = false, + this.onReadOnlyChanged, }); final Future Function() onNewDatabaseConnection; + final ConnectionRow? activeConnection; + final VoidCallback? onConnect; + final VoidCallback? onReconnect; + final VoidCallback? onDisconnect; + final VoidCallback? onDisconnectAll; + final VoidCallback? onDisconnectOthers; + final bool isReadOnly; + final VoidCallback? onReadOnlyChanged; @visibleForTesting static Color titleBarBackground(BuildContext context) => @@ -147,39 +164,48 @@ class QueryaWindowTitleBar extends StatelessWidget { ), const MenuDivider(), MenuButton( - enabled: false, + enabled: activeConnection != null, leading: const material.Icon( material.Icons.power_rounded, size: 18), - onPressed: (_) {}, + onPressed: (_) => onConnect?.call(), child: const Text('Connect'), ), MenuButton( + enabled: activeConnection != null, leading: const material.Icon( material.Icons.refresh_rounded, size: 18), - onPressed: (_) {}, + onPressed: (_) => onReconnect?.call(), child: const Text('Invalidate/Reconnect'), ), MenuButton( + enabled: activeConnection != null, leading: const material.Icon( material.Icons.power_off_rounded, size: 18), - onPressed: (_) {}, + onPressed: (_) => onDisconnect?.call(), child: const Text('Disconnect'), ), MenuButton( - onPressed: (_) {}, + onPressed: (_) => onDisconnectAll?.call(), child: const Text('Disconnect All')), MenuButton( - onPressed: (_) {}, + enabled: activeConnection != null, + onPressed: (_) => onDisconnectOthers?.call(), child: const Text('Disconnect Others')), const MenuDivider(), MenuButton( + enabled: activeConnection != null, leading: const material.Icon( material.Icons.lock_outline_rounded, size: 18), - onPressed: (_) {}, + trailing: isReadOnly + ? const material.Icon( + material.Icons.check_rounded, + size: 16) + : null, + onPressed: (_) => onReadOnlyChanged?.call(), child: const Text('Read-only'), ), ], diff --git a/lib/features/main_screen/workspace_panel.dart b/lib/features/main_screen/workspace_panel.dart index 9e5acd11..80d89b72 100644 --- a/lib/features/main_screen/workspace_panel.dart +++ b/lib/features/main_screen/workspace_panel.dart @@ -60,11 +60,13 @@ class WorkspacePanel extends StatefulWidget { this.mysqlSqlTabRequestToken = 0, this.selectedSqliteObject, this.sqliteSqlTabRequestToken = 0, + this.isReadOnly = false, this.onRequestNewConnection, }); /// Currently selected connection from the sidebar. final ConnectionRow? activeConnection; + final bool isReadOnly; /// When set, the user selected a specific Redis database in the sidebar tree. /// null = show stats, non-null = show data explorer for that db. @@ -159,6 +161,7 @@ class _WorkspacePanelState extends State { postgresSqlEditorContextToken: widget.postgresSqlEditorContextToken, sqlTabRequestToken: widget.postgresSqlTabRequestToken, + isReadOnly: widget.isReadOnly, ) : buildPostgresObjectWorkspace( connection: activeConn, @@ -172,6 +175,7 @@ class _WorkspacePanelState extends State { key: ValueKey('mysql_home_${activeConn.id}'), connectionRow: activeConn, sqlTabRequestToken: widget.mysqlSqlTabRequestToken, + isReadOnly: widget.isReadOnly, ) : MysqlTableView( key: ValueKey( @@ -216,6 +220,7 @@ class _WorkspacePanelState extends State { key: ValueKey('sqlite_home_${activeConn.id}'), connectionRow: activeConn, sqlTabRequestToken: widget.sqliteSqlTabRequestToken, + isReadOnly: widget.isReadOnly, ) : SqliteTableView( key: ValueKey( diff --git a/lib/features/mysql/mysql_sql_workspace.dart b/lib/features/mysql/mysql_sql_workspace.dart index 339128c3..42c0aac1 100644 --- a/lib/features/mysql/mysql_sql_workspace.dart +++ b/lib/features/mysql/mysql_sql_workspace.dart @@ -25,9 +25,11 @@ class MysqlSqlWorkspace extends material.StatefulWidget { const MysqlSqlWorkspace({ super.key, required this.connectionRow, + this.isReadOnly = false, }); final ConnectionRow connectionRow; + final bool isReadOnly; @override material.State createState() => _MysqlSqlWorkspaceState(); @@ -66,6 +68,15 @@ class _MysqlSqlWorkspaceState extends material.State { }); } + @override + void didUpdateWidget(covariant MysqlSqlWorkspace oldWidget) { + super.didUpdateWidget(oldWidget); + if (oldWidget.isReadOnly != widget.isReadOnly) { + _lease?.release(); + _lease = null; + } + } + Future _loadWorkspaceSettings() async { final t = await AppSettings.instance.getMysqlSqlStmtTimeoutSeconds(); final rows = await AppSettings.instance.getSqlResultMaxRows(); @@ -94,7 +105,7 @@ class _MysqlSqlWorkspaceState extends material.State { final lease = await MysqlService.instance.acquire( widget.connectionRow, database: _poolDatabaseKey(), - mode: MysqlSessionMode.readWrite, + mode: widget.isReadOnly ? MysqlSessionMode.readOnly : MysqlSessionMode.readWrite, ); if (!mounted) { lease.release(); @@ -116,7 +127,7 @@ class _MysqlSqlWorkspaceState extends material.State { MysqlService.instance.interrupt( widget.connectionRow, database: _poolDatabaseKey(), - mode: MysqlSessionMode.readWrite, + mode: widget.isReadOnly ? MysqlSessionMode.readOnly : MysqlSessionMode.readWrite, ); } _lease?.release(); diff --git a/lib/features/mysql/mysql_workspace_home.dart b/lib/features/mysql/mysql_workspace_home.dart index 81f9f8ad..9005ad9a 100644 --- a/lib/features/mysql/mysql_workspace_home.dart +++ b/lib/features/mysql/mysql_workspace_home.dart @@ -15,9 +15,11 @@ class MysqlWorkspaceHome extends material.StatefulWidget { super.key, required this.connectionRow, this.sqlTabRequestToken = 0, + this.isReadOnly = false, }); final ConnectionRow connectionRow; + final bool isReadOnly; /// Parent increments to switch to the SQL tab (e.g. context menu on connection). final int sqlTabRequestToken; @@ -74,6 +76,14 @@ class _MysqlWorkspaceHomeState extends material.State { child: material.Row( children: [ const Text('MySQL').semiBold().small(), + if (widget.isReadOnly) ...[ + const Gap(6), + material.Icon( + material.Icons.lock_outline_rounded, + size: 14, + color: theme.colorScheme.mutedForeground, + ), + ], const Spacer(), ...List.generate(2, (i) { final labels = ['Server', 'SQL']; @@ -120,6 +130,7 @@ class _MysqlWorkspaceHomeState extends material.State { MysqlSqlWorkspace( key: ValueKey('mysql_sql_${widget.connectionRow.id}'), connectionRow: widget.connectionRow, + isReadOnly: widget.isReadOnly, ), ], ), diff --git a/lib/features/postgresql/postgres_sql_workspace.dart b/lib/features/postgresql/postgres_sql_workspace.dart index f6ecb83f..22bc209f 100644 --- a/lib/features/postgresql/postgres_sql_workspace.dart +++ b/lib/features/postgresql/postgres_sql_workspace.dart @@ -37,9 +37,11 @@ class PostgresSqlWorkspace extends material.StatefulWidget { this.transactionOpenNotifier, this.postgresSqlEditorContext, this.postgresSqlEditorContextToken = 0, + this.isReadOnly = false, }); final ConnectionRow connectionRow; + final bool isReadOnly; /// Updated when transaction state changes (for tab-switch warnings). final material.ValueNotifier? transactionOpenNotifier; @@ -113,6 +115,9 @@ class _PostgresSqlWorkspaceState extends material.State { if (oldWidget.connectionRow.id != widget.connectionRow.id) { _lastAppliedSqlContextToken = -1; } + if (oldWidget.isReadOnly != widget.isReadOnly) { + _dropLease(); + } _syncPostgresSqlTreeContext(); } @@ -178,7 +183,7 @@ class _PostgresSqlWorkspaceState extends material.State { final lease = await PostgresService.instance.acquire( widget.connectionRow, database: db, - mode: PgSessionMode.readWrite, + mode: widget.isReadOnly ? PgSessionMode.readOnly : PgSessionMode.readWrite, ); if (!mounted) { lease.release(); @@ -259,7 +264,7 @@ class _PostgresSqlWorkspaceState extends material.State { PostgresService.instance.interrupt( widget.connectionRow, database: _interruptDatabase ?? _effectiveSessionDatabase(), - mode: PgSessionMode.readWrite, + mode: widget.isReadOnly ? PgSessionMode.readOnly : PgSessionMode.readWrite, ); } _dropLease(); diff --git a/lib/features/postgresql/postgres_workspace_home.dart b/lib/features/postgresql/postgres_workspace_home.dart index e2cea5b7..bcf1b99f 100644 --- a/lib/features/postgresql/postgres_workspace_home.dart +++ b/lib/features/postgresql/postgres_workspace_home.dart @@ -18,9 +18,11 @@ class PostgresWorkspaceHome extends material.StatefulWidget { this.postgresSqlEditorContext, this.postgresSqlEditorContextToken = 0, this.sqlTabRequestToken = 0, + this.isReadOnly = false, }); final ConnectionRow connectionRow; + final bool isReadOnly; /// Set when opening SQL from the tree (e.g. "Open in SQL") to seed session DB + template. final ({ @@ -120,6 +122,14 @@ class _PostgresWorkspaceHomeState child: material.Row( children: [ const Text('PostgreSQL').semiBold().small(), + if (widget.isReadOnly) ...[ + const Gap(6), + material.Icon( + material.Icons.lock_outline_rounded, + size: 14, + color: theme.colorScheme.mutedForeground, + ), + ], const Spacer(), ...List.generate(2, (i) { final labels = ['Server', 'SQL']; @@ -170,6 +180,7 @@ class _PostgresWorkspaceHomeState postgresSqlEditorContext: widget.postgresSqlEditorContext, postgresSqlEditorContextToken: widget.postgresSqlEditorContextToken, + isReadOnly: widget.isReadOnly, ), ], ), diff --git a/lib/features/sqlite/sqlite_sql_workspace.dart b/lib/features/sqlite/sqlite_sql_workspace.dart index 45d38d52..2f34b117 100644 --- a/lib/features/sqlite/sqlite_sql_workspace.dart +++ b/lib/features/sqlite/sqlite_sql_workspace.dart @@ -21,9 +21,11 @@ class SqliteSqlWorkspace extends material.StatefulWidget { const SqliteSqlWorkspace({ super.key, required this.connectionRow, + this.isReadOnly = false, }); final ConnectionRow connectionRow; + final bool isReadOnly; @override material.State createState() => _SqliteSqlWorkspaceState(); @@ -60,6 +62,15 @@ class _SqliteSqlWorkspaceState extends material.State { }); } + @override + void didUpdateWidget(covariant SqliteSqlWorkspace oldWidget) { + super.didUpdateWidget(oldWidget); + if (oldWidget.isReadOnly != widget.isReadOnly) { + _lease?.release(); + _lease = null; + } + } + Future _loadWorkspaceSettings() async { final rows = await AppSettings.instance.getSqlResultMaxRows(); final hist = await AppSettings.instance.getSqlHistoryMaxEntries(); @@ -78,7 +89,7 @@ class _SqliteSqlWorkspaceState extends material.State { _lease = null; final lease = await SqliteService.instance.acquire( widget.connectionRow, - mode: SqliteSessionMode.readWrite, + mode: widget.isReadOnly ? SqliteSessionMode.readOnly : SqliteSessionMode.readWrite, ); if (!mounted) { lease.release(); diff --git a/lib/features/sqlite/sqlite_workspace_home.dart b/lib/features/sqlite/sqlite_workspace_home.dart index 67991e09..94fc438a 100644 --- a/lib/features/sqlite/sqlite_workspace_home.dart +++ b/lib/features/sqlite/sqlite_workspace_home.dart @@ -8,10 +8,12 @@ class SqliteWorkspaceHome extends material.StatefulWidget { super.key, required this.connectionRow, this.sqlTabRequestToken = 0, + this.isReadOnly = false, }); final ConnectionRow connectionRow; final int sqlTabRequestToken; + final bool isReadOnly; @override material.State createState() => _SqliteWorkspaceHomeState(); @@ -33,6 +35,14 @@ class _SqliteWorkspaceHomeState extends material.State { child: material.Row( children: [ const Text('SQLite').semiBold().small(), + if (widget.isReadOnly) ...[ + const Gap(6), + material.Icon( + material.Icons.lock_outline_rounded, + size: 14, + color: theme.colorScheme.mutedForeground, + ), + ], const Spacer(), material.Container( padding: const material.EdgeInsets.symmetric(horizontal: 12, vertical: 8), @@ -50,6 +60,7 @@ class _SqliteWorkspaceHomeState extends material.State { child: SqliteSqlWorkspace( key: ValueKey('sqlite_sql_${widget.connectionRow.id}'), connectionRow: widget.connectionRow, + isReadOnly: widget.isReadOnly, ), ), ], diff --git a/test/features/connections/connections_panel_layout_test.dart b/test/features/connections/connections_panel_layout_test.dart index 3f68e14b..4a67b096 100644 --- a/test/features/connections/connections_panel_layout_test.dart +++ b/test/features/connections/connections_panel_layout_test.dart @@ -246,4 +246,124 @@ void main() { _expectTextCount('Mongo local', 1); }); }); + + group('ConnectionsPanel state control methods', () { + late Directory stateTempDir; + + setUp(() async { + stateTempDir = await Directory.systemTemp.createTemp('querya_conn_state_test_'); + PathProviderPlatform.instance = _FakePathProvider(stateTempDir.path); + await LocalDb.instance.close(); + await LocalDb.instance.addConnection( + ConnectionRow( + type: 'generic', + name: 'Conn 1', + createdAt: _isoNow(), + ), + ); + await LocalDb.instance.addConnection( + ConnectionRow( + type: 'generic', + name: 'Conn 2', + createdAt: _isoNow(), + ), + ); + await FoldersStorage.instance.reload(); + }); + + tearDown(() async { + await LocalDb.instance.close(); + if (await stateTempDir.exists()) { + await stateTempDir.delete(recursive: true); + } + }); + + testWidgets('connect, disconnect, disconnectAll, and disconnectOthers update state', (tester) async { + await tester.pumpWidget( + ShadcnApp( + theme: AppTheme.dark, + home: material.SizedBox.expand( + child: ConnectionsPanel( + skipInitialDbLoadForTest: true, + onPostgresOpenSqlWorkspace: (_, {database, schema, name, kind}) {}, + ), + ), + ), + ); + await tester.pump(); + + final panelState = tester.state( + find.byType(ConnectionsPanel), + ); + + await tester.runAsync(() async { + await panelState.reloadConnectionsFromDb(); + }); + await tester.pump(); + + late final List conns; + await tester.runAsync(() async { + conns = await LocalDb.instance.getConnections(); + }); + final conn1 = conns.firstWhere((c) => c.name == 'Conn 1'); + final conn2 = conns.firstWhere((c) => c.name == 'Conn 2'); + final id1 = conn1.id!; + final id2 = conn2.id!; + + // 1. Initial state + expect(panelState.isConnectionExpanded(id1), false); + expect(panelState.isConnectionExpanded(id2), false); + + // 2. Connect + panelState.connect(id1); + await tester.pump(); + expect(panelState.isConnectionExpanded(id1), true); + expect(panelState.isConnectionExpanded(id2), false); + + // 3. Disconnect Others + panelState.connect(id2); + await tester.pump(); + expect(panelState.isConnectionExpanded(id1), true); + expect(panelState.isConnectionExpanded(id2), true); + + await tester.runAsync(() async { + await panelState.disconnectOthers(conn1); + }); + await tester.pump(); + expect(panelState.isConnectionExpanded(id1), true); + expect(panelState.isConnectionExpanded(id2), false); + + // 4. Disconnect + await tester.runAsync(() async { + await panelState.disconnect(conn1); + }); + await tester.pump(); + expect(panelState.isConnectionExpanded(id1), false); + + // 5. Disconnect All + panelState.connect(id1); + panelState.connect(id2); + await tester.pump(); + expect(panelState.isConnectionExpanded(id1), true); + expect(panelState.isConnectionExpanded(id2), true); + + await tester.runAsync(() async { + await panelState.disconnectAll(); + }); + await tester.pump(); + expect(panelState.isConnectionExpanded(id1), false); + expect(panelState.isConnectionExpanded(id2), false); + + // 6. Reconnect + panelState.connect(id1); + await tester.pump(); + expect(panelState.isConnectionExpanded(id1), true); + + await tester.runAsync(() async { + await panelState.reconnect(conn1); + }); + await tester.pump(const Duration(milliseconds: 100)); + expect(panelState.isConnectionExpanded(id1), true); + }); + }); } diff --git a/test/features/main_screen/main_screen_workspace_state_test.dart b/test/features/main_screen/main_screen_workspace_state_test.dart index 4023ef9e..411eac6c 100644 --- a/test/features/main_screen/main_screen_workspace_state_test.dart +++ b/test/features/main_screen/main_screen_workspace_state_test.dart @@ -166,5 +166,23 @@ void main() { ); expect(a, isNot(c)); }); + + test('read-only state toggle and reset', () { + var state = MainScreenWorkspaceState.empty; + expect(state.isReadOnly, isFalse); + + state = state.toggleReadOnly(); + expect(state.isReadOnly, isTrue); + + state = state.toggleReadOnly(); + expect(state.isReadOnly, isFalse); + + state = state.toggleReadOnly(); + expect(state.isReadOnly, isTrue); + + // Selecting a new connection should reset isReadOnly to false + state = state.selectConnection(mysqlConn); + expect(state.isReadOnly, isFalse); + }); }); } From cc39154fc8ea04cb504209639b1547424fbfecdb Mon Sep 17 00:00:00 2001 From: ZhuchkaTriplesix Date: Mon, 22 Jun 2026 12:42:48 +0300 Subject: [PATCH 11/12] docs: document SQLite support, read-only mode, and connection actions --- docs/user-guide.md | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/docs/user-guide.md b/docs/user-guide.md index 78cc25ee..12997878 100644 --- a/docs/user-guide.md +++ b/docs/user-guide.md @@ -4,9 +4,18 @@ 1. Start the app. 2. Create a connection: **Connection → New Database Connection** (or right-click **Servers** in the tree). -3. Pick **PostgreSQL**, **MySQL**, **Redis**, or **MongoDB** and fill in host, port, and credentials. +3. Pick **PostgreSQL**, **MySQL**, **SQLite**, **Redis**, or **MongoDB**. For SQLite, select the database file; for other databases, fill in host, port, and credentials. + - **Read-only mode**: You can toggle the **Read-only** option to prevent write operations (a lock icon will appear in the workspace). 4. Saved connections appear in the left tree. +## Connection Actions + +The application menu and window title bar provide actions for managing database connections: +- **Connect**: Connect to the selected database connection. +- **Invalidate/Reconnect**: Refresh and re-establish the connection to the database. +- **Disconnect**: Safely close the active session. +- **Read-only**: Toggle read-only mode for the current session. + ## Where data is stored - **Connection list and settings**: local SQLite (`querya.db` under the app support directory). @@ -33,6 +42,6 @@ Preferences (except secrets) live in the same local SQLite file as connection me ## Supported capabilities -High-level feature depth varies by database type. PostgreSQL and MySQL include rich object trees and SQL workspaces; Redis and MongoDB focus on data exploration and commands suitable for day-to-day development. +High-level feature depth varies by database type. PostgreSQL, MySQL, and SQLite include rich object trees and SQL workspaces (with SQLite utilizing local `.db` files); Redis and MongoDB focus on data exploration and commands suitable for day-to-day development. For troubleshooting build/run issues, see the main [README.md](../README.md). From c2d0a5441e141f2287b8406ade885ad5da4f02ef Mon Sep 17 00:00:00 2001 From: ZhuchkaTriplesix Date: Mon, 22 Jun 2026 14:02:46 +0300 Subject: [PATCH 12/12] chore(release): prepare pre-release 0.4.7-a --- CHANGELOG.md | 12 ++++++++++++ pubspec.yaml | 2 +- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 5e691251..b418fe2f 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -7,6 +7,18 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ## [Unreleased] +## [0.4.7-a] - 2026-06-22 + +### Added + +- **Connection Invalidate/Reconnect** — added Invalidate/Reconnect action in the application menu and window title bar. +- **Connection Read-only Mode** — added Read-only mode for PostgreSQL and SQLite database connections to prevent write operations (displays a lock icon in the workspace). +- **Connection Lifecycle Management** — added Connect and Disconnect menu items and corresponding title bar controls. + +### Documentation + +- Added comprehensive setup guides for SQLite, Read-only connection mode, and connection lifecycle controls in the English user guide and Obsidian Russian notes. + ## [0.4.7] - 2026-06-21 Local extension discovery and manifest foundation release. Git tag **`0.4.7`**. diff --git a/pubspec.yaml b/pubspec.yaml index 33f52043..835024d8 100644 --- a/pubspec.yaml +++ b/pubspec.yaml @@ -1,6 +1,6 @@ name: querya_desktop description: Lightweight desktop SQL/NoSQL client. Flutter (Dart). -version: 0.4.7+13 +version: 0.4.7-a