1+ using System . Collections . Generic ;
12using System . Collections . Specialized ;
23using System . IO ;
34using System . Linq ;
5+ using System . Text . RegularExpressions ;
46using NLog ;
57using NzbDrone . Common . Disk ;
68using NzbDrone . Common . Extensions ;
@@ -25,23 +27,89 @@ public class ImportScriptService : IImportScript
2527 private readonly IProcessProvider _processProvider ;
2628 private readonly IConfigService _configService ;
2729 private readonly ITagRepository _tagRepository ;
30+ private readonly IDiskProvider _diskProvider ;
2831 private readonly Logger _logger ;
2932
3033 public ImportScriptService ( IProcessProvider processProvider ,
3134 IVideoFileInfoReader videoFileInfoReader ,
3235 IConfigService configService ,
3336 IConfigFileProvider configFileProvider ,
3437 ITagRepository tagRepository ,
38+ IDiskProvider diskProvider ,
3539 Logger logger )
3640 {
3741 _processProvider = processProvider ;
3842 _videoFileInfoReader = videoFileInfoReader ;
3943 _configService = configService ;
4044 _configFileProvider = configFileProvider ;
4145 _tagRepository = tagRepository ;
46+ _diskProvider = diskProvider ;
4247 _logger = logger ;
4348 }
4449
50+ private static readonly Regex OutputRegex = new Regex ( @"^(?:\[(?:(?<mediaFile>MediaFile)|(?<extraFile>ExtraFile))\]\s?(?<fileName>.+)|(?<preventExtraImport>\[PreventExtraImport\])|\[MoveStatus\]\s?(?:(?<deferMove>DeferMove)|(?<moveComplete>MoveComplete)|(?<renameRequested>RenameRequested)))$" , RegexOptions . Compiled ) ;
51+
52+ private ScriptImportInfo ProcessOutput ( List < ProcessOutputLine > processOutputLines )
53+ {
54+ var possibleExtraFiles = new List < string > ( ) ;
55+ string mediaFile = null ;
56+ var decision = ScriptImportDecision . MoveComplete ;
57+ var importExtraFiles = true ;
58+
59+ foreach ( var line in processOutputLines )
60+ {
61+ var match = OutputRegex . Match ( line . Content ) ;
62+
63+ if ( match . Groups [ "mediaFile" ] . Success )
64+ {
65+ if ( mediaFile is not null )
66+ {
67+ throw new ScriptImportException ( "Script output contains multiple media files. Only one media file can be returned." ) ;
68+ }
69+
70+ mediaFile = match . Groups [ "fileName" ] . Value ;
71+
72+ if ( ! MediaFileExtensions . Extensions . Contains ( Path . GetExtension ( mediaFile ) ) )
73+ {
74+ throw new ScriptImportException ( "Script output contains invalid media file: {0}" , mediaFile ) ;
75+ }
76+ else if ( ! _diskProvider . FileExists ( mediaFile ) )
77+ {
78+ throw new ScriptImportException ( "Script output contains non-existent media file: {0}" , mediaFile ) ;
79+ }
80+ }
81+ else if ( match . Groups [ "extraFile" ] . Success )
82+ {
83+ var fileName = match . Groups [ "fileName" ] . Value ;
84+
85+ if ( ! _diskProvider . FileExists ( fileName ) )
86+ {
87+ _logger . Warn ( "Script output contains non-existent possible extra file: {0}" , fileName ) ;
88+ }
89+
90+ possibleExtraFiles . Add ( fileName ) ;
91+ }
92+ else if ( match . Groups [ "moveComplete" ] . Success )
93+ {
94+ decision = ScriptImportDecision . MoveComplete ;
95+ }
96+ else if ( match . Groups [ "renameRequested" ] . Success )
97+ {
98+ decision = ScriptImportDecision . RenameRequested ;
99+ }
100+ else if ( match . Groups [ "deferMove" ] . Success )
101+ {
102+ decision = ScriptImportDecision . DeferMove ;
103+ }
104+ else if ( match . Groups [ "preventExtraImport" ] . Success )
105+ {
106+ importExtraFiles = false ;
107+ }
108+ }
109+
110+ return new ScriptImportInfo ( possibleExtraFiles , mediaFile , decision , importExtraFiles ) ;
111+ }
112+
45113 public ScriptImportDecision TryImport ( string sourcePath , string destinationFilePath , LocalMovie localMovie , MovieFile movieFile , TransferMode mode )
46114 {
47115 var movie = localMovie . Movie ;
@@ -111,22 +179,37 @@ public ScriptImportDecision TryImport(string sourcePath, string destinationFileP
111179
112180 var processOutput = _processProvider . StartAndCapture ( _configService . ScriptImportPath , $ "\" { sourcePath } \" \" { destinationFilePath } \" ", environmentVariables ) ;
113181
114- _logger . Debug ( "Executed external script: {0} - Status: {1}" , _configService . ScriptImportPath , processOutput . ExitCode ) ;
115182 _logger . Debug ( "Script Output: \r \n {0}" , string . Join ( "\r \n " , processOutput . Lines ) ) ;
116183
117- switch ( processOutput . ExitCode )
184+ if ( processOutput . ExitCode != 0 )
185+ {
186+ throw new ScriptImportException ( "Script exited with non-zero exit code: {0}" , processOutput . ExitCode ) ;
187+ }
188+
189+ var scriptImportInfo = ProcessOutput ( processOutput . Lines ) ;
190+
191+ var mediaFile = scriptImportInfo . MediaFile ?? destinationFilePath ;
192+ localMovie . PossibleExtraFiles = scriptImportInfo . PossibleExtraFiles ;
193+
194+ movieFile . RelativePath = movie . Path . GetRelativePath ( mediaFile ) ;
195+ movieFile . Path = mediaFile ;
196+
197+ var exitCode = processOutput . ExitCode ;
198+
199+ localMovie . ShouldImportExtras = scriptImportInfo . ImportExtraFiles ;
200+
201+ if ( scriptImportInfo . Decision != ScriptImportDecision . DeferMove )
118202 {
119- case 0 : // Copy complete
120- return ScriptImportDecision . MoveComplete ;
121- case 2 : // Copy complete, file potentially changed, should try renaming again
122- movieFile . MediaInfo = _videoFileInfoReader . GetMediaInfo ( destinationFilePath ) ;
123- movieFile . Path = null ;
124- return ScriptImportDecision . RenameRequested ;
125- case 3 : // Let Radarr handle it
126- return ScriptImportDecision . DeferMove ;
127- default : // Error, fail to import
128- throw new ScriptImportException ( "Moving with script failed! Exit code {0}" , processOutput . ExitCode ) ;
203+ localMovie . ScriptImported = true ;
129204 }
205+
206+ if ( scriptImportInfo . Decision == ScriptImportDecision . RenameRequested )
207+ {
208+ movieFile . MediaInfo = _videoFileInfoReader . GetMediaInfo ( mediaFile ) ;
209+ movieFile . Path = null ;
210+ }
211+
212+ return scriptImportInfo . Decision ;
130213 }
131214 }
132215}
0 commit comments