telco_server/test/gen_test.go

package test

import (
	"crypto/rand"
	"crypto/rsa"
	"crypto/x509"
	"encoding/base64"
	"encoding/pem"
	"fmt"
	"go-admin/app/admin/models/bus_models"
	"go-admin/tools/crypto"
	"io"
	"os"

	"testing"
	//"text/template"
)

func TestGoModelTemplate(t *testing.T) {
	//t1, err := template.ParseFiles("model.go.template")
	//if err != nil {
	//	t.Error(err)
	//}
	//table := tools.SysTables{}
	//table.TBName = "sys_tables"
	//tab, err := table.Get()
	//if err != nil {
	//	t.Error(err)
	//}
	//file, err := os.Create("models/" + table.PackageName + ".go")
	//if err != nil {
	//	t.Error(err)
	//}
	//defer file.Close()
	//
	//_ = t1.Execute(file, tab)
	//t.Log("")

	// 加密
	data := "{\"import_serial_number\":\"8766f066-2880-48f0-a295-121f981a779b\",\"phone_list\":[\"15019236666\"],\"sms_content\":\"123\",\"send_time\":\"2025-04-17T15:30:00+08:00\"}"
	encryptedData, err := crypto.AESEncryptJson(bus_models.AESKey, data)
	fmt.Println("加密后的文本:", encryptedData)

	// 解密
	decryptedText, err := crypto.AESDecryptJson(bus_models.AESKey, encryptedData)
	if err != nil {
		fmt.Println("解密失败:", err)
		return
	}
	fmt.Println("解密后的文本:", decryptedText)

	//// 解密
	//encryptedText := "C0ItV2oeoDcr+T0ySc0ol5ux2uDiE3KAB6I0oqX8P6lb4FPaY4eQhB/BlEvQM9YnSeGmV/xJVtpai2tyHsg25K/zuaep24FPjDY/KZ4T0iwmTRpDREI34FjLT2QY/NHh7DFdgVfloWkW3KkuAcEoQuEWUZOajHObXyQKcdiYeBgONM8tyuCAU+fd9oC2lUImuOU8x50xsccSkouqliit8LhSNtlnnlRfUhMuPJ36sIa4+3IBgCtHrXaFv1X9L3a15xQXtq8xhmTLSNo3KUHvo1YsjvWbyjgERlmBby6avq3tmCehXhqUMAk0QUMZQBaWwRbDCpCEk6F+ZF/eUz+3eMCeLPj2dK7L71iIIQ49tjgiSW9QTALHXOt1vCExKcDHXC8lCJ3wbcI1jj4hRylu4M4W6JJxb1an2n6JPViiwes6uTFDjOBWxEk53G64oo2UdCdlHvFfPgVMUqrHJ2cXH4CwXomv3hMXRA+KCDWUaCCQo9jQnH5Q11VUBn+7Ji08MIDNcBGwLP8E54YQSslLtOxLsWLIZWMV/MYB/QmXz+yeLprt+qL9DZNBvrk/pDjBXKL+haO3034qxsbx8vOiPe4Utcmh4mZqCZJqyDzQdTo="
	//
	//decryptedText, err := crypto.AESDecryptJson(bus_models.AESKey, encryptedText)
	//if err != nil {
	//	fmt.Println("解密失败:", err)
	//	return
	//}
	//fmt.Println("解密后的文本:", decryptedText)

}

func TestGoApiTemplate(t *testing.T) {
	//t1, err := template.ParseFiles("api.go.template")
	//if err != nil {
	//	t.Error(err)
	//}
	//table := tools.SysTables{}
	//table.TBName = "sys_tables"
	//tab, _ := table.Get()
	//file, err := os.Create("apis/" + table.PackageName + ".go")
	//if err != nil {
	//	t.Error(err)
	//}
	//defer file.Close()
	//
	//_ = t1.Execute(file, tab)
	//t.Log("")

	dir, err := os.Getwd()
	if err != nil {
		fmt.Println("Error getting current directory:", err)
	}
	fmt.Println("Current working directory:", dir)

	// 生成 AES 密钥
	aesKey, err := GenerateAESKey()
	if err != nil {
		fmt.Printf("生成 AES 密钥失败: %v", err)
	}

	// 打印 AES 密钥
	fmt.Printf("生成的 AES 密钥: %x\n", aesKey)

	publicKeyPath := "/Users/max/Documents/code/deovo/telco_server/config/sms/public.pem"

	// 使用公钥加密 AES 密钥
	encryptedKey, err := EncryptWithRSA(publicKeyPath, aesKey)
	if err != nil {
		fmt.Printf("加密 AES 密钥失败: %v", err)
	}

	// 打印加密后的 AES 密钥（Base64 编码）
	fmt.Printf("加密后的 AES 密钥（Base64 编码）: %s\n", encryptedKey)
}

// GenerateAESKey 生成随机的 AES 密钥
func GenerateAESKey() ([]byte, error) {
	// 生成 256 位（32 字节）的 AES 密钥
	key := make([]byte, 32)
	_, err := rand.Read(key)
	if err != nil {
		return nil, err
	}
	return key, nil
}

// EncryptWithRSA 公钥加密 AES 密钥
func EncryptWithRSA(publicKeyPath string, aesKey []byte) (string, error) {
	// 读取公钥文件
	pubFile, err := os.Open(publicKeyPath)
	if err != nil {
		return "", err
	}
	defer pubFile.Close()

	// 解析公钥
	pubBytes, err := io.ReadAll(pubFile)
	if err != nil {
		return "", err
	}

	block, _ := pem.Decode(pubBytes)
	if block == nil {
		return "", fmt.Errorf("failed to parse PEM block containing the public key")
	}

	pubKey, err := x509.ParsePKCS1PublicKey(block.Bytes)
	if err != nil {
		return "", err
	}

	// 使用公钥加密 AES 密钥
	encryptedKey, err := rsa.EncryptPKCS1v15(rand.Reader, pubKey, aesKey)
	if err != nil {
		return "", err
	}

	// 返回加密后的密钥（base64 编码）
	return base64.StdEncoding.EncodeToString(encryptedKey), nil
}