MaysWind
35 lines
846 B
Go
35 lines
846 B
Go
package middlewares
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/mayswind/ezbookkeeping/pkg/core"
|
|
"github.com/mayswind/ezbookkeeping/pkg/settings"
|
|
)
|
|
|
|
const settingsCookieName = "ebk_server_settings"
|
|
|
|
// ServerSettingsCookie adds server settings to cookies in response
|
|
func ServerSettingsCookie(config *settings.Config) core.MiddlewareHandlerFunc {
|
|
return func(c *core.Context) {
|
|
settingsArr := []string{
|
|
buildBooleanSetting("r", config.EnableUserRegister),
|
|
buildBooleanSetting("e", config.EnableDataExport),
|
|
}
|
|
|
|
bundledSettings := strings.Join(settingsArr, "_")
|
|
c.SetCookie(settingsCookieName, bundledSettings, int(config.TokenExpiredTime), "", "", false, false)
|
|
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
func buildBooleanSetting(key string, value bool) string {
|
|
if value {
|
|
return fmt.Sprintf("%s.1", key)
|
|
} else {
|
|
return fmt.Sprintf("%s.0", key)
|
|
}
|
|
}
|