初始化2

bj_power_workstation/internal/handler/proxy.go

@@ -0,0 +1,121 @@
+package handler
+
+import (
+	"io"
+	"net/http"
+	"net/url"
+	"os"
+	"strconv"
+	"strings"
+
+	"bj_power_workstation/internal/mes"
+	"bj_power_workstation/internal/pdfcache"
+	"bj_power_workstation/internal/svc"
+)
+
+// taskCurrentHandler 代理 MES GET /api/internal/station/task?dockCode=。
+// 响应体原样透传（MES 已按 {code,message,data} 封装，不再二次包装）。
+func taskCurrentHandler(ctx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		dockCode := strings.TrimSpace(r.URL.Query().Get("dockCode"))
+		if dockCode == "" {
+			fail(w, http.StatusBadRequest, "缺少 dockCode 参数")
+			return
+		}
+
+		body, _, err := ctx.Mes.Get("/api/internal/station/task", url.Values{"dockCode": {dockCode}})
+		if err != nil {
+			fail(w, http.StatusBadGateway, "MES 服务不可达："+errString(err))
+			return
+		}
+		w.Header().Set("Content-Type", "application/json; charset=utf-8")
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write(body)
+	}
+}
+
+// ---------- 工艺 PDF 直通代理（含本地预缓存） ----------
+//
+// 前端 iframe/window.open 无法携带 Authorization，故提供免 JWT 前缀 /pdfopen/<name>，
+// 内部仍走 MES /api/internal/files/<name>（由 X-API-TOKEN 保护）。
+// 启用本地预缓存后，首次访问回源下载并落盘，后续直接读本地缓存（断网也能查看）。
+// 两端点均把文件流原样转发并透传 Content-Type，不套 {code,message,data}。
+
+// pdfQueryHandler GET /api/pdf?name=<file>（需登录）
+func pdfQueryHandler(ctx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		forwardFile(w, ctx.Mes, ctx.PdfCache, strings.TrimSpace(r.URL.Query().Get("name")))
+	}
+}
+
+// pdfOpenHandler GET /pdfopen/<name>（免鉴权，供 iframe 直链）
+func pdfOpenHandler(ctx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		raw := strings.TrimPrefix(r.URL.EscapedPath(), "/pdfopen/")
+		name, err := url.PathUnescape(raw)
+		if err != nil {
+			fail(w, http.StatusBadRequest, "非法文件名")
+			return
+		}
+		forwardFile(w, ctx.Mes, ctx.PdfCache, name)
+	}
+}
+
+// forwardFile 校验单段文件名后拉取 MES 文件流并原样转发；优先命中本地预缓存。
+func forwardFile(w http.ResponseWriter, mesc *mes.Client, cache *pdfcache.Cache, name string) {
+	name = strings.TrimSpace(name)
+	if name == "" {
+		fail(w, http.StatusBadRequest, "缺少文件名")
+		return
+	}
+	if strings.ContainsAny(name, `/\`) || name == "." || name == ".." || strings.Contains(name, "..") {
+		fail(w, http.StatusBadRequest, "非法文件名")
+		return
+	}
+
+	// 1) 命中本地预缓存 → 直接回源本地文件
+	if p, ok := cache.Get(name); ok {
+		f, err := os.Open(p)
+		if err == nil {
+			stream(f, w, name, "")
+			return
+		}
+		f.Close()
+	}
+
+	// 2) 回源 MES 下载
+	body, header, err := mesc.Get("/api/internal/files/"+url.PathEscape(name), nil)
+	if err != nil {
+		fail(w, http.StatusBadGateway, "工艺文件获取失败："+errString(err))
+		return
+	}
+
+	// 3) 写入本地预缓存（失败不影响本次响应）
+	_, _ = cache.Put(name, body)
+	streamBytes(body, w, name, header.Get("Content-Type"))
+}
+
+func stream(f io.ReadCloser, w http.ResponseWriter, name, contentType string) {
+	defer f.Close()
+	ct := contentType
+	if ct == "" {
+		ct = "application/pdf"
+	}
+	w.Header().Set("Content-Type", ct)
+	w.Header().Set("X-Original-Filename", strconv.Quote(name))
+	w.Header().Set("X-PDF-Cache", "hit")
+	w.WriteHeader(http.StatusOK)
+	_, _ = io.Copy(w, f)
+}
+
+func streamBytes(body []byte, w http.ResponseWriter, name, contentType string) {
+	ct := contentType
+	if ct == "" {
+		ct = "application/pdf"
+	}
+	w.Header().Set("Content-Type", ct)
+	w.Header().Set("X-Original-Filename", strconv.Quote(name))
+	w.Header().Set("X-PDF-Cache", "miss")
+	w.WriteHeader(http.StatusOK)
+	_, _ = w.Write(body)
+}