iris/_future/macros_test.go

// I would be grateful if I had the chance to see the whole work-in-progress in a codebase when I started.
// You have the chance to learn faster nowdays, don't underestimate that, that's the only reason that this "_future" folder exists now.
//
//
// The whole "router" package is a temp place to test my ideas and implementations for future iris' features.
// Young developers can understand and see how ideas can be transform to real implementations on a software like Iris,
// watching the history of a "dirty" code can be useful for some of you.
//
package router

import (
	"reflect"
	"regexp"
	"strconv"
	"testing"

	"gopkg.in/kataras/iris.v6"
	"gopkg.in/kataras/iris.v6/adaptors/httprouter"
	"gopkg.in/kataras/iris.v6/httptest"
)

// a helper to return a macro from a simple regexp
// it compiles the regexp  and after returns the macro, for obviously performance reasons.
func fromRegexp(expr string) _macrofn {
	if expr == "" {
		panic("empty expr on regex")
	}

	// add the last $ if missing (and not wildcard(?))
	if i := expr[len(expr)-1]; i != '$' && i != '*' {
		expr += "$"
	}

	r, err := regexp.Compile(expr)
	if err != nil {
		panic(err)
	}
	return func(pathParamValue string, _ *iris.Context) bool { return r.MatchString(pathParamValue) }
}

// link the path tmpl with macros, at .Boot time, before Listen.
// make it a as middleware from the beginning and prepend that before the main handler.
func link(path string, mac _macros) iris.HandlerFunc {
	tmpl, err := ParsePath(path)
	if err != nil {
		panic(err)
	}
	// link the path,  based on its template with a macro
	// and return a new compiled macro or a list of iris handlers
	// in order to be prepended on the original route or make a different function for that?
	// we'll see.

	var h iris.HandlerFunc // we could add an empty handler but we wouldn't know where to ctx.Next if this path doesn't uses macros.

	createH := func(paramName string, validator _macrofn, failStatus int, prevH iris.HandlerFunc) iris.HandlerFunc {
		return func(ctx *iris.Context) {
			if prevH != nil {
				prevH(ctx)
			}
			paramValue := ctx.Param(paramName)
			if paramValue != "" {
				valid := validator(paramValue, ctx)
				if !valid {
					// print("not valid for validator on paramValue= '" + paramValue + "' ctx.Pos = ")
					// println(ctx.Pos) // it should be always 0.
					ctx.EmitError(failStatus)
					return
				}
			}
			// remember: router already matches the path, so here if a path param is missing then it was allowed by the router.
			ctx.Next()
		}
	}

	for i := range tmpl.Params {
		p := tmpl.Params[i]
		if m, found := mac[p.Param.Macro.Name]; found && m.eval != nil {
			prevH := h
			eval := m.eval
			for _, fi := range m.funcs {
				for _, mi := range p.Param.Macro.Funcs {
					hasFunc := fi.name == mi.Name
					if !hasFunc {
						for _, gb := range mac[global_macro].funcs {
							if gb.name == fi.name {
								hasFunc = true
								break
							}
						}
					}

					if hasFunc {
						prevEval := eval
						macroFuncEval := fi.eval(mi.Params)
						eval = func(pvalue string, ctx *iris.Context) bool {
							if prevEval(pvalue, ctx) {
								return macroFuncEval(pvalue, ctx)
							}
							return false
						}
						continue
					}
				}
			}

			h = createH(p.Param.Name, eval, p.Param.FailStatusCode, prevH)
		}
	}

	if h == nil {
		// println("h is nil")
		return func(ctx *iris.Context) {
			ctx.Next() // is ok, the route doesn't contains any valid macros
		}
	}

	return h
}

