#!/bin/bash
# ============================================================
# 重置种子账号口令（由 MySQL 容器 initdb 自动执行）
#
# 01-schema.sql 里的密码哈希源自旧库，不能用于开源分发。
# 这里按 MYSQL_ADMIN_PASSWORD 统一重算，登录后请逐个修改。
#
# 哈希算法（见 iotlcm-admin-svr server/internal/library/utils/utils.go）：
#   md5( hex(md5(明文)) + md5(salt) )
# salt 取自每行 sys_user.salt 列。
# ============================================================
set -euo pipefail

if [ -z "${MYSQL_ADMIN_PASSWORD:-}" ]; then
    echo "[initdb] 错误：未设置 MYSQL_ADMIN_PASSWORD" >&2
    exit 1
fi

PWD_MD5=$(printf '%s' "$MYSQL_ADMIN_PASSWORD" | md5sum | awk '{print $1}')

# 逐个账号按其自身 salt 计算哈希（salt 每行不同）
mysql --protocol=socket -uroot -p"${MYSQL_ROOT_PASSWORD}" <<SQL
SET NAMES utf8mb4;
USE \`db_iotlcm\`;

DROP PROCEDURE IF EXISTS sp_reset_all_passwords;
DELIMITER \$\$
CREATE PROCEDURE sp_reset_all_passwords()
BEGIN
  DECLARE done INT DEFAULT 0;
  DECLARE v_id BIGINT;
  DECLARE v_salt VARCHAR(64);
  DECLARE v_salt_md5 VARCHAR(64);
  DECLARE v_final VARCHAR(64);
  DECLARE cur CURSOR FOR SELECT user_id, salt FROM sys_user;
  DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;

  OPEN cur;
  read_loop: LOOP
    FETCH cur INTO v_id, v_salt;
    IF done THEN
      LEAVE read_loop;
    END IF;
    SET v_salt_md5 = MD5(v_salt);
    SET v_final = MD5(CONCAT('${PWD_MD5}', v_salt_md5));
    UPDATE sys_user SET password = v_final WHERE user_id = v_id;
  END LOOP;
  CLOSE cur;
END\$\$
DELIMITER ;

CALL sp_reset_all_passwords();
DROP PROCEDURE sp_reset_all_passwords;
SQL

echo "[initdb] 种子账号口令已统一重置"