diff --git a/response.go b/response.go index 85b36d4..c0e4d58 100644 --- a/response.go +++ b/response.go @@ -1,61 +1,68 @@ package impart import ( "encoding/json" "net/http" "strconv" ) type ( // Envelope contains metadata and optional data for a response object. // Responses will always contain a status code and either: // - response Data on a 2xx response, or // - an ErrorMessage on non-2xx responses // // ErrorType is not currently used. Envelope struct { Code int `json:"code"` ErrorType string `json:"error_type,omitempty"` ErrorMessage string `json:"error_msg,omitempty"` Data interface{} `json:"data,omitempty"` } ) func writeBody(w http.ResponseWriter, body []byte, status int, contentType string) error { w.Header().Set("Content-Type", contentType+"; charset=UTF-8") w.Header().Set("Content-Length", strconv.Itoa(len(body))) w.WriteHeader(status) _, err := w.Write(body) return err } func renderJSON(w http.ResponseWriter, value interface{}, status int) error { body, err := json.Marshal(value) if err != nil { return err } return writeBody(w, body, status, "application/json") } func renderString(w http.ResponseWriter, status int, msg string) error { return writeBody(w, []byte(msg), status, "text/plain") } // WriteSuccess writes the successful data and metadata to the ResponseWriter as // JSON. func WriteSuccess(w http.ResponseWriter, data interface{}, status int) error { env := &Envelope{ Code: status, Data: data, } return renderJSON(w, env, status) } // WriteError writes the error to the ResponseWriter as JSON. func WriteError(w http.ResponseWriter, e HTTPError) error { env := &Envelope{ Code: e.Status, ErrorMessage: e.Message, } return renderJSON(w, env, e.Status) } + +// WriteRedirect sends a redirect +func WriteRedirect(w http.ResponseWriter, e HTTPError) int { + w.Header().Set("Location", e.Message) + w.WriteHeader(e.Status) + return e.Status +}