surrealpatch/web/rpc.go

// Copyright © 2016 Abcum Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package web

import (
	"github.com/abcum/fibre"
	"github.com/abcum/surreal/cnf"
	"github.com/abcum/surreal/db"
	"github.com/abcum/surreal/sql"
	"github.com/abcum/surreal/util/data"
	"github.com/abcum/surreal/util/rand"
)

type rpc struct{}

func (r *rpc) Uniq(c *fibre.Context) (interface{}, error) {
	return rand.String(128), nil
}

func (r *rpc) Info(c *fibre.Context) (interface{}, error) {
	return c.Get("auth").(*cnf.Auth).Data, nil
}

func (r *rpc) Auth(c *fibre.Context, auth string) (interface{}, error) {
	return c.Get("auth").(*cnf.Auth).Data, checkBearer(c, auth, ignore)
}

func (r *rpc) Let(c *fibre.Context, key string, val interface{}) (interface{}, error) {
	return c.Get("keep").(*data.Doc).Set(val, key)
}

func (r *rpc) Query(c *fibre.Context, sql string, vars map[string]interface{}) (interface{}, error) {
	return db.Execute(c, sql, vars)
}

func (r *rpc) Kill(c *fibre.Context, query string) (interface{}, error) {
	return db.Execute(c, "KILL $query", map[string]interface{}{
		"query": query,
	})
}

func (r *rpc) Live(c *fibre.Context, class string) (interface{}, error) {
	return db.Execute(c, "LIVE SELECT * FROM $class", map[string]interface{}{
		"class": sql.NewTable(class),
	})
}

func (r *rpc) Select(c *fibre.Context, class string, thing interface{}) (interface{}, error) {
	switch thing := thing.(type) {
	case *fibre.RPCNull:
		return db.Execute(c, "SELECT * FROM $class", map[string]interface{}{
			"class": sql.NewTable(class),
		})
	case []interface{}:
		return db.Execute(c, "SELECT * FROM $batch", map[string]interface{}{
			"batch": sql.NewBatch(class, thing),
		})
	default:
		return db.Execute(c, "SELECT * FROM $thing", map[string]interface{}{
			"thing": sql.NewThing(class, thing),
		})
	}
}

func (r *rpc) Create(c *fibre.Context, class string, thing interface{}, data map[string]interface{}) (interface{}, error) {
	switch thing := thing.(type) {
	case *fibre.RPCNull:
		return db.Execute(c, "CREATE $class CONTENT $data RETURN AFTER", map[string]interface{}{
			"class": sql.NewTable(class),
			"data":  data,
		})
	case []interface{}:
		return db.Execute(c, "CREATE $batch CONTENT $data RETURN AFTER", map[string]interface{}{
			"batch": sql.NewBatch(class, thing),
			"data":  data,
		})
	default:
		return db.Execute(c, "CREATE $thing CONTENT $data RETURN AFTER", map[string]interface{}{
			"thing": sql.NewThing(class, thing),
			"data":  data,
		})
	}
}

func (r *rpc) Update(c *fibre.Context, class string, thing interface{}, data map[string]interface{}) (interface{}, error) {
	switch thing := thing.(type) {
	case *fibre.RPCNull:
		return db.Execute(c, "UPDATE $class CONTENT $data RETURN AFTER", map[string]interface{}{
			"class": sql.NewTable(class),
			"data":  data,
		})
	case []interface{}:
		return db.Execute(c, "UPDATE $batch CONTENT $data RETURN AFTER", map[string]interface{}{
			"batch": sql.NewBatch(class, thing),
			"data":  data,
		})
	default:
		return db.Execute(c, "UPDATE $thing CONTENT $data RETURN AFTER", map[string]interface{}{
			"thing": sql.NewThing(class, thing),
			"data":  data,
		})
	}
}

func (r *rpc) Change(c *fibre.Context, class string, thing interface{}, data map[string]interface{}) (interface{}, error) {
	switch thing := thing.(type) {
	case *fibre.RPCNull:
		return db.Execute(c, "UPDATE $class MERGE $data RETURN AFTER", map[string]interface{}{
			"class": sql.NewTable(class),
			"data":  data,
		})
	case []interface{}:
		return db.Execute(c, "UPDATE $batch MERGE $data RETURN AFTER", map[string]interface{}{
			"batch": sql.NewBatch(class, thing),
			"data":  data,
		})
	default:
		return db.Execute(c, "UPDATE $thing MERGE $data RETURN AFTER", map[string]interface{}{
			"thing": sql.NewThing(class, thing),
			"data":  data,
		})
	}
}

func (r *rpc) Modify(c *fibre.Context, class string, thing interface{}, data map[string]interface{}) (interface{}, error) {
	switch thing := thing.(type) {
	case *fibre.RPCNull:
		return db.Execute(c, "UPDATE $class DIFF $data RETURN AFTER", map[string]interface{}{
			"class": sql.NewTable(class),
			"data":  data,
		})
	case []interface{}:
		return db.Execute(c, "UPDATE $batch DIFF $data RETURN AFTER", map[string]interface{}{
			"batch": sql.NewBatch(class, thing),
			"data":  data,
		})
	default:
		return db.Execute(c, "UPDATE $thing DIFF $data RETURN AFTER", map[string]interface{}{
			"thing": sql.NewThing(class, thing),
			"data":  data,
		})
	}
}

func (r *rpc) Delete(c *fibre.Context, class string, thing interface{}) (interface{}, error) {
	switch thing := thing.(type) {
	case *fibre.RPCNull:
		return db.Execute(c, "DELETE $class", map[string]interface{}{
			"class": sql.NewTable(class),
		})
	case []interface{}:
		return db.Execute(c, "DELETE $batch", map[string]interface{}{
			"batch": sql.NewBatch(class, thing),
		})
	default:
		return db.Execute(c, "DELETE $thing", map[string]interface{}{
			"thing": sql.NewThing(class, thing),
		})
	}
}