98 lines
1.9 KiB
Go
98 lines
1.9 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"path"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
var adminLink string
|
|
|
|
func init() {
|
|
flag.StringVar(&adminLink, "admin-link", adminLink, "URL to admin interface, to use its features as replacement")
|
|
}
|
|
|
|
func fwdAdmin(method, urlpath string, body io.Reader, out interface{}) error {
|
|
u, err := url.Parse(adminLink)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var user, pass string
|
|
if u.User != nil {
|
|
user = u.User.Username()
|
|
pass, _ = u.User.Password()
|
|
u.User = nil
|
|
}
|
|
|
|
u.Path = path.Join(u.Path, urlpath)
|
|
|
|
r, err := http.NewRequest(method, u.String(), body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(user) != 0 || len(pass) != 0 {
|
|
r.SetBasicAuth(user, pass)
|
|
}
|
|
|
|
resp, err := http.DefaultClient.Do(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
dec := json.NewDecoder(resp.Body)
|
|
return dec.Decode(&out)
|
|
}
|
|
|
|
func fwdAdminRequest(c *gin.Context, urlpath string) {
|
|
u, err := url.Parse(adminLink)
|
|
if err != nil {
|
|
c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
|
|
return
|
|
}
|
|
|
|
var user, pass string
|
|
if u.User != nil {
|
|
user = u.User.Username()
|
|
pass, _ = u.User.Password()
|
|
u.User = nil
|
|
}
|
|
|
|
if len(urlpath) > 0 {
|
|
u.Path = path.Join(u.Path, urlpath)
|
|
} else {
|
|
u.Path = path.Join(u.Path, c.Request.URL.Path)
|
|
}
|
|
|
|
r, err := http.NewRequest(c.Request.Method, u.String(), c.Request.Body)
|
|
if err != nil {
|
|
c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
|
|
return
|
|
}
|
|
|
|
if len(user) != 0 || len(pass) != 0 {
|
|
r.SetBasicAuth(user, pass)
|
|
}
|
|
|
|
resp, err := http.DefaultClient.Do(r)
|
|
if err != nil {
|
|
c.AbortWithStatusJSON(http.StatusBadGateway, gin.H{"errmsg": err.Error()})
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
headers := map[string]string{}
|
|
for key := range resp.Header {
|
|
headers[key] = resp.Header.Get(key)
|
|
}
|
|
|
|
c.DataFromReader(resp.StatusCode, resp.ContentLength, resp.Header.Get("Content-Type"), resp.Body, headers)
|
|
}
|