Merge branch 'master' of github.com:massCodeIO/massCode

Author: antonreshetov

src/main/db/index.ts

@@ -14,6 +14,19 @@ const isDev = process.env.NODE_ENV === 'development'
 let db: Database.Database | null = null
 let backupTimer: NodeJS.Timeout | null = null
 
+function tableExists(db: Database.Database, table: string): boolean {
+  const row = db
+    .prepare(
+      `
+    SELECT name 
+    FROM sqlite_master 
+    WHERE type='table' AND name=?
+  `,
+    )
+    .get(table)
+  return !!row
+}
+
 export function useDB() {
   if (db)
     return db
@@ -144,15 +157,26 @@ export function clearDB() {
   try {
     const db = useDB()
     const stmt = db.transaction(() => {
-      // Сначала удаляем записи из таблиц со внешними ключами
-      db.prepare('DELETE FROM snippet_tags').run()
-      db.prepare('DELETE FROM snippet_contents').run()
-      db.prepare('DELETE FROM snippets').run()
-      // Затем удаляем записи из основных таблиц
-      db.prepare('DELETE FROM tags').run()
-      db.prepare('DELETE FROM folders').run()
-      // Сбрасываем автоинкрементные счетчики, чтобы id начинались с 1
-      db.prepare('DELETE FROM sqlite_sequence').run()
+      const tables = [
+        // Таблицы со внешними ключами должны быть первыми
+        'snippet_tags',
+        'snippet_contents',
+        'snippets',
+        // Остальные таблицы можно удалить в любом порядке
+        'tags',
+        'folders',
+      ]
+
+      for (const table of tables) {
+        if (tableExists(db, table)) {
+          db.prepare(`DELETE FROM ${table}`).run()
+        }
+      }
+
+      // Сброс автоинкремента — тоже только если таблица есть
+      if (tableExists(db, 'sqlite_sequence')) {
+        db.prepare('DELETE FROM sqlite_sequence').run()
+      }
     })
 
     stmt()

src/main/db/migrate.ts

@@ -48,12 +48,12 @@ export function migrateJsonToSqlite(jsonData: JSONDB) {
         // Миграция папок
         jsonData.folders.forEach((folder) => {
           const result = insertFolderStmt.run(
-            folder.name,
+            folder.name || 'Untitled Folder',
             folder.defaultLanguage || 'plain_text',
             null, // parentId обновим позже
             folder.isOpen ? 1 : 0,
-            folder.createdAt,
-            folder.updatedAt,
+            folder.createdAt || Date.now(),
+            folder.updatedAt || Date.now(),
             folder.icon || null,
             folder.index ?? 0,
           )
@@ -74,9 +74,9 @@ export function migrateJsonToSqlite(jsonData: JSONDB) {
         // Миграция тегов
         jsonData.tags.forEach((tag) => {
           const result = insertTagStmt.run(
-            tag.name,
-            tag.createdAt,
-            tag.updatedAt,
+            tag.name || 'Untitled Tag',
+            tag.createdAt || Date.now(),
+            tag.updatedAt || Date.now(),
           )
           tagIdMap[tag.id] = Number(result.lastInsertRowid)
         })
@@ -86,13 +86,13 @@ export function migrateJsonToSqlite(jsonData: JSONDB) {
           // Определяем новый id папки для сниппета
           const mappedFolderId = folderIdMap[snippet.folderId] || null
           const result = insertSnippetStmt.run(
-            snippet.name,
+            snippet.name || 'Untitled Snippet',
             snippet.description || null,
             mappedFolderId,
             snippet.isDeleted ? 1 : 0,
             snippet.isFavorites ? 1 : 0,
-            snippet.createdAt,
-            snippet.updatedAt,
+            snippet.createdAt || Date.now(),
+            snippet.updatedAt || Date.now(),
           )
           const newSnippetId = Number(result.lastInsertRowid)
           snippetIdMap[snippet.id] = newSnippetId