Improved error handling

DatabaseService->GetSnippets
MainViewModel->InitializeAsync

Author: mx7b7

src/CodeSnip/MainViewModel.cs

@@ -237,11 +237,20 @@ public async Task InitializeAsync()
             {
                 var languages = await Task.Run(() => _databaseService.GetSnippets());
 
-                PopulateLanguagesCollection(languages);
+                var languageList = languages.ToList();
 
-                if (settingsService.LastSnippet != null)
+                if (languageList.Count == 0 && _databaseService.GetSnippets().Any()) // Check if loading failed
                 {
-                    RestoreSelectedSnippetState(settingsService.LastSnippet);
+                    await DialogService.Instance.ShowMessageAsync("Database Load Error",
+                        "Could not load snippets. The database file might be corrupted or the schema might have changed.");
+                }
+                else
+                {
+                    PopulateLanguagesCollection(languageList);
+                    if (settingsService.LastSnippet != null)
+                    {
+                        RestoreSelectedSnippetState(settingsService.LastSnippet);
+                    }
                 }
             }
             await UpdateDatabaseHealthStatusAsync();

src/CodeSnip/Services/DatabaseService.cs

@@ -128,9 +128,11 @@ INSERT INTO Snippets (CategoryId, Title, Code, Description, Tag)
 
         public IEnumerable<Language> GetSnippets()
         {
-            using var conn = CreateConnection();
+            try
+            {
+                using var conn = CreateConnection();
 
-            var sql = @"
+                var sql = @"
  SELECT 
     L.ID AS LanguageID,
     L.Code AS LanguageCode,
@@ -151,61 +153,67 @@ FROM Languages L
 LEFT JOIN Snippets S ON S.CategoryId = C.ID
 ORDER BY L.Name, C.Name, S.Title";
 
-            var lookup = new Dictionary<int, Language>();
-            //var result = conn.Query(sql).ToList();
-            var result = conn.Query<dynamic>(sql).ToList();
+                var lookup = new Dictionary<int, Language>();
+                //var result = conn.Query(sql).ToList();
+                var result = conn.Query<dynamic>(sql).ToList();
 
-            foreach (var row in result)
-            {
-                // Find or create a language
-                int langId = (int)row.LanguageID;
-                if (!lookup.TryGetValue(langId, out var language))
-                {
-                    language = new Language
-                    {
-                        Id = langId,
-                        Code = (string)row.LanguageCode,
-                        Name = (string)row.LanguageName,
-                        Categories = new ObservableCollection<Category>() // Collection initialization
-                    };
-                    lookup.Add(langId, language);
-                }
-                // Pronađi ili kreiraj kategoriju
-                if (row.CategoryID != null)
+                foreach (var row in result)
                 {
-                    int catId = (int)row.CategoryID;
-                    var category = language.Categories.FirstOrDefault(c => c.Id == catId);
-                    if (category == null)
+                    // Find or create a language
+                    int langId = (int)row.LanguageID;
+                    if (!lookup.TryGetValue(langId, out var language))
                     {
-                        category = new Category
+                        language = new Language
                         {
-                            Id = catId,
-                            LanguageId = (int)row.CategoryLanguageId,
-                            Name = (string)row.CategoryName,
-                            Language = language,
-                            Snippets = new ObservableCollection<Snippet>() // Collection initialization
+                            Id = langId,
+                            Code = (string)row.LanguageCode,
+                            Name = (string)row.LanguageName,
+                            Categories = new ObservableCollection<Category>() // Collection initialization
                         };
-                        language.Categories.Add(category);
+                        lookup.Add(langId, language);
                     }
-                    // Find or create a snippet
-                    if (row.SnippetID != null)
+                    // Find or create a category
+                    if (row.CategoryID != null)
                     {
-                        var snippet = new Snippet
+                        int catId = (int)row.CategoryID;
+                        var category = language.Categories.FirstOrDefault(c => c.Id == catId);
+                        if (category == null)
                         {
-                            Id = (int)row.SnippetID,
-                            CategoryId = (int)row.SnippetCategoryId,
-                            Title = (string)row.SnippetTitle,
-                            //Code = (string)row.SnippetCode,
-                            Code = string.Empty, // Initially empty
-                            Description = (string)row.SnippetDescription,
-                            Tag = (string)row.SnippetTag,
-                            Category = category
-                        };
-                        category.Snippets.Add(snippet);
+                            category = new Category
+                            {
+                                Id = catId,
+                                LanguageId = (int)row.CategoryLanguageId,
+                                Name = (string)row.CategoryName,
+                                Language = language,
+                                Snippets = new ObservableCollection<Snippet>() // Collection initialization
+                            };
+                            language.Categories.Add(category);
+                        }
+                        // Find or create a snippet
+                        if (row.SnippetID != null)
+                        {
+                            var snippet = new Snippet
+                            {
+                                Id = (int)row.SnippetID,
+                                CategoryId = (int)row.SnippetCategoryId,
+                                Title = (string)row.SnippetTitle,
+                                //Code = (string)row.SnippetCode,
+                                Code = string.Empty, // Initially empty
+                                Description = (string)row.SnippetDescription,
+                                Tag = (string)row.SnippetTag,
+                                Category = category
+                            };
+                            category.Snippets.Add(snippet);
+                        }
                     }
                 }
+                return lookup.Values;
+            }
+            catch (Exception ex)
+            {
+                System.Diagnostics.Debug.WriteLine($"[FATAL] Failed to load snippets from database: {ex}");
+                return Enumerable.Empty<Language>(); // Return an empty list on failure
             }
-            return lookup.Values;
         }
 
         public string GetSnippetCode(int snippetId)