#1-4 生成タスク周りをリファクタリング

OUCC · Apr 4, 2024 · 2f3069a · 2f3069a
1 parent 2dd3ea7
commit 2f3069a
Show file tree

Hide file tree

Showing 8 changed files with 79 additions and 63 deletions.
diff --git a/Epub/KoeBook.Epub/Services/AnalyzerService.cs b/Epub/KoeBook.Epub/Services/AnalyzerService.cs
@@ -15,9 +15,10 @@ public partial class AnalyzerService(IScraperSelectorService scrapingService, IE
     private readonly ILlmAnalyzerService _llmAnalyzerService = llmAnalyzerService;
     private Dictionary<string, string> _rubyReplacements = new Dictionary<string, string>();
 
-    public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties, string tempDirectory, string coverFilePath, CancellationToken cancellationToken)
-    {
-        coverFilePath = Path.Combine(tempDirectory, "Cover.png");
+    public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties, string tempDirectory, CancellationToken cancellationToken)
+    { 
+        Directory.CreateDirectory(tempDirectory);
+        var coverFilePath = Path.Combine(tempDirectory, "Cover.png");
         using var fs = File.Create(coverFilePath);
         await fs.WriteAsync(CoverFile.ToArray(), cancellationToken);
         await fs.FlushAsync(cancellationToken);
@@ -27,9 +28,13 @@ public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties,
         {
             document = await _scrapingService.ScrapingAsync(bookProperties.Source, coverFilePath, tempDirectory, bookProperties.Id, cancellationToken);
         }
+        catch (EbookException)
+        {
+            throw;
+        }
         catch (Exception ex)
         {
-            EbookException.Throw(ExceptionType.WebScrapingFailed, "", ex);
+            EbookException.Throw(ExceptionType.WebScrapingFailed, innerException: ex);
             return default;
         }
         _epubDocumentStoreService.Register(document, cancellationToken);
@@ -45,7 +50,7 @@ public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties,
                     {
                         var line = paragraph.Text;
                         // rubyタグがあればルビのdictionaryに登録
-                        var rubyDict = ExtractRuby(line);
+                        var rubyDict = ExtractRuby(line).ToDictionary();
 
                         foreach (var ruby in rubyDict)
                         {
@@ -85,20 +90,11 @@ public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties,
         return bookScripts;
     }
 
-    private static Dictionary<string, string> ExtractRuby(string text)
+    private static IEnumerable<KeyValuePair<string, string>> ExtractRuby(string text)
     {
-        var rubyDict = new Dictionary<string, string>();
-        var rubyRegex = new Regex("<ruby><rb>(.*?)</rb><rp>（</rp><rt>(.*?)</rt><rp>）</rp></ruby>");
-
-        foreach (Match match in rubyRegex.Matches(text))
-        {
-            if (!rubyDict.ContainsKey(match.Groups[1].Value))
-            {
-                rubyDict.Add(match.Groups[1].Value, match.Groups[2].Value);
-            }
-        }
-
-        return rubyDict;
+        return RubyRegex()
+            .Matches(text)
+            .Select(m => KeyValuePair.Create(m.Groups[1].Value, m.Groups[2].Value));
     }
 
     private static string ReplaceBaseTextWithRuby(string text, Dictionary<string, string> rubyDict)
@@ -113,4 +109,7 @@ private static string ReplaceBaseTextWithRuby(string text, Dictionary<string, st
 
         return resultText;
     }
+
+    [GeneratedRegex("<ruby><rb>(.*?)</rb><rp>（</rp><rt>(.*?)</rt><rp>）</rp></ruby>")]
+    private static partial Regex RubyRegex();
 }
diff --git a/Epub/KoeBook.Epub/Services/EpubGenerateService.cs b/Epub/KoeBook.Epub/Services/EpubGenerateService.cs
@@ -1,7 +1,6 @@
 using KoeBook.Core;
 using KoeBook.Core.Contracts.Services;
 using KoeBook.Core.Models;
-using KoeBook.Epub;
 using KoeBook.Epub.Contracts.Services;
 using KoeBook.Epub.Models;
 
@@ -17,8 +16,7 @@ public async ValueTask<string> GenerateEpubAsync(BookScripts bookScripts, string
     {
         cancellationToken.ThrowIfCancellationRequested();
 
-        var document = _documentStoreService.Documents.Where(doc => doc.Id == bookScripts.BookProperties.Id).FirstOrDefault()
-            ?? throw new InvalidOperationException($"The epub document ({bookScripts.BookProperties.Id}) can't be found.");
+        var document = _documentStoreService.Documents.Single(d => d.Id == bookScripts.BookProperties.Id);
 
         foreach (var scriptLine in bookScripts.ScriptLines)
         {

diff --git a/KoeBook.Core/Contracts/Services/IAnalyzerService.cs b/KoeBook.Core/Contracts/Services/IAnalyzerService.cs
@@ -8,5 +8,5 @@ public interface IAnalyzerService
     /// 本の情報の取得・解析を行います
     /// </summary>
     /// <returns>編集前の読み上げテキスト</returns>
-    ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties, string tempDirectory, string coverFilePath, CancellationToken cancellationToken);
+    ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties, string tempDirectory,  CancellationToken cancellationToken);
 }
diff --git a/KoeBook.Core/Contracts/Services/IDisplayStateChangeService.cs b/KoeBook.Core/Contracts/Services/IDisplayStateChangeService.cs
@@ -9,6 +9,8 @@ public interface IDisplayStateChangeService
     /// </summary>
     void UpdateState(BookProperties bookProperties, GenerationState state);
 
+    void UpdateTitle(BookProperties bookProperties, string title);
+
     /// <summary>
     /// プログレスバーを更新します
     /// </summary>

diff --git a/KoeBook.Core/Helpers/IDisplayStateChangeEx.cs b/KoeBook.Core/Helpers/IDisplayStateChangeEx.cs
@@ -24,9 +24,23 @@ public class DisplayStateChanging(IDisplayStateChangeService displayStateChangeS
 
         private readonly int _maximum = maximum;
 
+        private int _progress;
+
         public void UpdateProgress(int progress)
         {
-            _displayStateChangeService.UpdateProgress(_bookProperties, progress, _maximum);
+            _displayStateChangeService.UpdateProgress(_bookProperties, _progress = progress, _maximum);
+        }
+
+        public void IncrementProgress()
+        {
+            _progress++;
+            _displayStateChangeService.UpdateProgress(_bookProperties, _progress, _maximum);
+        }
+
+        public void Finish()
+        {
+            _progress = _maximum;
+            _displayStateChangeService.UpdateProgress(_bookProperties, _progress, _maximum);
         }
     }
 }
