forked from openatx/uiautomator2
/
proxyhttp.go
39 lines (33 loc) · 926 Bytes
/
proxyhttp.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
package main
import (
"log"
"net/http"
"net/http/httputil"
"net/url"
"github.com/koding/websocketproxy"
)
// TODO(ssx): not tested yet.
type HTTPWSProxy struct {
forwardedAddr string
wsProxy *websocketproxy.WebsocketProxy
httpProxy *httputil.ReverseProxy
}
// NewHTTPWSProxy return Proxy instance
func NewHTTPWSProxy(forwardedAddr string) *HTTPWSProxy {
wsURL, _ := url.Parse("ws://" + forwardedAddr)
httpURL, _ := url.Parse("http://" + forwardedAddr)
return &HTTPWSProxy{
forwardedAddr: forwardedAddr,
wsProxy: websocketproxy.NewProxy(wsURL),
httpProxy: httputil.NewSingleHostReverseProxy(httpURL),
}
}
func (p *HTTPWSProxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("Upgrade") == "websocket" {
log.Println("proxy websocket", r.RequestURI)
p.wsProxy.ServeHTTP(w, r)
return
}
log.Println("proxy http", r.RequestURI)
p.httpProxy.ServeHTTP(w, r)
}