package utils

import (
	"encoding/base64"

	"github.com/golang/protobuf/proto"
)

func MarshalBytes(req proto.Message) ([]byte, error) {
	data, err := proto.Marshal(req)
	return data, err
}
func UnmarshalBytes(pdata []byte, req proto.Message) error {
	err := proto.Unmarshal(pdata, req)
	return err
}

// take a proto and convert it into a base64 string
func Marshal(req proto.Message) (string, error) {
	data, err := proto.Marshal(req)
	if err != nil {
		return "", err
	}
	b64 := base64.StdEncoding.EncodeToString(data)
	return b64, nil
}

// take a base64 string and convert it into the proto
func Unmarshal(b64string string, req proto.Message) error {
	pdata, err := base64.StdEncoding.DecodeString(b64string)
	if err != nil {
		return err
	}
	err = proto.Unmarshal(pdata, req)
	if err != nil {
		return err
	}

	return nil
}

// write a proto to disk
func WriteProto(filename string, req proto.Message) error {
	data, err := proto.Marshal(req)
	if err != nil {
		return err
	}
	err = WriteFile(filename, data)
	if err != nil {
		return err
	}
	return nil
}

// read a proto from disk
func ReadProto(filename string, req proto.Message) error {
	b, err := ReadFile(filename)
	if err != nil {
		return err
	}
	err = UnmarshalBytes(b, req)
	if err != nil {
		return err
	}
	return nil
}