FullAutoWaterCheck/全自动水压检测仪/DATA/DatabaseInitializer.cs

using Dapper;
using MySql.Data.MySqlClient;
using System;
using System.Diagnostics;

namespace 全自动水压检测仪.DATA
{
    /// <summary>
    /// 数据库初始化辅助类
    /// 用于创建用户表和初始化默认管理员账户
    /// </summary>
    public static class DatabaseInitializer
    {
        private static readonly string _connectionString = "Server=localhost;Database=fullautowaterpressure;User=root;Password=123456;port=3306;charset=utf8;";

        /// <summary>
        /// 初始化用户表和默认管理员账户
        /// </summary>
        public static void Initialize()
        {
            try
            {
                CreateUsersTable();
                CreateDefaultAdminIfNotExists();
            }
            catch (Exception ex)
            {
                throw new Exception($"数据库初始化失败：{ex.Message}", ex);
            }
        }

        /// <summary>
        /// 强制重置默认用户（删除并重新创建）
        /// </summary>
        public static void ForceResetDefaultUsers()
        {
            using (var connection = new MySqlConnection(_connectionString))
            {
                connection.Open();

                try
                {
                    // 删除现有的默认用户
                    connection.Execute("DELETE FROM sys_users WHERE username IN ('admin', 'csi')");
                    Debug.WriteLine("已删除现有默认用户");

                    // 重新创建
                    CreateUserIfNotExists(connection, "admin", "admin123", 1);
                    CreateUserIfNotExists(connection, "csi", "123456", 0);

                    Debug.WriteLine("默认用户重置完成！");
                }
                catch (Exception ex)
                {
                    Debug.WriteLine($"重置用户失败: {ex.Message}");
                    throw;
                }
            }
        }

        /// <summary>
        /// 创建用户表
        /// </summary>
        private static void CreateUsersTable()
        {
            using (var connection = new MySqlConnection(_connectionString))
            {
                connection.Open();

                // 先检查表是否存在，如果不存在则创建
                string checkTableSql = @"SELECT COUNT(*) FROM information_schema.tables
                                      WHERE table_schema = @database AND table_name = 'sys_users'";
                int tableExists = connection.ExecuteScalar<int>(checkTableSql, new { database = "fullautowaterpressure" });

                if (tableExists == 0)
                {
                    string sql = @"
                        CREATE TABLE `sys_users` (
                          `id` INT NOT NULL AUTO_INCREMENT COMMENT '用户ID',
                          `username` VARCHAR(50) NOT NULL COMMENT '用户名',
                          `password_hash` VARCHAR(255) NOT NULL COMMENT '密码哈希值',
                          `salt` VARCHAR(64) NOT NULL COMMENT '盐值',
                          `user_role` TINYINT NOT NULL DEFAULT 0 COMMENT '用户角色：0=普通用户，1=管理员',
                          `status` TINYINT NOT NULL DEFAULT 1 COMMENT '状态：0=禁用，1=启用',
                          `create_time` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
                          `update_time` DATETIME DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
                          `last_login_time` DATETIME DEFAULT NULL COMMENT '最后登录时间',
                          PRIMARY KEY (`id`),
                          UNIQUE KEY `uk_username` (`username`)
                        ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='系统用户表'";

                    connection.Execute(sql);
                }
            }
        }

        /// <summary>
        /// 创建默认管理员账户（如果不存在）
        /// 默认账户：admin / admin123
        /// </summary>
        private static void CreateDefaultAdminIfNotExists()
        {
            using (var connection = new MySqlConnection(_connectionString))
            {
                connection.Open();

                try
                {
                    // 先检查 sys_users 表是否存在
                    string checkTableSql = @"SELECT COUNT(*) FROM information_schema.tables
                                           WHERE table_schema = @database AND table_name = 'sys_users'";
                    int tableExists = connection.ExecuteScalar<int>(checkTableSql, new { database = "fullautowaterpressure" });

                    if (tableExists == 0)
                    {
                        Debug.WriteLine("sys_users table does not exist, skipping admin creation");
                        return;
                    }

                    // 创建默认管理员账户 admin / admin123
                    CreateUserIfNotExists(connection, "admin", "admin123", 1);

                    // 创建默认普通用户账户 csi / 123456
                    CreateUserIfNotExists(connection, "csi", "123456", 0);
                }
                catch (Exception ex)
                {
                    // 捕获其他异常
                    Debug.WriteLine($"✗ Error creating default users: {ex.GetType().Name} - {ex.Message}");
                }
            }
        }

        /// <summary>
        /// 创建用户（如果不存在）
        /// </summary>
        /// <param name="connection">数据库连接</param>
        /// <param name="username">用户名</param>
        /// <param name="password">密码（明文）</param>
        /// <param name="userRole">用户角色：0=普通用户，1=管理员</param>
        private static void CreateUserIfNotExists(MySqlConnection connection, string username, string password, int userRole)
        {
            try
            {
                // 检查用户是否已存在
                string checkUserSql = "SELECT COUNT(*) FROM `sys_users` WHERE `username` = @username";
                int userExists = connection.ExecuteScalar<int>(checkUserSql, new { username });

                if (userExists > 0)
                {
                    Debug.WriteLine($"用户 '{username}' 已存在，跳过创建");
                    return;
                }

                // 简单明文密码存储
                // password_hash 字段存储明文密码
                // salt 字段存储空字符串（保持兼容性）
                string passwordHash = password;  // 直接使用明文
                string salt = "";  // 空字符串

                // 调试输出
                Debug.WriteLine($"\n=== 创建用户: {username} ===");
                Debug.WriteLine($"  密码: {password}");
                Debug.WriteLine($"  角色: {(userRole == 1 ? "管理员" : "普通用户")}");

                // 插入用户
                string insertSql = @"
                    INSERT INTO `sys_users`
                    (`username`, `password_hash`, `salt`, `user_role`, `status`, `create_time`)
                    VALUES (@username, @passwordHash, @salt, @userRole, @status, NOW())";

                int rowsAffected = connection.Execute(insertSql, new
                {
                    username = username,
                    passwordHash = passwordHash,
                    salt = salt,
                    userRole = userRole,
                    status = 1  // 启用
                });

                if (rowsAffected > 0)
                {
                    string roleText = userRole == 1 ? "管理员" : "普通用户";
                    Debug.WriteLine($"✓ 用户创建成功: {username} / {password} ({roleText})");
                }
                else
                {
                    Debug.WriteLine($"✗ 创建用户失败: {username}");
                }
            }
            catch (MySqlException ex)
            {
                Debug.WriteLine($"✗ MySQL错误 (创建用户 '{username}'): [{ex.Number}] {ex.Message}");
                
                if (ex.Number == 1062)
                {
                    Debug.WriteLine($"用户 '{username}' 已存在（重复键）");
                }
            }
            catch (Exception ex)
            {
                Debug.WriteLine($"✗ 异常 (创建用户 '{username}'): {ex.Message}");
            }
        }
    }
}