feat: 完善相册 AI 修图模板选择流程

2026-08-11 15:22:08 +08:00
parent 43c75c8e36
commit e64dcec094
11 changed files with 261 additions and 202 deletions
@@ -29,6 +29,9 @@ protocol TravelAlbumServing {
        isPurchased: Int?
    ) async throws -> TravelAlbumListResponse<TravelAlbumMaterial>

+    /// 拉取单个相册素材及其最新关联图片信息。
+    func materialInfo(userEquityTravelId: Int, materialId: Int) async throws -> TravelAlbumMaterial
+
    /// 上传并登记旅拍相册素材。
    func uploadMaterial(_ request: TravelAlbumUploadMaterialRequest) async throws -> TravelAlbumMaterial

@@ -122,6 +125,20 @@ final class TravelAlbumAPI: TravelAlbumServing {
        )
    }

+    /// 拉取单个相册素材及其最新关联图片信息。
+    func materialInfo(userEquityTravelId: Int, materialId: Int) async throws -> TravelAlbumMaterial {
+        try await client.send(
+            APIRequest(
+                method: .get,
+                path: "\(basePath)/material-info",
+                queryItems: [
+                    URLQueryItem(name: "user_equity_travel_id", value: String(userEquityTravelId)),
+                    URLQueryItem(name: "material_id", value: String(materialId)),
+                ]
+            )
+        )
+    }
+
    /// 上传并登记旅拍相册素材。
    func uploadMaterial(_ request: TravelAlbumUploadMaterialRequest) async throws -> TravelAlbumMaterial {
        try await client.send(APIRequest(method: .post, path: "\(basePath)/upload-material", body: request))
@@ -428,22 +428,35 @@ struct TravelAlbumAIRetouchTemplatesResponse: Decodable, Sendable, Equatable {
    let refinedTemplates: [TravelAlbumAIRetouchTemplate]
    let atmosphereTemplates: [TravelAlbumAIRetouchTemplate]
    let coverTemplates: [TravelAlbumAIRetouchTemplate]
+    let remainingQuota: Int

    enum CodingKeys: String, CodingKey {
        case refinedTemplates = "refined_templates"
        case atmosphereTemplates = "atmosphere_templates"
        case coverTemplates = "cover_templates"
+        case remainingQuota = "remaining_quota"
    }

    /// 创建分组模板响应，默认各组为空。
    init(
        refinedTemplates: [TravelAlbumAIRetouchTemplate] = [],
        atmosphereTemplates: [TravelAlbumAIRetouchTemplate] = [],
-        coverTemplates: [TravelAlbumAIRetouchTemplate] = []
+        coverTemplates: [TravelAlbumAIRetouchTemplate] = [],
+        remainingQuota: Int = 0
    ) {
        self.refinedTemplates = refinedTemplates
        self.atmosphereTemplates = atmosphereTemplates
        self.coverTemplates = coverTemplates
+        self.remainingQuota = max(0, remainingQuota)
+    }
+
+    /// 解码模板和剩余额度；旧响应缺少额度时按零处理，避免误提交付费任务。
+    init(from decoder: any Decoder) throws {
+        let container = try decoder.container(keyedBy: CodingKeys.self)
+        refinedTemplates = try container.decode([TravelAlbumAIRetouchTemplate].self, forKey: .refinedTemplates)
+        atmosphereTemplates = try container.decode([TravelAlbumAIRetouchTemplate].self, forKey: .atmosphereTemplates)
+        coverTemplates = try container.decode([TravelAlbumAIRetouchTemplate].self, forKey: .coverTemplates)
+        remainingQuota = max(0, try container.decodeIfPresent(Int.self, forKey: .remainingQuota) ?? 0)
    }
 }

@@ -13,6 +13,7 @@ final class TravelAlbumAIRetouchTemplateViewModel {
    private(set) var selectedRefinedTemplateId: Int?
    private(set) var selectedAtmosphereTemplateId: Int?
    private(set) var selectedCoverTemplateId: Int?
+    private(set) var remainingQuota: Int?
    private(set) var isLoading = false
    private(set) var isSubmitting = false
    private(set) var loadErrorMessage: String?
@@ -56,6 +57,22 @@ final class TravelAlbumAIRetouchTemplateViewModel {
        workflow.isOptional(category)
    }

+    /// 当前选择预计消耗的修图次数；封面模板免费，不计入额度。
+    var requiredQuota: Int {
+        switch workflow {
+        case .initial(_, let materialIds):
+            let outputCount = 1 + (selectedAtmosphereTemplateId == nil ? 0 : 1)
+            return materialIds.count * outputCount
+        case .reretouch(_, _, let type):
+            switch type {
+            case .refined, .atmosphere:
+                return 1
+            case .all:
+                return 1 + (selectedAtmosphereTemplateId == nil ? 0 : 1)
+            }
+        }
+    }
+
    /// 当前必选模板或业务参数缺失时用于底部提示的文案。
    var validationMessage: String? {
        guard !isLoading, loadErrorMessage == nil else { return nil }
@@ -70,6 +87,12 @@ final class TravelAlbumAIRetouchTemplateViewModel {
                return unavailableMessage(for: category)
            }
        }
+        guard let remainingQuota else {
+            return "剩余修图次数获取失败，请刷新后重试"
+        }
+        if requiredQuota > remainingQuota {
+            return "剩余修图次数不足，需要\(requiredQuota)次，当前剩余\(remainingQuota)次"
+        }
        return nil
    }

@@ -92,6 +115,7 @@ final class TravelAlbumAIRetouchTemplateViewModel {
        guard !isLoading else { return }
        isLoading = true
        loadErrorMessage = nil
+        remainingQuota = nil
        notifyStateChange()

        do {
@@ -99,6 +123,7 @@ final class TravelAlbumAIRetouchTemplateViewModel {
            refinedTemplates = response.refinedTemplates
            atmosphereTemplates = response.atmosphereTemplates
            coverTemplates = response.coverTemplates
+            remainingQuota = response.remainingQuota
            selectedRefinedTemplateId = visibleCategories.contains(.refined) ? refinedTemplates.first?.id : nil
            selectedAtmosphereTemplateId = visibleCategories.contains(.atmosphere) && !isOptional(.atmosphere)
                ? atmosphereTemplates.first?.id
@@ -116,6 +141,7 @@ final class TravelAlbumAIRetouchTemplateViewModel {
            selectedRefinedTemplateId = nil
            selectedAtmosphereTemplateId = nil
            selectedCoverTemplateId = nil
+            remainingQuota = nil
            isLoading = false
            loadErrorMessage = error.localizedDescription.isEmpty ? "模板加载失败" : error.localizedDescription
            notifyStateChange()
@@ -207,41 +207,18 @@ final class TravelAlbumDetailViewModel {
        }
    }

-    /// 重新拉取当前筛选、排序下已经加载的分页范围，供全屏预览刷新关联图片。
-    func reloadLoadedMaterials(
-        api: any TravelAlbumServing
-    ) async throws -> TravelAlbumListResponse<TravelAlbumMaterial> {
-        let requestedPageCount = max(currentPage, 1)
-        var refreshed: [TravelAlbumMaterial] = []
-        var refreshedTotal = 0
-        var loadedPageCount = 0
-
-        for page in 1 ... requestedPageCount {
-            let response = try await api.materialList(
-                userEquityTravelId: albumId,
-                page: page,
-                pageSize: pageSize,
-                orderBy: sortOption.rawValue,
-                isPurchased: selectedTab == .purchased ? 1 : nil
-            )
-            if page == 1 { refreshedTotal = response.total }
-            refreshed.append(contentsOf: response.list)
-            loadedPageCount = page
-            if refreshed.count >= refreshedTotal || response.list.isEmpty { break }
+    /// 定向刷新单个素材，并同步替换网格列表中的对应缓存。
+    func refreshMaterial(id: Int, api: any TravelAlbumServing) async throws -> TravelAlbumMaterial {
+        let material = try await api.materialInfo(
+            userEquityTravelId: albumId,
+            materialId: id
+        )
+        guard material.id == id else { throw APIError.invalidResponse }
+        if let index = materials.firstIndex(where: { $0.id == id }) {
+            materials[index] = material
+            notifyStateChange()
        }
-
-        var seen = Set<Int>()
-        materials = refreshed.filter { seen.insert($0.id).inserted }
-        currentPage = max(loadedPageCount, 1)
-        canLoadMore = materials.count < refreshedTotal
-        selectedMaterialIds.formIntersection(materials.map(\.id))
-        if selectedTab == .all {
-            allPhotoCount = refreshedTotal
-        } else {
-            purchasedPhotoCount = refreshedTotal
-        }
-        notifyStateChange()
-        return TravelAlbumListResponse(total: refreshedTotal, list: materials)
+        return material
    }

    /// 切换选择模式。