diff --git a/KoeBook/Services/CoreMocks/AnalyzerServiceMock.cs b/KoeBook/Services/CoreMocks/AnalyzerServiceMock.cs
@@ -1,8 +1,6 @@
 using KoeBook.Core.Contracts.Services;
 using KoeBook.Core.Helpers;
 using KoeBook.Core.Models;
-using KoeBook.Epub;
-using KoeBook.Epub.Models;
 using static KoeBook.Core.Helpers.IDisplayStateChangeEx;
 
 namespace KoeBook.Services.CoreMocks;
@@ -11,7 +9,7 @@ public class AnalyzerServiceMock(IDisplayStateChangeService stateService) : IAna
 {
     private readonly IDisplayStateChangeService _stateService = stateService;
 
-    public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties, string tempDirectory, string coverFilePath, CancellationToken cancellationToken)
+    public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties, string tempDirectory, CancellationToken cancellationToken)
     {
         DisplayStateChanging stateChanging;
         if (bookProperties.SourceType == SourceType.Url)

diff --git a/KoeBook/Services/DisplayStateChangeService.cs b/KoeBook/Services/DisplayStateChangeService.cs
@@ -27,4 +27,13 @@ public void UpdateState(BookProperties bookProperties, GenerationState state)
             taskService.GetProcessingTask(bookProperties.Id).State = state;
         });
     }
