360 lines
8.7 KiB
Go
360 lines
8.7 KiB
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 (
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/abcum/fibre"
|
|
"github.com/abcum/fibre/mw"
|
|
"github.com/abcum/surreal/db"
|
|
"github.com/abcum/surreal/sql"
|
|
"github.com/abcum/surreal/util/build"
|
|
"github.com/abcum/surreal/util/show"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func limit(c *fibre.Context, i int64) int64 {
|
|
if s := c.Query("limit"); len(s) > 0 {
|
|
if x, err := strconv.ParseInt(s, 10, 64); err == nil {
|
|
return x
|
|
}
|
|
}
|
|
return i
|
|
}
|
|
|
|
func start(c *fibre.Context, i int64) int64 {
|
|
if s := c.Query("start"); len(s) > 0 {
|
|
if x, err := strconv.ParseInt(s, 10, 64); err == nil {
|
|
return x
|
|
}
|
|
}
|
|
return i
|
|
}
|
|
|
|
func versn(c *fibre.Context, i time.Time) time.Time {
|
|
if s := c.Query("version"); len(s) > 0 {
|
|
if x, err := time.Parse(sql.RFCNano, s); err == nil {
|
|
return x
|
|
}
|
|
}
|
|
return i
|
|
}
|
|
|
|
func routes(s *fibre.Fibre) {
|
|
|
|
s.Dir("/", "app/")
|
|
|
|
// --------------------------------------------------
|
|
// Endpoint for health checks
|
|
// --------------------------------------------------
|
|
|
|
s.Get("/info", func(c *fibre.Context) error {
|
|
return c.Send(200, build.GetInfo())
|
|
})
|
|
|
|
// --------------------------------------------------
|
|
// Endpoints for submitting rpc queries
|
|
// --------------------------------------------------
|
|
|
|
s.Rpc("/rpc", &rpc{})
|
|
|
|
// --------------------------------------------------
|
|
// Endpoints for authentication signup
|
|
// --------------------------------------------------
|
|
|
|
s.Options("/signup", func(c *fibre.Context) error {
|
|
return c.Code(200)
|
|
})
|
|
|
|
s.Post("/signup", func(c *fibre.Context) error {
|
|
return signup(c)
|
|
})
|
|
|
|
s.Use(mw.Quit(&mw.QuitOpts{
|
|
Timeout: 10 * time.Second,
|
|
}).PathIs("/signup"))
|
|
|
|
s.Use(mw.Type(&mw.TypeOpts{
|
|
AllowedContent: map[string]bool{
|
|
"application/json": true,
|
|
"application/cbor": true,
|
|
"application/msgpack": true,
|
|
"application/x-www-form-urlencoded": true,
|
|
},
|
|
}).PathIs("/signup"))
|
|
|
|
// --------------------------------------------------
|
|
// Endpoints for authentication signin
|
|
// --------------------------------------------------
|
|
|
|
s.Options("/signin", func(c *fibre.Context) error {
|
|
return c.Code(200)
|
|
})
|
|
|
|
s.Post("/signin", func(c *fibre.Context) error {
|
|
return signin(c)
|
|
})
|
|
|
|
s.Use(mw.Quit(&mw.QuitOpts{
|
|
Timeout: 10 * time.Second,
|
|
}).PathIs("/signin"))
|
|
|
|
s.Use(mw.Type(&mw.TypeOpts{
|
|
AllowedContent: map[string]bool{
|
|
"application/json": true,
|
|
"application/cbor": true,
|
|
"application/msgpack": true,
|
|
"application/x-www-form-urlencoded": true,
|
|
},
|
|
}).PathIs("/signin"))
|
|
|
|
// --------------------------------------------------
|
|
// Endpoints for import and exporting data
|
|
// --------------------------------------------------
|
|
|
|
s.Get("/export", func(c *fibre.Context) error {
|
|
return exporter(c)
|
|
})
|
|
|
|
s.Post("/import", func(c *fibre.Context) error {
|
|
return importer(c)
|
|
})
|
|
|
|
s.Use(mw.Type(&mw.TypeOpts{
|
|
AllowedContent: map[string]bool{
|
|
"application/octet-stream": true,
|
|
},
|
|
}).PathIs("/export", "/import"))
|
|
|
|
// --------------------------------------------------
|
|
// Endpoints for submitting sql queries
|
|
// --------------------------------------------------
|
|
|
|
s.Options("/sql", func(c *fibre.Context) error {
|
|
return c.Code(200)
|
|
})
|
|
|
|
s.Post("/sql", func(c *fibre.Context) error {
|
|
res, err := db.Execute(c, c.Request().Body, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return c.Send(200, res)
|
|
})
|
|
|
|
s.Use(mw.Type(&mw.TypeOpts{
|
|
AllowedContent: map[string]bool{
|
|
"application/json": true,
|
|
"application/cbor": true,
|
|
"application/msgpack": true,
|
|
},
|
|
}).PathIs("/sql"))
|
|
|
|
// --------------------------------------------------
|
|
// Endpoints for submitting websocket sql
|
|
// --------------------------------------------------
|
|
|
|
s.Get("/sql", func(c *fibre.Context) error {
|
|
if err := c.Upgrade(); err != nil {
|
|
return err
|
|
}
|
|
for {
|
|
_, msg, err := c.Socket().Read()
|
|
if err != nil {
|
|
if websocket.IsCloseError(err, 1000, 1001, 1005) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
if res, err := db.Execute(c, msg, nil); err != nil {
|
|
c.Socket().SendText(err.Error())
|
|
} else {
|
|
c.Socket().SendJSON(res)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
|
|
// --------------------------------------------------
|
|
// Endpoints for manipulating multiple records
|
|
// --------------------------------------------------
|
|
|
|
s.Options("/key/:class", func(c *fibre.Context) error {
|
|
return c.Code(200)
|
|
})
|
|
|
|
s.Get("/key/:class", func(c *fibre.Context) error {
|
|
|
|
txt := "SELECT * FROM $class LIMIT $limit START $start"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"class": sql.NewTable(c.Param("class")),
|
|
"limit": limit(c, 100),
|
|
"start": start(c, 0),
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.Many, show.Select, res, err)
|
|
|
|
})
|
|
|
|
s.Post("/key/:class", func(c *fibre.Context) error {
|
|
|
|
var data interface{}
|
|
|
|
if err := c.Bind(&data); err != nil {
|
|
return fibre.NewHTTPError(422)
|
|
}
|
|
|
|
txt := "CREATE $class CONTENT $data RETURN AFTER"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"class": sql.NewTable(c.Param("class")),
|
|
"data": data,
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.Many, show.Create, res, err)
|
|
|
|
})
|
|
|
|
s.Delete("/key/:class", func(c *fibre.Context) error {
|
|
|
|
txt := "DELETE FROM $class"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"class": sql.NewTable(c.Param("class")),
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.Many, show.Delete, res, err)
|
|
|
|
})
|
|
|
|
// --------------------------------------------------
|
|
// Endpoints for manipulating a single record
|
|
// --------------------------------------------------
|
|
|
|
s.Options("/key/:class/:id", func(c *fibre.Context) error {
|
|
return c.Code(200)
|
|
})
|
|
|
|
s.Get("/key/:class/:id", func(c *fibre.Context) error {
|
|
|
|
txt := "SELECT * FROM $thing VERSION $versn"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"thing": sql.NewThing(c.Param("class"), c.Param("id")),
|
|
"versn": versn(c, time.Now()),
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.One, show.Select, res, err)
|
|
|
|
})
|
|
|
|
s.Post("/key/:class/:id", func(c *fibre.Context) error {
|
|
|
|
var data interface{}
|
|
|
|
if err := c.Bind(&data); err != nil {
|
|
return fibre.NewHTTPError(422)
|
|
}
|
|
|
|
txt := "CREATE $thing CONTENT $data RETURN AFTER"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"thing": sql.NewThing(c.Param("class"), c.Param("id")),
|
|
"data": data,
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.One, show.Create, res, err)
|
|
|
|
})
|
|
|
|
s.Put("/key/:class/:id", func(c *fibre.Context) error {
|
|
|
|
var data interface{}
|
|
|
|
if err := c.Bind(&data); err != nil {
|
|
return fibre.NewHTTPError(422)
|
|
}
|
|
|
|
txt := "UPDATE $thing CONTENT $data RETURN AFTER"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"thing": sql.NewThing(c.Param("class"), c.Param("id")),
|
|
"data": data,
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.One, show.Update, res, err)
|
|
|
|
})
|
|
|
|
s.Patch("/key/:class/:id", func(c *fibre.Context) error {
|
|
|
|
var data interface{}
|
|
|
|
if err := c.Bind(&data); err != nil {
|
|
return fibre.NewHTTPError(422)
|
|
}
|
|
|
|
txt := "UPDATE $thing DIFF $data RETURN DIFF"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"thing": sql.NewThing(c.Param("class"), c.Param("id")),
|
|
"data": data,
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.One, show.Modify, res, err)
|
|
|
|
})
|
|
|
|
s.Trace("/key/:class/:id", func(c *fibre.Context) error {
|
|
|
|
txt := "SELECT HISTORY FROM $thing"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"thing": sql.NewThing(c.Param("class"), c.Param("id")),
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.One, show.Trace, res, err)
|
|
|
|
})
|
|
|
|
s.Delete("/key/:class/:id", func(c *fibre.Context) error {
|
|
|
|
txt := "DELETE $thing"
|
|
|
|
res, err := db.Execute(c, txt, map[string]interface{}{
|
|
"thing": sql.NewThing(c.Param("class"), c.Param("id")),
|
|
})
|
|
|
|
return show.Output(c, c.Param("class"), show.One, show.Delete, res, err)
|
|
|
|
})
|
|
|
|
s.Use(mw.Quit(&mw.QuitOpts{
|
|
Timeout: 15 * time.Second,
|
|
}).PathBegsWith("/key/"))
|
|
|
|
s.Use(mw.Type(&mw.TypeOpts{
|
|
AllowedContent: map[string]bool{
|
|
"application/json": true,
|
|
"application/cbor": true,
|
|
"application/msgpack": true,
|
|
"application/vnd.api+json": true,
|
|
},
|
|
}).PathBegsWith("/key/"))
|
|
|
|
}
|