328 lines
10 KiB
Go
328 lines
10 KiB
Go
// Copyright 2022 E99p1ant. All rights reserved.
|
||
// Use of this source code is governed by a MIT-style
|
||
// license that can be found in the LICENSE file.
|
||
|
||
package context
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"net/http"
|
||
"reflect"
|
||
"strings"
|
||
|
||
"github.com/flamego/csrf"
|
||
"github.com/flamego/flamego"
|
||
"github.com/flamego/session"
|
||
"github.com/flamego/template"
|
||
"github.com/pkg/errors"
|
||
"github.com/unknwon/com"
|
||
"github.com/wuhan005/govalid"
|
||
"go.opentelemetry.io/otel/attribute"
|
||
"go.opentelemetry.io/otel/trace"
|
||
|
||
"github.com/tamakyi/TamaBox/internal/conf"
|
||
"github.com/tamakyi/TamaBox/internal/db"
|
||
"github.com/tamakyi/TamaBox/internal/i18n"
|
||
"github.com/tamakyi/TamaBox/internal/shared"
|
||
templatepkg "github.com/tamakyi/TamaBox/internal/template"
|
||
)
|
||
|
||
type EndpointType string
|
||
|
||
const (
|
||
EndpointAPI EndpointType = "api"
|
||
EndpointWeb EndpointType = "web"
|
||
)
|
||
|
||
func (e EndpointType) IsAPI() bool {
|
||
return e == EndpointAPI
|
||
}
|
||
|
||
func (e EndpointType) IsWeb() bool {
|
||
return e == EndpointWeb
|
||
}
|
||
|
||
func APIEndpoint(ctx Context) {
|
||
ctx.Map(EndpointAPI)
|
||
}
|
||
|
||
// Context represents context of a request.
|
||
type Context struct {
|
||
flamego.Context
|
||
|
||
Data template.Data
|
||
Session session.Session
|
||
Template template.Template
|
||
|
||
User *db.User
|
||
IsLogged bool
|
||
}
|
||
|
||
// HasError returns true if error occurs in form validation.
|
||
func (c *Context) HasError() bool {
|
||
hasErr, ok := c.Data["HasError"]
|
||
if !ok {
|
||
return false
|
||
}
|
||
return hasErr.(bool)
|
||
}
|
||
|
||
func (c *Context) SetError(err error, f ...interface{}) {
|
||
// 防御性兜底：调用方可能传入 nil error（如从某个可能返回 nil 的函数直接透传），
|
||
// 此时下方 err.Error() 会触发 nil 指针 panic。统一兜底为「未知错误」，避免整页崩溃。
|
||
if err == nil {
|
||
err = errors.New("发生未知错误")
|
||
}
|
||
|
||
c.Data["HasError"] = true
|
||
|
||
// 表单校验（govalid）错误：其 Error() 在字段带 msg 标签时为 i18n 键，
|
||
// 在此按当前界面语言翻译；其余业务错误（errors.New(i18n.T(...)) 等）原样保留。
|
||
if ge, ok := err.(*govalid.ErrContext); ok {
|
||
c.Data["Error"] = i18n.T(c.Lang(), ge.Error())
|
||
} else {
|
||
c.Data["Error"] = err.Error()
|
||
}
|
||
|
||
// Set back the form data.
|
||
if len(f) > 0 {
|
||
form := f[0]
|
||
typ := reflect.TypeOf(form)
|
||
val := reflect.ValueOf(form)
|
||
|
||
if typ.Kind() == reflect.Ptr {
|
||
typ = typ.Elem()
|
||
val = val.Elem()
|
||
}
|
||
|
||
for i := 0; i < typ.NumField(); i++ {
|
||
field := typ.Field(i)
|
||
fieldName := com.ToSnakeCase(field.Name)
|
||
|
||
c.Data[fieldName] = val.Field(i).Interface()
|
||
}
|
||
}
|
||
|
||
span := trace.SpanFromContext(c.Request().Context())
|
||
if span.IsRecording() {
|
||
span.SetAttributes(
|
||
attribute.String("tamabox.error", err.Error()),
|
||
)
|
||
}
|
||
}
|
||
|
||
func (c *Context) SetInternalError(f ...interface{}) {
|
||
span := trace.SpanFromContext(c.Request().Context())
|
||
traceID := span.SpanContext().TraceID()
|
||
|
||
c.Data["FlashTip"] = fmt.Sprintf("若问题一直出现，请带上该段字符 %s 提交反馈。", traceID.String())
|
||
c.SetError(errors.New(i18n.T(c.Lang(), "internal.errFlash")), f...)
|
||
}
|
||
|
||
// Success renders HTML template with given name with 200 OK status code.
|
||
func (c *Context) Success(templateName string) {
|
||
c.Template.HTML(http.StatusOK, templateName)
|
||
}
|
||
|
||
func (c *Context) SetTitle(title string) {
|
||
c.Data["Title"] = title
|
||
}
|
||
|
||
func (c *Context) Refresh() {
|
||
c.Redirect(c.Request().URL.Path)
|
||
}
|
||
|
||
func (c *Context) JSON(data interface{}) error {
|
||
resp := map[string]interface{}{
|
||
"code": 0,
|
||
"data": data,
|
||
"message": "success",
|
||
}
|
||
|
||
return json.NewEncoder(c.ResponseWriter()).Encode(resp)
|
||
}
|
||
|
||
func (c *Context) ServerError() error {
|
||
return c.JSONError(50000, i18n.T(c.Lang(), "internal.errServer"))
|
||
}
|
||
|
||
func (c *Context) JSONError(errorCode int, message string) error {
|
||
span := trace.SpanFromContext(c.Request().Context())
|
||
|
||
resp := map[string]interface{}{
|
||
"code": errorCode,
|
||
"message": message,
|
||
"trace_id": span.SpanContext().TraceID().String(),
|
||
}
|
||
|
||
statusCode := errorCode / 100
|
||
if statusCode < 100 || statusCode > 999 {
|
||
statusCode = http.StatusInternalServerError
|
||
}
|
||
c.ResponseWriter().WriteHeader(statusCode)
|
||
|
||
return json.NewEncoder(c.ResponseWriter()).Encode(resp)
|
||
}
|
||
|
||
// Contexter initializes a classic context for a request.
|
||
func Contexter() flamego.Handler {
|
||
return func(ctx flamego.Context, data template.Data, session session.Session, x csrf.CSRF, t template.Template, flash session.Flash) {
|
||
c := Context{
|
||
Context: ctx,
|
||
Data: data,
|
||
Session: session,
|
||
Template: t,
|
||
}
|
||
|
||
// POST 类请求统一校验 CSRF（像素追踪代理已改为 GET 且仅转发只读请求，
|
||
// 不再需要 CSRF 豁免，故此处不再对 /api/v1/pixel/ 做例外）。
|
||
if ctx.Request().Method == http.MethodPost {
|
||
x.Validate(ctx)
|
||
}
|
||
|
||
// Get user from session or header when possible
|
||
c.User = authenticatedUser(c.Context, c.Session)
|
||
|
||
var userID uint
|
||
if c.User != nil {
|
||
c.IsLogged = true
|
||
c.Data["IsLogged"] = c.IsLogged
|
||
c.Data["LoggedUser"] = c.User
|
||
c.Data["LoggedUserID"] = c.User.ID
|
||
c.Data["LoggedUserName"] = c.User.Name
|
||
|
||
userID = c.User.ID
|
||
} else {
|
||
c.Data["LoggedUserID"] = 0
|
||
c.Data["LoggedUserName"] = ""
|
||
}
|
||
|
||
c.Data["IsPixel"] = ctx.Request().URL.Path == "/pixel"
|
||
|
||
span := trace.SpanFromContext(ctx.Request().Context())
|
||
if span.IsRecording() {
|
||
span.SetAttributes(
|
||
attribute.Bool("tamabox.user.is-login", c.IsLogged),
|
||
attribute.Int("tamabox.user.id", int(userID)),
|
||
)
|
||
}
|
||
c.ResponseWriter().Header().Set("Trace-ID", span.SpanContext().TraceID().String())
|
||
|
||
if flash != nil {
|
||
flash, ok := flash.(Flash)
|
||
if ok {
|
||
switch flash.Type {
|
||
case "success":
|
||
c.Data["Success"] = flash.Message
|
||
case "error":
|
||
c.Data["HasError"] = true
|
||
c.Data["Error"] = flash.Message
|
||
case "info":
|
||
c.Data["Info"] = flash.Message
|
||
case "warning":
|
||
c.Data["Warning"] = flash.Message
|
||
}
|
||
|
||
c.Data["FlashTip"] = flash.FlashTip
|
||
}
|
||
}
|
||
|
||
// ---- i18n 语言注入 ----
|
||
// 解析优先级（按页面类型）：
|
||
// - 个人页（/user 后台、/_/{domain} 提问箱）：永远跟随主人 profile 语言，忽略访客 cookie。
|
||
// /user 后台即登录用户本人的配置，直接采用 c.User.Language；
|
||
// /_/{domain} 提问箱页由 question.Pager 中间件按 pageUser.Language 覆盖（见 route/question/page.go）。
|
||
// - 公开页（首页/探索/登录等）：访客 lang cookie 优先；未设置 cookie 时登录用户资料语言次之；
|
||
// 再回退站点默认语言（shared.App）> i18n.Default。
|
||
path := ctx.Request().URL.Path
|
||
isUserPage := strings.HasPrefix(path, "/user") || strings.HasPrefix(path, "/_/")
|
||
c.Data["IsUserPage"] = isUserPage
|
||
|
||
// SEO：仅公开页（首页/注册/登录/explore）允许被搜索引擎收录；
|
||
// 提问内容（提问箱页、单题页）、密码页、私信、用户后台等隐私页一律 noindex。
|
||
isPublicPage := path == "/" || path == "/register" || path == "/login" || strings.HasPrefix(path, "/explore")
|
||
if !isPublicPage {
|
||
c.Data["NoIndex"] = true
|
||
}
|
||
|
||
lang := ""
|
||
if isUserPage {
|
||
// 个人页：永远跟随 profile，忽略 cookie（/user 直接采用本人配置；
|
||
// /_/{domain} 由 question.Pager 覆盖，此处留空交由其处理）。
|
||
if strings.HasPrefix(path, "/user") && c.IsLogged && c.User.Language != "" {
|
||
lang = i18n.Resolve(c.User.Language)
|
||
}
|
||
} else {
|
||
// 公开页：会话内显式选择的语言优先，其次登录用户资料语言，最后回退站点默认。
|
||
// 语言偏好存于 session（由 session 中间件托管 cookie）；此前独立 lang cookie 会被
|
||
// session 中间件的 Set-Cookie 整体覆盖而失效，故此处改读 session。
|
||
if v := c.Session.Get("lang"); v != nil {
|
||
if s, ok := v.(string); ok && s != "" {
|
||
lang = i18n.Resolve(s)
|
||
}
|
||
}
|
||
if lang == "" && c.IsLogged && c.User.Language != "" {
|
||
lang = i18n.Resolve(c.User.Language)
|
||
}
|
||
}
|
||
if lang == "" {
|
||
lang = shared.App.Lang()
|
||
}
|
||
c.Data["Lang"] = lang
|
||
c.Data["LangCode"] = lang
|
||
c.Data["SupportedLangs"] = i18n.Supported()
|
||
|
||
c.SetTitle(i18n.T(lang, "site.title"))
|
||
c.Data["CSRFToken"] = x.Token()
|
||
c.Data["CSRFTokenHTML"] = templatepkg.Safe(`<input type="hidden" name="_csrf" value="` + x.Token() + `">`)
|
||
|
||
c.Data["RecaptchaDomain"] = conf.Recaptcha.Domain
|
||
c.Data["RecaptchaSiteKey"] = conf.Recaptcha.SiteKey
|
||
c.Data["RecaptchaTurnstileStyle"] = conf.Recaptcha.TurnstileStyle
|
||
c.Data["CaptchaType"] = conf.Recaptcha.CaptchaType
|
||
|
||
c.Data["CurrentURI"] = ctx.Request().Request.RequestURI
|
||
c.Data["ExternalURL"] = conf.App.ExternalURL
|
||
|
||
// 默认头像/背景图兜底：未设置自定义头像的箱主，其二维码 logo 应与页面头像一致回退到 default_avatar。
|
||
// 模板按原始 URL 直连（easy.qrcode 以 crossOrigin:'anonymous' 加载，配合对象存储桶 CORS
|
||
// 返回 Access-Control-Allow-Origin，保证 canvas 不被污染、可正常导出）。
|
||
c.Data["DefaultAvatar"] = conf.Upload.DefaultAvatarURL
|
||
c.Data["DefaultBackgroundImage"] = conf.Upload.DefaultBackgroundImageURL
|
||
|
||
// [limits] 段全站上限的模板同源入口：所有页面可直接引用
|
||
// {{.Limits.QuestionContentLen}} / {{.Limits.AnswerContentLen}} /
|
||
// {{.Limits.FollowUpContentLen}} / {{.Limits.ChatMessageLen}} 等，
|
||
// 作为输入框 maxlength 等与后端校验的同一数据源，避免各处写死数字漂移。
|
||
c.Data["Limits"] = conf.Limits
|
||
|
||
// ⚠️ VConsole can only be enabled for the first user for security reasons.
|
||
c.Data["VConsole"] = ctx.Query("debug") == "on" && c.IsLogged && c.User.ID == 1
|
||
|
||
// 🚨 SECURITY: Prevent MIME type sniffing in some browsers,
|
||
c.ResponseWriter().Header().Set("X-Content-Type-Options", "nosniff")
|
||
c.ResponseWriter().Header().Set("X-Frame-Options", "DENY")
|
||
|
||
ctx.Map(c)
|
||
ctx.Map(EndpointWeb)
|
||
|
||
if conf.App.MaintenanceMode {
|
||
c.ResponseWriter().Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
|
||
c.Success("maintenance-mode")
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
// Lang 返回当前请求解析出的界面语言代码（已规范化，必定是字典可查的规范代码）。
|
||
// 解析优先级见上方 Data["Lang"] 注入逻辑（个人页跟随箱主语言；公开页跟随 lang cookie /
|
||
// 登录用户资料语言 / 站点默认语言）。若中间件尚未注入（极少数非标准路径），回退 i18n.Default。
|
||
// 用于后端 flash / 错误信息按访客当前语言本地化，避免硬编码中文绕过 i18n。
|
||
func (c *Context) Lang() string {
|
||
if l, ok := c.Data["Lang"].(string); ok && l != "" {
|
||
return l
|
||
}
|
||
return i18n.Default
|
||
}
|