293 lines
11 KiB
Go
293 lines
11 KiB
Go
package cmd
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"net/http"
|
||
"net/url"
|
||
"os"
|
||
"reflect"
|
||
"strings"
|
||
"unsafe"
|
||
_ "unsafe"
|
||
|
||
"github.com/tiger1103/gfast/v3/api/v1/common/coryCommon"
|
||
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/net/ghttp"
|
||
"github.com/gogf/gf/v2/net/goai"
|
||
"github.com/gogf/gf/v2/os/gcmd"
|
||
"github.com/gogf/gf/v2/os/gctx"
|
||
"github.com/gogf/gf/v2/os/glog"
|
||
"github.com/gogf/gf/v2/os/gres"
|
||
"github.com/gogf/gf/v2/os/gsession"
|
||
"github.com/gogf/gf/v2/os/gspath"
|
||
"github.com/gogf/gf/v2/os/gview"
|
||
"github.com/gogf/gf/v2/text/gstr"
|
||
"github.com/tiger1103/gfast/v3/api/saft_hat"
|
||
"github.com/tiger1103/gfast/v3/api/video_hat"
|
||
"github.com/tiger1103/gfast/v3/internal/consts"
|
||
"github.com/tiger1103/gfast/v3/internal/router"
|
||
"github.com/tiger1103/gfast/v3/library/libValidate"
|
||
"github.com/tiger1103/gfast/v3/library/upload"
|
||
"github.com/tiger1103/gfast/v3/task"
|
||
)
|
||
|
||
var Main = gcmd.Command{
|
||
Name: "main",
|
||
Usage: "main",
|
||
Brief: "start http server",
|
||
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
|
||
g.Log().SetFlags(glog.F_ASYNC | glog.F_TIME_DATE | glog.F_TIME_TIME | glog.F_FILE_LONG)
|
||
g.Log().Info(ctx, "GFast version:", consts.Version)
|
||
s := g.Server()
|
||
getwd, err := os.Getwd()
|
||
getwd = strings.ReplaceAll(getwd, "\\", "/")
|
||
|
||
//s.Group("/file", func(group *ghttp.RouterGroup) {
|
||
// group.Middleware(commonService.Middleware().MiddlewareCORS)
|
||
//})
|
||
|
||
s.AddStaticPath("/resource", getwd+"/resource") // 指向当前资源
|
||
s.AddStaticPath(coryCommon.GlobalFile, getwd+"/resource/public") // 指向pc
|
||
s.AddStaticPath("/wxfile", getwd+"/resource/public") // 指向小程序
|
||
|
||
s.Group("/", func(group *ghttp.RouterGroup) {
|
||
router.R.BindController(ctx, group)
|
||
})
|
||
|
||
//// zm/api/wxApplet/wxApplet/
|
||
//s.BindHookHandler("/zm/api/wxApplet/wxApplet/*", ghttp.HookBeforeServe, bandWxApp)
|
||
|
||
s.BindHookHandler("/file/*", ghttp.HookBeforeServe, reStaticFile)
|
||
|
||
// 重新配置swaggerUI静态页面--start--,若要使用原版gf字段swaggerUI请删除或注释此段
|
||
s.BindHookHandler(g.Cfg().MustGet(ctx, "server.swaggerPath").String()+"/*", ghttp.HookBeforeServe, func(r *ghttp.Request) {
|
||
content := gstr.ReplaceByMap(consts.SwaggerUITemplate, map[string]string{
|
||
`{SwaggerUIDocUrl}`: g.Cfg().MustGet(ctx, "server.openapiPath").String(),
|
||
`{SwaggerUIDocNamePlaceHolder}`: gstr.TrimRight(fmt.Sprintf(`//%s`, r.Host)),
|
||
})
|
||
r.Response.Write(content)
|
||
r.ExitAll()
|
||
})
|
||
|
||
// 重新配置swaggerUI静态页面--end--
|
||
enhanceOpenAPIDoc(s)
|
||
// 注册相关组件
|
||
register()
|
||
key := g.Cfg().MustGet(gctx.New(), "wx.key")
|
||
crt := g.Cfg().MustGet(gctx.New(), "wx.crt")
|
||
if len(key.String()) > 0 && len(crt.String()) > 0 {
|
||
s.EnableHTTPS(crt.String(), key.String())
|
||
}
|
||
s.Run()
|
||
return nil
|
||
},
|
||
}
|
||
|
||
type staticFile struct {
|
||
File *gres.File // Resource file object.
|
||
Path string // File path.
|
||
IsDir bool // Is directory.
|
||
}
|
||
type middleware struct {
|
||
served bool // Is the request served, which is used for checking response status 404.
|
||
request *Request // The request object pointer.
|
||
handlerIndex int // Index number for executing sequence purpose for handler items.
|
||
handlerMDIndex int // Index number for executing sequence purpose for bound middleware of handler item.
|
||
}
|
||
|
||
type Request struct {
|
||
*http.Request
|
||
Server *ghttp.Server // Server.
|
||
Cookie *ghttp.Cookie // Cookie.
|
||
Session *gsession.Session // Session.
|
||
Response *ghttp.Response // Corresponding Response of this request.
|
||
Router *ghttp.Router // Matched Router for this request. Note that it's not available in HOOK handler.
|
||
EnterTime int64 // Request starting time in milliseconds.
|
||
LeaveTime int64 // Request to end time in milliseconds.
|
||
Middleware *middleware // Middleware manager.
|
||
StaticFile *staticFile // Static file object for static file serving.
|
||
|
||
// =================================================================================================================
|
||
// Private attributes for internal usage purpose.
|
||
// =================================================================================================================
|
||
|
||
handlers []*ghttp.HandlerItemParsed // All matched handlers containing handler, hook and middleware for this request.
|
||
serveHandler *ghttp.HandlerItemParsed // Real handler serving for this request, not hook or middleware.
|
||
handlerResponse interface{} // Handler response object for Request/Response handler.
|
||
hasHookHandler bool // A bool marking whether there's hook handler in the handlers for performance purpose.
|
||
hasServeHandler bool // A bool marking whether there's serving handler in the handlers for performance purpose.
|
||
parsedQuery bool // A bool marking whether the GET parameters parsed.
|
||
parsedBody bool // A bool marking whether the request body parsed.
|
||
parsedForm bool // A bool marking whether request Form parsed for HTTP method PUT, POST, PATCH.
|
||
paramsMap map[string]interface{} // Custom parameters map.
|
||
routerMap map[string]string // Router parameters map, which might be nil if there are no router parameters.
|
||
queryMap map[string]interface{} // Query parameters map, which is nil if there's no query string.
|
||
formMap map[string]interface{} // Form parameters map, which is nil if there's no form of data from the client.
|
||
bodyMap map[string]interface{} // Body parameters map, which might be nil if their nobody content.
|
||
error error // Current executing error of the request.
|
||
exitAll bool // A bool marking whether current request is exited.
|
||
parsedHost string // The parsed host name for current host used by GetHost function.
|
||
clientIp string // The parsed client ip for current host used by GetClientIp function.
|
||
bodyContent []byte // Request body content.
|
||
isFileRequest bool // A bool marking whether current request is file serving.
|
||
viewObject *gview.View // Custom template view engine object for this response.
|
||
viewParams gview.Params // Custom template view variables for this response.
|
||
originUrlPath string // Original URL path that passed from client.
|
||
}
|
||
|
||
type staticPathItem struct {
|
||
Prefix string
|
||
Path string
|
||
}
|
||
|
||
//// bandWxApp 禁止访问小程序
|
||
//func bandWxApp(r *ghttp.Request) {
|
||
// // 如果不为 /zm/api/wxApplet/wxApplet/sysUserProjectRelevancy/list
|
||
// if r.URL.Path != "/zm/api/wxApplet/wxApplet/sysUserProjectRelevancy/list" {
|
||
// r.Response.WriteJsonExit(g.Map{"code": 403, "msg": "禁止访问"})
|
||
// }
|
||
//}
|
||
|
||
// staticFile 重赋值中间件
|
||
func reStaticFile(r *ghttp.Request) {
|
||
isURLEncoded := func(s string) bool {
|
||
// 如果 s 为空，直接返回 false
|
||
if s == "" {
|
||
return false
|
||
}
|
||
|
||
_, err := url.QueryUnescape(s)
|
||
return err == nil
|
||
}
|
||
|
||
if !isURLEncoded(r.URL.RawPath) {
|
||
return
|
||
}
|
||
|
||
// 解码 URL
|
||
r.URL.Path, _ = url.QueryUnescape(r.URL.RawPath)
|
||
|
||
fileRequestField := reflect.ValueOf(r).Elem().FieldByName("isFileRequest")
|
||
reflect.NewAt(fileRequestField.Type(), unsafe.Pointer(fileRequestField.UnsafeAddr())).Elem().Set(reflect.ValueOf(true))
|
||
|
||
configField := reflect.ValueOf(r.Server)
|
||
|
||
staticPathList := []staticPathItem{}
|
||
staticPaths := configField.Elem().FieldByName("config").FieldByName("StaticPaths")
|
||
for i := range make([]struct{}, staticPaths.Len()) {
|
||
prefix := staticPaths.Index(i).FieldByName("Prefix").String()
|
||
path := staticPaths.Index(i).FieldByName("Path").String()
|
||
|
||
staticPathList = append(staticPathList, staticPathItem{
|
||
Prefix: prefix,
|
||
Path: path,
|
||
})
|
||
}
|
||
|
||
indexFileList := []string{}
|
||
IndexFiles := configField.Elem().FieldByName("config").FieldByName("IndexFiles")
|
||
for i := range make([]struct{}, IndexFiles.Len()) {
|
||
indexFileList = append(indexFileList, IndexFiles.Index(i).String())
|
||
}
|
||
|
||
SearchPathsList := []string{}
|
||
SearchPaths := configField.Elem().FieldByName("config").FieldByName("SearchPaths")
|
||
for i := range make([]struct{}, SearchPaths.Len()) {
|
||
SearchPathsList = append(SearchPathsList, SearchPaths.Index(i).String())
|
||
}
|
||
|
||
(*Request)(unsafe.Pointer(r)).StaticFile = searchStaticFile(r.URL.Path, staticPathList, SearchPathsList, indexFileList)
|
||
}
|
||
|
||
func searchStaticFile(uri string, StaticPaths []staticPathItem, SearchPaths, indexFileList []string) *staticFile {
|
||
var (
|
||
file *gres.File
|
||
path string
|
||
dir bool
|
||
)
|
||
// Firstly search the StaticPaths mapping.
|
||
if len(StaticPaths) > 0 {
|
||
for _, item := range StaticPaths {
|
||
if len(uri) >= len(item.Prefix) && strings.EqualFold(item.Prefix, uri[0:len(item.Prefix)]) {
|
||
// To avoid case like: /static/style -> /static/style.css
|
||
if len(uri) > len(item.Prefix) && uri[len(item.Prefix)] != '/' {
|
||
continue
|
||
}
|
||
file = gres.GetWithIndex(item.Path+uri[len(item.Prefix):], indexFileList)
|
||
if file != nil {
|
||
return &staticFile{
|
||
File: file,
|
||
IsDir: file.FileInfo().IsDir(),
|
||
}
|
||
}
|
||
path, dir = gspath.Search(item.Path, uri[len(item.Prefix):], indexFileList...)
|
||
if path != "" {
|
||
return &staticFile{
|
||
Path: path,
|
||
IsDir: dir,
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
// Secondly search the root and searching paths.
|
||
if len(SearchPaths) > 0 {
|
||
for _, p := range SearchPaths {
|
||
file = gres.GetWithIndex(p+uri, indexFileList)
|
||
if file != nil {
|
||
return &staticFile{
|
||
File: file,
|
||
IsDir: file.FileInfo().IsDir(),
|
||
}
|
||
}
|
||
if path, dir = gspath.Search(p, uri, indexFileList...); path != "" {
|
||
return &staticFile{
|
||
Path: path,
|
||
IsDir: dir,
|
||
}
|
||
}
|
||
}
|
||
}
|
||
// Lastly search the resource manager.
|
||
if len(StaticPaths) == 0 && len(SearchPaths) == 0 {
|
||
if file = gres.GetWithIndex(uri, indexFileList); file != nil {
|
||
return &staticFile{
|
||
File: file,
|
||
IsDir: file.FileInfo().IsDir(),
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 相关组件注册
|
||
func register() {
|
||
// 注册上传组件
|
||
upload.Register()
|
||
// 注册自定义验证规则
|
||
libValidate.Register()
|
||
// 执行计划任务
|
||
task.Run()
|
||
go video_hat.WsConnection() // 连接视频安全帽平台
|
||
// go saft_hat.StartWs() // 开启安全帽 WS 推送，端口8222
|
||
go saft_hat.HeartCheck() // 开启安全帽设备数据的心跳检测
|
||
}
|
||
|
||
func enhanceOpenAPIDoc(s *ghttp.Server) {
|
||
openapi := s.GetOpenApi()
|
||
openapi.Config.CommonResponse = ghttp.DefaultHandlerResponse{}
|
||
openapi.Config.CommonResponseDataField = `Data`
|
||
|
||
// API description.
|
||
openapi.Info = goai.Info{
|
||
Title: consts.OpenAPITitle,
|
||
Description: consts.OpenAPIDescription,
|
||
Contact: &goai.Contact{
|
||
Name: consts.OpenAPIContactName,
|
||
URL: consts.OpenAPIContactUrl,
|
||
},
|
||
}
|
||
}
|