summaryrefslogtreecommitdiff
path: root/shrine/utils/meta/request.go
blob: 3104f06316c19826c3c1ea0cfab7a8e84d3fa8d9 (plain)
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
88
89
90
package meta

import (
	"shrine/types/hypertext"
	"shrine/utils/logger"

	"github.com/gofiber/fiber/v2"
)

const RequestKey = "__request_context"

func Request(context *fiber.Ctx) facade {
	request, ok := context.Locals(RequestKey).(hypertext.Request)
	if !ok {
		logger.Errorf("META", "RequestContext missing in fiber locals")
		return facade{}
	}
	return facade{Request: request, context: context}
}

func (self facade) Param(key string) (string, bool) {
	if self.context != nil {
		val := self.context.Params(key)
		if val != "" {
			return val, true
		}
	}
	return "", false
}

func (self facade) Query(key string) (string, bool) {
	return findParam(self.Request.Query, key)
}

func (self facade) Header(key string) (string, bool) {
	return findParam(self.Request.Headers, key)
}

func (self required) Param(key string) string {
	if self.context != nil {
		val := self.context.Params(key)
		if val != "" {
			return val
		}
	}
	logger.Errorf("META", "missing required param: %s", key)
	return ""
}

func (self required) Query(key string) string {
	value, found := findParam(self.request.Query, key)
	if !found {
		logger.Errorf("META", "missing required query: %s", key)
	}
	return value
}

func (self required) Header(key string) string {
	value, found := findParam(self.request.Headers, key)
	if !found {
		logger.Errorf("META", "missing required header: %s", key)
	}
	return value
}

func (self withDefault) Param(key string) string {
	if self.context != nil {
		val := self.context.Params(key)
		if val != "" {
			return val
		}
	}
	return self.defaults
}

func (self withDefault) Query(key string) string {
	value, found := findParam(self.request.Query, key)
	if found {
		return value
	}
	return self.defaults
}

func (self withDefault) Header(key string) string {
	value, found := findParam(self.request.Headers, key)
	if found {
		return value
	}
	return self.defaults
}