mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 08:49:13 +00:00
85fa941cd3
* feat(macros): expose parsed dates from rls string * feat(macros)!: add parsed year BREAKING CHANGE: add prefix Current to - year, month, day, hour, min, second
76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package action
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
)
|
|
|
|
type Macro struct {
|
|
TorrentName string
|
|
TorrentPathName string
|
|
TorrentHash string
|
|
TorrentUrl string
|
|
Indexer string
|
|
Title string
|
|
Resolution string
|
|
Source string
|
|
HDR string
|
|
Season int
|
|
Episode int
|
|
Year int
|
|
CurrentYear int
|
|
CurrentMonth int
|
|
CurrentDay int
|
|
CurrentHour int
|
|
CurrentMinute int
|
|
CurrentSecond int
|
|
}
|
|
|
|
func NewMacro(release domain.Release) Macro {
|
|
currentTime := time.Now()
|
|
|
|
ma := Macro{
|
|
TorrentName: release.TorrentName,
|
|
TorrentUrl: release.TorrentURL,
|
|
TorrentPathName: release.TorrentTmpFile,
|
|
TorrentHash: release.TorrentHash,
|
|
Indexer: release.Indexer,
|
|
Title: release.Title,
|
|
Resolution: release.Resolution,
|
|
Source: release.Source,
|
|
HDR: strings.Join(release.HDR, ", "),
|
|
Season: release.Season,
|
|
Episode: release.Episode,
|
|
Year: release.Year,
|
|
CurrentYear: currentTime.Year(),
|
|
CurrentMonth: int(currentTime.Month()),
|
|
CurrentDay: currentTime.Day(),
|
|
CurrentHour: currentTime.Hour(),
|
|
CurrentMinute: currentTime.Minute(),
|
|
CurrentSecond: currentTime.Second(),
|
|
}
|
|
|
|
return ma
|
|
}
|
|
|
|
// Parse takes a string and replaces valid vars
|
|
func (m Macro) Parse(text string) (string, error) {
|
|
|
|
// setup template
|
|
tmpl, err := template.New("macro").Parse(text)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
var tpl bytes.Buffer
|
|
err = tmpl.Execute(&tpl, m)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return tpl.String(), nil
|
|
}
|