context.go

package router

import (
	"io"
	"net/http"
)

// RequestContext is...
type RequestContext struct {
	Env    map[interface{}]interface{}
	Params []Param
	req    *http.Request
}

type iogoContextReadCloser struct {
	io.ReadCloser
	context *RequestContext
}

func SetContext(req *http.Request) *RequestContext {
	c, ok := req.Body.(iogoContextReadCloser)
	if !ok {
		c = iogoContextReadCloser{
			ReadCloser: req.Body,
			context:    &RequestContext{Env: make(map[interface{}]interface{})},
		}
		req.Body = c
	}
	return c.context
}

// Context is...
func Context(req *http.Request) *RequestContext {
	return req.Body.(iogoContextReadCloser).context
}

// Param is...
func (c *RequestContext) Param(name string) string {
	for _, e := range c.Params {
		if e.Name == name {
			return e.Value
		}
	}
	return ""
}

// SetParams is...
func (c *RequestContext) setParams(params []Param) {
	c.Params = params
}