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"
)

type rpc struct{}

// --------------------------------------------------
// Methods for authentication
// --------------------------------------------------

func (r *rpc) Ping(c *fibre.Context) (interface{}, error) {
	return "OK", nil
}

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

func (r *rpc) Signup(c *fibre.Context, vars map[string]interface{}) (interface{}, error) {
	return signupRpc(c, vars)
}

func (r *rpc) Signin(c *fibre.Context, vars map[string]interface{}) (interface{}, error) {
	return signinRpc(c, vars)
}

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

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

// --------------------------------------------------
// Methods for live queries
// --------------------------------------------------

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),
	})
}

// --------------------------------------------------
// Methods for static queries
// --------------------------------------------------

func (r *rpc) Let(c *fibre.Context, key string, val interface{}) (interface{}, error) {
	switch val := val.(type) {
	case *fibre.RPCNull:
		vars := c.Get("vars").(map[string]interface{})
		delete(vars, key)
		c.Set("vars", vars)
		return vars, nil
	default:
		vars := c.Get("vars").(map[string]interface{})
		vars[key] = val
		c.Set("vars", vars)
		return vars, nil
	}
}

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

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 []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),
		})
	}
}