add entire query and headers as well

This commit is contained in:
Adnan Hajdarevic 2015-06-06 14:25:32 +02:00
parent 9c545a745f
commit cbe2440cda

View file

@ -21,6 +21,8 @@ const (
SourcePayload string = "payload"
SourceString string = "string"
SourceEntirePayload string = "entire-payload"
SourceEntireQuery string = "entire-query"
SourceEntireHeaders string = "entire-headers"
)
// CheckPayloadSignature calculates and verifies SHA1 signature of the given payload
@ -159,6 +161,22 @@ func (ha *Argument) Get(headers, query, payload *map[string]interface{}) (string
return "", false
}
return string(r), true
case SourceEntireHeaders:
r, err := json.Marshal(headers)
if err != nil {
return "", false
}
return string(r), true
case SourceEntireQuery:
r, err := json.Marshal(query)
if err != nil {
return "", false
}
return string(r), true
}