mirror of https://github.com/grafana/grafana.git
102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package pluginproxy
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"net/url"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/log"
|
|
"github.com/grafana/grafana/pkg/middleware"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/plugins"
|
|
"github.com/grafana/grafana/pkg/util"
|
|
)
|
|
|
|
type templateData struct {
|
|
JsonData map[string]interface{}
|
|
SecureJsonData map[string]string
|
|
}
|
|
|
|
func getHeaders(route *plugins.AppPluginRoute, orgId int64, appId string) (http.Header, error) {
|
|
result := http.Header{}
|
|
|
|
query := m.GetPluginSettingByIdQuery{OrgId: orgId, PluginId: appId}
|
|
|
|
if err := bus.Dispatch(&query); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
data := templateData{
|
|
JsonData: query.Result.JsonData,
|
|
SecureJsonData: query.Result.SecureJsonData.Decrypt(),
|
|
}
|
|
|
|
err := addHeaders(&result, route, data)
|
|
return result, err
|
|
}
|
|
|
|
func NewApiPluginProxy(ctx *middleware.Context, proxyPath string, route *plugins.AppPluginRoute, appId string) *httputil.ReverseProxy {
|
|
targetUrl, _ := url.Parse(route.Url)
|
|
|
|
director := func(req *http.Request) {
|
|
|
|
req.URL.Scheme = targetUrl.Scheme
|
|
req.URL.Host = targetUrl.Host
|
|
req.Host = targetUrl.Host
|
|
|
|
req.URL.Path = util.JoinUrlFragments(targetUrl.Path, proxyPath)
|
|
|
|
// clear cookie headers
|
|
req.Header.Del("Cookie")
|
|
req.Header.Del("Set-Cookie")
|
|
|
|
// clear X-Forwarded Host/Port/Proto headers
|
|
req.Header.Del("X-Forwarded-Host")
|
|
req.Header.Del("X-Forwarded-Port")
|
|
req.Header.Del("X-Forwarded-Proto")
|
|
|
|
// set X-Forwarded-For header
|
|
if req.RemoteAddr != "" {
|
|
remoteAddr, _, err := net.SplitHostPort(req.RemoteAddr)
|
|
if err != nil {
|
|
remoteAddr = req.RemoteAddr
|
|
}
|
|
if req.Header.Get("X-Forwarded-For") != "" {
|
|
req.Header.Set("X-Forwarded-For", req.Header.Get("X-Forwarded-For")+", "+remoteAddr)
|
|
} else {
|
|
req.Header.Set("X-Forwarded-For", remoteAddr)
|
|
}
|
|
}
|
|
|
|
// Create a HTTP header with the context in it.
|
|
ctxJson, err := json.Marshal(ctx.SignedInUser)
|
|
if err != nil {
|
|
ctx.JsonApiErr(500, "failed to marshal context to json.", err)
|
|
return
|
|
}
|
|
|
|
req.Header.Add("X-Grafana-Context", string(ctxJson))
|
|
|
|
if len(route.Headers) > 0 {
|
|
headers, err := getHeaders(route, ctx.OrgId, appId)
|
|
if err != nil {
|
|
ctx.JsonApiErr(500, "Could not generate plugin route header", err)
|
|
return
|
|
}
|
|
|
|
for key, value := range headers {
|
|
log.Trace("setting key %v value %v", key, value[0])
|
|
req.Header.Set(key, value[0])
|
|
}
|
|
}
|
|
|
|
// reqBytes, _ := httputil.DumpRequestOut(req, true);
|
|
// log.Trace("Proxying plugin request: %s", string(reqBytes))
|
|
}
|
|
|
|
return &httputil.ReverseProxy{Director: director}
|
|
}
|