rest.go

package parse

import (
	"compress/gzip"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"reflect"
	"strings"
	"time"
)

const (
	SchemeHttp         = "http"
	SchemeHttps        = "https"
	AppIdHeader        = "X-Parse-Application-Id"
	RestKeyHeader      = "X-Parse-REST-API-Key"
	MasterKeyHeader    = "X-Parse-Master-Key"
	SessionTokenHeader = "X-Parse-Session-Token"
	UserAgentHeader    = "User-Agent"
)

var fieldNameCache map[reflect.Type]map[string]string = make(map[reflect.Type]map[string]string)
var fieldCache = make(map[reflect.Type]reflect.StructField)

type requestT interface {
	method() string
	endpoint() (string, error)
	body() (string, error)
	useMasterKey() bool
	session() *sessionT
	contentType() string
}

type ParseError interface {
	error
	Code() int
	Message() string
}

type parseErrorT struct {
	ErrorCode    int    `json:"code" parse:"code"`
	ErrorMessage string `json:"error" parse:"error"`
}

func (e *parseErrorT) Error() string {
	return fmt.Sprintf("error %d - %s", e.ErrorCode, e.ErrorMessage)
}

func (e *parseErrorT) Code() int {
	return e.ErrorCode
}

func (e *parseErrorT) Message() string {
	return e.ErrorMessage
}

// ClientConfig parsesdk配置
type ClientConfig struct {
	Schema     string
	Host       string
	PathPrefix string
	Version    string
	AppID      string
	RestKey    string
	MasterKey  string
}

type clientT struct {
	userAgent  string
	config     *ClientConfig
	httpClient *http.Client
	limiter    limiter
}

var defaultClient *clientT
var isInit = false

// Initialize the parse library with your API keys
func Initialize(config *ClientConfig) {
	if config == nil {
		panic("Parse Config Can't be nil")
	}
	if config.Schema == "" ||
		config.Host == "" ||
		config.AppID == "" {
		panic("Parse Config filed 'Schema' 'Host' 'AppID' can not be empty")
	}
	defaultClient = &clientT{
		config:     config,
		httpClient: &http.Client{},
	}
}

// Initialize2 the parse library with your API keys
func Initialize2(config map[string]interface{}) {
	if config == nil {
		panic("Parse Config Can't be nil")
	}

	clientConfig := &ClientConfig{
		Schema:     config["Schema"].(string),
		Host:       config["Host"].(string),
		PathPrefix: config["PathPrefix"].(string),
		Version:    config["Version"].(string),
		AppID:      config["AppID"].(string),
		RestKey:    config["RestKey"].(string),
		MasterKey:  config["MasterKey"].(string),
	}
	Initialize(clientConfig)
}

// Set the timeout for requests to Parse
//
// Returns an error if called before parse.Initialize
func SetHTTPTimeout(t time.Duration) error {
	if defaultClient == nil {
		return errors.New("parse.Initialize must be called before parse.SetHTTPTimeout")
	}

	defaultClient.httpClient.Timeout = t
	return nil
}

// Set the User Agent to be specified for requests against Parse
//
// Returns an error if called before parse.Initialize
func SetUserAgent(ua string) error {
	if defaultClient == nil {
		return errors.New("parse.Initialize must be called before parse.SetUserAgent")
	}

	defaultClient.userAgent = ua
	return nil
}

// Set the maximum number of requests per second, with an optional
// burst rate.
//
// Returns an error if called before parse.Initialize
//
// If this option is set, this library will restrict calling code to
// a maximum number of requests per second. Requests exceeding this limit
// will block for the appropriate period of time.
func SetRateLimit(limit, burst uint) error {
	if defaultClient == nil {
		return errors.New("parse.Initialize must be called before parse.SetHTTPTimeout")
	}

	defaultClient.limiter = newRateLimiter(limit, burst)
	return nil
}

func SetHTTPClient(c *http.Client) error {
	if defaultClient == nil {
		return errors.New("parse.Initialize must be called before parse.SetHTTPTimeout")
	}

	defaultClient.httpClient = c
	return nil
}

