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

feat(actions): watch dir rename file (#515)

Browse source
This commit is contained in:
ze0s 2022-10-28 00:29:34 +02:00 • committed by GitHub
parent 716784e7a6
commit 547b01a50d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

32
internal/action/run.go
View file

@ -67,7 +67,7 @@ func (s *service) RunAction(action *domain.Action, release domain.Release) ([]st
rejections, err = s.whisparr(*action, release)
case domain.ActionTypeReadarr:
rejections, err = s.readarr(*action, release)
rejections, err = s.readarr(*action, release)
default:
s.log.Warn().Msgf("unsupported action type: %v", action.Type)
@ -154,6 +154,10 @@ func (s *service) watchFolder(action domain.Action, release domain.Release) erro
m := domain.NewMacro(release)
// parse and replace values in argument string before continuing
// /mnt/watch/{{.Indexer}}
// /mnt/watch/mock
// /mnt/watch/{{.Indexer}}-{{.TorrentName}}.torrent
// /mnt/watch/mock-Torrent.Name-GROUP.torrent
watchFolderArgs, err := m.Parse(action.WatchFolder)
if err != nil {
return errors.Wrap(err, "could not parse watch folder macro: %v", action.WatchFolder)
@ -168,27 +172,37 @@ func (s *service) watchFolder(action domain.Action, release domain.Release) erro
}
defer original.Close()
_, tmpFileName := filepath.Split(release.TorrentTmpFile)
fullFileName := filepath.Join(watchFolderArgs, tmpFileName+".torrent")
// default dir to watch folder
dir := watchFolderArgs
newFileName := watchFolderArgs
// if watchFolderArgs does not contain .torrent, create
if !strings.Contains(watchFolderArgs, ".torrent") {
_, tmpFileName := filepath.Split(release.TorrentTmpFile)
newFileName = filepath.Join(watchFolderArgs, tmpFileName+".torrent")
} else {
dir, _ = filepath.Split(watchFolderArgs)
}
// Create folder
if err = os.MkdirAll(watchFolderArgs, os.ModePerm); err != nil {
return errors.Wrap(err, "could not create new folders %v", fullFileName)
if err = os.MkdirAll(dir, os.ModePerm); err != nil {
return errors.Wrap(err, "could not create new folders %v", dir)
}
// Create new file
newFile, err := os.Create(fullFileName)
newFile, err := os.Create(newFileName)
if err != nil {
return errors.Wrap(err, "could not create new file %v", fullFileName)
return errors.Wrap(err, "could not create new file %v", newFileName)
}
defer newFile.Close()
// Copy file
if _, err := io.Copy(newFile, original); err != nil {
return errors.Wrap(err, "could not copy file %v to watch folder", fullFileName)
return errors.Wrap(err, "could not copy file %v to watch folder", newFileName)
}
s.log.Info().Msgf("saved file to watch folder: %v", fullFileName)
s.log.Info().Msgf("saved file to watch folder: %v", newFileName)
return nil
}