68 lines
1.2 KiB
Go
68 lines
1.2 KiB
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] < '0' || f.Value[0] > '9' {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
v, err := crypto.Encrypt(f.Value, &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)
|
||
|
|
}
|