package main

import (
	"git.xdrm.io/example/aicra/db"
	"git.xdrm.io/go/aicra/api"
	"git.xdrm.io/go/aicra/driver"
	e "git.xdrm.io/go/aicra/err"
)

func main() {}

type RootController int

func Export() driver.Controller { return new(RootController) }

// Redirects to an url from a key
func (rctl RootController) Get(d api.Arguments) api.Response {

	r := api.NewResponse()

	/* (1) Init redis connection */
	cli := db.Connect()
	if cli == nil {
		r.Err = e.Failure
		return *r
	}

	/* (2) Extract api input */
	key, err := d.GetString("url")

	if err != nil {
		r.Err = e.InvalidParam
		r.Err.Put("url")
		r.Err.Put(err.Error())
		return *r
	}

	/* (3) Check if match for this key */
	val := cli.Get(db.DATA, key)
	if val == nil {
		r.Err = e.NoMatchFound
		return *r
	}

	/* (4) Redirect to value */
	r.Set("_REDIRECT_", string(val))
	r.Err = e.Success
	return *r
}

// Stores a new tinyurl/fullurl combination
func (rctl RootController) Post(d api.Arguments) api.Response {

	r := api.NewResponse()
	/* (1) Init redis connection */
	cli := db.Connect()
	if cli == nil {
		r.Err = e.Failure
		return *r
	}

	/* (2) Extract api input */
	target, err := d.GetString("target")
	if err != nil {
		r.Err = e.InvalidParam
		r.Err.Put("target")
		r.Err.Put(err.Error())
		return *r
	}

	url, err := d.GetString("url")

	if err != nil {
		r.Err = e.InvalidParam
		r.Err.Put("url")
		r.Err.Put(err.Error())
		return *r
	}

	/* (3) Check if key already used */
	if cli.Get(db.DATA, url) != nil {
		r.Err = e.AlreadyExists
		r.Err.Put("url")
		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
func (rctl RootController) Put(d api.Arguments) api.Response {

	r := api.NewResponse()

	/* (1) Init redis connection */
	cli := db.Connect()
	if cli == nil {
		r.Err = e.Failure
		return *r
	}

	/* (2) Extract api input */
	target, err := d.GetString("target")
	if err != nil {
		r.Err = e.InvalidParam
		r.Err.Put("target")
		r.Err.Put(err.Error())
		return *r
	}

	url, err := d.GetString("url")

	if err != nil {
		r.Err = e.InvalidParam
		r.Err.Put("url")
		r.Err.Put(err.Error())
		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
func (rctl RootController) Delete(d api.Arguments) api.Response {

	r := api.NewResponse()

	/* (1) Init redis connection */
	cli := db.Connect()
	if cli == nil {
		r.Err = e.Failure
		return *r
	}

	/* (2) Extract api input */
	url, err := d.GetString("url")

	if err != nil {
		r.Err = e.InvalidParam
		r.Err.Put("url")
		r.Err.Put(err.Error())
		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) {
		r.Err = e.Failure
		return *r
	}

	r.Err = e.Success
	return *r
}