/
apiproxy.go
45 lines (39 loc) · 1.13 KB
/
apiproxy.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
package apiproxy
import (
"net/http"
"net/http/httputil"
"net/url"
"runtime"
"github.com/cockroachdb/errors"
"golang.org/x/oauth2"
"encr.dev/internal/conf"
"encr.dev/internal/version"
)
func New(targetURL string) (*httputil.ReverseProxy, error) {
target, err := url.Parse(targetURL)
if err != nil {
return nil, errors.Wrap(err, "parse target url")
}
proxy := &httputil.ReverseProxy{
Transport: &oauth2.Transport{
Base: http.DefaultTransport,
Source: oauth2.ReuseTokenSource(nil, conf.DefaultTokenSource),
},
ErrorHandler: func(writer http.ResponseWriter, request *http.Request, err error) {
if errors.Is(err, conf.ErrNotLoggedIn) {
writer.WriteHeader(http.StatusUnauthorized)
return
}
writer.WriteHeader(http.StatusBadGateway)
},
Rewrite: func(r *httputil.ProxyRequest) {
r.Out.URL = target
r.Out.Header.Set("User-Agent", "EncoreCLI/"+version.Version)
r.Out.Header.Set("X-Encore-Dev-Dash", "true")
r.Out.Header.Set("X-Encore-Version", version.Version)
r.Out.Header.Set("X-Encore-GOOS", runtime.GOOS)
r.Out.Header.Set("X-Encore-GOARCH", runtime.GOARCH)
},
}
return proxy, nil
}