pangolin_mirror/badger
1
0
Fork 0
mirror of https://github.com/fosrl/badger.git synced 2026-09-01 03:31:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

support custom client error responses

Browse Source
This commit is contained in:
miloschwartz
2026-08-12 12:35:38 -04:00
parent 57b436a140
commit f5000d8b18
1 changed files with 34 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
main.go
View File

@@ -62,20 +62,27 @@ type VerifyBody struct {
BadgerVersion string `json:"badgerVersion,omitempty"`
}
type ClientErrorResponse struct {
StatusCode int `json:"statusCode,omitempty"`
ContentType string `json:"contentType,omitempty"`
Body string `json:"body"`
}
type VerifyResponse struct {
Data struct {
HeaderAuthChallenged bool `json:"headerAuthChallenged"`
Valid bool `json:"valid"`
RedirectURL *string `json:"redirectUrl"`
UserId *string `json:"userId,omitempty"`
DontStripSession bool `json:"dontStripSession,omitempty"`
Username *string `json:"username,omitempty"`
Email *string `json:"email,omitempty"`
Name *string `json:"name,omitempty"`
Role *string `json:"role,omitempty"`
VirtualApiKeyId *string `json:"virtualApiKeyId,omitempty"`
ResponseHeaders map[string]string `json:"responseHeaders,omitempty"`
PangolinVersion *string `json:"pangolinVersion,omitempty"`
HeaderAuthChallenged bool `json:"headerAuthChallenged"`
Valid bool `json:"valid"`
RedirectURL *string `json:"redirectUrl"`
UserId *string `json:"userId,omitempty"`
DontStripSession bool `json:"dontStripSession,omitempty"`
Username *string `json:"username,omitempty"`
Email *string `json:"email,omitempty"`
Name *string `json:"name,omitempty"`
Role *string `json:"role,omitempty"`
VirtualApiKeyId *string `json:"virtualApiKeyId,omitempty"`
ResponseHeaders map[string]string `json:"responseHeaders,omitempty"`
PangolinVersion *string `json:"pangolinVersion,omitempty"`
ClientError *ClientErrorResponse `json:"clientError,omitempty"`
} `json:"data"`
}
@@ -359,6 +366,21 @@ func (p *Badger) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
return
}
if result.Data.ClientError != nil && result.Data.ClientError.Body != "" {
statusCode := result.Data.ClientError.StatusCode
if statusCode == 0 {
statusCode = http.StatusUnauthorized
}
contentType := result.Data.ClientError.ContentType
if contentType == "" {
contentType = "application/json"
}
rw.Header().Set("Content-Type", contentType)
rw.WriteHeader(statusCode)
rw.Write([]byte(result.Data.ClientError.Body))
return
}
http.Error(rw, "Unauthorized", http.StatusUnauthorized)
}