mirror of
https://github.com/wangwangit/qywx-push.git
synced 2026-08-29 04:06:38 +08:00
Initial commit
This commit is contained in:
wangwangit
@@ -0,0 +1,251 @@
|
||||
// 数据库初始化与操作模块
|
||||
// 管理SQLite数据库连接和表结构
|
||||
|
||||
const sqlite3 = require('sqlite3').verbose();
|
||||
const path = require('path');
|
||||
|
||||
class Database {
|
||||
constructor(dbPath) {
|
||||
this.dbPath = dbPath;
|
||||
this.db = null;
|
||||
}
|
||||
|
||||
// 初始化数据库连接和表结构
|
||||
async init() {
|
||||
return new Promise((resolve, reject) => {
|
||||
// 确保数据库目录存在
|
||||
const dbDir = path.dirname(this.dbPath);
|
||||
const fs = require('fs');
|
||||
if (!fs.existsSync(dbDir)) {
|
||||
fs.mkdirSync(dbDir, { recursive: true });
|
||||
}
|
||||
|
||||
// 创建数据库连接
|
||||
this.db = new sqlite3.Database(this.dbPath, (err) => {
|
||||
if (err) {
|
||||
console.error('数据库连接失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
console.log('SQLite数据库连接成功');
|
||||
|
||||
// 创建configurations表
|
||||
this.createTables()
|
||||
.then(() => resolve())
|
||||
.catch(reject);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 创建数据表
|
||||
async createTables() {
|
||||
return new Promise((resolve, reject) => {
|
||||
const createTableSQL = `
|
||||
CREATE TABLE IF NOT EXISTS configurations (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
code TEXT UNIQUE NOT NULL,
|
||||
corpid TEXT NOT NULL,
|
||||
encrypted_corpsecret TEXT NOT NULL,
|
||||
agentid INTEGER NOT NULL,
|
||||
touser TEXT NOT NULL,
|
||||
description TEXT,
|
||||
callback_token TEXT,
|
||||
encrypted_encoding_aes_key TEXT,
|
||||
callback_enabled BOOLEAN DEFAULT 0,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
UNIQUE(corpid, agentid, touser)
|
||||
)
|
||||
`;
|
||||
|
||||
this.db.run(createTableSQL, (err) => {
|
||||
if (err) {
|
||||
console.error('创建表失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
console.log('数据表创建成功');
|
||||
resolve();
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 保存配置
|
||||
async saveConfiguration(config) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const {
|
||||
code, corpid, encrypted_corpsecret, agentid, touser, description,
|
||||
callback_token, encrypted_encoding_aes_key, callback_enabled
|
||||
} = config;
|
||||
const sql = `
|
||||
INSERT INTO configurations (
|
||||
code, corpid, encrypted_corpsecret, agentid, touser, description,
|
||||
callback_token, encrypted_encoding_aes_key, callback_enabled
|
||||
)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
`;
|
||||
|
||||
this.db.run(sql, [
|
||||
code, corpid, encrypted_corpsecret, agentid, touser, description,
|
||||
callback_token, encrypted_encoding_aes_key, callback_enabled || 0
|
||||
], function(err) {
|
||||
if (err) {
|
||||
console.error('保存配置失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
console.log('配置保存成功, ID:', this.lastID);
|
||||
resolve({ id: this.lastID, code });
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 根据code获取配置
|
||||
async getConfigurationByCode(code) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const sql = `SELECT * FROM configurations WHERE code = ?`;
|
||||
|
||||
this.db.get(sql, [code], (err, row) => {
|
||||
if (err) {
|
||||
console.error('查询配置失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
resolve(row);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 更新配置
|
||||
async updateConfiguration(config) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const {
|
||||
code, corpid, encrypted_corpsecret, agentid, touser, description,
|
||||
callback_token, encrypted_encoding_aes_key, callback_enabled
|
||||
} = config;
|
||||
const sql = `
|
||||
UPDATE configurations
|
||||
SET corpid = ?, encrypted_corpsecret = ?, agentid = ?, touser = ?, description = ?,
|
||||
callback_token = ?, encrypted_encoding_aes_key = ?, callback_enabled = ?
|
||||
WHERE code = ?
|
||||
`;
|
||||
this.db.run(sql, [
|
||||
corpid, encrypted_corpsecret, agentid, touser, description,
|
||||
callback_token, encrypted_encoding_aes_key, callback_enabled,
|
||||
code
|
||||
], function(err) {
|
||||
if (err) {
|
||||
console.error('更新配置失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
console.log('配置更新成功, code:', code);
|
||||
resolve({ code });
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 根据字段查询配置
|
||||
async getConfigurationByFields(corpid, agentid, touser) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const sql = `SELECT * FROM configurations WHERE corpid = ? AND agentid = ? AND touser = ?`;
|
||||
this.db.get(sql, [corpid, agentid, touser], (err, row) => {
|
||||
if (err) {
|
||||
console.error('查询配置失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
resolve(row);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 根据完整字段查询配置（包括回调配置）
|
||||
async getConfigurationByCompleteFields(corpid, agentid, touser, callback_enabled, callback_token) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const sql = `SELECT * FROM configurations WHERE corpid = ? AND agentid = ? AND touser = ? AND callback_enabled = ? AND (callback_token = ? OR (callback_token IS NULL AND ? IS NULL))`;
|
||||
this.db.get(sql, [corpid, agentid, touser, callback_enabled, callback_token, callback_token], (err, row) => {
|
||||
if (err) {
|
||||
console.error('查询完整配置失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
resolve(row);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 保存回调配置（第一步）
|
||||
async saveCallbackConfiguration(config) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const { code, corpid, callback_token, encrypted_encoding_aes_key } = config;
|
||||
const sql = `
|
||||
INSERT INTO configurations (
|
||||
code, corpid, callback_token, encrypted_encoding_aes_key, callback_enabled,
|
||||
encrypted_corpsecret, agentid, touser, description
|
||||
)
|
||||
VALUES (?, ?, ?, ?, 1, '', 0, '', '')
|
||||
`;
|
||||
|
||||
this.db.run(sql, [code, corpid, callback_token, encrypted_encoding_aes_key], function(err) {
|
||||
if (err) {
|
||||
console.error('保存回调配置失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
console.log('回调配置保存成功, ID:', this.lastID);
|
||||
resolve({ id: this.lastID, code });
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 查询回调配置
|
||||
async getCallbackConfiguration(corpid, callback_token) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const sql = `SELECT * FROM configurations WHERE corpid = ? AND callback_token = ? AND callback_enabled = 1`;
|
||||
this.db.get(sql, [corpid, callback_token], (err, row) => {
|
||||
if (err) {
|
||||
console.error('查询回调配置失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
resolve(row);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 完善配置（第二步）
|
||||
async completeConfiguration(config) {
|
||||
return new Promise((resolve, reject) => {
|
||||
const { code, encrypted_corpsecret, agentid, touser, description } = config;
|
||||
const sql = `
|
||||
UPDATE configurations
|
||||
SET encrypted_corpsecret = ?, agentid = ?, touser = ?, description = ?
|
||||
WHERE code = ?
|
||||
`;
|
||||
this.db.run(sql, [encrypted_corpsecret, agentid, touser, description, code], function(err) {
|
||||
if (err) {
|
||||
console.error('完善配置失败:', err.message);
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
console.log('配置完善成功, code:', code);
|
||||
resolve({ code });
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// 关闭数据库连接
|
||||
close() {
|
||||
if (this.db) {
|
||||
this.db.close((err) => {
|
||||
if (err) {
|
||||
console.error('关闭数据库失败:', err.message);
|
||||
} else {
|
||||
console.log('数据库连接已关闭');
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = Database;
|
||||
Reference in New Issue
Block a user