mirror of
				https://github.com/TeaOSLab/EdgeAdmin.git
				synced 2025-11-04 13:10:26 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			98 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			98 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package actionutils
 | 
						||
 | 
						||
import (
 | 
						||
	"fmt"
 | 
						||
	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
 | 
						||
	rpcerrors "github.com/TeaOSLab/EdgeCommon/pkg/rpc/errors"
 | 
						||
	"github.com/iwind/TeaGo/actions"
 | 
						||
	"github.com/iwind/TeaGo/logs"
 | 
						||
	"net/http"
 | 
						||
	"os"
 | 
						||
	"path/filepath"
 | 
						||
	"reflect"
 | 
						||
	"runtime"
 | 
						||
	"strings"
 | 
						||
)
 | 
						||
 | 
						||
// Fail 提示服务器错误信息
 | 
						||
func Fail(action actions.ActionWrapper, err error) {
 | 
						||
	if err != nil {
 | 
						||
		logs.Println("[" + reflect.TypeOf(action).String() + "]" + findStack(err.Error()))
 | 
						||
	}
 | 
						||
	action.Object().Fail(teaconst.ErrServer + "(" + err.Error() + ")")
 | 
						||
}
 | 
						||
 | 
						||
// FailPage 提示页面错误信息
 | 
						||
func FailPage(action actions.ActionWrapper, err error) {
 | 
						||
	if err != nil {
 | 
						||
		logs.Println("[" + reflect.TypeOf(action).String() + "]" + findStack(err.Error()))
 | 
						||
	}
 | 
						||
	err = rpcerrors.HumanError(err)
 | 
						||
	action.Object().ResponseWriter.WriteHeader(http.StatusInternalServerError)
 | 
						||
	if len(action.Object().Request.Header.Get("X-Requested-With")) > 0 {
 | 
						||
		action.Object().WriteString(teaconst.ErrServer)
 | 
						||
	} else {
 | 
						||
		action.Object().WriteString(`<!DOCTYPE html>
 | 
						||
<html>
 | 
						||
	<head></head>
 | 
						||
	<body>
 | 
						||
	<div style="background: #eee; border: 1px #ccc solid; padding: 10px; font-size: 12px; line-height: 1.8">
 | 
						||
	` + teaconst.ErrServer + `
 | 
						||
	<div>可以通过查看 <strong><em>$安装目录/logs/run.log</em></strong> 日志文件查看具体的错误提示。</div>
 | 
						||
	<hr style="border-top: 1px #ccc solid"/>
 | 
						||
	<div style="color: red">Error: ` + err.Error() + `</pre>
 | 
						||
	</div>
 | 
						||
	</body>
 | 
						||
</html>`)
 | 
						||
	}
 | 
						||
}
 | 
						||
 | 
						||
// MatchPath 判断动作的文件路径是否相当
 | 
						||
func MatchPath(action *actions.ActionObject, path string) bool {
 | 
						||
	return action.Request.URL.Path == path
 | 
						||
}
 | 
						||
 | 
						||
// FindParentAction 查找父级Action
 | 
						||
func FindParentAction(actionPtr actions.ActionWrapper) *ParentAction {
 | 
						||
	parentActionValue := reflect.ValueOf(actionPtr).Elem().FieldByName("ParentAction")
 | 
						||
	if parentActionValue.IsValid() {
 | 
						||
		parentAction, isOk := parentActionValue.Interface().(ParentAction)
 | 
						||
		if isOk {
 | 
						||
			return &parentAction
 | 
						||
		}
 | 
						||
	}
 | 
						||
	return nil
 | 
						||
}
 | 
						||
 | 
						||
func findStack(err string) string {
 | 
						||
	_, currentFilename, _, currentOk := runtime.Caller(1)
 | 
						||
	if currentOk {
 | 
						||
		for i := 1; i < 32; i++ {
 | 
						||
			_, filename, lineNo, ok := runtime.Caller(i)
 | 
						||
			if !ok {
 | 
						||
				break
 | 
						||
			}
 | 
						||
 | 
						||
			if filename == currentFilename || filepath.Base(filename) == "parent_action.go" {
 | 
						||
				continue
 | 
						||
			}
 | 
						||
 | 
						||
			goPath := os.Getenv("GOPATH")
 | 
						||
			if len(goPath) > 0 {
 | 
						||
				absGoPath, err := filepath.Abs(goPath)
 | 
						||
				if err == nil {
 | 
						||
					filename = strings.TrimPrefix(filename, absGoPath)[1:]
 | 
						||
				}
 | 
						||
			} else if strings.Contains(filename, "src") {
 | 
						||
				filename = filename[strings.Index(filename, "src"):]
 | 
						||
			}
 | 
						||
 | 
						||
			err += "\n\t\t" + string(filename) + ":" + fmt.Sprintf("%d", lineNo)
 | 
						||
 | 
						||
			break
 | 
						||
		}
 | 
						||
	}
 | 
						||
 | 
						||
	return err
 | 
						||
}
 |