refactor: auto detect response type

recover.go

@@ -17,6 +17,7 @@ package recover
 import (
 	"fmt"
 	"net/http"
+	"strings"
 
 	"github.com/vicanso/hes"
 
@@ -28,15 +29,8 @@ const (
 	ErrCategory = "elton-recover"
 )
 
-type (
-	// Config json parser config
-	Config struct {
-		ResponseType string
-	}
-)
-
 // New new recover
-func New(config Config) elton.Handler {
+func New() elton.Handler {
 	return func(c *elton.Context) error {
 		defer func() {
 			// 可针对实际需求调整，如对于每个recover增加邮件通知等
@@ -64,7 +58,7 @@ func New(config Config) elton.Handler {
 				c.Committed = true
 				resp := c.Response
 				buf := []byte(err.Error())
-				if config.ResponseType == "json" {
+				if strings.Contains(c.GetRequestHeader("Accept"), "application/json") {
 					c.SetHeader(elton.HeaderContentType, elton.MIMEApplicationJSON)
 					buf = he.ToJSON()
 				}

recover_test.go

@@ -16,7 +16,7 @@ func TestRecover(t *testing.T) {
 		assert := assert.New(t)
 		var ctx *elton.Context
 		d := elton.New()
-		d.Use(New(Config{}))
+		d.Use(New())
 		d.GET("/", func(c *elton.Context) error {
 			ctx = c
 			panic("abc")
@@ -51,13 +51,12 @@ func TestRecover(t *testing.T) {
 	t.Run("response json", func(t *testing.T) {
 		assert := assert.New(t)
 		d := elton.New()
-		d.Use(New(Config{
-			ResponseType: "json",
-		}))
+		d.Use(New())
 		d.GET("/", func(c *elton.Context) error {
 			panic("abc")
 		})
 		req := httptest.NewRequest("GET", "https://aslant.site/", nil)
+		req.Header.Set("Accept", "application/json, text/plain, */*")
 		resp := httptest.NewRecorder()
 		d.ServeHTTP(resp, req)
 		assert.Equal(500, resp.Code)