func (c *clientT) doRequest(op requestT) ([]byte, error) {
	ep, err := op.endpoint()
	if err != nil {
		return nil, err
	}

	method := op.method()
	Debug("%s %s", method, ep)
	var body io.Reader
	if method == "POST" || method == "PUT" {
		b, err := op.body()
		if err != nil {
			return nil, err
		}

		body = strings.NewReader(b)
	}

	req, err := http.NewRequest(method, ep, body)
	if err != nil {
		return nil, err
	}

	req.Header.Add(UserAgentHeader, defaultClient.userAgent)
	Debug("%s : %s", UserAgentHeader, defaultClient.userAgent)
	req.Header.Add(AppIdHeader, defaultClient.config.AppID)
	Debug("%s : %s", AppIdHeader, defaultClient.config.AppID)
	if op.useMasterKey() && c.config.MasterKey != "" && op.session() == nil {
		req.Header.Add(MasterKeyHeader, c.config.MasterKey)
		Debug("%s : %s", MasterKeyHeader, c.config.MasterKey)
	} else {
		req.Header.Add(RestKeyHeader, c.config.RestKey)
		Debug("%s : %s", RestKeyHeader, c.config.RestKey)
		if s := op.session(); s != nil {
			req.Header.Add(SessionTokenHeader, s.sessionToken)
			Debug("%s : %s", SessionTokenHeader, s.sessionToken)
		}
	}

	if c := op.contentType(); c != "" {
		req.Header.Add("Content-Type", op.contentType())
		Debug("%s : %s", "Content-Type", op.contentType())
	}
	req.Header.Add("Accept-Encoding", "gzip")
	Debug("%s : %s", "Accept-Encoding", "gzip")
	b, _ := op.body()
	Debug("%s", b)

	if c.limiter != nil {
		c.limiter.limit()
	}

	resp, err := defaultClient.httpClient.Do(req)
	if err != nil {
		return nil, err
	}

	defer resp.Body.Close()
	var reader io.ReadCloser
	switch resp.Header.Get("Content-Encoding") {
	case "gzip":
		if r, err := gzip.NewReader(resp.Body); err != nil {
			return nil, err
		} else {
			reader = r
		}
	default:
		reader = resp.Body
	}

	respBody, err := ioutil.ReadAll(reader)
	if err != nil {
		return nil, err
	}

	// Error formats are consistent. If the response is an error,
	// return a ParseError
	if !(resp.StatusCode >= 200 && resp.StatusCode < 300) {
		ret := parseErrorT{}
		if err := json.Unmarshal(respBody, &ret); err != nil {
			return nil, err
		}
		Debug("%s", string(respBody))
		return nil, &ret
	}
	Debug("%s", string(respBody))
	return respBody, nil
}

func handleResponse(body []byte, dst interface{}) error {
	rv := reflect.ValueOf(dst)
	if rv.Kind() != reflect.Ptr || rv.IsNil() {
		return errors.New("v must be a non-nil pointer")
	}

	data := make(map[string]interface{})
	if err := json.Unmarshal(body, &data); err != nil {
		return err
	}

	if c, ok := data["count"]; ok {
		return populateValue(dst, c)
	} else if r, ok := data["results"]; ok {
		if rl, ok := r.([]interface{}); ok && len(rl) == 0 {
			return ErrNoRows
		}

		// Handle query results
		return populateValue(dst, r)
	} else {
		return populateValue(dst, data)
	}
}

func getFields(t reflect.Type) []reflect.StructField {
	fields := make([]reflect.StructField, 0)

	if t.Kind() == reflect.Ptr {
		t = t.Elem()
	}

	for i := 0; i < t.NumField(); i++ {
		f := t.Field(i)
		ft := f.Type
		if ft.Kind() == reflect.Ptr {
			ft = ft.Elem()
		}
		switch ft.Kind() {
		case reflect.Struct:
			if f.Anonymous {
				fields = append(fields, getFields(ft)...)
			} else if len(f.PkgPath) == 0 {
				fields = append(fields, f)
			}
		default:
			if len(f.PkgPath) == 0 {
				fields = append(fields, f)
			}
		}
	}

	return fields
}

