62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package step
|
|
|
|
import "io"
|
|
import "bytes"
|
|
import "net/http"
|
|
|
|
// HTTPData represents information about an ongoing HTTP request that is made
|
|
// available to templates as they are being executed.
|
|
type HTTPData struct {
|
|
// Res is like an http.ResponseWriter.
|
|
Res struct {
|
|
Header http.Header
|
|
WriteHeader func (statusCode int)
|
|
}
|
|
// Req is the HTTP request.
|
|
Req *http.Request
|
|
}
|
|
|
|
var _ http.ResponseWriter = new(HTTPResponseRecorder)
|
|
|
|
// HTTPResponseRecorder is an http.ResponseWriter that can buffer a response to
|
|
// be played back later.
|
|
type HTTPResponseRecorder struct {
|
|
statusCode int
|
|
header http.Header
|
|
buffer bytes.Buffer
|
|
}
|
|
|
|
func (this *HTTPResponseRecorder) Header () http.Header {
|
|
if this.header == nil {
|
|
this.header = make(http.Header)
|
|
}
|
|
return this.header
|
|
}
|
|
|
|
func (this *HTTPResponseRecorder) SetHeader (header http.Header) {
|
|
this.header = header
|
|
}
|
|
|
|
func (this *HTTPResponseRecorder) Write (buffer []byte) (int, error) {
|
|
return this.buffer.Write(buffer)
|
|
}
|
|
|
|
func (this *HTTPResponseRecorder) WriteHeader (statusCode int) {
|
|
this.statusCode = statusCode
|
|
}
|
|
|
|
// Play replays the response to the given http.ResponseWriter. This resets the
|
|
// recorder.
|
|
func (this *HTTPResponseRecorder) Play (res http.ResponseWriter) error {
|
|
defer this.Reset()
|
|
_, err := io.Copy(res, &this.buffer)
|
|
return err
|
|
}
|
|
|
|
// Reset resets this response recorder so it can be used again.
|
|
func (this *HTTPResponseRecorder) Reset () {
|
|
this.buffer.Reset()
|
|
this.header = nil
|
|
this.statusCode = http.StatusOK
|
|
}
|