forked from gizak/termui
-
Notifications
You must be signed in to change notification settings - Fork 1
/
router.go
112 lines (91 loc) · 2.53 KB
/
router.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
package router
import (
"github.com/pkg/errors"
"github.com/verdverm/tview"
"github.com/verdverm/vermui"
"github.com/verdverm/vermui/events"
"github.com/verdverm/vermui/mux"
)
type RoutePair struct {
Path string
Thing interface{}
}
type Routable interface {
Routings() []RoutePair
}
type Router struct {
*tview.Pages
// internal router
iRouter *mux.Router
}
func New() *Router {
r := &Router{
Pages: tview.NewPages(),
iRouter: mux.NewRouter(),
}
vermui.AddWidgetHandler(r.Pages, "/router/dispatch", func(ev events.Event) {
path := ev.Data.(*events.EventCustom).Data().(string)
context := map[string]interface{}{
"activation": "dispatch",
"path": path,
"data": ev.Data.(*events.EventCustom).Data(),
"event": ev,
}
r.SetActive(path, context)
})
return r
}
func (R *Router) SetNotFound(layout tview.Primitive) {
handler := func(req *mux.Request) (tview.Primitive, *mux.Request, error) {
return layout, req, nil
}
R.iRouter.NotFoundHandler = mux.NewDefaultHandler(handler)
R.AddPage(layout.Id(), layout, true, false)
}
func (R *Router) AddRoute(path string, thing interface{}) error {
switch t := thing.(type) {
case tview.Primitive:
R.AddRouteLayout(path, t)
case mux.HandlerFunc:
R.AddRouteHandlerFunc(path, t)
case mux.Handler:
R.AddRouteHandler(path, t)
default:
return errors.New("Unknown thing to be routed to...")
}
return nil
}
func (R *Router) AddRouteLayout(path string, layout tview.Primitive) error {
R.AddPage(layout.Id(), layout, true, false)
handler := func(req *mux.Request) (tview.Primitive, *mux.Request, error) {
return layout, req, nil
}
R.iRouter.Handle(path, mux.NewDefaultHandler(handler))
return nil
}
func (R *Router) AddRouteHandlerFunc(path string, handler mux.HandlerFunc) error {
R.iRouter.Handle(path, mux.NewDefaultHandler(handler))
return nil
}
func (R *Router) AddRouteHandler(path string, handler mux.Handler) error {
R.iRouter.Handle(path, handler)
return nil
}
func (R *Router) SetActive(path string, context map[string]interface{}) {
layout, req, err := R.iRouter.Dispatch(path, context)
if err != nil {
go events.SendCustomEvent("/console/error", errors.Wrap(err, "in dispatch handler"))
}
if layout != nil {
ctx := req.Context
req.Context = nil
ctx["req"] = req
R.setActive(layout, ctx)
} else {
go events.SendCustomEvent("/console/error", "nil layout in dispatch handler")
}
}
func (R *Router) setActive(layout tview.Primitive, context map[string]interface{}) {
R.Pages.SwitchToPage(layout.Id(), context)
vermui.Draw()
}