package pem

import (
	"crypto/ecdsa"
	"crypto/x509"
	"encoding/pem"
	"fmt"
	"os"
)

func LoadECPrivateKey(filename string) (*ecdsa.PrivateKey, error) {
	pb, err := os.ReadFile(filename)
	if err != nil {
		return nil, err
	}

	block, _ := pem.Decode(pb)
	if block == nil {
		return nil, fmt.Errorf("failed to decode pem bytes")
	}

	key, err := x509.ParseECPrivateKey(block.Bytes)
	if err != nil {
		return nil, err
	}

	return key, nil
}

func CertificateFromPemData(data []byte) (*x509.Certificate, error) {
	block, _ := pem.Decode(data)
	if block == nil {
		return nil, fmt.Errorf("Unable to decode PEM")
	}

	crt, err := x509.ParseCertificate(block.Bytes)
	if err != nil {
		return nil, err
	}

	return crt, nil
}

func CertificateRequestFromPemDataRaw(data []byte) (*x509.CertificateRequest, []byte, error) {
	block, _ := pem.Decode(data)
	if block == nil {
		return nil, nil, fmt.Errorf("Unable to decode PEM")
	}

	csr, err := x509.ParseCertificateRequest(block.Bytes)
	if err != nil {
		return nil, nil, err
	}

	return csr, block.Bytes, nil
}

func CertificateRequestFromPemData(data []byte) (*x509.CertificateRequest, error) {
	csr, _, err := CertificateRequestFromPemDataRaw(data)
	return csr, err
}