mirror of
https://github.com/ncarlier/webhookd.git
synced 2024-09-20 08:55:28 +00:00
parent
39ab72bb30
commit
aad75eba08
|
@ -12,6 +12,7 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/ncarlier/webhookd/pkg/config"
|
"github.com/ncarlier/webhookd/pkg/config"
|
||||||
|
"github.com/ncarlier/webhookd/pkg/helper"
|
||||||
"github.com/ncarlier/webhookd/pkg/hook"
|
"github.com/ncarlier/webhookd/pkg/hook"
|
||||||
"github.com/ncarlier/webhookd/pkg/worker"
|
"github.com/ncarlier/webhookd/pkg/worker"
|
||||||
)
|
)
|
||||||
|
@ -23,6 +24,8 @@ var (
|
||||||
outputDir string
|
outputDir string
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var supportedContentTypes = []string{"text/plain", "text/event-stream", "application/json", "text/*"}
|
||||||
|
|
||||||
func atoiFallback(str string, fallback int) int {
|
func atoiFallback(str string, fallback int) int {
|
||||||
if value, err := strconv.Atoi(str); err == nil && value > 0 {
|
if value, err := strconv.Atoi(str); err == nil && value > 0 {
|
||||||
return value
|
return value
|
||||||
|
@ -118,17 +121,14 @@ func triggerWebhook(w http.ResponseWriter, r *http.Request) {
|
||||||
// Put work in queue
|
// Put work in queue
|
||||||
worker.WorkQueue <- job
|
worker.WorkQueue <- job
|
||||||
|
|
||||||
// Use content negotiation to enable Server-Sent Events
|
// Use content negotiation
|
||||||
useSSE := r.Method == "GET" && r.Header.Get("Accept") == "text/event-stream"
|
ct = helper.NegotiateContentType(r, supportedContentTypes, "text/plain")
|
||||||
if useSSE {
|
|
||||||
// Send SSE response
|
// set respons headers
|
||||||
w.Header().Set("Content-Type", "text/event-stream")
|
w.Header().Set("Content-Type", ct+"; charset=utf-8")
|
||||||
} else {
|
|
||||||
// Send chunked response
|
|
||||||
w.Header().Set("X-Content-Type-Options", "nosniff")
|
|
||||||
}
|
|
||||||
w.Header().Set("Cache-Control", "no-cache")
|
w.Header().Set("Cache-Control", "no-cache")
|
||||||
w.Header().Set("Connection", "keep-alive")
|
w.Header().Set("Connection", "keep-alive")
|
||||||
|
w.Header().Set("X-Content-Type-Options", "nosniff")
|
||||||
w.Header().Set("X-Hook-ID", strconv.FormatUint(job.ID(), 10))
|
w.Header().Set("X-Hook-ID", strconv.FormatUint(job.ID(), 10))
|
||||||
|
|
||||||
for {
|
for {
|
||||||
|
@ -136,7 +136,7 @@ func triggerWebhook(w http.ResponseWriter, r *http.Request) {
|
||||||
if !open {
|
if !open {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
if useSSE {
|
if ct == "text/event-stream" {
|
||||||
fmt.Fprintf(w, "data: %s\n\n", msg) // Send SSE response
|
fmt.Fprintf(w, "data: %s\n\n", msg) // Send SSE response
|
||||||
} else {
|
} else {
|
||||||
fmt.Fprintf(w, "%s\n", msg) // Send chunked response
|
fmt.Fprintf(w, "%s\n", msg) // Send chunked response
|
||||||
|
|
151
pkg/helper/header/header.go
Normal file
151
pkg/helper/header/header.go
Normal file
|
@ -0,0 +1,151 @@
|
||||||
|
// Copyright 2013 The Go Authors. All rights reserved.
|
||||||
|
//
|
||||||
|
// Use of this source code is governed by a BSD-style
|
||||||
|
// license that can be found in the LICENSE file or at
|
||||||
|
// https://developers.google.com/open-source/licenses/bsd.
|
||||||
|
// Package header provides functions for parsing HTTP headers.
|
||||||
|
package header
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
var octetTypes [256]octetType
|
||||||
|
|
||||||
|
type octetType byte
|
||||||
|
|
||||||
|
const (
|
||||||
|
isToken octetType = 1 << iota
|
||||||
|
isSpace
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
// OCTET = <any 8-bit sequence of data>
|
||||||
|
// CHAR = <any US-ASCII character (octets 0 - 127)>
|
||||||
|
// CTL = <any US-ASCII control character (octets 0 - 31) and DEL (127)>
|
||||||
|
// CR = <US-ASCII CR, carriage return (13)>
|
||||||
|
// LF = <US-ASCII LF, linefeed (10)>
|
||||||
|
// SP = <US-ASCII SP, space (32)>
|
||||||
|
// HT = <US-ASCII HT, horizontal-tab (9)>
|
||||||
|
// <"> = <US-ASCII double-quote mark (34)>
|
||||||
|
// CRLF = CR LF
|
||||||
|
// LWS = [CRLF] 1*( SP | HT )
|
||||||
|
// TEXT = <any OCTET except CTLs, but including LWS>
|
||||||
|
// separators = "(" | ")" | "<" | ">" | "@" | "," | ";" | ":" | "\" | <">
|
||||||
|
// | "/" | "[" | "]" | "?" | "=" | "{" | "}" | SP | HT
|
||||||
|
// token = 1*<any CHAR except CTLs or separators>
|
||||||
|
// qdtext = <any TEXT except <">>
|
||||||
|
for c := 0; c < 256; c++ {
|
||||||
|
var t octetType
|
||||||
|
isCtl := c <= 31 || c == 127
|
||||||
|
isChar := 0 <= c && c <= 127
|
||||||
|
isSeparator := strings.ContainsRune(" \t\"(),/:;<=>?@[]\\{}", rune(c))
|
||||||
|
if strings.ContainsRune(" \t\r\n", rune(c)) {
|
||||||
|
t |= isSpace
|
||||||
|
}
|
||||||
|
if isChar && !isCtl && !isSeparator {
|
||||||
|
t |= isToken
|
||||||
|
}
|
||||||
|
octetTypes[c] = t
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Copy returns a shallow copy of the header.
|
||||||
|
func Copy(header http.Header) http.Header {
|
||||||
|
h := make(http.Header)
|
||||||
|
for k, vs := range header {
|
||||||
|
h[k] = vs
|
||||||
|
}
|
||||||
|
return h
|
||||||
|
}
|
||||||
|
|
||||||
|
// AcceptSpec describes an Accept* header.
|
||||||
|
type AcceptSpec struct {
|
||||||
|
Value string
|
||||||
|
Q float64
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseAccept parses Accept* headers.
|
||||||
|
func ParseAccept(header http.Header, key string) (specs []AcceptSpec) {
|
||||||
|
loop:
|
||||||
|
for _, s := range header[key] {
|
||||||
|
for {
|
||||||
|
var spec AcceptSpec
|
||||||
|
spec.Value, s = expectTokenSlash(s)
|
||||||
|
if spec.Value == "" {
|
||||||
|
continue loop
|
||||||
|
}
|
||||||
|
spec.Q = 1.0
|
||||||
|
s = skipSpace(s)
|
||||||
|
if strings.HasPrefix(s, ";") {
|
||||||
|
s = skipSpace(s[1:])
|
||||||
|
if !strings.HasPrefix(s, "q=") {
|
||||||
|
continue loop
|
||||||
|
}
|
||||||
|
spec.Q, s = expectQuality(s[2:])
|
||||||
|
if spec.Q < 0.0 {
|
||||||
|
continue loop
|
||||||
|
}
|
||||||
|
}
|
||||||
|
specs = append(specs, spec)
|
||||||
|
s = skipSpace(s)
|
||||||
|
if !strings.HasPrefix(s, ",") {
|
||||||
|
continue loop
|
||||||
|
}
|
||||||
|
s = skipSpace(s[1:])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func skipSpace(s string) (rest string) {
|
||||||
|
i := 0
|
||||||
|
for ; i < len(s); i++ {
|
||||||
|
if octetTypes[s[i]]&isSpace == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return s[i:]
|
||||||
|
}
|
||||||
|
|
||||||
|
func expectTokenSlash(s string) (token, rest string) {
|
||||||
|
i := 0
|
||||||
|
for ; i < len(s); i++ {
|
||||||
|
b := s[i]
|
||||||
|
if (octetTypes[b]&isToken == 0) && b != '/' {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return s[:i], s[i:]
|
||||||
|
}
|
||||||
|
|
||||||
|
func expectQuality(s string) (q float64, rest string) {
|
||||||
|
switch {
|
||||||
|
case len(s) == 0:
|
||||||
|
return -1, ""
|
||||||
|
case s[0] == '0':
|
||||||
|
q = 0
|
||||||
|
case s[0] == '1':
|
||||||
|
q = 1
|
||||||
|
default:
|
||||||
|
return -1, ""
|
||||||
|
}
|
||||||
|
s = s[1:]
|
||||||
|
if !strings.HasPrefix(s, ".") {
|
||||||
|
return q, s
|
||||||
|
}
|
||||||
|
s = s[1:]
|
||||||
|
i := 0
|
||||||
|
n := 0
|
||||||
|
d := 1
|
||||||
|
for ; i < len(s); i++ {
|
||||||
|
b := s[i]
|
||||||
|
if b < '0' || b > '9' {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
n = n*10 + int(b) - '0'
|
||||||
|
d *= 10
|
||||||
|
}
|
||||||
|
return q + float64(n)/float64(d), s[i:]
|
||||||
|
}
|
64
pkg/helper/negociate.go
Normal file
64
pkg/helper/negociate.go
Normal file
|
@ -0,0 +1,64 @@
|
||||||
|
package helper
|
||||||
|
|
||||||
|
// Copyright 2013 The Go Authors. All rights reserved.
|
||||||
|
//
|
||||||
|
// Use of this source code is governed by a BSD-style
|
||||||
|
// license that can be found in the LICENSE file or at
|
||||||
|
// https://developers.google.com/open-source/licenses/bsd.
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/ncarlier/webhookd/pkg/helper/header"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NegotiateContentType returns the best offered content type for the request's
|
||||||
|
// Accept header. If two offers match with equal weight, then the more specific
|
||||||
|
// offer is preferred. For example, text/* trumps */*. If two offers match
|
||||||
|
// with equal weight and specificity, then the offer earlier in the list is
|
||||||
|
// preferred. If no offers match, then defaultOffer is returned.
|
||||||
|
func NegotiateContentType(r *http.Request, offers []string, defaultOffer string) string {
|
||||||
|
bestOffer := defaultOffer
|
||||||
|
bestQ := -1.0
|
||||||
|
bestWild := 3
|
||||||
|
specs := header.ParseAccept(r.Header, "Accept")
|
||||||
|
for _, offer := range offers {
|
||||||
|
for _, spec := range specs {
|
||||||
|
switch {
|
||||||
|
case spec.Q == 0.0:
|
||||||
|
// ignore
|
||||||
|
case spec.Q < bestQ:
|
||||||
|
// better match found
|
||||||
|
case spec.Value == "*/*":
|
||||||
|
if spec.Q > bestQ || bestWild > 2 {
|
||||||
|
bestQ = spec.Q
|
||||||
|
bestWild = 2
|
||||||
|
bestOffer = offer
|
||||||
|
}
|
||||||
|
case strings.HasSuffix(spec.Value, "/*"):
|
||||||
|
if strings.HasPrefix(offer, spec.Value[:len(spec.Value)-1]) &&
|
||||||
|
(spec.Q > bestQ || bestWild > 1) {
|
||||||
|
bestQ = spec.Q
|
||||||
|
bestWild = 1
|
||||||
|
bestOffer = offer
|
||||||
|
}
|
||||||
|
case strings.HasSuffix(offer, "/*"):
|
||||||
|
if strings.HasPrefix(spec.Value, offer[:len(offer)-1]) &&
|
||||||
|
(spec.Q > bestQ || bestWild > 1) {
|
||||||
|
bestQ = spec.Q
|
||||||
|
bestWild = 1
|
||||||
|
bestOffer = spec.Value
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
if spec.Value == offer &&
|
||||||
|
(spec.Q > bestQ || bestWild > 0) {
|
||||||
|
bestQ = spec.Q
|
||||||
|
bestWild = 0
|
||||||
|
bestOffer = offer
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return bestOffer
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user