wood/proxy-go
1
0
Fork 0
mirror of https://github.com/woodchen-ink/proxy-go.git synced 2025-07-18 08:31:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(handler): enhance mirror proxy handler with URL parsing and CORS headers

Browse Source
This commit is contained in:
wood chen 2024-11-13 19:50:50 +08:00
parent a7753f90d5
commit 2653014c89
1 changed files with 37 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
internal/handler/mirror_proxy.go
View File

@ -2,9 +2,11 @@
package handler
import (
"fmt"
"io"
"log"
"net/http"
"net/url"
"proxy-go/internal/utils"
"strings"
"time"
@ -23,10 +25,8 @@ func (h *MirrorProxyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS, HEAD, PATCH")
w.Header().Set("Access-Control-Allow-Headers", "*")
// 如果需要允许发送凭证（cookies等），可以设置：
// w.Header().Set("Access-Control-Allow-Credentials", "true")
// 处理 OPTIONS 请求（预检请求）
// 处理 OPTIONS 请求
if r.Method == "OPTIONS" {
w.WriteHeader(http.StatusOK)
log.Printf("| %-6s | %3d | %12s | %15s | %10s | %-30s | CORS Preflight",
@ -36,18 +36,37 @@ func (h *MirrorProxyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
}
// 从路径中提取实际URL
// 例如：/mirror/https://example.com/path 变成 https://example.com/path
actualURL := strings.TrimPrefix(r.URL.Path, "/mirror/")
if actualURL == "" || actualURL == r.URL.Path {
http.Error(w, "Invalid URL", http.StatusBadRequest)
log.Printf("| %-6s | %3d | %12s | %15s | %10s | %-30s | Invalid URL",
r.Method, http.StatusBadRequest, time.Since(startTime),
utils.GetClientIP(r), "-", r.URL.Path)
return
}
// 添加原始请求中的查询参数和片段
if r.URL.RawQuery != "" {
actualURL += "?" + r.URL.RawQuery
}
// 解析目标 URL 以获取 host
parsedURL, err := url.Parse(actualURL)
if err != nil {
http.Error(w, "Invalid URL", http.StatusBadRequest)
log.Printf("| %-6s | %3d | %12s | %15s | %10s | %-30s | Parse URL error: %v",
r.Method, http.StatusBadRequest, time.Since(startTime),
utils.GetClientIP(r), "-", actualURL, err)
return
}
// 确保有 scheme
scheme := parsedURL.Scheme
if scheme == "" {
scheme = "https"
actualURL = "https://" + actualURL
parsedURL, _ = url.Parse(actualURL)
}
// 创建新的请求
proxyReq, err := http.NewRequest(r.Method, actualURL, r.Body)
if err != nil {
@ -61,11 +80,24 @@ func (h *MirrorProxyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
// 复制原始请求的header
copyHeader(proxyReq.Header, r.Header)
// 设置必要的请求头
proxyReq.Header.Set("Origin", fmt.Sprintf("%s://%s", scheme, parsedURL.Host))
proxyReq.Header.Set("Referer", fmt.Sprintf("%s://%s/", scheme, parsedURL.Host))
if ua := r.Header.Get("User-Agent"); ua != "" {
proxyReq.Header.Set("User-Agent", ua)
} else {
proxyReq.Header.Set("User-Agent", "Mozilla/5.0")
}
proxyReq.Header.Set("Host", parsedURL.Host)
proxyReq.Host = parsedURL.Host
// 发送请求
client := &http.Client{
Transport: &http.Transport{
DisableCompression: true,
// 可以添加其他传输设置，如TLS配置等
},
Timeout: 30 * time.Second,
}
resp, err := client.Do(proxyReq)
if err != nil {
@ -77,9 +109,6 @@ func (h *MirrorProxyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
}
defer resp.Body.Close()
// 设置CORS头
w.Header().Set("Access-Control-Allow-Origin", "*")
// 复制响应头
copyHeader(w.Header(), resp.Header)