fix: fix content-type header in response

arkbriar · arkbriar · commit 14b009aa3163 · 2025-01-21T15:44:43.000+08:00
Signed-off-by: arkbriar &lt;arkbriar@gmail.com&gt;
diff --git a/proxy.go b/proxy.go
@@ -90,13 +90,18 @@ func (p *proxy) do(writer http.ResponseWriter, request *http.Request) {
 	}
 	defer resp.Body.Close()
 
-	writer.WriteHeader(resp.StatusCode)
 	for key, values := range resp.Header {
 		for _, value := range values {
 			writer.Header().Add(key, value)
 		}
 	}
-	_, _ = io.Copy(writer, resp.Body)
+	writer.WriteHeader(resp.StatusCode)
+
+	_, err = io.Copy(writer, resp.Body)
+	if err != nil {
+		http.Error(writer, err.Error(), http.StatusInternalServerError)
+		return
+	}
 }
 
 func (p *proxy) proxyMatchTarget(path string, writer http.ResponseWriter, request *http.Request) {

Original file line number	Diff line number	Diff line change
`@@ -90,13 +90,18 @@ func (p proxy) do(writer http.ResponseWriter, request http.Request) {`
`90`	`90`	`}`
`91`	`91`	`defer resp.Body.Close()`
`92`	`92`
`93`		`- writer.WriteHeader(resp.StatusCode)`
`94`	`93`	`for key, values := range resp.Header {`
`95`	`94`	`for _, value := range values {`
`96`	`95`	`writer.Header().Add(key, value)`
`97`	`96`	`}`
`98`	`97`	`}`
`99`		`- _, _ = io.Copy(writer, resp.Body)`
	`98`	`+ writer.WriteHeader(resp.StatusCode)`
	`99`	`+`
	`100`	`+ _, err = io.Copy(writer, resp.Body)`
	`101`	`+ if err != nil {`
	`102`	`+ http.Error(writer, err.Error(), http.StatusInternalServerError)`
	`103`	`+ return`
	`104`	`+ }`
`100`	`105`	`}`
`101`	`106`
`102`	`107`	`func (p proxy) proxyMatchTarget(path string, writer http.ResponseWriter, request http.Request) {`