initial
This commit is contained in:
parent
6330e844bc
commit
c303b61eb2
2
.gitignore
vendored
2
.gitignore
vendored
@ -1,3 +1,3 @@
|
|||||||
.env
|
.env
|
||||||
deploy.sh
|
deploy.sh
|
||||||
palanticshelper
|
/palanticshelper
|
||||||
132
pkg/palanticshelper/server.go
Normal file
132
pkg/palanticshelper/server.go
Normal file
@ -0,0 +1,132 @@
|
|||||||
|
package palanticshelper
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"gitea.karlbreuer.com/karl/palanticshelper/pkg/settings"
|
||||||
|
"github.com/go-chi/chi"
|
||||||
|
"github.com/go-chi/chi/middleware"
|
||||||
|
"github.com/go-chi/cors"
|
||||||
|
)
|
||||||
|
|
||||||
|
func StartServer() {
|
||||||
|
r := chi.NewRouter()
|
||||||
|
r.Use(middleware.RealIP)
|
||||||
|
|
||||||
|
allowedOrigins := []string{"*." + settings.Settings.Domain}
|
||||||
|
if settings.Settings.Debug {
|
||||||
|
allowedOrigins = []string{"*"}
|
||||||
|
}
|
||||||
|
|
||||||
|
r.Use(cors.Handler(cors.Options{
|
||||||
|
AllowedOrigins: allowedOrigins,
|
||||||
|
AllowedHeaders: []string{"Accept", "Authorization", "Content-Type", "X-CSRF-Token", "Id"},
|
||||||
|
AllowedMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS", "Settings"},
|
||||||
|
ExposedHeaders: []string{"Link"},
|
||||||
|
AllowCredentials: false,
|
||||||
|
MaxAge: 300, // Maximum value not ignored by any of major browsers
|
||||||
|
}))
|
||||||
|
|
||||||
|
r.Get("/healthz", health)
|
||||||
|
r.Post("/*", forwardToTracker)
|
||||||
|
|
||||||
|
portNum, err := strconv.Atoi(strings.TrimSpace(settings.Settings.Port))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Invalid port number: %v", err)
|
||||||
|
}
|
||||||
|
srv := &http.Server{
|
||||||
|
Handler: r,
|
||||||
|
Addr: fmt.Sprintf(":%d", portNum),
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("runs on: %s", srv.Addr)
|
||||||
|
err = srv.ListenAndServe()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func forwardToTracker(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if err := r.ParseForm(); err != nil {
|
||||||
|
log.Printf("Error parsing form: %v", err)
|
||||||
|
RespondWithJSON(w, http.StatusBadRequest, struct{}{})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
forwardURL := "https://" + settings.Settings.TrackingServerURL + "/spur"
|
||||||
|
|
||||||
|
req, err := http.NewRequest(r.Method, forwardURL, nil)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Error creating request: %v", err)
|
||||||
|
RespondWithJSON(w, http.StatusInternalServerError, struct{}{})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
q := r.URL.Query()
|
||||||
|
|
||||||
|
realIp := getRealIP(r)
|
||||||
|
q.Add("ip", realIp)
|
||||||
|
|
||||||
|
req.URL.RawQuery = q.Encode()
|
||||||
|
|
||||||
|
for name, values := range r.Header {
|
||||||
|
for _, value := range values {
|
||||||
|
req.Header.Add(name, value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Error forwarding request: %v", err)
|
||||||
|
RespondWithJSON(w, http.StatusInternalServerError, struct{}{})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
RespondWithJSON(w, http.StatusOK, struct{}{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func health(w http.ResponseWriter, r *http.Request) {
|
||||||
|
RespondWithJSON(w, 200, struct{}{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func RespondWithJSON(w http.ResponseWriter, statuscode int, payload interface{}) {
|
||||||
|
var dat []byte
|
||||||
|
w.Header().Add("Content-Type", "application/json")
|
||||||
|
dat, _ = json.Marshal(payload)
|
||||||
|
w.WriteHeader(statuscode)
|
||||||
|
w.Write(dat)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getRealIP(r *http.Request) string {
|
||||||
|
// NGINX
|
||||||
|
ip := r.Header.Get("X-Real-IP")
|
||||||
|
if ip != "" {
|
||||||
|
return ip
|
||||||
|
}
|
||||||
|
// TRAEFIK
|
||||||
|
forwardedFor := r.Header.Get("X-Forwarded-For")
|
||||||
|
if forwardedFor != "" {
|
||||||
|
// X-Forwarded-For might contain multiple IPs - the leftmost is the original client
|
||||||
|
ips := strings.Split(forwardedFor, ",")
|
||||||
|
if len(ips) > 0 {
|
||||||
|
return strings.TrimSpace(ips[0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("ERROR getting forwared IP!")
|
||||||
|
ip, _, err := net.SplitHostPort(r.RemoteAddr)
|
||||||
|
if err != nil {
|
||||||
|
|
||||||
|
ip = r.RemoteAddr
|
||||||
|
}
|
||||||
|
|
||||||
|
return ip
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user