merp/merp.go

180 lines
3.8 KiB
Go
Raw Normal View History

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