astra/contextFuncBuilder.go at main · LS6-Events/astra · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package astra

import "github.com/ls6-events/astra/astTraversal"

type ContextFuncBuilder struct {
	Route           *Route
	Traverser       *astTraversal.CallExpressionTraverser
	paramOperations []func() (any, error)
}

func NewContextFuncBuilder(route *Route, traverser *astTraversal.CallExpressionTraverser) *ContextFuncBuilder {
	paramOperations := make([]func() (any, error), 0)
	return &ContextFuncBuilder{
		Route:           route,
		Traverser:       traverser,
		paramOperations: paramOperations,
	}
}

func (c *ContextFuncBuilder) getCurrentParamIndex() int {
	return len(c.paramOperations)
}

func (c *ContextFuncBuilder) Ignored() *ContextFuncBuilder {
	c.paramOperations = append(c.paramOperations, func() (any, error) {
		//nolint:nilnil // This is intentional
		return nil, nil
	})

	return c
}

func (c *ContextFuncBuilder) StatusCode() *ContextFuncBuilder {
	currIndex := c.getCurrentParamIndex()
	c.paramOperations = append(c.paramOperations, func() (any, error) {
		return c.Traverser.Traverser.ExtractStatusCode(c.Traverser.Node.Args[currIndex])
	})

	return c
}

func (c *ContextFuncBuilder) ExpressionResult() *ContextFuncBuilder {
	currIndex := c.getCurrentParamIndex()
	c.paramOperations = append(c.paramOperations, func() (any, error) {
		exprType, err := c.Traverser.Traverser.Expression(c.Traverser.Node.Args[currIndex]).Type()
		if err != nil {
			return nil, err
		}

		return c.Traverser.Traverser.Type(exprType, c.Traverser.File.Package).Result()
	})

	return c
}

func (c *ContextFuncBuilder) Value() *ContextFuncBuilder {
	currIndex := c.getCurrentParamIndex()
	c.paramOperations = append(c.paramOperations, func() (any, error) {
		expr := c.Traverser.Traverser.Expression(c.Traverser.Node.Args[currIndex])

		return expr.Value()
	})

	return c
}

func (c *ContextFuncBuilder) Build(mapper func(*Route, []any) (*Route, error)) (*Route, error) {
	results := make([]any, len(c.paramOperations))
	for i, operation := range c.paramOperations {
		result, err := operation()
		if err != nil {
			return nil, err
		}

		results[i] = result
	}

	return mapper(c.Route, results)
}

type CustomFunc func(contextVarName string, contextFuncBuilder *ContextFuncBuilder) (*Route, error)

func WithCustomFunc(customFunc CustomFunc) Option {
	return func(service *Service) {
		service.CustomFuncs = append(service.CustomFuncs, customFunc)
	}
}