mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-04 16:00:25 +08:00
104 lines
2.6 KiB
Go
104 lines
2.6 KiB
Go
package nodes
|
||
|
||
import (
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
|
||
"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
|
||
"github.com/TeaOSLab/EdgeNode/internal/utils"
|
||
"github.com/iwind/TeaGo/Tea"
|
||
"github.com/iwind/TeaGo/logs"
|
||
"net/http"
|
||
"os"
|
||
)
|
||
|
||
// 调用临时关闭页面
|
||
func (this *HTTPRequest) doShutdown() {
|
||
shutdown := this.web.Shutdown
|
||
if shutdown == nil {
|
||
return
|
||
}
|
||
|
||
if urlPrefixRegexp.MatchString(shutdown.URL) { // URL
|
||
this.doURL(http.MethodGet, shutdown.URL, "", shutdown.Status, true)
|
||
return
|
||
}
|
||
|
||
if len(shutdown.BodyType) == 0 || shutdown.BodyType == shared.BodyTypeURL {
|
||
// URL为空，则显示文本
|
||
if len(shutdown.URL) == 0 {
|
||
// 自定义响应Headers
|
||
if shutdown.Status > 0 {
|
||
this.processResponseHeaders(shutdown.Status)
|
||
this.writer.WriteHeader(shutdown.Status)
|
||
} else {
|
||
this.processResponseHeaders(http.StatusOK)
|
||
this.writer.WriteHeader(http.StatusOK)
|
||
}
|
||
_, err := this.writer.WriteString("The site have been shutdown.")
|
||
if err != nil {
|
||
logs.Error(err)
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
// 从本地文件中读取
|
||
file := Tea.Root + Tea.DS + shutdown.URL
|
||
fp, err := os.Open(file)
|
||
if err != nil {
|
||
logs.Error(err)
|
||
msg := "404 page not found: '" + shutdown.URL + "'"
|
||
|
||
this.writer.WriteHeader(http.StatusNotFound)
|
||
_, err = this.writer.Write([]byte(msg))
|
||
if err != nil {
|
||
logs.Error(err)
|
||
}
|
||
return
|
||
}
|
||
|
||
// 自定义响应Headers
|
||
if shutdown.Status > 0 {
|
||
this.processResponseHeaders(shutdown.Status)
|
||
this.writer.WriteHeader(shutdown.Status)
|
||
} else {
|
||
this.processResponseHeaders(http.StatusOK)
|
||
this.writer.WriteHeader(http.StatusOK)
|
||
}
|
||
buf := bytePool1k.Get()
|
||
_, err = utils.CopyWithFilter(this.writer, fp, buf, func(p []byte) []byte {
|
||
return []byte(this.Format(string(p)))
|
||
})
|
||
bytePool1k.Put(buf)
|
||
if err != nil {
|
||
if !this.canIgnore(err) {
|
||
remotelogs.Warn("HTTP_REQUEST_SHUTDOWN", "write to client failed: "+err.Error())
|
||
}
|
||
} else {
|
||
this.writer.SetOk()
|
||
}
|
||
|
||
err = fp.Close()
|
||
if err != nil {
|
||
remotelogs.Warn("HTTP_REQUEST_SHUTDOWN", "close file failed: "+err.Error())
|
||
}
|
||
} else if shutdown.BodyType == shared.BodyTypeHTML {
|
||
// 自定义响应Headers
|
||
if shutdown.Status > 0 {
|
||
this.processResponseHeaders(shutdown.Status)
|
||
this.writer.WriteHeader(shutdown.Status)
|
||
} else {
|
||
this.processResponseHeaders(http.StatusOK)
|
||
this.writer.WriteHeader(http.StatusOK)
|
||
}
|
||
|
||
_, err := this.writer.WriteString(this.Format(shutdown.Body))
|
||
if err != nil {
|
||
if !this.canIgnore(err) {
|
||
remotelogs.Warn("HTTP_REQUEST_SHUTDOWN", "write to client failed: "+err.Error())
|
||
}
|
||
} else {
|
||
this.writer.SetOk()
|
||
}
|
||
}
|
||
}
|