src_response.go

package hlfhr

import (
	"bytes"
	"io"
	"net/http"
	"time"
)

type response struct {
	resp http.Response
	body *bytes.Buffer
}

func newResponse() *response {
	return &response{
		resp: http.Response{
			ProtoMajor: 1,
			ProtoMinor: 1,
			Close:      true,
			Header: http.Header{
				"Date": []string{time.Now().UTC().Format(http.TimeFormat)},
			},
		},
	}
}

func (r *response) Header() http.Header {
	if r.resp.StatusCode != 0 {
		// wrote header
		return http.Header{}
	}
	return r.resp.Header
}

func (r *response) Write(b []byte) (int, error) {
	if r.body == nil {
		r.body = bytes.NewBuffer(b)
		return len(b), nil
	}
	return r.body.Write(b)
}

func (r *response) WriteString(s string) (int, error) {
	if r.body == nil {
		r.body = bytes.NewBufferString(s)
		return len(s), nil
	}
	return r.body.WriteString(s)
}

func (r *response) WriteHeader(statusCode int) {
	if r.resp.StatusCode == 0 {
		// not wrote header
		r.resp.StatusCode = statusCode
	}
}

// flush flushes buffered data to the client.
func (r *response) flush(w io.Writer) error {
	r.WriteHeader(400)
	if r.body != nil {
		r.resp.ContentLength = int64(r.body.Len())
		r.resp.Body = io.NopCloser(r.body)
	}
	return r.resp.Write(w)
}