-
Notifications
You must be signed in to change notification settings - Fork 0
/
routers.go
130 lines (112 loc) · 2.84 KB
/
routers.go
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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
/*
* Detectordag
*
* API for detectordag JAMStack dashboard
*
* API version: 1.0.0
* Contact: briggySmalls90@gmail.com
* Generated by: Swagger Codegen (https://github.com/swagger-api/swagger-codegen.git)
*/
package swagger
import (
"fmt"
"net/http"
"strings"
"github.com/briggysmalls/detectordag/shared/database"
"github.com/briggysmalls/detectordag/shared/email"
"github.com/briggysmalls/detectordag/shared/shadow"
"github.com/gorilla/mux"
)
const (
uuidRegex = `[0-9a-fA-F]{8}\-[0-9a-fA-F]{4}\-[0-9a-fA-F]{4}\-[0-9a-fA-F]{4}\-[0-9a-fA-F]{12}`
)
type Route struct {
Name string
Method string
Pattern string
HandlerFunc http.HandlerFunc
}
type Routes []Route
func NewRouter(config *Config, db database.Client, shadow shadow.Client, email email.Client) *mux.Router {
// Create the router
router := mux.NewRouter().StrictSlash(true)
// Create a server struct
s := server{
config: *config,
db: db,
shadow: shadow,
email: email,
}
// Prepare the routes
var routes = Routes{
Route{
"GetAccount",
strings.ToUpper("Get"),
fmt.Sprintf("/v1/accounts/{accountId:%s}", uuidRegex),
s.GetAccount,
},
Route{
"GetDevices",
strings.ToUpper("Get"),
fmt.Sprintf("/v1/accounts/{accountId:%s}/devices", uuidRegex),
s.GetDevices,
},
Route{
"UpdateAccount",
strings.ToUpper("Patch"),
fmt.Sprintf("/v1/accounts/{accountId:%s}", uuidRegex),
s.UpdateAccount,
},
Route{
"Auth",
strings.ToUpper("Post"),
"/v1/auth",
s.Auth,
},
Route{
"UpdateDevice",
strings.ToUpper("Patch"),
"/v1/devices/{deviceId}",
s.UpdateDevice,
},
}
// Build the router
for _, route := range routes {
var handler http.Handler
handler = route.HandlerFunc
router.
Methods(route.Method).
Path(route.Pattern).
Name(route.Name).
Handler(handler)
}
// Add routes to allow CORS
addOptionsRoutes(router, routes)
// Add CORS header on all responses
router.Use(corsMiddleware)
return router
}
func addOptionsRoutes(router *mux.Router, routes []Route) {
// Determine unique routes, and collect the methods on each
methods := make(map[string][]string)
for _, route := range routes {
// Append the method to the pattern
methods[route.Pattern] = append(methods[route.Pattern], route.Method)
}
// Add routes for the options method
for pattern, methods := range methods {
router.
Methods("OPTIONS").
Path(pattern).
Handler(OptionsHandlerFactory(methods))
}
}
//OptionsHandlerFactory creates a handler for a route
func OptionsHandlerFactory(methods []string) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// Add header to permit methods
w.Header().Set("Access-Control-Allow-Methods", strings.Join(methods, ","))
w.Header().Set("Access-Control-Allow-Headers", "Content-Type,Authorization")
w.WriteHeader(http.StatusOK)
})
}