tiny-url-ex/controller.plugin/ROOT/main.go

159 lines
2.6 KiB
Go
Raw Normal View History

package main
import (
"git.xdrm.io/example/aicra/db"
2018-10-01 13:15:02 +00:00
"git.xdrm.io/go/aicra/driver"
2018-07-05 17:19:39 +00:00
e "git.xdrm.io/go/aicra/err"
"git.xdrm.io/go/aicra/response"
i "git.xdrm.io/go/aicra/response"
)
2018-10-01 13:15:02 +00:00
func main() {}
type RootController int
func Export() driver.Controller { return new(RootController) }
2018-06-16 16:04:28 +00:00
// Redirects to an url from a key
2018-10-01 13:15:02 +00:00
func (rctl RootController) Get(d i.Arguments) i.Response {
r := response.New()
2018-06-16 16:04:28 +00:00
/* (1) Init redis connection */
cli := db.Connect()
if cli == nil {
2018-06-16 16:04:28 +00:00
r.Err = e.Failure
return *r
}
/* (2) Extract api input */
key, ok := d["url"].(string)
if !ok {
r.Err = e.InvalidParam
2018-10-01 19:21:33 +00:00
r.Err.Put("url")
2018-06-16 16:04:28 +00:00
return *r
}
/* (3) Check if match for this key */
val := cli.Get(db.DATA, key)
if val == nil {
2018-06-16 16:04:28 +00:00
r.Err = e.NoMatchFound
return *r
}
/* (4) Redirect to value */
r.Set("_REDIRECT_", string(val))
2018-06-16 16:04:28 +00:00
r.Err = e.Success
return *r
}
// Stores a new tinyurl/fullurl combination
2018-10-01 13:15:02 +00:00
func (rctl RootController) Post(d i.Arguments) i.Response {
r := response.New()
2018-06-16 16:04:28 +00:00
/* (1) Init redis connection */
cli := db.Connect()
if cli == nil {
2018-06-16 16:04:28 +00:00
r.Err = e.Failure
return *r
}
/* (2) Extract api input */
target, ok1 := d["target"].(string)
url, ok2 := d["url"].(string)
if !ok1 || !ok2 {
r.Err = e.InvalidParam
return *r
}
/* (3) Check if key already used */
if cli.Get(db.DATA, url) != nil {
2018-06-16 16:04:28 +00:00
r.Err = e.AlreadyExists
return *r
}
/* (4) Store */
if !cli.Set(db.DATA, url, target) {
r.Err = e.Failure
return *r
}
r.Err = e.Success
return *r
}
// Overrides a existing tinyurl with new target
2018-10-01 13:15:02 +00:00
func (rctl RootController) Put(d i.Arguments) i.Response {
r := response.New()
/* (1) Init redis connection */
cli := db.Connect()
if cli == nil {
r.Err = e.Failure
return *r
}
/* (2) Extract api input */
target, ok1 := d["target"].(string)
url, ok2 := d["url"].(string)
if !ok1 || !ok2 {
r.Err = e.InvalidParam
return *r
}
/* (3) Check if key already used */
if cli.Get(db.DATA, url) == nil {
r.Err = e.NoMatchFound
return *r
}
/* (4) Update */
if !cli.Set(db.DATA, url, target) {
r.Err = e.Failure
return *r
}
r.Err = e.Success
return *r
}
// Deletes an existing tinyurl
2018-10-01 13:15:02 +00:00
func (rctl RootController) Delete(d i.Arguments) i.Response {
r := response.New()
/* (1) Init redis connection */
cli := db.Connect()
if cli == nil {
r.Err = e.Failure
return *r
}
/* (2) Extract api input */
url, ok := d["url"].(string)
if !ok {
r.Err = e.InvalidParam
return *r
}
/* (3) Check if key already used */
if cli.Get(db.DATA, url) == nil {
r.Err = e.NoMatchFound
return *r
}
/* (4) Delete */
if !cli.Del(db.DATA, url) {
2018-06-16 16:04:28 +00:00
r.Err = e.Failure
return *r
}
r.Err = e.Success
return *r
}