// eval runs while serving paths
// instead of path it can receive the iris.Context and work as middleware
// if the macro passed completely then do ctx.Next() to continue to the main handler and the following,
// otherwise ctx.EmitError(pathTmpl.FailStatusCode) , which defaults to 404 for normal behavior on not found a route,
// but the developer can change that too,
// for example in order to fire the 402 if the compiled macro(I should think the name later) failed to be evaluted
// then the user should add !+statuscode, i.e "{id:int !402}".
// func eval(path string, tmpl *PathTmpl) bool {
// 	return false
// }
// <--- fun(c)k it, we will do it directly to be iris' middleware or create a new type which will save a macro and tries to eval it with a path
// only for test-cases? and after on iris we can make a middleware from this, I should think it more when I stop the drinking.

func testMacros(source string) error {
	return nil
}

// let's give the macro's funcs access to context, it will be great experimental to serve templates just with a path signature
type _macrofn func(pathParamValue string, ctx *iris.Context) bool

type _macrofunc struct {
	name string
	eval func([]string) _macrofn
}

type _macro struct {
	funcs []_macrofunc
	eval  _macrofn
}

// macros should be registered before .Listen
type _macros map[string]*_macro

var all_macros = _macros{}

func addMacro(name string, v _macrofn) {
	all_macros[name] = &_macro{eval: v}
}

func addMacroFunc(macroName string, funcName string, v func([]string) _macrofn) {
	if m, found := all_macros[macroName]; found {
		m.funcs = append(m.funcs, _macrofunc{name: funcName, eval: v})
	}
}

const global_macro = "any"

func macroFuncFrom(v interface{}) func(params []string) _macrofn {
	// this is executed once on boot time, not at serve time:
	vot := reflect.TypeOf(v)
	numFields := vot.NumIn()

	return func(params []string) _macrofn {
		if len(params) != numFields {
			panic("should accepts _numFields_ args")
		}
		var args []reflect.Value

		// check for accepting arguments
		for i := 0; i < numFields; i++ {
			field := vot.In(i)

			switch field.Kind() {
			case reflect.Int:
				val, err := strconv.Atoi(params[i])
				if err != nil {
					panic("invalid first parameter: " + err.Error())
				}
				args = append(args, reflect.ValueOf(val))
			case reflect.String:
			case reflect.Bool:
			default:
				panic("unsported type!")
			}
		}

		// check for the return type (only one ofc, which again is a function but it returns a boolean)
		// which accepts one argument which is the parameter value.
		if vot.NumOut() != 1 {
			panic("expecting to return only one (func)")
		}
		rof := vot.Out(0)
		if rof.Kind() != reflect.Func {
			panic("expecting to return a function!")
		}

		returnRof := rof.Out(0)
		if rof.NumOut() != 1 {
			panic("expecting to return only one (bool)")
		}
		if returnRof.Kind() != reflect.Bool {
			panic("expecting this func to return a boolean")
		}

		if rof.NumIn() != 1 {
			panic("expecting this func to receive one arg")
		}

		vof := reflect.ValueOf(v).Call(args)[0].Interface().(func(string) bool)
		//

		// this is executed when a route requested:
		return func(paramValue string, _ *iris.Context) bool {
			return vof(paramValue)
		}
		//
	}
}

