super-graph/serv/args.go

144 lines
2.8 KiB
Go
Raw Normal View History

2019-04-19 07:55:03 +02:00
package serv
import (
2019-09-05 06:09:56 +02:00
"bytes"
2019-11-25 08:22:33 +01:00
"context"
2019-12-25 07:24:30 +01:00
"encoding/json"
2019-11-25 08:22:33 +01:00
"errors"
2019-09-05 06:09:56 +02:00
"fmt"
2019-04-19 07:55:03 +02:00
"io"
2019-09-05 06:09:56 +02:00
"github.com/dosco/super-graph/jsn"
2019-04-19 07:55:03 +02:00
)
2019-11-25 08:22:33 +01:00
func argMap(ctx context.Context, vars []byte) func(w io.Writer, tag string) (int, error) {
2019-09-05 06:09:56 +02:00
return func(w io.Writer, tag string) (int, error) {
switch tag {
case "user_id_provider":
if v := ctx.Value(userIDProviderKey); v != nil {
return io.WriteString(w, v.(string))
}
2019-11-25 08:22:33 +01:00
return 0, errors.New("query requires variable $user_id_provider")
2019-09-05 06:09:56 +02:00
case "user_id":
if v := ctx.Value(userIDKey); v != nil {
return io.WriteString(w, v.(string))
2019-09-05 06:09:56 +02:00
}
2019-11-25 08:22:33 +01:00
return 0, errors.New("query requires variable $user_id")
case "user_role":
if v := ctx.Value(userRoleKey); v != nil {
return io.WriteString(w, v.(string))
2019-09-05 06:09:56 +02:00
}
2019-11-25 08:22:33 +01:00
return 0, errors.New("query requires variable $user_role")
2019-04-19 07:55:03 +02:00
}
2020-01-15 05:16:55 +01:00
fmt.Println("1>", tag)
fmt.Println("2>", string(vars))
2019-11-25 08:22:33 +01:00
fields := jsn.Get(vars, [][]byte{[]byte(tag)})
2020-01-15 05:16:55 +01:00
fmt.Println("2.1>", fields)
2019-09-05 06:09:56 +02:00
if len(fields) == 0 {
2019-11-25 08:22:33 +01:00
return 0, nil
2019-09-05 06:09:56 +02:00
}
v := fields[0].Value
if len(v) >= 2 && v[0] == '"' && v[len(v)-1] == '"' {
fields[0].Value = v[1 : len(v)-1]
}
2019-04-19 07:55:03 +02:00
2020-01-15 05:16:55 +01:00
fmt.Println("3>", string(fields[0].Value))
2020-01-14 07:02:12 +01:00
return w.Write(escQuote(fields[0].Value))
2019-04-19 07:55:03 +02:00
}
}
2019-07-29 07:13:33 +02:00
2019-11-25 08:22:33 +01:00
func argList(ctx *coreContext, args [][]byte) ([]interface{}, error) {
2019-09-05 06:09:56 +02:00
vars := make([]interface{}, len(args))
2019-12-25 07:24:30 +01:00
var fields map[string]json.RawMessage
2019-09-05 06:09:56 +02:00
var err error
2019-07-29 07:13:33 +02:00
2019-09-05 06:09:56 +02:00
if len(ctx.req.Vars) != 0 {
fields, _, err = jsn.Tree(ctx.req.Vars)
if err != nil {
2019-11-25 08:22:33 +01:00
return nil, err
2019-09-05 06:09:56 +02:00
}
2019-07-29 07:13:33 +02:00
}
for i := range args {
2019-09-05 06:09:56 +02:00
av := args[i]
2019-07-29 07:13:33 +02:00
2019-09-05 06:09:56 +02:00
switch {
case bytes.Equal(av, []byte("user_id")):
2019-07-29 07:13:33 +02:00
if v := ctx.Value(userIDKey); v != nil {
2019-09-05 06:09:56 +02:00
vars[i] = v.(string)
2019-11-25 08:22:33 +01:00
} else {
return nil, errors.New("query requires variable $user_id")
2019-07-29 07:13:33 +02:00
}
2019-09-05 06:09:56 +02:00
case bytes.Equal(av, []byte("user_id_provider")):
2019-07-29 07:13:33 +02:00
if v := ctx.Value(userIDProviderKey); v != nil {
2019-09-05 06:09:56 +02:00
vars[i] = v.(string)
2019-11-25 08:22:33 +01:00
} else {
return nil, errors.New("query requires variable $user_id_provider")
2019-07-29 07:13:33 +02:00
}
case bytes.Equal(av, []byte("user_role")):
if v := ctx.Value(userRoleKey); v != nil {
vars[i] = v.(string)
2019-11-25 08:22:33 +01:00
} else {
return nil, errors.New("query requires variable $user_role")
}
2019-09-05 06:09:56 +02:00
default:
if v, ok := fields[string(av)]; ok {
2019-12-25 07:24:30 +01:00
switch v[0] {
case '[', '{':
2020-01-14 07:02:12 +01:00
vars[i] = escQuote(v)
2019-12-25 07:24:30 +01:00
default:
var val interface{}
if err := json.Unmarshal(v, &val); err != nil {
return nil, err
}
vars[i] = val
}
2019-11-25 08:22:33 +01:00
} else {
return nil, fmt.Errorf("query requires variable $%s", string(av))
2019-07-29 07:13:33 +02:00
}
}
}
2019-11-25 08:22:33 +01:00
return vars, nil
2019-07-29 07:13:33 +02:00
}
2020-01-14 07:02:12 +01:00
func escQuote(b []byte) []byte {
f := false
for i := range b {
if b[i] == '\'' {
f = true
break
}
}
if !f {
return b
}
buf := &bytes.Buffer{}
s := 0
for i := range b {
if b[i] == '\'' {
buf.Write(b[s:i])
buf.WriteString(`''`)
s = i + 1
}
}
l := len(b)
if s < (l - 1) {
buf.Write(b[s:l])
}
return buf.Bytes()
}