diff options
Diffstat (limited to 'plugin/plaintext/common_out.go')
| -rw-r--r-- | plugin/plaintext/common_out.go | 136 |
1 files changed, 98 insertions, 38 deletions
diff --git a/plugin/plaintext/common_out.go b/plugin/plaintext/common_out.go index cc7ba41e..c6588c51 100644 --- a/plugin/plaintext/common_out.go +++ b/plugin/plaintext/common_out.go @@ -7,6 +7,7 @@ import ( "net" "os" "path/filepath" + "strings" "github.com/Loyalsoldier/geoip/lib" ) @@ -18,7 +19,7 @@ var ( defaultOutputDirForSurgeRuleSetOut = filepath.Join("./", "output", "surge") ) -type TextOut struct { +type textOut struct { Type string Action lib.Action Description string @@ -32,7 +33,84 @@ type TextOut struct { AddSuffixInLine string } -func newTextOut(iType string, iDesc string, action lib.Action, data json.RawMessage) (lib.OutputConverter, error) { +func NewTextOut(iType string, iDesc string, action lib.Action, opts ...lib.OutputOption) lib.OutputConverter { + t := &textOut{ + Type: iType, + Action: action, + Description: iDesc, + } + + for _, opt := range opts { + if opt != nil { + opt(t) + } + } + + return t +} + +func WithTextOutOutputDir(iType, dir string) lib.OutputOption { + return func(s lib.OutputConverter) { + dir = strings.TrimSpace(dir) + if dir == "" { + switch iType { + case TypeTextOut: + dir = defaultOutputDirForTextOut + case TypeClashRuleSetClassicalOut: + dir = defaultOutputDirForClashRuleSetClassicalOut + case TypeClashRuleSetIPCIDROut: + dir = defaultOutputDirForClashRuleSetIPCIDROut + case TypeSurgeRuleSetOut: + dir = defaultOutputDirForSurgeRuleSetOut + } + } + + s.(*textOut).OutputDir = dir + } +} + +func WithTextOutOutputExt(ext string) lib.OutputOption { + return func(s lib.OutputConverter) { + ext = strings.TrimSpace(ext) + if ext == "" { + ext = ".txt" + } + + s.(*textOut).OutputExt = ext + } +} + +func WithTextOutWantedList(lists []string) lib.OutputOption { + return func(s lib.OutputConverter) { + s.(*textOut).Want = lists + } +} + +func WithTextOutExcludedList(lists []string) lib.OutputOption { + return func(s lib.OutputConverter) { + s.(*textOut).Exclude = lists + } +} + +func WithTextOutOnlyIPType(onlyIPType lib.IPType) lib.OutputOption { + return func(s lib.OutputConverter) { + s.(*textOut).OnlyIPType = onlyIPType + } +} + +func WithTextOutAddPrefixInLine(prefix string) lib.OutputOption { + return func(s lib.OutputConverter) { + s.(*textOut).AddPrefixInLine = prefix + } +} + +func WithTextOutAddSuffixInLine(suffix string) lib.OutputOption { + return func(s lib.OutputConverter) { + s.(*textOut).AddSuffixInLine = suffix + } +} + +func NewTextOutFromBytes(iType string, iDesc string, action lib.Action, data []byte) (lib.OutputConverter, error) { var tmp struct { OutputDir string `json:"outputDir"` OutputExt string `json:"outputExtension"` @@ -50,39 +128,21 @@ func newTextOut(iType string, iDesc string, action lib.Action, data json.RawMess } } - if tmp.OutputDir == "" { - switch iType { - case TypeTextOut: - tmp.OutputDir = defaultOutputDirForTextOut - case TypeClashRuleSetClassicalOut: - tmp.OutputDir = defaultOutputDirForClashRuleSetClassicalOut - case TypeClashRuleSetIPCIDROut: - tmp.OutputDir = defaultOutputDirForClashRuleSetIPCIDROut - case TypeSurgeRuleSetOut: - tmp.OutputDir = defaultOutputDirForSurgeRuleSetOut - } - } - - if tmp.OutputExt == "" { - tmp.OutputExt = ".txt" - } - - return &TextOut{ - Type: iType, - Action: action, - Description: iDesc, - OutputDir: tmp.OutputDir, - OutputExt: tmp.OutputExt, - Want: tmp.Want, - Exclude: tmp.Exclude, - OnlyIPType: tmp.OnlyIPType, - - AddPrefixInLine: tmp.AddPrefixInLine, - AddSuffixInLine: tmp.AddSuffixInLine, - }, nil + return NewTextOut( + iType, + iDesc, + action, + WithTextOutOutputDir(iType, tmp.OutputDir), + WithTextOutOutputExt(tmp.OutputExt), + WithTextOutWantedList(tmp.Want), + WithTextOutExcludedList(tmp.Exclude), + WithTextOutOnlyIPType(tmp.OnlyIPType), + WithTextOutAddPrefixInLine(tmp.AddPrefixInLine), + WithTextOutAddSuffixInLine(tmp.AddSuffixInLine), + ), nil } -func (t *TextOut) marshalBytes(entry *lib.Entry) ([]byte, error) { +func (t *textOut) marshalBytes(entry *lib.Entry) ([]byte, error) { entryCidr, err := entry.MarshalText(lib.GetIgnoreIPType(t.OnlyIPType)) if err != nil { return nil, err @@ -108,7 +168,7 @@ func (t *TextOut) marshalBytes(entry *lib.Entry) ([]byte, error) { return buf.Bytes(), nil } -func (t *TextOut) marshalBytesForTextOut(buf *bytes.Buffer, entryCidr []string) error { +func (t *textOut) marshalBytesForTextOut(buf *bytes.Buffer, entryCidr []string) error { for _, cidr := range entryCidr { if t.AddPrefixInLine != "" { buf.WriteString(t.AddPrefixInLine) @@ -122,7 +182,7 @@ func (t *TextOut) marshalBytesForTextOut(buf *bytes.Buffer, entryCidr []string) return nil } -func (t *TextOut) marshalBytesForClashRuleSetClassicalOut(buf *bytes.Buffer, entryCidr []string) error { +func (t *textOut) marshalBytesForClashRuleSetClassicalOut(buf *bytes.Buffer, entryCidr []string) error { buf.WriteString("payload:\n") for _, cidr := range entryCidr { ip, _, err := net.ParseCIDR(cidr) @@ -141,7 +201,7 @@ func (t *TextOut) marshalBytesForClashRuleSetClassicalOut(buf *bytes.Buffer, ent return nil } -func (t *TextOut) marshalBytesForClashRuleSetIPCIDROut(buf *bytes.Buffer, entryCidr []string) error { +func (t *textOut) marshalBytesForClashRuleSetIPCIDROut(buf *bytes.Buffer, entryCidr []string) error { buf.WriteString("payload:\n") for _, cidr := range entryCidr { buf.WriteString(" - '") @@ -152,7 +212,7 @@ func (t *TextOut) marshalBytesForClashRuleSetIPCIDROut(buf *bytes.Buffer, entryC return nil } -func (t *TextOut) marshalBytesForSurgeRuleSetOut(buf *bytes.Buffer, entryCidr []string) error { +func (t *textOut) marshalBytesForSurgeRuleSetOut(buf *bytes.Buffer, entryCidr []string) error { for _, cidr := range entryCidr { ip, _, err := net.ParseCIDR(cidr) if err != nil { @@ -173,7 +233,7 @@ func (t *TextOut) marshalBytesForSurgeRuleSetOut(buf *bytes.Buffer, entryCidr [] return nil } -func (t *TextOut) writeFile(filename string, data []byte) error { +func (t *textOut) writeFile(filename string, data []byte) error { if err := os.MkdirAll(t.OutputDir, 0755); err != nil { return err } |