package jwt

import (
	
	
	
	
	
)

var (
	// Sadly this is missing from crypto/ecdsa compared to crypto/rsa
	ErrECDSAVerification = errors.New("crypto/ecdsa: verification error")
)

// SigningMethodECDSA implements the ECDSA family of signing methods.
// Expects *ecdsa.PrivateKey for signing and *ecdsa.PublicKey for verification
type SigningMethodECDSA struct {
	Name      string
	Hash      crypto.Hash
	KeySize   int
	CurveBits int
}

// Specific instances for EC256 and company
var (
	SigningMethodES256 *SigningMethodECDSA
	SigningMethodES384 *SigningMethodECDSA
	SigningMethodES512 *SigningMethodECDSA
)

func init() {
	// ES256
	SigningMethodES256 = &SigningMethodECDSA{"ES256", crypto.SHA256, 32, 256}
	RegisterSigningMethod(SigningMethodES256.Alg(), func() SigningMethod {
		return SigningMethodES256
	})

	// ES384
	SigningMethodES384 = &SigningMethodECDSA{"ES384", crypto.SHA384, 48, 384}
	RegisterSigningMethod(SigningMethodES384.Alg(), func() SigningMethod {
		return SigningMethodES384
	})

	// ES512
	SigningMethodES512 = &SigningMethodECDSA{"ES512", crypto.SHA512, 66, 521}
	RegisterSigningMethod(SigningMethodES512.Alg(), func() SigningMethod {
		return SigningMethodES512
	})
}

func ( *SigningMethodECDSA) () string {
	return .Name
}

// Verify implements token verification for the SigningMethod.
// For this verify method, key must be an ecdsa.PublicKey struct
func ( *SigningMethodECDSA) (,  string,  interface{}) error {
	var  error

	// Decode the signature
	var  []byte
	if ,  = DecodeSegment();  != nil {
		return 
	}

	// Get the key
	var  *ecdsa.PublicKey
	switch k := .(type) {
	case *ecdsa.PublicKey:
		 = 
	default:
		return ErrInvalidKeyType
	}

	if len() != 2*.KeySize {
		return ErrECDSAVerification
	}

	 := big.NewInt(0).SetBytes([:.KeySize])
	 := big.NewInt(0).SetBytes([.KeySize:])

	// Create hasher
	if !.Hash.Available() {
		return ErrHashUnavailable
	}
	 := .Hash.New()
	.Write([]byte())

	// Verify the signature
	if  := ecdsa.Verify(, .Sum(nil), , );  {
		return nil
	}

	return ErrECDSAVerification
}

// Sign implements token signing for the SigningMethod.
// For this signing method, key must be an ecdsa.PrivateKey struct
func ( *SigningMethodECDSA) ( string,  interface{}) (string, error) {
	// Get the key
	var  *ecdsa.PrivateKey
	switch k := .(type) {
	case *ecdsa.PrivateKey:
		 = 
	default:
		return "", ErrInvalidKeyType
	}

	// Create the hasher
	if !.Hash.Available() {
		return "", ErrHashUnavailable
	}

	 := .Hash.New()
	.Write([]byte())

	// Sign the string and return r, s
	if , ,  := ecdsa.Sign(rand.Reader, , .Sum(nil));  == nil {
		 := .Curve.Params().BitSize

		if .CurveBits !=  {
			return "", ErrInvalidKey
		}

		 :=  / 8
		if %8 > 0 {
			 += 1
		}

		// We serialize the outputs (r and s) into big-endian byte arrays
		// padded with zeros on the left to make sure the sizes work out.
		// Output must be 2*keyBytes long.
		 := make([]byte, 2*)
		.FillBytes([0:]) // r is assigned to the first half of output.
		.FillBytes([:])  // s is assigned to the second half of output.

		return EncodeSegment(), nil
	} else {
		return "", 
	}
}