2024-01-12 19:18:33 +08:00
|
|
|
package nf
|
|
|
|
|
|
|
|
|
|
const (
|
|
|
|
|
banner = " _ _ _ ___ _ \n | \\| |___| |_ | __|__ _ _ _ _ __| |\n | .` / _ \\ _| | _/ _ \\ || | ' \\/ _` |\n |_|\\_\\___/\\__| |_|\\___/\\_,_|_||_\\__,_|\n "
|
2024-01-19 20:15:36 +08:00
|
|
|
_404 = "<!doctype html><html lang=\"en\"><head><meta charset=\"UTF-8\"><meta name=\"viewport\" content=\"width=device-width,user-scalable=no,initial-scale=1,maximum-scale=1,minimum-scale=1\"><meta http-equiv=\"X-UA-Compatible\" content=\"ie=edge\"><title>Not Found</title><style>body{background:#333;margin:0;color:#ccc;display:flex;align-items:center;max-height:100vh;height:100vh;justify-content:center}textarea{min-height:5rem;min-width:20rem;text-align:center;border:none;background:0 0;color:#ccc;resize:none;user-input:none;user-select:none;cursor:default;-webkit-user-select:none;-webkit-touch-callout:none;-moz-user-select:none;-ms-user-select:none;outline:0}</style></head><body><textarea id=\"banner\" readonly=\"readonly\"></textarea><script type=\"text/javascript\">let htmlCodes = [\n ' _ _ _ ___ _ ',\n '| \\\\| |___| |_ | __|__ _ _ _ _ __| |',\n '| .` / _ \\\\ _| | _/ _ \\\\ || | \\' \\\\/ _` |',\n '|_|\\\\_\\\\___/\\\\__| |_|\\\\___/\\\\_,_|_||_\\\\__,_|'\n].join('\\n');\ndocument.querySelector('#banner').value = htmlCodes</script></body></html>"
|
2024-01-12 19:18:33 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
type Map map[string]interface{}
|
|
|
|
|
|
|
|
|
|
type Config struct {
|
|
|
|
|
// Default: 4 * 1024 * 1024
|
2024-01-13 18:54:23 +08:00
|
|
|
BodyLimit int64 `json:"-"`
|
|
|
|
|
|
|
|
|
|
// if report http.ErrServerClosed as run err
|
|
|
|
|
ErrServeClose bool `json:"-"`
|
|
|
|
|
|
2024-01-13 20:36:11 +08:00
|
|
|
DisableBanner bool `json:"-"`
|
|
|
|
|
DisableLogger bool `json:"-"`
|
|
|
|
|
DisableRecover bool `json:"-"`
|
|
|
|
|
DisableHttpErrorLog bool `json:"-"`
|
2024-01-19 20:15:36 +08:00
|
|
|
|
2024-01-29 15:48:09 +08:00
|
|
|
EnableNotImplementHandler bool `json:"-"`
|
|
|
|
|
NotFoundHandler HandlerFunc `json:"-"`
|
2024-01-12 19:18:33 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var (
|
|
|
|
|
defaultConfig = &Config{
|
|
|
|
|
BodyLimit: 4 * 1024 * 1024,
|
2024-01-19 20:15:36 +08:00
|
|
|
NotFoundHandler: func(c *Ctx) error {
|
|
|
|
|
c.Set("Content-Type", MIMETextHTML)
|
|
|
|
|
_, err := c.Status(404).Write([]byte(_404))
|
|
|
|
|
return err
|
|
|
|
|
},
|
2024-01-12 19:18:33 +08:00
|
|
|
}
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func New(config ...Config) *App {
|
|
|
|
|
app := &App{
|
|
|
|
|
router: newRouter(),
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if len(config) > 0 {
|
|
|
|
|
app.config = &config[0]
|
|
|
|
|
if app.config.BodyLimit == 0 {
|
|
|
|
|
app.config.BodyLimit = defaultConfig.BodyLimit
|
|
|
|
|
}
|
2024-01-19 20:15:36 +08:00
|
|
|
|
|
|
|
|
if app.config.NotFoundHandler == nil {
|
|
|
|
|
app.config.NotFoundHandler = defaultConfig.NotFoundHandler
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-12 19:18:33 +08:00
|
|
|
} else {
|
|
|
|
|
app.config = defaultConfig
|
|
|
|
|
}
|
|
|
|
|
|
2024-02-19 16:00:56 +08:00
|
|
|
app.RouterGroup = &RouterGroup{app: app, prefix: "/"}
|
2024-01-12 19:18:33 +08:00
|
|
|
app.groups = []*RouterGroup{app.RouterGroup}
|
|
|
|
|
|
2024-01-14 19:10:05 +08:00
|
|
|
if !app.config.DisableLogger {
|
|
|
|
|
app.Use(NewLogger())
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-12 19:18:33 +08:00
|
|
|
if !app.config.DisableRecover {
|
|
|
|
|
app.Use(NewRecover(true))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return app
|
|
|
|
|
}
|
