mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-11-02 11:15:03 +03:00
MHSanaei
f1e1b0fb0c
InitLocalizer filled the shared bundle in place, and the Telegram bot runs it again while the panel is serving, so a request could read the bundle's maps while they were being written and the process died on a concurrent map access. Build the bundle privately and publish it, and the bot and web localizers, through atomic pointers; the lazy load in the middleware publishes only if nothing else did first.
55 lines
1.9 KiB
Go
55 lines
1.9 KiB
Go
package tgbot
|
|
|
|
import (
|
|
"encoding/json"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mhsanaei/3x-ui/v3/internal/database/dbtest"
|
|
"github.com/mhsanaei/3x-ui/v3/internal/web/locale"
|
|
"github.com/mhsanaei/3x-ui/v3/internal/xray"
|
|
|
|
"github.com/nicksnyder/go-i18n/v2/i18n"
|
|
"golang.org/x/text/language"
|
|
)
|
|
|
|
// clientInfoLocalizer renders the lines clientInfoMsg prints with the templates
|
|
// the translation files carry; without it I18n returns the bare keys.
|
|
func clientInfoLocalizer(t *testing.T) {
|
|
t.Helper()
|
|
bundle := i18n.NewBundle(language.MustParse("en-US"))
|
|
bundle.RegisterUnmarshalFunc("json", json.Unmarshal)
|
|
_ = bundle.AddMessages(language.MustParse("en-US"),
|
|
&i18n.Message{ID: "tgbot.messages.email", Other: "Email: {{ .Email }}\r\n"},
|
|
&i18n.Message{ID: "tgbot.days", Other: "Days"},
|
|
&i18n.Message{ID: "tgbot.messages.expireIn", Other: "Expire In: {{ .Time }}\r\n"},
|
|
&i18n.Message{ID: "tgbot.messages.expire", Other: "Expire Date: {{ .Time }}\r\n"},
|
|
&i18n.Message{ID: "tgbot.wentWrong", Other: "went wrong"},
|
|
)
|
|
orig := locale.SwapBotLocalizer(i18n.NewLocalizer(bundle, "en-US"))
|
|
t.Cleanup(func() { locale.SwapBotLocalizer(orig) })
|
|
}
|
|
|
|
// Regression test: a start-after-first-use client is stored as a negative duration,
|
|
// and a disabled one rendered it as a 1969 date.
|
|
func TestClientInfoShowsStartAfterFirstUseWhenDisabled(t *testing.T) {
|
|
dbtest.InitDB(t, filepath.Join(t.TempDir(), "x-ui.db"))
|
|
clientInfoLocalizer(t)
|
|
|
|
traffic := &xray.ClientTraffic{
|
|
Email: "trial@example.com",
|
|
Enable: false,
|
|
ExpiryTime: -30 * 24 * 60 * 60000,
|
|
}
|
|
|
|
out := (&Tgbot{}).clientInfoMsg(traffic, false, false, false, true, false, false)
|
|
|
|
if strings.Contains(out, "1969") {
|
|
t.Errorf("client info = %q, want the days left, not a 1969 date", out)
|
|
}
|
|
if !strings.Contains(out, "Expire In: 30 Days") {
|
|
t.Errorf("client info = %q, want it to contain %q", out, "Expire In: 30 Days")
|
|
}
|
|
}
|