package middleware

import (
	"bufio"
	"bytes"
	"net"
	"net/http"
	"strconv"
	"strings"

	"github.com/gin-gonic/gin"
	ginSwagger "github.com/swaggo/gin-swagger"
	swaggerFiles "github.com/swaggo/files"
)

func SwaggerHandler() gin.HandlerFunc {
	return ginSwagger.WrapHandler(swaggerFiles.Handler)
}

func SwaggerCSSBranding() gin.HandlerFunc {
	return func(c *gin.Context) {
		if !strings.HasSuffix(c.Request.URL.Path, "swagger-ui.css") {
			c.Next()
			return
		}

		underlying := c.Writer
		crw := &captureResponseWriter{
			ginWriter: underlying,
			code:      200,
		}
		c.Writer = crw

		c.Next()

		c.Writer = underlying

		if crw.code == 200 && crw.buf.Len() > 0 {
			body := append(crw.buf.Bytes(), []byte(brandingCSS)...)
			h := underlying.Header()
			h.Set("Content-Length", strconv.Itoa(len(body)))
			h.Set("Content-Type", "text/css; charset=utf-8")
			underlying.WriteHeader(crw.code)
			underlying.Write(body)
			return
		}

		if crw.buf.Len() > 0 {
			underlying.WriteHeader(crw.code)
			underlying.Write(crw.buf.Bytes())
			return
		}

		underlying.WriteHeader(crw.code)
	}
}

type captureResponseWriter struct {
	ginWriter gin.ResponseWriter
	code      int
	buf       bytes.Buffer
}

func (w *captureResponseWriter) Header() http.Header         { return w.ginWriter.Header() }
func (w *captureResponseWriter) Write(b []byte) (int, error)  { return w.buf.Write(b) }
func (w *captureResponseWriter) WriteHeader(code int)         { w.code = code }
func (w *captureResponseWriter) WriteString(s string) (int, error) { return w.buf.WriteString(s) }
func (w *captureResponseWriter) Status() int                  { return w.code }
func (w *captureResponseWriter) Size() int                    { return w.buf.Len() }
func (w *captureResponseWriter) Written() bool                { return w.buf.Len() > 0 }
func (w *captureResponseWriter) WriteHeaderNow()              {}
func (w *captureResponseWriter) Pusher() http.Pusher          { return nil }
func (w *captureResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
	return w.ginWriter.Hijack()
}
func (w *captureResponseWriter) CloseNotify() <-chan bool      { return w.ginWriter.CloseNotify() }
func (w *captureResponseWriter) Flush()                       {}

const brandingCSS = `
/* MengStack API branding — replace Swagger SmartBear logo */
.topbar {
  display: flex !important;
  align-items: center !important;
  background-color: #1b1b1f !important;
  padding: 8px 20px !important;
  min-height: 50px !important;
  position: relative !important;
}
.topbar > * {
  visibility: hidden !important;
  width: 0 !important;
  height: 0 !important;
  overflow: hidden !important;
  margin: 0 !important;
  padding: 0 !important;
  flex: 0 !important;
}
.topbar::before {
  content: "MengStack API \6587\6863" !important;
  visibility: visible !important;
  width: auto !important;
  height: auto !important;
  overflow: visible !important;
  flex: 0 0 auto !important;
  color: #fff !important;
  font-size: 20px !important;
  font-weight: 700 !important;
  font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif !important;
  letter-spacing: 1px !important;
  text-decoration: none !important;
  position: absolute !important;
  left: 20px !important;
  top: 50% !important;
  transform: translateY(-50%) !important;
}
.topbar::after {
  content: "v0.3.0" !important;
  visibility: visible !important;
  width: auto !important;
  height: auto !important;
  overflow: visible !important;
  flex: 0 0 auto !important;
  color: rgba(255,255,255,0.5) !important;
  font-size: 12px !important;
  font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif !important;
  position: absolute !important;
  left: 230px !important;
  top: 50% !important;
  transform: translateY(-50%) !important;
}
.swagger-ui {
  margin-top: 0 !important;
}
`