package aesx_test

import (
	"fmt"
	"testing"

	"code.yun.ink/pkg/encryptx/aesx"
)

func TestGCM(t *testing.T) {
	key := "example key 1234"
	plaintext := "exampleplaintext"

	// aesx.AesEncrypt(plaintext,key)
	// aesx.AesDecrypt()

	ciphertext, err := aesx.AESCBCEncrypt(key, plaintext)
	fmt.Println(ciphertext, err)

	plaintext, err = aesx.AESCBCDecrypter(key, ciphertext)
	fmt.Println(plaintext, err)
	///GCM
	noncetext := ""
	ciphertext, noncetext, err = aesx.EncryptGCM(key, plaintext)
	fmt.Println(ciphertext, err)

	plaintext, err = aesx.DecryptGCM(key, ciphertext, noncetext)
	fmt.Println(plaintext, err)
}

func TestECB(t *testing.T) {
	key := "example key 1234"
	plaintext := "exampleplaintext"
	
	// 测试ECB加密和解密
	encrypted, err := aesx.EncryptECBHex(plaintext, key)
	if err != nil {
		t.Fatalf("ECB加密失败: %v", err)
	}
	fmt.Printf("ECB加密结果: %s\n", encrypted)

	decrypted, err := aesx.DecryptECBHex(encrypted, key)
	if err != nil {
		t.Fatalf("ECB解密失败: %v", err)
	}
	fmt.Printf("ECB解密结果: %s\n", decrypted)

	if decrypted != plaintext {
		t.Fatalf("ECB解密结果不匹配，原文: %s, 解密结果: %s", plaintext, decrypted)
	}
}

func TestCBC(t *testing.T) {
	key := "example key 1234"
	plaintext := "exampleplaintext"

	// 测试CBC加密和解密
	encrypted, err := aesx.EncryptCBCHex(plaintext, key)
	if err != nil {
		t.Fatalf("CBC加密失败: %v", err)
	}
	fmt.Printf("CBC加密结果: %s\n", encrypted)

	decrypted, err := aesx.DecryptCBCHex(encrypted, key)
	if err != nil {
		t.Fatalf("CBC解密失败: %v", err)
	}
	fmt.Printf("CBC解密结果: %s\n", decrypted)

	if decrypted != plaintext {
		t.Fatalf("CBC解密结果不匹配，原文: %s, 解密结果: %s", plaintext, decrypted)
	}
}

func TestCFB(t *testing.T) {
	key := "example key 1234"
	plaintext := "exampleplaintext"

	// 测试CFB加密和解密
	encrypted, err := aesx.EncryptCFBHex(plaintext, key)
	if err != nil {
		t.Fatalf("CFB加密失败: %v", err)
	}
	fmt.Printf("CFB加密结果: %s\n", encrypted)

	decrypted, err := aesx.DecryptCFBHex(encrypted, key)
	if err != nil {
		t.Fatalf("CFB解密失败: %v", err)
	}
	fmt.Printf("CFB解密结果: %s\n", decrypted)

	if decrypted != plaintext {
		t.Fatalf("CFB解密结果不匹配，原文: %s, 解密结果: %s", plaintext, decrypted)
	}
}

func TestOFB(t *testing.T) {
	key := "example key 1234"
	plaintext := "exampleplaintext"

	// 测试OFB加密和解密
	encrypted, err := aesx.EncryptOFBHex(plaintext, key)
	if err != nil {
		t.Fatalf("OFB加密失败: %v", err)
	}
	fmt.Printf("OFB加密结果: %s\n", encrypted)

	decrypted, err := aesx.DecryptOFBHex(encrypted, key)
	if err != nil {
		t.Fatalf("OFB解密失败: %v", err)
	}
	fmt.Printf("OFB解密结果: %s\n", decrypted)

	if decrypted != plaintext {
		t.Fatalf("OFB解密结果不匹配，原文: %s, 解密结果: %s", plaintext, decrypted)
	}
}