package wg_core

import (
	"errors"
	"fmt"
	"os"

	"gitee.com/carlmax_my/console-core-go/pkg/file"
	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
)

func (p *WgPeer) GetWgTypesKeyBytes(encodedKey string) (*wgtypes.Key, error) {
	key, err := wgtypes.ParseKey(encodedKey)
	if err != nil {
		return nil, err
	}
	return &key, nil
}
func (p *WgPeer) GetWgTypesPublicKeyBytes() (*wgtypes.Key, error) {
	return p.GetWgTypesKeyBytes(p.PublicKey)
}
func (p *WgPeer) GetWgTypesPSKeyBytes() (*wgtypes.Key, error) {
	return p.GetWgTypesKeyBytes(p.PSK)
}

func (p *WgInterface) GetWgTypesKeyBytes(encodedKey string) (*wgtypes.Key, error) {
	key, err := wgtypes.ParseKey(encodedKey)
	if err != nil {
		return nil, err
	}
	return &key, nil
}
func (p *WgInterface) GetWgTypesPrivateKeyBytes() (*wgtypes.Key, error) {
	return p.GetWgTypesKeyBytes(p.PrivateKey)
}

func ReadPrivateKey(keyFile string) (*wgtypes.Key, error) {
	if _, b := file.IsExists(keyFile); b {
		if bytes, err := os.ReadFile(keyFile); err != nil {
			return nil, err
		} else {
			keyPair, err := wgtypes.ParseKey(string(bytes))
			if err != nil {
				return nil, err
			}

			return &keyPair, nil
		}
	}

	return nil, errors.New("key file not exist")
}

func GenerateKeyPairByPath(keyFile string) (privateKey, publicKey string, err error) {
	var keyPair wgtypes.Key

	if keyFile != "" {
		kp, err := ReadPrivateKey(keyFile)
		if err != nil {
			return "", "", fmt.Errorf("ReadPrivateKey error:%+v", err.Error())
		}
		keyPair = *kp
	} else {
		keyPair, err = wgtypes.GeneratePrivateKey()
		if err != nil {
			return "", "", fmt.Errorf("generate private key error:%s", err.Error())
		}
	}

	privateKey = keyPair.String()
	publicKey = keyPair.PublicKey().String()

	return
}

func GenerateKeyPair() (privateKey, publicKey string, err error) {
	keyPair, err := wgtypes.GeneratePrivateKey()
	if err != nil {
		return "", "", fmt.Errorf("generate private key error:%s", err.Error())
	}

	privateKey = keyPair.String()
	publicKey = keyPair.PublicKey().String()

	return
}