-
Notifications
You must be signed in to change notification settings - Fork 28
/
authflow_enter_recovery_code.go
90 lines (74 loc) · 2.58 KB
/
authflow_enter_recovery_code.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
package webapp
import (
"net/http"
"github.com/authgear/authgear-server/pkg/auth/handler/webapp/viewmodels"
"github.com/authgear/authgear-server/pkg/auth/webapp"
"github.com/authgear/authgear-server/pkg/util/httproute"
"github.com/authgear/authgear-server/pkg/util/template"
"github.com/authgear/authgear-server/pkg/util/validation"
)
var TemplateWebAuthflowEnterRecoveryCodeHTML = template.RegisterHTML(
"web/authflow_enter_recovery_code.html",
Components...,
)
var AuthflowEnterRecoveryCodeSchema = validation.NewSimpleSchema(`
{
"type": "object",
"properties": {
"x_recovery_code": {
"type": "string",
"format": "x_recovery_code"
}
},
"required": ["x_recovery_code"]
}
`)
func ConfigureAuthflowEnterRecoveryCodeRoute(route httproute.Route) httproute.Route {
return route.
WithMethods("OPTIONS", "POST", "GET").
WithPathPattern(webapp.AuthflowRouteEnterRecoveryCode)
}
type AuthflowEnterRecoveryCodeHandler struct {
Controller *AuthflowController
BaseViewModel *viewmodels.BaseViewModeler
Renderer Renderer
}
func (h *AuthflowEnterRecoveryCodeHandler) GetData(w http.ResponseWriter, r *http.Request, s *webapp.Session, screen *webapp.AuthflowScreenWithFlowResponse) (map[string]interface{}, error) {
data := make(map[string]interface{})
baseViewModel := h.BaseViewModel.ViewModelForAuthFlow(r, w)
viewmodels.Embed(data, baseViewModel)
branchViewModel := viewmodels.NewAuthflowBranchViewModel(screen)
viewmodels.Embed(data, branchViewModel)
return data, nil
}
func (h *AuthflowEnterRecoveryCodeHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
var handlers AuthflowControllerHandlers
handlers.Get(func(s *webapp.Session, screen *webapp.AuthflowScreenWithFlowResponse) error {
data, err := h.GetData(w, r, s, screen)
if err != nil {
return err
}
h.Renderer.RenderHTML(w, r, TemplateWebAuthflowEnterRecoveryCodeHTML, data)
return nil
})
handlers.PostAction("", func(s *webapp.Session, screen *webapp.AuthflowScreenWithFlowResponse) error {
err := AuthflowEnterRecoveryCodeSchema.Validator().ValidateValue(FormToJSON(r.Form))
if err != nil {
return err
}
recoveryCode := r.Form.Get("x_recovery_code")
requestDeviceToken := r.Form.Get("x_device_token") == "true"
input := map[string]interface{}{
"authentication": "recovery_code",
"recovery_code": recoveryCode,
"request_device_token": requestDeviceToken,
}
result, err := h.Controller.AdvanceWithInput(r, s, screen, input, nil)
if err != nil {
return err
}
result.WriteResponse(w, r)
return nil
})
h.Controller.HandleStep(w, r, &handlers)
}