💾 Archived View for source.community › ckaznocha › gemini › blob › main › mux.go captured on 2024-02-05 at 09:52:45. Gemini links have been rewritten to link to archived content
⬅️ Previous capture (2023-01-29)
-=-=-=-=-=-=-
.
,-. ,-. . . ,-. ,-. ,-. ,-. ,-. ,-,-. ,-,-. . . ,-. . |- . .
`-. | | | | | | |-' | | | | | | | | | | | | | | | | |
`-' `-' `-^ ' `-' `-' :: `-' `-' ' ' ' ' ' ' `-^ ' ' ' `' `-|
/|
`-'
git clone https://source.community/ckaznocha/gemini.git
View raw contents of /mux.go (main)
────╮
1│ package gemini
2│
3│ import (
4│ "context"
5│ "strings"
6│ "sync"
7│ )
8│
9│ // ServeMux is a Gemini request multiplexer. It will match requests to handlers
10│ // based on the URI path. The longest match will be the one returned. Patterns
11│ // ending with `/` will be matched exactly. Patterns without a trailing `/` will
12│ // be treated as a prefix match.
13│ type ServeMux struct {
14│ nodeToPattern map[*muxNode]string
15│ nodes *muxNode
16│ mu sync.RWMutex
17│ }
18│
19│ // NewServeMux returns a new ServeMux ready to be used.
20│ func NewServeMux() *ServeMux {
21│ return &ServeMux{
22│ nodeToPattern: map[*muxNode]string{},
23│ nodes: &muxNode{
24│ handler: nil,
25│ nodes: map[string]*muxNode{},
26│ },
27│ }
28│ }
29│
30│ // Handle adds a new pattern/Handler pair to the ServeMux.
31│ func (sm *ServeMux) Handle(pattern string, handler Handler) {
32│ sm.mu.Lock()
33│ defer sm.mu.Unlock()
34│
35│ sm.nodeToPattern[sm.nodes.insert(pattern, handler)] = pattern
36│ }
37│
38│ // HandleFunc adds a new pattern/HandleFunc pair to the ServeMux.
39│ func (sm *ServeMux) HandleFunc(pattern string, handler func(context.Context, ResponseWriter, *Request)) {
40│ sm.Handle(pattern, HandlerFunc(handler))
41│ }
42│
43│ // Handler looks up a matching Handler based on a Request. It returns the patter
44│ // that matched in addition to the Hander.
45│ func (sm *ServeMux) Handler(r *Request) (h Handler, pattern string) {
46│ sm.mu.RLock()
47│ defer sm.mu.RUnlock()
48│
49│ if !strings.HasPrefix(r.URI.Path, "/") {
50│ r.URI.Path = "/" + r.URI.Path
51│ }
52│
53│ n := sm.nodes.find(r.URI.Path)
54│ if n.handler == nil {
55│ return HandlerFunc(func(c context.Context, rw ResponseWriter, r *Request) {
56│ rw.Failure(c, StatusNotFound, StatusNotFound.Description())
57│ }), ""
58│ }
59│
60│ return n.handler, sm.nodeToPattern[n]
61│ }
62│
63│ // ServeGemini implements the Handler interface.
64│ func (sm *ServeMux) ServeGemini(ctx context.Context, w ResponseWriter, r *Request) {
65│ handlr, _ := sm.Handler(r)
66│ handlr.ServeGemini(ctx, w, r)
67│ }
68│
69│ type muxNode struct {
70│ handler Handler
71│ nodes map[string]*muxNode
72│ }
73│
74│ func (mn *muxNode) insert(pattern string, handler Handler) *muxNode {
75│ if pattern == "" {
76│ mn.handler = handler
77│
78│ return mn
79│ }
80│
81│ idx := nextIndex(pattern)
82│
83│ child, ok := mn.nodes[pattern[:idx]]
84│ if !ok {
85│ child = &muxNode{nodes: map[string]*muxNode{}}
86│ mn.nodes[pattern[:idx]] = child
87│ }
88│
89│ return child.insert(pattern[idx:], handler)
90│ }
91│
92│ func (mn *muxNode) find(pattern string) *muxNode {
93│ if child, ok := mn.nodes[pattern]; ok {
94│ return child
95│ }
96│
97│ idx := nextIndex(pattern)
98│
99│ if child, ok := mn.nodes[pattern[:idx]]; ok {
100│ return child.find(pattern[idx:])
101│ }
102│
103│ return mn
104│ }
105│
106│ func nextIndex(pattern string) int {
107│ idx := strings.IndexRune(pattern[1:], '/')
108│ if idx == -1 {
109│ return len(pattern)
110│ }
111│
112│ return idx + 1
113│ }
────╯
· · ·
© 2024 source.community