101 lines
2.1 KiB
Go
101 lines
2.1 KiB
Go
|
package response
|
||
|
|
|
||
|
|
import(
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
|
||
|
|
"httpfromtcp/internal/headers"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Response struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
type Writer struct{
|
||
|
|
writer io.Writer
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewWriter(writer io.Writer)*Writer{
|
||
|
|
return &Writer{writer:writer}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *Writer) WriteStatusLine(statusCode StatusCode) error{
|
||
|
|
statusLine := []byte{}
|
||
|
|
|
||
|
|
switch statusCode{
|
||
|
|
case StatusOK: statusLine = []byte("HTTP/1.1 200 OK\r\n")
|
||
|
|
case StatusBadRequest: statusLine = []byte("HTTP/1.1 400 Bad Request\r\n")
|
||
|
|
case StatusInternalServerError: statusLine = []byte("HTTP/1.1 500 Internal Server Error\r\n")
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("unrecognized error code")
|
||
|
|
}
|
||
|
|
|
||
|
|
_,err := w.writer.Write(statusLine)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *Writer) WriteHeaders(h headers.Headers) error{
|
||
|
|
b := []byte{}
|
||
|
|
h.ForEach(func(n,v string){
|
||
|
|
b = fmt.Appendf(b, "%s: %s\r\n",n,v)
|
||
|
|
})
|
||
|
|
b = fmt.Appendf(b,"\r\n")
|
||
|
|
_,err := w.writer.Write(b)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *Writer) WriteBody(p []byte) (int, error){
|
||
|
|
return w.writer.Write(p)
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
type StatusCode int
|
||
|
|
|
||
|
|
const (
|
||
|
|
StatusOK StatusCode = 200
|
||
|
|
StatusBadRequest StatusCode = 400
|
||
|
|
StatusInternalServerError StatusCode = 500
|
||
|
|
)
|
||
|
|
|
||
|
|
func WriteHeaders(w io.Writer, h *headers.Headers) error{
|
||
|
|
b := []byte{}
|
||
|
|
h.ForEach(func(n,v string){
|
||
|
|
b = fmt.Appendf(b, "%s: %s\r\n",n,v)
|
||
|
|
})
|
||
|
|
b = fmt.Appendf(b,"\r\n")
|
||
|
|
_,err := w.Write(b)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetDefaultHeaders(contentLen int) *headers.Headers{
|
||
|
|
h := headers.NewHeaders()
|
||
|
|
h.Set("Content-Length", fmt.Sprintf("%d",contentLen))
|
||
|
|
h.Set("Connection", "close")
|
||
|
|
h.Set("Content-Type","text/plain")
|
||
|
|
|
||
|
|
return h
|
||
|
|
}
|
||
|
|
|
||
|
|
func WriteStatusLine(w io.Writer, statusCode StatusCode) error{
|
||
|
|
statusLine := []byte{}
|
||
|
|
|
||
|
|
switch statusCode{
|
||
|
|
case StatusOK: statusLine = []byte("HTTP/1.1 200 OK\r\n")
|
||
|
|
case StatusBadRequest: statusLine = []byte("HTTP/1.1 400 Bad Request\r\n")
|
||
|
|
case StatusInternalServerError: statusLine = []byte("HTTP/1.1 500 Internal Server Error\r\n")
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("unrecognized error code")
|
||
|
|
}
|
||
|
|
|
||
|
|
_,err := w.Write(statusLine)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
//func (w *Writer) WriteChunkedBody(p []byte) (int,error){
|
||
|
|
|
||
|
|
//}
|
||
|
|
|
||
|
|
//func (w *Writer) WriteChunkedBodyDone() (int,error){
|
||
|
|
|
||
|
|
//}
|