mirror of
https://github.com/krau/SaveAny-Bot.git
synced 2026-05-11 22:29:41 +08:00
refactor: refactor task logic for better scalability (#76)
* refactor: a big refactor. wip * refactor: port handle file * refactor: place all handlers * fix: task info nil pointer * feat: enhance task progress tracking and context management * feat: cancel task * feat: stream mode * feat: silent mode * feat: dir cmd * refactor: remove unused old file * feat: rule cmd * feat: handle silent mode * feat: batch task * fix: batch task progress and temp file cleanup * refactor: update file creation and cleanup methods for better resource management * feat: add save command with silent mode handling * feat: message link * feat: update message prompts to include file count in storage selection * feat: slient save links * refactor: reduce dup code * feat: rule type * feat: chose dir * feat: refactor file handling and storage rules, improve error handling and logging * feat: rule mode * feat: telegraph pics * fix: tphpics nil pointer and inaccurate dirpath * feat: silent save telegraph * feat: add suffix to avoid file overwrite * feat: new storage telegram * chore: tidy go mod
This commit is contained in:
45
pkg/rule/filename_regex.go
Normal file
45
pkg/rule/filename_regex.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package rule
|
||||
|
||||
import (
|
||||
"regexp"
|
||||
|
||||
ruleenum "github.com/krau/SaveAny-Bot/pkg/enums/rule"
|
||||
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
||||
)
|
||||
|
||||
type RuleFileNameRegex struct {
|
||||
storInfo
|
||||
regex *regexp.Regexp
|
||||
}
|
||||
|
||||
var _ RuleClass[tfile.TGFile] = (*RuleFileNameRegex)(nil)
|
||||
|
||||
func (r RuleFileNameRegex) Type() ruleenum.RuleType {
|
||||
return ruleenum.FileNameRegex
|
||||
}
|
||||
|
||||
func (r RuleFileNameRegex) Match(input tfile.TGFile) (bool, error) {
|
||||
return r.regex.MatchString(input.Name()), nil
|
||||
}
|
||||
|
||||
func (r RuleFileNameRegex) StorageName() string {
|
||||
return r.storName
|
||||
}
|
||||
|
||||
func (r RuleFileNameRegex) StoragePath() string {
|
||||
return r.storPath
|
||||
}
|
||||
|
||||
func NewRuleFileNameRegex(storName, storPath, regexStr string) (*RuleFileNameRegex, error) {
|
||||
regex, err := regexp.Compile(regexStr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &RuleFileNameRegex{
|
||||
storInfo: storInfo{
|
||||
storName: storName,
|
||||
storPath: storPath,
|
||||
},
|
||||
regex: regex,
|
||||
}, nil
|
||||
}
|
||||
43
pkg/rule/message_regex.go
Normal file
43
pkg/rule/message_regex.go
Normal file
@@ -0,0 +1,43 @@
|
||||
package rule
|
||||
|
||||
import (
|
||||
"regexp"
|
||||
|
||||
ruleenum "github.com/krau/SaveAny-Bot/pkg/enums/rule"
|
||||
)
|
||||
|
||||
var _ RuleClass[string] = (*RuleMessageRegex)(nil)
|
||||
|
||||
type RuleMessageRegex struct {
|
||||
storInfo
|
||||
regex *regexp.Regexp
|
||||
}
|
||||
|
||||
func (r RuleMessageRegex) Type() ruleenum.RuleType {
|
||||
return ruleenum.MessageRegex
|
||||
}
|
||||
|
||||
func (r RuleMessageRegex) Match(input string) (bool, error) {
|
||||
return r.regex.MatchString(input), nil
|
||||
}
|
||||
|
||||
func (r RuleMessageRegex) StorageName() string {
|
||||
return r.storName
|
||||
}
|
||||
func (r RuleMessageRegex) StoragePath() string {
|
||||
return r.storPath
|
||||
}
|
||||
|
||||
func NewRuleMessageRegex(storName, storPath, regexStr string) (*RuleMessageRegex, error) {
|
||||
regex, err := regexp.Compile(regexStr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &RuleMessageRegex{
|
||||
storInfo: storInfo{
|
||||
storName: storName,
|
||||
storPath: storPath,
|
||||
},
|
||||
regex: regex,
|
||||
}, nil
|
||||
}
|
||||
17
pkg/rule/rule.go
Normal file
17
pkg/rule/rule.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package rule
|
||||
|
||||
import (
|
||||
ruleenum "github.com/krau/SaveAny-Bot/pkg/enums/rule"
|
||||
)
|
||||
|
||||
type RuleClass[InputType any] interface {
|
||||
Type() ruleenum.RuleType
|
||||
Match(input InputType) (bool, error)
|
||||
StorageName() string
|
||||
StoragePath() string
|
||||
}
|
||||
|
||||
type storInfo struct {
|
||||
storName string
|
||||
storPath string
|
||||
}
|
||||
Reference in New Issue
Block a user