feat: add cherry-stuido-db project

packages/database/src/agents.js

@@ -0,0 +1,45 @@
+const sqlite3 = require('sqlite3').verbose()
+const fs = require('fs')
+
+// 连接到数据库
+const db = new sqlite3.Database('./data/CherryStudio.sqlite3', (err) => {
+  if (err) {
+    console.error('Error connecting to the database:', err.message)
+    return
+  }
+  console.log('Connected to the database.')
+})
+
+// 查询数据并转换为JSON
+db.all('SELECT * FROM agents', [], (err, rows) => {
+  if (err) {
+    console.error('Error querying the database:', err.message)
+    return
+  }
+
+  // 将 ID 类型转换为字符串
+  for (const row of rows) {
+    row.id = row.id.toString()
+  }
+
+  // 将查询结果转换为JSON字符串
+  const jsonData = JSON.stringify(rows, null, 2)
+
+  // 将JSON数据写入文件
+  fs.writeFile('../../src/renderer/src/config/agents.json', jsonData, (err) => {
+    if (err) {
+      console.error('Error writing to file:', err.message)
+      return
+    }
+    console.log('Data has been written to agents.json')
+  })
+
+  // 关闭数据库连接
+  db.close((err) => {
+    if (err) {
+      console.error('Error closing the database:', err.message)
+      return
+    }
+    console.log('Database connection closed.')
+  })
+})

packages/database/src/csv.js

@@ -0,0 +1,77 @@
+const fs = require('fs')
+const csv = require('csv-parser')
+const sqlite3 = require('sqlite3').verbose()
+
+// 连接到 SQLite 数据库
+const db = new sqlite3.Database('./data/CherryStudio.sqlite3', (err) => {
+  if (err) {
+    console.error('Error opening database', err)
+    return
+  }
+  console.log('Connected to the SQLite database.')
+})
+
+// 创建一个数组来存储 CSV 数据
+const results = []
+
+// 读取 CSV 文件
+fs.createReadStream('./data/data.csv')
+  .pipe(csv())
+  .on('data', (data) => results.push(data))
+  .on('end', () => {
+    // 准备 SQL 插入语句，使用 INSERT OR IGNORE
+    const stmt = db.prepare('INSERT OR IGNORE INTO emails (email, github, sent) VALUES (?, ?, ?)')
+
+    // 插入每一行数据
+    let inserted = 0
+    let skipped = 0
+    let emptyEmail = 0
+
+    db.serialize(() => {
+      // 开始一个事务以提高性能
+      db.run('BEGIN TRANSACTION')
+
+      results.forEach((row) => {
+        // 检查 email 是否为空
+        if (!row.email || row.email.trim() === '') {
+          emptyEmail++
+          return // 跳过这一行
+        }
+
+        stmt.run(row.email, row['user-href'], 0, function (err) {
+          if (err) {
+            console.error('Error inserting row', err)
+          } else {
+            if (this.changes === 1) {
+              inserted++
+            } else {
+              skipped++
+            }
+          }
+        })
+      })
+
+      // 提交事务
+      db.run('COMMIT', (err) => {
+        if (err) {
+          console.error('Error committing transaction', err)
+        } else {
+          console.log(
+            `Insertion complete. Inserted: ${inserted}, Skipped (duplicate): ${skipped}, Skipped (empty email): ${emptyEmail}`
+          )
+        }
+
+        // 完成插入
+        stmt.finalize()
+
+        // 关闭数据库连接
+        db.close((err) => {
+          if (err) {
+            console.error('Error closing database', err)
+          } else {
+            console.log('Database connection closed.')
+          }
+        })
+      })
+    })
+  })

packages/database/src/email.js

@@ -0,0 +1,36 @@
+const sqlite3 = require('sqlite3').verbose()
+
+// 连接到数据库
+const db = new sqlite3.Database('./data/CherryStudio.sqlite3', (err) => {
+  if (err) {
+    console.error('Error connecting to the database:', err.message)
+    return
+  }
+})
+
+// 查询数据并转换为JSON
+db.all('SELECT * FROM emails WHERE sent = 0', [], (err, rows) => {
+  if (err) {
+    console.error('Error querying the database:', err.message)
+    return
+  }
+
+  for (const row of rows) {
+    console.log(row.email)
+    // Update row set sent = 1
+    db.run('UPDATE emails SET sent = 1 WHERE id = ?', [row.id], (err) => {
+      if (err) {
+        console.error('Error updating the database:', err.message)
+        return
+      }
+    })
+  }
+
+  // 关闭数据库连接
+  db.close((err) => {
+    if (err) {
+      console.error('Error closing the database:', err.message)
+      return
+    }
+  })
+})