81 lines
1.5 KiB
Go
81 lines
1.5 KiB
Go
package cborutil
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"io"
|
|
"math"
|
|
|
|
cbor "github.com/ipfs/go-ipld-cbor"
|
|
ipld "github.com/ipfs/go-ipld-format"
|
|
logging "github.com/ipfs/go-log"
|
|
cbg "github.com/whyrusleeping/cbor-gen"
|
|
)
|
|
|
|
var log = logging.Logger("cborrrpc")
|
|
|
|
const Debug = false
|
|
|
|
func init() {
|
|
if Debug {
|
|
log.Warn("CBOR-RPC Debugging enabled")
|
|
}
|
|
}
|
|
|
|
func WriteCborRPC(w io.Writer, obj interface{}) error {
|
|
if m, ok := obj.(cbg.CBORMarshaler); ok {
|
|
// TODO: impl debug
|
|
return m.MarshalCBOR(w)
|
|
}
|
|
data, err := cbor.DumpObject(obj)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if Debug {
|
|
log.Infof("> %s", hex.EncodeToString(data))
|
|
}
|
|
|
|
_, err = w.Write(data)
|
|
return err
|
|
}
|
|
|
|
func ReadCborRPC(r io.Reader, out interface{}) error {
|
|
if um, ok := out.(cbg.CBORUnmarshaler); ok {
|
|
return um.UnmarshalCBOR(r)
|
|
}
|
|
return cbor.DecodeReader(r, out)
|
|
}
|
|
|
|
func Dump(obj interface{}) ([]byte, error) {
|
|
var out bytes.Buffer
|
|
if err := WriteCborRPC(&out, obj); err != nil {
|
|
return nil, err
|
|
}
|
|
return out.Bytes(), nil
|
|
}
|
|
|
|
// TODO: this is a bit ugly, and this package is not exactly the best place
|
|
func AsIpld(obj interface{}) (ipld.Node, error) {
|
|
if m, ok := obj.(cbg.CBORMarshaler); ok {
|
|
b, err := Dump(m)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return cbor.Decode(b, math.MaxUint64, -1)
|
|
}
|
|
return cbor.WrapObject(obj, math.MaxUint64, -1)
|
|
}
|
|
|
|
func Equals(a cbg.CBORMarshaler, b cbg.CBORMarshaler) (bool, error) {
|
|
ab, err := Dump(a)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
bb, err := Dump(b)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return bytes.Equal(ab, bb), nil
|
|
}
|