him411/UA3F
1
0
Fork 0
mirror of https://github.com/SunBK201/UA3F.git synced 2025-12-16 08:44:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: rename UAPattern to UARegex for consistency

Browse Source
This commit is contained in:
SunBK201 2025-11-05 21:37:04 +08:00
parent f6f693ad11
commit 47a2a1d29a
4 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/internal/config/config.go
View File

@ -20,7 +20,7 @@ type Config struct {
ListenAddr string
LogLevel string
PayloadUA string
UAPattern string
UARegex string
EnablePartialReplace bool
}
@ -31,7 +31,7 @@ func Parse() (*Config, bool) {
port int
loglevel string
payloadUA string
uaPattern string
uaRegx string
partial bool
showVer bool
)
@ -40,7 +40,7 @@ func Parse() (*Config, bool) {
flag.StringVar(&bindAddr, "b", "127.0.0.1", "bind address (default: 127.0.0.1)")
flag.IntVar(&port, "p", 1080, "port")
flag.StringVar(&payloadUA, "f", "FFF", "User-Agent")
flag.StringVar(&uaPattern, "r", "", "UA-Pattern")
flag.StringVar(&uaRegx, "r", "", "UA-Pattern")
flag.BoolVar(&partial, "s", false, "Enable Regex Partial Replace")
flag.StringVar(&loglevel, "l", "info", "Log level (default: info)")
flag.BoolVar(&showVer, "v", false, "show version")
@ -53,7 +53,7 @@ func Parse() (*Config, bool) {
ListenAddr: fmt.Sprintf("%s:%d", bindAddr, port),
LogLevel: loglevel,
PayloadUA: payloadUA,
UAPattern: uaPattern,
UARegex: uaRegx,
EnablePartialReplace: partial,
}
if serverMode == ServerModeRedirect {

2
src/internal/log/log.go
View File

@ -73,7 +73,7 @@ func LogHeader(version string, cfg *config.Config) {
logrus.Info("Server Mode: " + cfg.ServerMode)
logrus.Infof("Listen on %s", cfg.ListenAddr)
logrus.Infof("User-Agent: %s", cfg.PayloadUA)
logrus.Infof("User-Agent Regex: '%s'", cfg.UAPattern)
logrus.Infof("User-Agent Regex: '%s'", cfg.UARegex)
logrus.Infof("Partial Replace: %v", cfg.EnablePartialReplace)
logrus.Infof("Log level: %s", cfg.LogLevel)
}

4
src/internal/rewrite/rewriter.go
View File

@ -35,7 +35,7 @@ type Rewriter struct {
// New constructs a Rewriter from config. Compiles regex and allocates cache.
func New(cfg *config.Config) (*Rewriter, error) {
// UA pattern is compiled with case-insensitive prefix (?i)
pattern := "(?i)" + cfg.UAPattern
pattern := "(?i)" + cfg.UARegex
uaRegex, err := regexp2.Compile(pattern, regexp2.None)
if err != nil {
return nil, err
@ -43,7 +43,7 @@ func New(cfg *config.Config) (*Rewriter, error) {
return &Rewriter{
payloadUA: cfg.PayloadUA,
pattern: cfg.UAPattern,
pattern: cfg.UARegex,
partialReplace: cfg.EnablePartialReplace,
uaRegex: uaRegex,
Cache: expirable.NewLRU[string, struct{}](1024, nil, 30*time.Minute),

6
src/internal/rewrite/rewriter_test.go
View File

@ -26,7 +26,7 @@ func (m *mockConn) SetWriteDeadline(t time.Time) error { return nil }
func newTestRewriter(t *testing.T) *Rewriter {
cfg := &config.Config{
UAPattern: "TestUA",
UARegex: "TestUA",
PayloadUA: "MockUA/1.0",
EnablePartialReplace: false,
}
@ -37,14 +37,14 @@ func newTestRewriter(t *testing.T) *Rewriter {
func TestNewRewriter(t *testing.T) {
cfg := &config.Config{
UAPattern: "TestUA",
UARegex: "TestUA",
PayloadUA: "FFF0",
EnablePartialReplace: false,
}
rewriter, err := New(cfg)
assert.NoError(t, err)
assert.Equal(t, cfg.PayloadUA, rewriter.payloadUA)
assert.Equal(t, cfg.UAPattern, rewriter.pattern)
assert.Equal(t, cfg.UARegex, rewriter.pattern)
assert.Equal(t, cfg.EnablePartialReplace, rewriter.partialReplace)
assert.NotNil(t, rewriter.uaRegex)
assert.NotNil(t, rewriter.Cache)