feat: add AI retouch task center

2026-08-14 15:06:57 +08:00
parent 0f3b26991e
commit 30bfe0313f
26 changed files with 3679 additions and 50 deletions
@@ -25,7 +25,7 @@ final class TravelAlbumAIRetouchTemplateViewController: BaseViewController {

    private let viewModel: TravelAlbumAIRetouchTemplateViewModel
    private let api: any TravelAlbumServing
-    private let onSubmitted: () -> Void
+    private let onSubmitted: (TravelAlbumAIJobSubmission) -> Void

    private lazy var collectionView = UICollectionView(frame: .zero, collectionViewLayout: makeLayout())
    private var dataSource: UICollectionViewDiffableDataSource<Section, Item>!
@@ -45,7 +45,7 @@ final class TravelAlbumAIRetouchTemplateViewController: BaseViewController {
    init(
        viewModel: TravelAlbumAIRetouchTemplateViewModel,
        api: any TravelAlbumServing,
-        onSubmitted: @escaping () -> Void
+        onSubmitted: @escaping (TravelAlbumAIJobSubmission) -> Void
    ) {
        self.viewModel = viewModel
        self.api = api
@@ -182,10 +182,10 @@ final class TravelAlbumAIRetouchTemplateViewController: BaseViewController {
        viewModel.onShowMessage = { [weak self] message in
            Task { @MainActor in self?.showToast(message) }
        }
-        viewModel.onSubmitted = { [weak self] in
+        viewModel.onSubmitted = { [weak self] submission in
            Task { @MainActor in
                guard let self else { return }
-                self.dismiss(animated: true, completion: self.onSubmitted)
+                self.dismiss(animated: true) { self.onSubmitted(submission) }
            }
        }
    }