scriptforge/backend/internal/handler/raw.go

package handler

import (
	"errors"
	"fmt"
	"net/http"

	"github.com/gin-gonic/gin"

	"gitea.kmux.cn/zhilv/scriptforge/internal/config"
	"gitea.kmux.cn/zhilv/scriptforge/internal/service"
)

func GetRawScript(svc *service.ScriptService) gin.HandlerFunc {
	return func(c *gin.Context) {
		id := c.Param("id")

		script, err := svc.GetByID(id)
		if errors.Is(err, service.ErrNotFound) {
			c.String(http.StatusNotFound, "script not found or expired\n")
			return
		}
		if err != nil {
			c.String(http.StatusInternalServerError, "internal error\n")
			return
		}

		rt, _ := config.GetRuntime(script.Runtime)
		mime := rt.MIMEType
		ext := rt.Extension
		if mime == "" {
			mime = "text/plain"
			ext = ".sh"
		}

		c.Header("Content-Type", mime+"; charset=utf-8")
		c.Header("Content-Disposition", fmt.Sprintf(`attachment; filename="script%s"`, ext))
		c.Header("X-Content-Type-Options", "nosniff")
		c.String(http.StatusOK, script.Content)
	}
}