super-graph/serv/cursor.go

package serv

import (
	"bytes"
	"encoding/base64"

	"github.com/dosco/super-graph/crypto"
	"github.com/dosco/super-graph/jsn"
	"github.com/dosco/super-graph/qcode"
)

func encryptCursor(qc *qcode.QCode, data []byte) ([]byte, error) {
	var keys [][]byte

	for _, s := range qc.Selects {
		if s.Paging.Type != qcode.PtOffset {
			var buf bytes.Buffer

			buf.WriteString(s.FieldName)
			buf.WriteString("_cursor")
			keys = append(keys, buf.Bytes())
		}
	}

	if len(keys) == 0 {
		return data, nil
	}

	from := jsn.Get(data, keys)
	to := make([]jsn.Field, len(from))

	for i, f := range from {
		to[i].Key = f.Key

		if f.Value[0] != '"' || f.Value[len(f.Value)-1] != '"' {
			continue
		}

		v, err := crypto.Encrypt(f.Value[1:len(f.Value)-1], &internalKey)
		if err != nil {
			return nil, err
		}

		var buf bytes.Buffer
		buf.WriteByte('"')
		buf.WriteString(base64.StdEncoding.EncodeToString(v))
		buf.WriteByte('"')

		to[i].Value = buf.Bytes()
	}

	var buf bytes.Buffer

	if err := jsn.Replace(&buf, data, from, to); err != nil {
		return nil, err
	}

	return buf.Bytes(), nil
}

func decrypt(data string) ([]byte, error) {
	v, err := base64.StdEncoding.DecodeString(data)
	if err != nil {
		return nil, err
	}
	return crypto.Decrypt(v, &internalKey)
}