+
+    public void UpdateTitle(BookProperties bookProperties, string title)
+    {
+        var taskService = _taskService; // thisをキャプチャしないようにする
+        _ = App.MainWindow.DispatcherQueue.TryEnqueue(() =>
+        {
+            taskService.GetProcessingTask(bookProperties.Id).Title = title;
+        });
+    }
 }
diff --git a/KoeBook/Services/GenerationTaskRunnerService.cs b/KoeBook/Services/GenerationTaskRunnerService.cs
@@ -48,50 +48,45 @@ private async void TasksChanged(GenerationTask task, ChangedEvents changedEvents
 
     private async ValueTask RunAsync(GenerationTask task)
     {
-        try
-        {
-            var scripts = await _analyzerService.AnalyzeAsync(new(task.Id, task.Source, task.SourceType), _tempFolder, "", task.CancellationToken);
-            task.BookScripts = scripts;
-            task.State = GenerationState.Editting;
-            task.Progress = 0;
-            task.MaximumProgress = 0;
-            if (task.SkipEdit)
-            {
-                var resultPath = await _epubGenService.GenerateEpubAsync(scripts, _tempFolder, task.CancellationToken);
-                task.State = GenerationState.Completed;
-                task.Progress = 1;
-                task.MaximumProgress = 1;
-                var fileName = Path.GetFileName(resultPath);
-                File.Copy(resultPath, Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments), "KoeBook", fileName), true);
-            }
-        }
-        catch (OperationCanceledException)
-        {
-            task.State = GenerationState.Failed;
-        }
-        catch (EbookException e)
-        {
-            task.State = GenerationState.Failed;
-            await _dialogService.ShowInfoAsync("生成失敗", e.ExceptionType.GetEnumMemberValue()!, "OK", default);
-        }
-        catch
-        {
-            task.State = GenerationState.Failed;
-        }
+        if (task.CancellationToken.IsCancellationRequested || task.State == GenerationState.Failed)
+            return;
+
+        await RunAsyncCore(task, true);
+        await RunAsyncCore(task, false);
     }
 
     public async void RunGenerateEpubAsync(GenerationTask task)
     {
         if (task.CancellationToken.IsCancellationRequested || task.State == GenerationState.Failed || task.BookScripts is null)
             return;
+
+        await RunAsyncCore(task, false);
+    }
+
+    private async ValueTask RunAsyncCore(GenerationTask task, bool firstStep)
+    {
+        var tempDirectory = Path.Combine(_tempFolder, task.Id.ToString());
         try
         {
-            var resultPath = await _epubGenService.GenerateEpubAsync(task.BookScripts, _tempFolder, task.CancellationToken);
-            task.State = GenerationState.Completed;
-            task.Progress = 1;
-            task.MaximumProgress = 1;
-            var fileName = Path.GetFileName(resultPath);
-            File.Copy(resultPath, Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments), "KoeBook", fileName), true);
+            if (firstStep)
+            {
+                var scripts = await _analyzerService.AnalyzeAsync(new(task.Id, task.Source, task.SourceType), tempDirectory, task.CancellationToken);
+                task.BookScripts = scripts;
+                task.State = GenerationState.Editting;
+                task.Progress = 0;
+                task.MaximumProgress = 0;
+            }
+            else if (task.BookScripts is not null)
+            {
+                var resultPath = await _epubGenService.GenerateEpubAsync(task.BookScripts, tempDirectory, task.CancellationToken);
+                task.State = GenerationState.Completed;
+                task.Progress = 1;
+                task.MaximumProgress = 1;
+                var fileName = Path.GetFileName(resultPath);
+                File.Move(resultPath, Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments), "KoeBook", fileName), true);
+            }
+            else
+                throw new InvalidOperationException();
         }
         catch (OperationCanceledException)
         {
@@ -102,9 +97,10 @@ public async void RunGenerateEpubAsync(GenerationTask task)
             task.State = GenerationState.Failed;
             await _dialogService.ShowInfoAsync("生成失敗", e.ExceptionType.GetEnumMemberValue()!, "OK", default);
         }
-        catch
+        catch (Exception e)
         {
             task.State = GenerationState.Failed;
+            await _dialogService.ShowInfoAsync("生成失敗", $"不明なエラーが発生しました。\n{e.Message}", "OK", default);
         }
     }