func TestMacros(t *testing.T) {
	addMacro("int", fromRegexp("[1-9]+$"))

	// {id:int range(42,49)}
	// addMacroFunc("int", "range", func(params []string) _macrofn {
	// 	// start: .Boot time, before .Listen
	// 	allowedParamsLen := 2
	// 	// params:  42,49 (including first and second)
	// 	if len(params) != allowedParamsLen {
	// 		panic("range accepts two parameters")
	// 	}

	// 	min, err := strconv.Atoi(params[0])
	// 	if err != nil {
	// 		panic("invalid first parameter: " + err.Error())
	// 	}
	// 	max, err := strconv.Atoi(params[1])
	// 	if err != nil {
	// 		panic("invalid second parameter: " + err.Error())
	// 	}
	// 	// end

	// 	return func(paramValue string, _ *iris.Context) bool {
	// 		paramValueInt, err := strconv.Atoi(paramValue)
	// 		if err != nil {
	// 			return false
	// 		}
	// 		if paramValueInt >= min && paramValueInt <= max {
	// 			return true
	// 		}
	// 		return false
	// 	}
	// })
	// result should be like that in the final feature implementation, using reflection BEFORE .Listen on .Boot time,
	// so no performance cost(done) =>
	addMacroFunc("int", "range", macroFuncFrom(func(min int, max int) func(string) bool {
		return func(paramValue string) bool {
			paramValueInt, err := strconv.Atoi(paramValue)
			if err != nil {
				return false
			}
			if paramValueInt >= min && paramValueInt <= max {
				return true
			}
			return false
		}
	}))

	addMacroFunc("int", "even", func(params []string) _macrofn {
		return func(paramValue string, _ *iris.Context) bool {
			paramValueInt, err := strconv.Atoi(paramValue)
			if err != nil {
				return false
			}
			if paramValueInt%2 == 0 {
				return true
			}
			return false
		}
	})

	// "any" will contain macros functions
	// which are available to all other, we will need some functions to be 'globally' registered when don't care about
	// what macro is used, for example let's try the markdown(#hello), yes serve files without even call a function:)
	addMacro("any", fromRegexp(".*"))
	addMacroFunc("any", "markdown", func(params []string) _macrofn {
		if len(params) != 1 {
			panic("markdown expected 1 arg")
		}

		contents := params[0]
		// we don't care about path's parameter here
		return func(_ string, ctx *iris.Context) bool {
			ctx.Markdown(iris.StatusOK, contents)
			return true
		}
	})

	path := "/api/users/{id:int range(42,49) even() !600}/posts"
	app := iris.New()
	app.Adapt(httprouter.New())

	hv := link(path, all_macros)
	// 600 is a custom virtual error code to handle "int" param invalids
	// it sends a custom error message with a 404 (not found) http status code.
	app.OnError(600, func(ctx *iris.Context) {
		ctx.SetStatusCode(404) // throw a raw 404 not found
		ctx.Writef("Expecting an integer in range between and 42-49, should be even number too")
		// println("600 -> 404 from " + ctx.Path())
	})
	app.Get("/api/users/:id/posts", hv, func(ctx *iris.Context) {
		ctx.ResponseWriter.WriteString(ctx.Path())
	})

	path2 := "/markdown/{anything:any markdown(**hello**)}"
	hv2 := link(path2, all_macros)
	app.Get("/markdown/*anything", hv2)

	e := httptest.New(app, t)

	e.GET("/api/users/42/posts").Expect().Status(iris.StatusOK).Body().Equal("/api/users/42/posts")
	e.GET("/api/users/50/posts").Expect().Status(iris.StatusNotFound).Body().Equal("Expecting an integer in range between and 42-49, should be even number too") // remember, it accepts 1-9 not matched if zero.
	e.GET("/api/users/0/posts").Expect().Status(iris.StatusNotFound)
	e.GET("/api/users/_/posts").Expect().Status(iris.StatusNotFound)
	e.GET("/api/users/s/posts").Expect().Status(iris.StatusNotFound)
	e.GET("/api/users/posts").Expect().Status(iris.StatusNotFound)
	// macro func invalidate test with a non-zero value between 1-9 but bigger than the max(49)
	e.GET("/api/users/51/posts").Expect().Status(iris.StatusNotFound)
	// macro func invalidate "even" with a non-zero value but 49 is not an even number
	e.GET("/api/users/49/posts").Expect().Status(iris.StatusNotFound)

	// test any and global
	// response with "path language" only no need of handler too.
	// As it goes I love the idea and users will embrace and built awesome things on top of it.
	// maybe I have to 'rename' the final feature on something like iris expression language and document it as much as I can, people will love that
	e.GET("/markdown/something").Expect().Status(iris.StatusOK).ContentType("text/html", "utf-8").Body().Equal("<p><strong>hello</strong></p>\n")
}