merp/merp.go
2019-11-08 04:41:18 -08:00

180 lines
3.8 KiB
Go

package merp
import "encoding/json"
import "net/http"
import "regexp"
import "time"
import "github.com/gin-gonic/gin"
import "github.com/google/uuid"
import "github.com/rs/zerolog/log"
import "github.com/sneak/merp/models"
func thingRegex() *regexp.Regexp {
ThingRegex, e := regexp.Compile(`^[a-zA-Z0-9\_\-]+$`)
if e != nil {
panic(e)
}
return ThingRegex
}
func decodeJSON(in []byte) interface{} {
var out interface{}
err := json.Unmarshal(in, &out)
if err != nil {
log.Error().Msg("error decoding json")
return nil
}
return out
}
func getLatestMerps() gin.HandlerFunc {
ThingRegex := thingRegex()
orm := models.GetOrmObject()
h := func(c *gin.Context) {
thing := c.Param("thing")
if ThingRegex.MatchString(thing) == false {
c.JSON(http.StatusPreconditionFailed, gin.H{
"this": "failed",
"status": http.StatusPreconditionFailed,
"because": "invalid thing format, try [a-zA-Z0-9-_]",
})
return
}
var merps []*models.Merp
qs := orm.QueryTable("merp").Filter("thing", thing).OrderBy("-created").Limit(50)
qs.All(&merps)
output := make([]map[string]interface{}, 0)
for _, merp := range merps {
outelem := make(map[string]interface{})
outelem["thing"] = merp.Thing
outjs := decodeJSON([]byte(merp.Content))
if outjs == nil {
outelem["content"] = gin.H{}
} else {
outelem["content"] = outjs
}
outelem["created"] = merp.Created
output = append(output, outelem)
}
c.JSON(http.StatusOK, gin.H{
"this": "succeeded",
"by": "getting",
"the": "merps",
"with": output,
})
}
return h
}
func getLatestMerp() gin.HandlerFunc {
ThingRegex := thingRegex()
h := func(c *gin.Context) {
thing := c.Param("thing")
if ThingRegex.MatchString(thing) == false {
c.JSON(http.StatusPreconditionFailed, gin.H{
"this": "failed",
"status": http.StatusPreconditionFailed,
"because": "invalid thing format, try [a-zA-Z0-9-_]",
})
return
}
c.JSON(http.StatusOK, gin.H{
"this": "succeeded",
})
}
return h
}
func handleNewMerp() gin.HandlerFunc {
// server startup time
ThingRegex := thingRegex()
// establish db connection *first*, before requests
orm := models.GetOrmObject()
h := func(c *gin.Context) {
// request time
thing := c.Param("thing")
if ThingRegex.MatchString(thing) == false {
c.JSON(http.StatusPreconditionFailed, gin.H{
"this": "failed",
"status": http.StatusPreconditionFailed,
"because": "invalid thing format, try a-zA-Z0-9-_",
})
return
}
//web.Get(`/merp/for/([A-Za-z0-9\-\_\.]+)`, merpHandler)
// FIXME rate limit this a bit on thing+clientip+json to cut down on
// repeated messages
content := make(map[string]interface{})
respContent := gin.H{}
// FIXME support POST data as well
for k, v := range c.Request.URL.Query() {
content[k] = v[0]
respContent[k] = v[0]
}
u := uuid.New()
at := time.Now().UTC()
atString := at.Format(time.RFC3339)
serialized, jsonerr := json.MarshalIndent(content, "", " ")
if jsonerr != nil {
c.JSON(
http.StatusPreconditionFailed,
gin.H{
"this": "failed",
"status": http.StatusPreconditionFailed,
"because": jsonerr.Error(),
},
)
return
}
merp := models.Merp{
Created: at,
Thing: thing,
Content: string(serialized),
UUID: u.String(),
}
_, err := orm.Insert(&merp)
if err != nil {
c.JSON(
http.StatusPreconditionFailed,
gin.H{
"this": "failed",
"status": http.StatusPreconditionFailed,
"because": err.Error(),
},
)
return
}
c.JSON(http.StatusOK, gin.H{
"this": "succeeded",
"by": "merping",
"the": "merp",
"with": gin.H{
"thing": thing,
"created": atString,
"content": respContent,
"transaction": u.String(),
},
})
}
return h
}