mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-04 05:00:25 +08:00
40 lines
919 B
Go
40 lines
919 B
Go
package waf
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/ttlcache"
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
|
"github.com/iwind/TeaGo/types"
|
|
"strconv"
|
|
)
|
|
|
|
type ExportDownloadAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *ExportDownloadAction) Init() {
|
|
this.Nav("", "", "")
|
|
}
|
|
|
|
func (this *ExportDownloadAction) RunGet(params struct {
|
|
Key string
|
|
PolicyId int64
|
|
}) {
|
|
item := ttlcache.DefaultCache.Read(params.Key)
|
|
if item == nil || item.Value == nil {
|
|
this.WriteString("找不到要导出的内容")
|
|
return
|
|
}
|
|
|
|
ttlcache.DefaultCache.Delete(params.Key)
|
|
|
|
data, ok := item.Value.([]byte)
|
|
if ok {
|
|
this.AddHeader("Content-Disposition", "attachment; filename=\"WAF-"+types.String(params.PolicyId)+".json\";")
|
|
this.AddHeader("Content-Length", strconv.Itoa(len(data)))
|
|
_, _ = this.Write(data)
|
|
} else {
|
|
this.WriteString("找不到要导出的内容")
|
|
return
|
|
}
|
|
}
|