73 lines
1.4 KiB
Go
73 lines
1.4 KiB
Go
package core
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
|
|
"github.com/dosco/super-graph/core/internal/crypto"
|
|
"github.com/dosco/super-graph/jsn"
|
|
"github.com/dosco/super-graph/core/internal/qcode"
|
|
)
|
|
|
|
func (sg *SuperGraph) 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
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
|
|
if len(f.Value) > 2 {
|
|
v, err := crypto.Encrypt(f.Value[1:len(f.Value)-1], &sg.encKey)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
buf.WriteByte('"')
|
|
buf.WriteString(base64.StdEncoding.EncodeToString(v))
|
|
buf.WriteByte('"')
|
|
} else {
|
|
buf.WriteString(`null`)
|
|
}
|
|
|
|
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 (sg *SuperGraph) decrypt(data string) ([]byte, error) {
|
|
v, err := base64.StdEncoding.DecodeString(data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return crypto.Decrypt(v, &sg.encKey)
|
|
}
|