idanoo/autobrr
1
0
Fork 0
mirror of https://github.com/idanoo/autobrr synced 2025-07-23 08:49:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(macros): add sprig functions to macros (#733)

Browse source
This commit is contained in:
vekvoid 2023-02-26 18:02:41 +00:00 • committed by GitHub
parent d100703784
commit e29cd9d80a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
internal/domain/macros.go
View file

@ -7,6 +7,8 @@ import (
"time" "time"
"github.com/autobrr/autobrr/pkg/errors" "github.com/autobrr/autobrr/pkg/errors"
"github.com/Masterminds/sprig/v3"
) )
type Macro struct { type Macro struct {
@ -70,7 +72,7 @@ func (m Macro) Parse(text string) (string, error) {
} }
// setup template // setup template
tmpl, err := template.New("macro").Parse(text) tmpl, err := template.New("macro").Funcs(sprig.TxtFuncMap()).Parse(text)
if err != nil { if err != nil {
return "", errors.Wrap(err, "could parse macro template") return "", errors.Wrap(err, "could parse macro template")
} }
@ -91,7 +93,7 @@ func (m Macro) MustParse(text string) string {
} }
// setup template // setup template
tmpl, err := template.New("macro").Parse(text) tmpl, err := template.New("macro").Funcs(sprig.TxtFuncMap()).Parse(text)
if err != nil { if err != nil {
return "" return ""
} }

18
internal/domain/macros_test.go
View file

@ -166,6 +166,24 @@ func TestMacros_Parse(t *testing.T) {
want: "movies-2021", want: "movies-2021",
wantErr: false, wantErr: false,
}, },
{
name: "test_size_formating",
release: Release{
Size: 3834225472,
},
args: args{text: "{{printf \"%.2f GB\" (divf .Size 1073741824)}}"},
want: "3.57 GB",
wantErr: false,
},
{
name: "test_text_manipulation",
release: Release{
TorrentName: "Title Name 2 - Keyword [Blu-ray][MKV][h264 10-bit][1080p][FLAC 2.0][Dual Audio][Softsubs (Sub Group)][Freeleech]",
},
args: args{text: "{{join \"\" (regexSplit \"^.+- Keyword \" .TorrentName -1)}}"},
want: "[Blu-ray][MKV][h264 10-bit][1080p][FLAC 2.0][Dual Audio][Softsubs (Sub Group)][Freeleech]",
wantErr: false,
},
} }
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {