mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-04 05:00:25 +08:00
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
|
|
|
|
package cluster
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
|
"github.com/iwind/TeaGo/Tea"
|
|
"github.com/iwind/TeaGo/types"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"regexp"
|
|
)
|
|
|
|
type DownloadInstallerAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *DownloadInstallerAction) Init() {
|
|
this.Nav("", "", "")
|
|
}
|
|
|
|
func (this *DownloadInstallerAction) RunGet(params struct {
|
|
Name string
|
|
}) {
|
|
if len(params.Name) == 0 {
|
|
this.ResponseWriter.WriteHeader(http.StatusNotFound)
|
|
this.WriteString("file not found")
|
|
return
|
|
}
|
|
|
|
// 检查文件名
|
|
// 以防止路径穿越等风险
|
|
if !regexp.MustCompile(`^[a-zA-Z0-9.-]+$`).MatchString(params.Name) {
|
|
this.ResponseWriter.WriteHeader(http.StatusNotFound)
|
|
this.WriteString("file not found")
|
|
return
|
|
}
|
|
|
|
var zipFile = Tea.Root + "/edge-api/deploy/" + params.Name
|
|
fp, err := os.OpenFile(zipFile, os.O_RDWR, 0444)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
this.ResponseWriter.WriteHeader(http.StatusNotFound)
|
|
this.WriteString("file not found")
|
|
return
|
|
}
|
|
|
|
this.ResponseWriter.WriteHeader(http.StatusInternalServerError)
|
|
this.WriteString("file can not be opened")
|
|
|
|
return
|
|
}
|
|
|
|
defer func() {
|
|
_ = fp.Close()
|
|
}()
|
|
|
|
stat, err := fp.Stat()
|
|
if err != nil {
|
|
this.ResponseWriter.WriteHeader(http.StatusInternalServerError)
|
|
this.WriteString("file can not be opened")
|
|
return
|
|
}
|
|
|
|
this.AddHeader("Content-Disposition", "attachment; filename=\""+params.Name+"\";")
|
|
this.AddHeader("Content-Type", "application/zip")
|
|
this.AddHeader("Content-Length", types.String(stat.Size()))
|
|
_, _ = io.Copy(this.ResponseWriter, fp)
|
|
}
|