func getFieldNameMap(v reflect.Value) map[string]string {
	// Get the actual type we care about. Indirect any pointers, and handle
	ind := reflect.Indirect(v)
	t := ind.Type()
	if t.Kind() == reflect.Slice || t.Kind() == reflect.Array {
		t = t.Elem()
		if t.Kind() == reflect.Ptr {
			t = t.Elem()
		}
	}

	if f, ok := fieldNameCache[t]; ok {
		return f
	}

	fields := getFields(t)

	fieldMap := make(map[string]string)
	for _, f := range fields {
		if name, _ := parseTag(f.Tag.Get("parse")); name != "" && name != "-" {
			fieldMap[name] = f.Name
		}
	}

	fieldNameCache[t] = fieldMap
	return fieldMap
}

func populateValue(dst interface{}, src interface{}) (err error) {
	defer func() {
		if r := recover(); r != nil {
			if e, ok := r.(error); ok {
				err = e
			} else {
				err = fmt.Errorf("error populating struct: %v", r)
			}
		}
	}()

	dv := reflect.ValueOf(dst)
	dvi := reflect.Indirect(dv)

	if src == nil {
		dvi.Set(reflect.Zero(dvi.Type()))
		return nil
	}

	sv := reflect.ValueOf(src)
	svi := reflect.Indirect(sv)

	switch dvi.Kind() {
	case reflect.Slice, reflect.Array:
		if sv.Kind() == reflect.Slice || sv.Kind() == reflect.Array {
			dt := dvi.Type().Elem()
			dvi.Set(reflect.MakeSlice(reflect.SliceOf(dt), 0, sv.Len()))
			for i := 0; i < sv.Len(); i++ {
				var newV reflect.Value
				if dt.Kind() == reflect.Ptr {
					newV = reflect.New(dt.Elem())
				} else {
					newV = reflect.New(dt)
				}

				err := populateValue(newV.Interface(), sv.Index(i).Interface())
				if err != nil {
					return err
				}
				if dt.Kind() == reflect.Ptr {
					dvi = reflect.Append(dvi, newV)
				} else {
					dvi = reflect.Append(dvi, reflect.Indirect(newV))
				}
				dv.Elem().Set(dvi)
			}
		} else {
			return fmt.Errorf("expected slice, got %s", sv.Kind())
		}
	case reflect.Struct: // TODO: Handle other Parse object types ?
		if dvi.Type() == reflect.TypeOf(time.Time{}) || dvi.Type() == reflect.TypeOf(Date{}) {
			if s, ok := src.(string); ok {
				if t, err := parseTime(s); err != nil {
					return err
				} else {
					dvi.Set(reflect.ValueOf(t).Convert(dvi.Type()))
				}
			} else if m, ok := src.(map[string]interface{}); ok {
				if t, ok := m["__type"]; ok {
					if t == "Date" {
						if ds, ok := m["iso"]; ok {
							if t, err := parseTime(ds.(string)); err != nil {
								return err
							} else {
								dvi.Set(reflect.ValueOf(t).Convert(dvi.Type()))
							}
						} else {
							return fmt.Errorf("malformed Date type: %v", m)
						}
					} else {
						return fmt.Errorf("expected Date type got %s", t)
					}
				} else {
					return fmt.Errorf("no __type in object: %v", m)
				}
			} else if svi.Type().ConvertibleTo(dvi.Type()) {
				dvi.Set(sv.Convert(dvi.Type()))
			} else {
				return fmt.Errorf("expected string or Date type, got %s", sv.Type())
			}
		} else if svi.Kind() == reflect.Map {
			fieldNameMap := getFieldNameMap(dvi)
			if m, ok := src.(map[string]interface{}); ok {
				if f := dvi.FieldByName("Extra"); f.IsValid() && f.CanSet() && f.IsNil() {
					f.Set(reflect.ValueOf(make(map[string]interface{})))
				}

				for k, v := range m {
					if k == "__type" || k == "className" {
						continue
					}

					if nk, ok := fieldNameMap[k]; ok {
						k = nk
					}

					k1 := firstToUpper(k)

					if f := dvi.FieldByName(k1); f.IsValid() && v != nil {
						if f.Kind() == reflect.Ptr {
							if f.IsNil() {
								f.Set(reflect.New(f.Type().Elem()))
							}
						}

						fi := reflect.Indirect(f)
						if fi.CanSet() {
							var err error
							if f.Kind() == reflect.Ptr {
								err = populateValue(f.Interface(), v)
							} else {
								fptr := f.Addr()
								err = populateValue(fptr.Interface(), v)
							}
							if err != nil {
								return fmt.Errorf("can not set field %s - %s", k, err)
							}
						}
					} else if f := dvi.FieldByName("Extra"); f.IsValid() && f.Kind() == reflect.Map {
						f.SetMapIndex(reflect.ValueOf(k), reflect.ValueOf(v))
					}
				}
			} else {
				return fmt.Errorf("expected map[string]interface{} got %s", sv.Type())
			}
		} else if svi.Type().AssignableTo(dvi.Type()) {
			dvi.Set(svi)
		} else if p, ok := src.(Pointer); ok {
			newv := reflect.New(dvi.Type())
			newvi := reflect.Indirect(newv)
			if f := newvi.FieldByName("Id"); f.CanSet() {
				f.Set(reflect.ValueOf(p.Id))
			}
			return populateValue(dst, newv.Interface())
		} else {
			return fmt.Errorf("expected map, got %s", svi.Kind())
		}
	case reflect.Interface:
		if _, ok := dst.(*ACL); ok {
			if a, ok := src.(map[string]interface{}); ok {
				acl := aclT{
					read:  map[string]bool{},
					write: map[string]bool{},
				}
				for ak, av := range a {
					avm := av.(map[string]interface{})
					if ak == "*" {
						if r, ok := avm["read"]; ok && r.(bool) {
							acl.publicReadAccess = true
						}
						if w, ok := avm["write"]; ok && w.(bool) {
							acl.publicWriteAccess = true
						}
					} else {
						if r, ok := avm["read"]; ok && r.(bool) {
							acl.read[ak] = true
						}
						if w, ok := avm["write"]; ok && w.(bool) {
							acl.write[ak] = true
						}
					}
				}
				dvi.Set(reflect.ValueOf(&acl))
			} else if _, ok := src.(ACL); ok {
				dvi.Set(sv)
			} else {
				return fmt.Errorf("can not set field ACL - expected type map[string]interface{} - got: %v", reflect.TypeOf(src))
			}
		} else if m, ok := src.(map[string]interface{}); ok {
			if c, ok := m["className"]; ok {
				if t, ok := registeredTypes[c.(string)]; ok {
					tv := reflect.New(t)
					if err := populateValue(tv.Interface(), src); err != nil {
						return err
					}
					dvi.Set(tv)
					return nil
				}
			} else if t, ok := m["__type"]; ok && t == "File" {
				f := File{}
				if err := populateValue(&f, m); err != nil {
					return err
				}
				dvi.Set(reflect.ValueOf(&f))
				return nil
			}
		}
		fallthrough
	default:
		if dvi.Kind() == reflect.Ptr {
			if dvi.IsNil() {
				dvi = reflect.New(dvi.Type())
			}

			dvi = dvi.Elem()
		}

		if sv.Type().AssignableTo(dvi.Type()) {
			if dvi.CanSet() {
				dvi.Set(sv)
			}
			return nil
		} else if sv.Type().ConvertibleTo(dvi.Type()) {
			newV := sv.Convert(dvi.Type())
			if dvi.CanSet() {
				dvi.Set(newV)
			}
			return nil
		}
	}

	return nil
}

func parseTime(s string) (time.Time, error) {
	return time.Parse(time.RFC3339Nano, s)
}