Merge branch 'main' into feat/#23

OUCC · Apr 20, 2024 · 990c31e · 990c31e
2 parents 456ff7d + de81a0a
commit 990c31e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/KoeBook/Services/CoreMocks/AnalyzerServiceMock.cs b/KoeBook/Services/CoreMocks/AnalyzerServiceMock.cs
@@ -11,6 +11,7 @@ public class AnalyzerServiceMock(IDisplayStateChangeService stateService) : IAna
 
     public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties, string tempDirectory, CancellationToken cancellationToken)
     {
+        Directory.CreateDirectory(tempDirectory);
         DisplayStateChanging stateChanging;
         if (bookProperties.SourceType == SourceType.Url)
         {

diff --git a/KoeBook/Services/GenerationTaskRunnerService.cs b/KoeBook/Services/GenerationTaskRunnerService.cs
@@ -52,7 +52,8 @@ private async ValueTask RunAsync(GenerationTask task)
             return;
 
         await RunAsyncCore(task, true);
-        await RunAsyncCore(task, false);
+        if (task.SkipEdit)
+            await RunAsyncCore(task, false);
     }
 
     public async void RunGenerateEpubAsync(GenerationTask task)
@@ -83,7 +84,10 @@ private async ValueTask RunAsyncCore(GenerationTask task, bool firstStep)
                 task.Progress = 1;
                 task.MaximumProgress = 1;
                 var fileName = Path.GetFileName(resultPath);
-                File.Move(resultPath, Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments), "KoeBook", fileName), true);
+                var resultDirectory = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments), "KoeBook");
+                if (!Directory.Exists(resultDirectory))
+                    Directory.CreateDirectory(resultDirectory);
+                File.Move(resultPath, Path.Combine(resultDirectory, fileName), true);
             }
             else
                 throw new InvalidOperationException();