mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-22 10:57:42 +08:00
3f1cbe2f43
ServeOptions take the node and muxer, they should get the listener too as sometimes they need to operate on the listener address. License: MIT Signed-off-by: Juan Batiz-Benet <juan@benet.ai>
35 lines
1014 B
Go
35 lines
1014 B
Go
package corehttp
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
|
|
isd "github.com/ipfs/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-is-domain"
|
|
"github.com/ipfs/go-ipfs/Godeps/_workspace/src/golang.org/x/net/context"
|
|
"github.com/ipfs/go-ipfs/core"
|
|
)
|
|
|
|
// IPNSHostnameOption rewrites an incoming request if its Host: header contains
|
|
// an IPNS name.
|
|
// The rewritten request points at the resolved name on the gateway handler.
|
|
func IPNSHostnameOption() ServeOption {
|
|
return func(n *core.IpfsNode, _ net.Listener, mux *http.ServeMux) (*http.ServeMux, error) {
|
|
childMux := http.NewServeMux()
|
|
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
ctx, cancel := context.WithCancel(n.Context())
|
|
defer cancel()
|
|
|
|
host := strings.SplitN(r.Host, ":", 2)[0]
|
|
if len(host) > 0 && isd.IsDomain(host) {
|
|
name := "/ipns/" + host
|
|
if _, err := n.Namesys.Resolve(ctx, name); err == nil {
|
|
r.URL.Path = name + r.URL.Path
|
|
}
|
|
}
|
|
childMux.ServeHTTP(w, r)
|
|
})
|
|
return childMux, nil
|
|
}
|
|
}
|