package utils_test

import (
	"crypto/md5"
	"encoding/base64"
	"encoding/hex"
	"fmt"
	"net"
	"testing"

	"next-terminal/server/utils"

	"github.com/stretchr/testify/assert"
)

func TestTcping(t *testing.T) {
	localhost4 := "127.0.0.1"
	localhost6 := "::1"
	conn, err := net.Listen("tcp", ":9999")
	assert.NoError(t, err)
	ip4resfalse := utils.Tcping(localhost4, 22)
	assert.Equal(t, false, ip4resfalse)

	ip4res := utils.Tcping(localhost4, 9999)
	assert.Equal(t, true, ip4res)

	ip6res := utils.Tcping(localhost6, 9999)
	assert.Equal(t, true, ip6res)

	ip4resWithBracket := utils.Tcping("["+localhost4+"]", 9999)
	assert.Equal(t, true, ip4resWithBracket)

	ip6resWithBracket := utils.Tcping("["+localhost6+"]", 9999)
	assert.Equal(t, true, ip6resWithBracket)

	defer func() {
		_ = conn.Close()
	}()
}

func TestAesEncryptCBC(t *testing.T) {
	origData := []byte("Hello Next Terminal") // 待加密的数据
	key := []byte("qwertyuiopasdfgh")         // 加密的密钥
	encryptedCBC, err := utils.AesEncryptCBC(origData, key)
	assert.NoError(t, err)
	assert.Equal(t, "s2xvMRPfZjmttpt+x0MzG9dsWcf1X+h9nt7waLvXpNM=", base64.StdEncoding.EncodeToString(encryptedCBC))
}

func TestAesDecryptCBC(t *testing.T) {
	origData, err := base64.StdEncoding.DecodeString("s2xvMRPfZjmttpt+x0MzG9dsWcf1X+h9nt7waLvXpNM=") // 待解密的数据
	assert.NoError(t, err)
	key := []byte("qwertyuiopasdfgh") // 解密的密钥
	decryptCBC, err := utils.AesDecryptCBC(origData, key)
	assert.NoError(t, err)
	assert.Equal(t, "Hello Next Terminal", string(decryptCBC))
}

func TestPbkdf2(t *testing.T) {
	pbkdf2, err := utils.Pbkdf2("1234")
	assert.NoError(t, err)
	println(hex.EncodeToString(pbkdf2))
}

func TestAesEncryptCBCWithAnyKey(t *testing.T) {
	origData := []byte("admin")                                        // 待加密的数据
	key := []byte(fmt.Sprintf("%x", md5.Sum([]byte("next-terminal")))) // 加密的密钥
	encryptedCBC, err := utils.AesEncryptCBC(origData, key)
	assert.NoError(t, err)
	assert.Equal(t, "3qwawlPxghyiLS5hdr/p0g==", base64.StdEncoding.EncodeToString(encryptedCBC))
}

func TestAesDecryptCBCWithAnyKey(t *testing.T) {
	origData, err := base64.StdEncoding.DecodeString("3qwawlPxghyiLS5hdr/p0g==") // 待解密的数据
	assert.NoError(t, err)
	key := []byte(fmt.Sprintf("%x", md5.Sum([]byte("next-terminal")))) // 加密的密钥
	decryptCBC, err := utils.AesDecryptCBC(origData, key)
	assert.NoError(t, err)
	assert.Equal(t, "admin", string(decryptCBC))
}