package main

import (
	"fmt"
	"golang.org/x/crypto/bcrypt"
)

// GeneratePassword 生成bcrypt加密密码
func GeneratePassword(password string) (string, error) {
	hashedPassword, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
	if err != nil {
		return "", err
	}
	return string(hashedPassword), nil
}

func main() {
	// 为测试数据生成加密密码
	passwords := []string{
		"admin123",  // 企业管理员密码
		"user123",   // 普通用户密码
	}

	fmt.Println("生成加密密码：")
	fmt.Println("=====================================")
	
	for i, pwd := range passwords {
		hashed, err := GeneratePassword(pwd)
		if err != nil {
			fmt.Printf("生成密码失败: %v\n", err)
			continue
		}
		fmt.Printf("%d. 原始密码: %s\n", i+1, pwd)
		fmt.Printf("   加密后: %s\n\n", hashed)
	}
	
	fmt.Println("=====================================")
	fmt.Println("使用说明：")
	fmt.Println("1. 复制上面的加密密码")
	fmt.Println("2. 在 test_data_ai_wht.sql 中替换对应的密码占位符")
	fmt.Println("3. 执行 SQL 文件导入测试数据")
}