package bson

import (
	"fmt"

	"github.com/keep-network/paillier"
	"gopkg.in/mgo.v2/bson"
)

type SerializablePublicKey paillier.PublicKey

// Serializes PublicKey to BSON
func SerializePublicKey(publicKey *paillier.PublicKey) ([]byte, error) {
	return bson.Marshal(toSerializablePublicKey(publicKey))
}

// Deserializes BSON to PublicKey
func DeserializePublicKey(data []byte) (*paillier.PublicKey, error) {
	serializable := new(SerializablePublicKey)
	if err := bson.Unmarshal(data, serializable); err != nil {
		return nil, err
	}

	return toOriginalPublicKey(serializable), nil
}

func toSerializablePublicKey(publicKey *paillier.PublicKey) *SerializablePublicKey {
	serializable := SerializablePublicKey(*publicKey)
	return &serializable
}

func toOriginalPublicKey(serializable *SerializablePublicKey) *paillier.PublicKey {
	original := paillier.PublicKey(*serializable)
	return &original
}

type dbPublicKey struct {
	N string `bson:",omitempty"`
}

func (publicKey *SerializablePublicKey) GetBSON() (interface{}, error) {
	m := make(map[string]string)
	m["n"] = fmt.Sprintf("%x", publicKey.N)
	return m, nil
}

func (publicKey *SerializablePublicKey) SetBSON(raw bson.Raw) error {
	var err error = nil
	c := new(dbPublicKey)
	raw.Unmarshal(c)

	if c.N != "" {
		publicKey.N, err = fromHex(c.N)
		if err != nil {
			return err
		}
	}

	return err
}