1
0
mirror of https://github.com/amir20/dozzle.git synced 2025-12-27 15:41:45 +01:00

chore(refactor): refactors go code (#2443)

This commit is contained in:
Amir Raminfar
2023-10-24 06:28:16 -07:00
committed by GitHub
parent 74e44bd5f2
commit 2a07ca39db
31 changed files with 18 additions and 22 deletions

58
internal/auth/proxy.go Normal file
View File

@@ -0,0 +1,58 @@
package auth
import (
"context"
"crypto/md5"
"encoding/hex"
"net/http"
"strings"
log "github.com/sirupsen/logrus"
)
type contextKey string
const RemoteUser contextKey = "remoteUser"
type User struct {
Username string `json:"username"`
Email string `json:"email"`
Name string `json:"name"`
Avatar string `json:"avatar,omitempty"`
}
func hashEmail(email string) string {
email = strings.TrimSpace(email)
email = strings.ToLower(email)
hash := md5.Sum([]byte(email))
return hex.EncodeToString(hash[:])
}
func newUser(username, email, name string) *User {
avatar := ""
if email != "" {
avatar = "https://gravatar.com/avatar/" + hashEmail(email)
}
return &User{
Username: username,
Email: email,
Name: name,
Avatar: avatar,
}
}
func ForwardProxyAuthorizationRequired(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("Remote-Email") == "" {
log.Error("Unable to find remote email. Please check your proxy configuration. Expecting header 'Remote-Email'")
http.Error(w, "Unauthorized", http.StatusUnauthorized)
return
}
user := newUser(r.Header.Get("Remote-User"), r.Header.Get("Remote-Email"), r.Header.Get("Remote-Name"))
ctx := context.WithValue(r.Context(), RemoteUser, user)
next.ServeHTTP(w, r.WithContext(ctx))
})
}