34 lines
1.0 KiB
Go
34 lines
1.0 KiB
Go
package middleware
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/QuantumNous/new-api/common"
|
|
"github.com/QuantumNous/new-api/constant"
|
|
"github.com/QuantumNous/new-api/i18n"
|
|
"github.com/QuantumNous/new-api/relaykit/dto"
|
|
)
|
|
|
|
// I18n middleware detects and sets the language preference for the request
|
|
func I18n() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
lang := detectLanguage(c)
|
|
c.Set(string(constant.ContextKeyLanguage), lang)
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
// detectLanguage returns the language the request states, or "" when it states
|
|
// none. Priority: 1. User setting (if logged in) -> 2. Accept-Language header
|
|
func detectLanguage(c *gin.Context) string {
|
|
// 1. Try to get language from user setting (set by auth middleware)
|
|
if userSetting, ok := common.GetContextKeyType[dto.UserSetting](c, constant.ContextKeyUserSetting); ok {
|
|
if userSetting.Language != "" && i18n.IsSupported(userSetting.Language) {
|
|
return userSetting.Language
|
|
}
|
|
}
|
|
|
|
// 2. Parse Accept-Language header
|
|
return i18n.ParseAcceptLanguage(c.GetHeader("Accept-Language"))
|
|
}
|