10 Commits
23 changed files with 1289 additions and 277 deletions
+282
View File
@@ -0,0 +1,282 @@
# AI 重新修图接口优化需求
## 1. 背景
图片预览页面支持用户在“原图”Tab 上重新选择修图模板。
用户可能只生成过一种关联图片,例如之前仅生成了“原图精修”,尚未生成“氛围感”图片。此时用户重新修图时,可以同时选择:
- 原图精修模板
- 氛围感修图模板
客户端会调用现有 `ai-reretouch` 接口,并通过 `type = 3` 提交两个模板,希望一次创建两个修图输出任务。
## 2. 问题复现
### 前置状态
- 原图素材 ID:`702`
- 原 AI 修图批次 ID:`61`
- 已存在并完成:原图精修
- 从未生成:氛围感图片
### 请求
```http
POST /api/yf-handset-app/photog/travel-album/ai-reretouch
```
```json
{
"ai_retouch_batch_id": 61,
"id": 702,
"type": 3,
"refined_template_id": 2,
"atmosphere_template_id": 14
}
```
字段含义:
| 字段 | 值 | 说明 |
| --- | ---: | --- |
| `ai_retouch_batch_id` | `61` | 原 AI 修图批次 ID |
| `id` | `702` | 原图素材 ID |
| `type` | `3` | 同时处理精修和氛围感 |
| `refined_template_id` | `2` | 本次选择的精修模板 |
| `atmosphere_template_id` | `14` | 本次选择的氛围感模板 |
### 当前响应
```json
{
"code": 100099,
"data": [],
"msg": "仅已完成修图的照片可重新修图",
"time": "2026-08-14 17:41:17"
}
```
## 3. 当前问题
从响应判断,后端可能在 `type = 3` 时要求精修图和氛围感图都已经存在且完成,才允许重新修图。
但本场景中:
- 精修图已经存在,本次需要重新生成,并在成功后覆盖旧精修图。
- 氛围感图从未生成,本次需要创建新的关联图片。
由于氛围感图历史上不存在,接口拒绝了整个请求,导致客户端无法通过一次请求同时完成“覆盖已有精修图”和“新增氛围感图”。
客户端当前请求已经正确传递 `type = 3` 和两个模板 ID,无需修改请求结构。
## 4. 期望接口行为
建议将 `ai-reretouch` 调整为按输出类型分别执行的 **upsert(存在则覆盖,不存在则新增)** 语义。
对于本次请求:
### 原图精修
- 请求传入了 `refined_template_id`。
- 旧精修图已经存在。
- 后端创建新的精修任务。
- 新任务成功后覆盖旧精修图。
- 新任务处理中或失败时保留旧精修图。
### 氛围感修图
- 请求传入了 `atmosphere_template_id`。
- 旧氛围感图不存在。
- 后端创建新的氛围感任务。
- 任务成功后,将结果保存为原图的新关联图片。
### 整体任务
- 一次请求创建两个输出任务:`refined` 和 `atmosphere`。
- 按两个实际创建的输出预占 2 张修图额度。
- 两个输出分别执行、分别记录状态、分别结算额度。
- 单个输出失败不应影响另一个已经成功的输出。
## 5. 建议的后端处理规则
### 5.1 基础校验
建议校验:
- 原图素材存在。
- 原图素材属于指定相册或批次。
- 原图当前状态允许提交 AI 修图任务。
- 请求中至少传入一个与 `type` 匹配的有效模板 ID。
- 用户剩余额度满足本次实际创建的输出数量。
不建议校验:
- `type = 3` 时要求精修和氛围感历史结果都必须存在。
- 某一种关联图从未生成时直接拒绝整个请求。
### 5.2 按模板字段创建任务
后端根据本次实际传入的模板字段创建对应任务:
| 请求情况 | 期望处理 |
| --- | --- |
| 只传 `refined_template_id` | 只创建精修任务 |
| 只传 `atmosphere_template_id` | 只创建氛围感任务 |
| 两个模板 ID 都传 | 同时创建精修和氛围感任务 |
| 某个模板 ID 未传 | 不处理、不覆盖该类型的已有结果 |
### 5.3 按输出类型处理已有结果
对于每个实际提交的输出类型,分别判断:
| 历史状态 | 建议处理 |
| --- | --- |
| 已存在成功结果 | 创建重修任务,新结果成功后替换旧结果 |
| 从未生成 | 创建新任务,成功后新增关联图片 |
| 历史任务失败或已取消 | 允许重新创建任务 |
| 当前已有处理中或排队任务 | 拒绝该类型重复提交,或通过幂等机制复用已有任务 |
### 5.4 替换时机
继续采用现有 `replace_on_success` 语义:
- 新结果成功后,才替换同类型旧结果。
- 新任务排队、处理中或失败时,保留旧结果。
- 精修和氛围感分别替换,互不影响。
## 6. `type` 参数建议语义
保持现有请求类型和接口路径不变:
| `type` | 语义 | 模板规则 |
| ---: | --- | --- |
| `1` | 只处理精修 | 必须传 `refined_template_id`,忽略氛围感模板 |
| `2` | 只处理氛围感 | 必须传 `atmosphere_template_id`,忽略精修模板 |
| `3` | 同时处理精修和氛围感 | 根据实际传入的两个模板分别创建任务 |
对于 `type = 3`:
- 必须传 `refined_template_id`。
- `atmosphere_template_id` 可以选传。
- 传入氛围感模板时创建氛围感任务。
- 未传氛围感模板时,只重修精修图,并保留已有氛围感图。
## 7. 建议处理伪代码
```text
validateOriginalMaterial(request.id)
validateBatch(request.ai_retouch_batch_id)
outputs = []
if request.refined_template_id is not null:
outputs.add(
createTask(
outputType = refined,
templateId = request.refined_template_id,
mode = existingRefinedResult ? replace_on_success : create
)
)
if request.atmosphere_template_id is not null:
outputs.add(
createTask(
outputType = atmosphere,
templateId = request.atmosphere_template_id,
mode = existingAtmosphereResult ? replace_on_success : create
)
)
if outputs is empty:
return invalid_template_error
reserveQuota(outputs.count)
submitTasks(outputs)
```
注意:实际是否存在旧结果,应按照各自的 `output_type` 独立查询,不能用“两个类型都已完成”作为 `type = 3` 的整体前置条件。
## 8. 期望响应
接口成功响应建议继续返回任务批次信息,并明确本次创建的输出类型。例如:
```json
{
"code": 100000,
"data": {
"ai_retouch_batch_id": 61,
"source_material_id": 702,
"outputs": [
{
"type": "refined",
"mode": "replace_on_success",
"status": "queued"
},
{
"type": "atmosphere",
"mode": "create",
"status": "queued"
}
],
"reserved_units": 2
},
"msg": "AI修图任务已提交"
}
```
响应结构可沿用现有实现,不强制增加上述字段;关键要求是一次请求能够成功创建两个对应的修图任务。
## 9. 验收用例
### 用例一:只有旧精修图,同时提交两个模板
- 历史状态:精修已完成,氛围感不存在。
- 请求:`type = 3`,同时传精修和氛围感模板。
- 期望:创建两个任务;精修成功后覆盖,氛围感成功后新增。
### 用例二:只有旧氛围感图,同时提交两个模板
- 历史状态:精修不存在,氛围感已完成。
- 请求:`type = 3`,同时传精修和氛围感模板。
- 期望:创建两个任务;精修成功后新增,氛围感成功后覆盖。
### 用例三:两种旧结果都存在
- 历史状态:精修、氛围感都已完成。
- 请求:`type = 3`,同时传两个模板。
- 期望:创建两个任务,分别在成功后覆盖各自旧结果。
### 用例四:原图 Tab 只选择精修模板
- 历史状态:精修、氛围感均可能存在。
- 请求:`type = 3`,只传 `refined_template_id`。
- 期望:只创建精修任务;不处理、不删除、不覆盖旧氛围感图;预占 1 张额度。
### 用例五:精修后 Tab 重新修图
- 请求:`type = 1`,只传 `refined_template_id`。
- 期望:只创建精修任务,新精修成功后覆盖旧精修图。
### 用例六:氛围感 Tab 重新修图
- 请求:`type = 2`,只传 `atmosphere_template_id`。
- 期望:只创建氛围感任务,新氛围感成功后覆盖旧氛围感图。
### 用例七:其中一个输出失败
- 请求创建精修和氛围感两个任务。
- 精修成功,氛围感失败。
- 期望:保留成功的新精修结果;氛围感旧图存在时继续保留旧图,不存在时不生成关联图;额度分别结算。
### 用例八:防止重复提交
- 同一原图、同一输出类型已经存在排队中或处理中的任务。
- 用户再次提交相同任务。
- 期望:通过幂等机制返回原任务,或仅拒绝重复的输出类型,避免重复扣减额度和重复生成结果。
## 10. 总结
希望 `ai-reretouch` 支持以下统一语义:
> 对本次传入模板对应的每种输出独立处理:已有结果则在新任务成功后覆盖,没有结果则新增;未传模板的类型不处理。不要因为某一种关联图历史上从未生成,就拒绝整个重新修图请求。
+54
View File
@@ -0,0 +1,54 @@
# 工作周报(2026.08.10—2026.08.14)
## 本周概况
本周围绕旅拍相册和 AI 修图功能开展开发,共完成 **23 次提交**,涉及 **37 个文件**,代码变更约新增 **11,495 行**、删除 **1,120 行**。
## 本周完成
### 1. 完善旅拍相册图片预览
- 新增 iOS 端全屏图片预览。
- 完善预览工具栏、图片索引胶囊、分段控件及关闭按钮样式。
- 支持原图与 AI 修图版本切换,并修复切换时出现黑屏的问题。
- 增加相册管理页下拉刷新。
- 接入旅拍相册批量删除接口。
### 2. 打通 AI 修图业务流程
- 从相册及图片预览页接入 AI 修图入口。
- 完成图片选择、模板选择、模板预览和任务提交流程。
- 优化修图标签、页面布局及提交反馈。
- 新增 AI 修图前后效果对比功能。
### 3. 新增 AI 修图任务中心
- 完成任务列表及任务详情页面。
- 接入 AI 修图任务相关接口和状态模型。
- 支持处理中、成功、失败等任务状态展示。
- 打通消息中心、推送通知与任务详情跳转链路。
### 4. 优化相册数据同步
- 修复从 OTG 页面返回后旅拍相册未及时刷新的问题。
- 优化修图提交后的页面反馈及状态更新。
- 完善异常状态和页面切换场景下的数据刷新逻辑。
### 5. 补充文档与测试
- 补充 AI 修图需求、接口和交互设计文档。
- 完善相册 API、数据模型、ViewModel、消息中心及推送相关单元测试。
- 增加图片预览、任务中心和修图流程的回归测试覆盖。
## 风险与待验证项
- AI 修图任务中心涉及接口、推送、消息中心和页面跳转,需要结合测试环境继续进行全链路验证。
- 需重点回归任务处理中、成功、失败及网络异常等状态。
- OTG 返回刷新、批量删除和图片版本切换需要在真机及大相册场景下继续验证。
## 下周计划
- 联调 AI 修图任务全生命周期及推送跳转。
- 完成旅拍相册与 AI 修图功能的真机回归。
- 优化大图加载、任务刷新和弱网场景下的交互体验。
- 根据测试反馈修复问题,推进功能验收与发布准备。
+63
View File
@@ -0,0 +1,63 @@
# 工作周报(2026.08.17—2026.08.21)
## 本周概况
本周围绕旅拍相册和 AI 重修流程开展功能完善与体验优化,共完成 **7 次提交**,涉及 **20 个文件**,代码及文档变更约新增 **1,101 行**、删除 **240 行**;同时完成版本号升级,为 **1.3.1** 版本发布做准备。
## 本周完成
### 1. 完善 AI 重修业务规则
- 根据是否已有 AI 结果区分首次修图与重新修图流程。
- 已有 AI 结果时,统一支持精修和氛围感模板按需选择,并校验至少选择一种模板。
- 优化单图、多图场景下的模板必选规则、生成数量与额度提示。
- 对齐 AI 重修请求参数、模板范围及结果覆盖规则,确保未选结果类型保持原有版本不变。
### 2. 优化相册素材状态与删除保护
- 将购买状态和 AI 修图状态拆分展示,避免不同业务状态互相覆盖。
- 扩展素材选择能力,支持跨相册 Tab 进入选择模式。
- 增加已购素材删除保护,批量删除时仅提交未购买素材。
- 优化删除确认提示,明确已选、已购及实际可删除数量;全部为已购素材时阻止删除并给出反馈。
### 3. 改进任务详情与图片预览体验
- 优化 AI 修图任务详情中的结果类型标签、状态区域和失败信息布局,提升信息辨识度。
- 调整任务目标标题、模板和操作按钮布局,改善长文本与不同状态下的适配表现。
- 完善图片预览初始版本选择,当前版本不可用时自动回退到有效图片。
- 新增高对比度延迟加载指示器,减少快速加载时的闪烁,并优化未知文件大小的展示。
### 4. 完善相册刷新能力
- 为旅拍相册入口增加下拉刷新。
- 补充刷新状态管理,确保成功、失败及取消场景均能正确结束刷新动画。
### 5. 补充文档与测试覆盖
- 新增 AI 重新修图接口优化需求文档,并同步更新 AI 修图需求及接口说明。
- 完善 AI 修图模板、重修请求、相册素材模型、删除保护及页面布局相关单元测试。
- 增加首次修图、已有结果重修、模板选填和异常状态等关键场景的回归覆盖。
### 6. 推进版本发布准备
- 将项目版本号升级至 **1.3.1**。
- 整理本轮 AI 重修规则、交互调整及测试范围,为后续联调验收提供依据。
## 进行中
- 补充 AI 重修模板占位图的多倍图资源配置。
- 继续验证 AI 重修、批量删除及图片预览在真机和真实数据下的完整表现。
## 风险与待验证项
- AI 重修规则涉及模板范围、额度计算和结果覆盖,需要结合测试环境验证服务端行为是否与最新规则一致。
- 已购与未购素材混合选择时,需要重点回归删除参数、数量提示及刷新后的数据一致性。
- 图片预览需继续验证弱网、大图及不同 AI 结果组合下的加载、回退和切换体验。
- 本周测试代码已补充,仍需在已连接的 iPhone 真机上完成完整测试回归并记录结果。
## 下周计划
- 完成 1.3.1 版本真机测试、问题修复及发布验收。
- 联调 AI 重修模板选择、额度扣减、任务生成与结果覆盖全链路。
- 回归相册下拉刷新、素材状态展示和已购素材删除保护。
- 补齐 AI 重修模板多倍图资源,优化弱网和大图加载体验。
+6 -6
View File
@@ -436,7 +436,7 @@
ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
CODE_SIGN_ENTITLEMENTS = suixinkan/suixinkan.entitlements;
CODE_SIGN_STYLE = Automatic;
CURRENT_PROJECT_VERSION = 1020101;
CURRENT_PROJECT_VERSION = 1030101;
DEVELOPMENT_TEAM = 56GVN5RNVN;
ENABLE_USER_SCRIPT_SANDBOXING = NO;
"FRAMEWORK_SEARCH_PATHS[sdk=iphoneos*]" = (
@@ -460,7 +460,7 @@
"$(inherited)",
"@executable_path/Frameworks",
);
MARKETING_VERSION = 1.2.1;
MARKETING_VERSION = 1.3.1;
OTHER_LDFLAGS = (
"$(inherited)",
"-ObjC",
@@ -502,7 +502,7 @@
ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
CODE_SIGN_ENTITLEMENTS = suixinkan/suixinkan.entitlements;
CODE_SIGN_STYLE = Automatic;
CURRENT_PROJECT_VERSION = 1020101;
CURRENT_PROJECT_VERSION = 1030101;
DEVELOPMENT_TEAM = 56GVN5RNVN;
ENABLE_USER_SCRIPT_SANDBOXING = NO;
"FRAMEWORK_SEARCH_PATHS[sdk=iphoneos*]" = (
@@ -526,7 +526,7 @@
"$(inherited)",
"@executable_path/Frameworks",
);
MARKETING_VERSION = 1.2.1;
MARKETING_VERSION = 1.3.1;
OTHER_LDFLAGS = (
"$(inherited)",
"-ObjC",
@@ -741,7 +741,7 @@
ASSETCATALOG_COMPILER_GLOBAL_ACCENT_COLOR_NAME = AccentColor;
CODE_SIGN_ENTITLEMENTS = suixinkan/suixinkan.entitlements;
CODE_SIGN_STYLE = Automatic;
CURRENT_PROJECT_VERSION = 1020101;
CURRENT_PROJECT_VERSION = 1030101;
DEVELOPMENT_TEAM = 56GVN5RNVN;
ENABLE_USER_SCRIPT_SANDBOXING = NO;
GENERATE_INFOPLIST_FILE = YES;
@@ -761,7 +761,7 @@
"$(inherited)",
"@executable_path/Frameworks",
);
MARKETING_VERSION = 1.2.1;
MARKETING_VERSION = 1.3.1;
OTHER_LDFLAGS = (
"$(inherited)",
"-ObjC",
@@ -4,6 +4,14 @@
"filename" : "ai_retouch_template_placeholder.png",
"idiom" : "universal",
"scale" : "1x"
},
{
"idiom" : "universal",
"scale" : "2x"
},
{
"idiom" : "universal",
"scale" : "3x"
}
],
"info" : {
@@ -13,7 +13,7 @@ AI 修图属于异步长耗时任务。当前用户提交后只能等待或主
1. 当前账号全部相册的 AI 修图任务列表。
2. 单次 AI 修图任务详情。
3. AI 修图终态消息推送。
4. 从推送、任务列表和提交成功提示进入对应任务的完整导航链路。
4. 从推送和任务列表进入对应任务的完整导航链路;提交成功后仅 Toast 提示并返回相册管理。
本期解决“看得到进度、完成会通知、结果可直达”的问题,不增加任务取消、批量重试、历史版本管理或后台供应商诊断能力。
@@ -65,7 +65,8 @@ AI 修图推送优先跳转到对应任务详情页。
相册管理选择照片
→ 提交 AI 修图
→ 后端返回 ai_retouch_batch_id
→ 客户端提示“AI修图任务已提交”并提供“查看任务”
→ 客户端 Toast 提示“AI修图任务已提交,完成后将通过消息通知”
→ 关闭模板页及可能存在的照片预览页,回到相册管理并刷新列表
→ 用户可离开页面
→ 任务进入终态后收到 type = 14 推送
→ 点击推送进入任务详情
@@ -75,7 +76,8 @@ AI 修图推送优先跳转到对应任务详情页。
### 3.2 页面入口
- 相册管理页导航栏右侧增加“修图任务”,进入当前账号的全局任务列表;新增相册页不展示该入口。
- AI 修图提交成功提示提供“查看任务”,直接进入本次任务详情。
- AI 修图提交成功仅展示自动消失的 Toast,不弹出查看任务确认框,也不提供立即跳转操作。
- 提交成功后回到相册管理页,并刷新相册摘要、数量和当前素材列表。
- 任务列表点击卡片进入对应任务详情。
- `type = 14` 推送携带有效任务 ID 时直达详情,否则进入任务列表。
@@ -86,10 +86,10 @@ GET /api/yf-handset-app/photog/travel-album/ai-retouch-options?user_equity_trave
| `scope` | 使用场景 |
|---|---|
| `batch` | 网格多选 AI 修图 |
| `all_variants` | 预览页无 Tab,或当前是原图 Tab |
| `refined_only` | 当前是精修后 Tab |
| `atmosphere_only` | 当前是氛围感 Tab |
| `batch` | 网格多选 AI 修图,或预览页尚无 AI 结果 Tab 时首次修图 |
| `all_variants` | 预览页已有 AI 结果 Tab;当前为原图、精修后或氛围感均使用此范围 |
| `refined_only` | 仅重修精修结果的接口能力,当前客户端无独立入口 |
| `atmosphere_only` | 仅重修氛围感结果的接口能力,当前客户端无独立入口 |
`user_equity_travel_id`、`scope` 和 `source_count` 均必填。`source_count` 用于计算封面模板是否显示和必选。
@@ -226,7 +226,7 @@ Content-Type: application/json
}
```
#### 5.4.2 预览页原图 Tab
#### 5.4.2 预览页已有 AI 结果 Tab
```json
{
@@ -234,10 +234,6 @@ Content-Type: application/json
"scope": "all_variants",
"source_material_ids": ["2031"],
"outputs": [
{
"type": "refined",
"template_id": "tpl_refined_15"
},
{
"type": "atmosphere",
"template_id": "tpl_atmosphere_09"
@@ -246,7 +242,7 @@ Content-Type: application/json
}
```
如用户未选氛围感模板,`outputs` 中不传 `atmosphere`。缺失表示“本次不处理”,不表示删除现有氛围感图。
已有精修或氛围感结果时,无论当前位于原图、精修后还是氛围感 Tab,精修和氛围感模板均为选填,但至少选择一种。`outputs` 只传本次选中的结果类型;缺失表示“本次不处理”,不表示删除或覆盖对应的已有结果。
#### 5.4.3 只重新精修
@@ -285,7 +281,7 @@ Content-Type: application/json
| `scope` | 原图数 | 精修 | 氛围感 | 封面 |
|---|---:|---|---|---|
| `batch` | 1 至 50 | 必选 | 可选 | 少于 4 张禁止;4 张及以上必选 |
| `all_variants` | 必须为 1 | 必选 | 可选 | 禁止 |
| `all_variants` | 必须为 1 | 可选(与氛围感至少一项) | 可选(与精修至少一项) | 禁止 |
| `refined_only` | 必须为 1 | 必选 | 禁止 | 禁止 |
| `atmosphere_only` | 必须为 1 | 禁止 | 必选 | 禁止 |
@@ -114,9 +114,9 @@ Cell 左上角展示稳定的修图状态:
| 当前情况 | 模板要求 | 生成与覆盖规则 |
|---|---|---|
| 不显示 Tab | 精修必选,氛围感可选 | 首次生成关联的精修图,可选生成氛围感图 |
| 已显示 Tab,当前为“原图” | 精修必选,氛围感可选 | 重新生成精修图;如选氛围感则也重新生成。新结果成功后原子替换对应旧结果 |
| 已显示 Tab,当前为“精修后” | 精修必选 | 只重新生成精修图,成功后覆盖旧精修图,不影响氛围感图 |
| 已显示 Tab,当前为“氛围感” | 氛围感必选 | 只重新生成氛围感图,成功后覆盖旧氛围感图,不影响精修图 |
| 已显示 Tab,当前为“原图” | 精修、氛围感均可选,至少选择一种 | 只重新生成本次选中的结果类型;新结果成功后替换对应旧结果,未选类型保持不变 |
| 已显示 Tab,当前为“精修后” | 精修、氛围感均可选,至少选择一种 | 只重新生成本次选中的结果类型;新结果成功后替换对应旧结果,未选类型保持不变 |
| 已显示 Tab,当前为“氛围感” | 精修、氛围感均可选,至少选择一种 | 只重新生成本次选中的结果类型;新结果成功后替换对应旧结果,未选类型保持不变 |
重新修图期间应保留上一版成功图片可见;只有新结果成功时才原子替换当前版本。失败时继续保留旧版本,并返回可展示的错误信息。
@@ -224,7 +224,7 @@ struct TravelAlbumMaterial: Decodable, Sendable, Equatable, Hashable, Identifiab
}
}
/// 相册素材网格角标类别,用于稳定映射文案优先级和语义颜色。
/// 相册素材网格角标类别,用于稳定映射独立业务状态和语义颜色。
enum TravelAlbumMaterialBadgeKind: Sendable, Equatable {
case purchased
case pending
@@ -241,14 +241,15 @@ struct TravelAlbumMaterialBadgePresentation: Sendable, Equatable {
}
extension TravelAlbumMaterial {
/// 按 AI 修图状态和购买状态生成网格角标;返回 nil 时隐藏角标。
var badgePresentation: TravelAlbumMaterialBadgePresentation? {
if aiRetouchStatus == 0 {
return isPurchased
/// 生成购买状态角标;购买状态与修图状态互不覆盖。
var purchaseBadgePresentation: TravelAlbumMaterialBadgePresentation? {
isPurchased
? TravelAlbumMaterialBadgePresentation(kind: .purchased, text: "已购")
: nil
}
/// 生成 AI 修图状态角标;未进入修图流程时返回 nil。
var aiRetouchBadgePresentation: TravelAlbumMaterialBadgePresentation? {
let statusName = aiRetouchStatusName.trimmingCharacters(in: .whitespacesAndNewlines)
switch aiRetouchStatus {
case 1:
@@ -386,11 +387,17 @@ enum TravelAlbumAIRetouchWorkflow: Sendable, Equatable {
}
}
/// 当前分组是否允许不选择;仅首次修图的氛围感模板选填。
/// 当前分组是否允许不选择;首次修图仅氛围感选填,已有 AI 结果的预览重修两类模板均选填。
func isOptional(_ category: TravelAlbumAIRetouchTemplateCategory) -> Bool {
if case .initial = self, category == .atmosphere { return true }
switch self {
case .initial:
return category == .atmosphere
case .reretouch(_, _, .all):
return category == .refined || category == .atmosphere
case .reretouch:
return false
}
}
/// 工作流目标是否满足接口的最小参数要求。
var isValid: Bool {
@@ -175,7 +175,7 @@ struct TravelAlbumPreviewProject: Identifiable, Sendable, Hashable {
self.assets = assets.filter { seenKinds.insert($0.kind).inserted }
}
/// 根据当前 Tab 生成首次修图或覆盖重修工作流。
/// 根据是否已有 AI 结果生成首次修图或双模板覆盖重修工作流。
func aiRetouchWorkflow(
albumId: Int,
selectedKind: TravelAlbumPreviewAssetKind
@@ -183,16 +183,8 @@ struct TravelAlbumPreviewProject: Identifiable, Sendable, Hashable {
guard hasVariants else {
return .initial(albumId: albumId, materialIds: [originalMaterialId])
}
switch selectedKind {
case .original:
guard selectedKind != .cover else { return nil }
return .reretouch(materialId: originalMaterialId, batchId: aiRetouchBatchId, type: .all)
case .retouched:
return .reretouch(materialId: originalMaterialId, batchId: aiRetouchBatchId, type: .refined)
case .atmosphere:
return .reretouch(materialId: originalMaterialId, batchId: aiRetouchBatchId, type: .atmosphere)
case .cover:
return nil
}
}
}
@@ -66,9 +66,12 @@ final class TravelAlbumAIRetouchTemplateViewModel {
return false
}
/// 首次修图页固定展示的氛围感与封面生成规则说明。
/// 首次修图页按照当前选图数量展示氛围感与封面生成规则。
var initialTipsText: String {
"Tips:氛围感修图为选填,可横向选择一种样式;选中后每张照片会额外生成1个独立结果,第一张照片仍另生成封面。"
if visibleCategories.contains(.cover) {
return "Tips:氛围感修图为选填;封面风格为必选,将使用第一张照片另生成封面。"
}
return "Tips:氛围感修图为选填,选中后每张照片会额外生成1个独立结果。"
}
/// 当前分组是否为选填。
@@ -87,7 +90,8 @@ final class TravelAlbumAIRetouchTemplateViewModel {
case .refined, .atmosphere:
return 1
case .all:
return 1 + (selectedAtmosphereTemplateId == nil ? 0 : 1)
return (selectedRefinedTemplateId == nil ? 0 : 1)
+ (selectedAtmosphereTemplateId == nil ? 0 : 1)
}
}
}
@@ -106,6 +110,11 @@ final class TravelAlbumAIRetouchTemplateViewModel {
return unavailableMessage(for: category)
}
}
if case .reretouch(_, _, .all) = workflow,
selectedRefinedTemplateId == nil,
selectedAtmosphereTemplateId == nil {
return "请至少选择一个修图模板"
}
guard let remainingQuota else {
return "剩余修图次数获取失败,请刷新后重试"
}
@@ -143,7 +152,9 @@ final class TravelAlbumAIRetouchTemplateViewModel {
atmosphereTemplates = response.atmosphereTemplates
coverTemplates = response.coverTemplates
remainingQuota = response.remainingQuota
selectedRefinedTemplateId = visibleCategories.contains(.refined) ? refinedTemplates.first?.id : nil
selectedRefinedTemplateId = visibleCategories.contains(.refined) && !isOptional(.refined)
? refinedTemplates.first?.id
: nil
selectedAtmosphereTemplateId = visibleCategories.contains(.atmosphere) && !isOptional(.atmosphere)
? atmosphereTemplates.first?.id
: nil
@@ -185,14 +196,14 @@ final class TravelAlbumAIRetouchTemplateViewModel {
}
}
/// 选择模板;仅选填分组允许再次点击取消,必选分组保持单选。
/// 选择模板;选填分组允许再次点击取消,必选分组保持单选。
func toggleTemplate(id: Int, category: TravelAlbumAIRetouchTemplateCategory) {
guard visibleCategories.contains(category),
templates(for: category).contains(where: { $0.id == id })
else { return }
switch category {
case .refined:
selectedRefinedTemplateId = id
selectedRefinedTemplateId = isOptional(category) && selectedRefinedTemplateId == id ? nil : id
case .atmosphere:
selectedAtmosphereTemplateId = isOptional(category) && selectedAtmosphereTemplateId == id ? nil : id
case .cover:
@@ -48,6 +48,7 @@ final class TravelAlbumDetailViewModel {
private(set) var isRefreshing = false
private(set) var isSelectionMode = false
private(set) var selectedMaterialIds: Set<Int> = []
private var selectedMaterialPurchaseStates: [Int: Bool] = [:]
var onStateChange: (() -> Void)?
var onShowMessage: ((String) -> Void)?
@@ -142,12 +143,29 @@ final class TravelAlbumDetailViewModel {
selectedTab == .all ? allPhotoCount : purchasedPhotoCount
}
/// 当前选中素材中的已购买数量。
var selectedPurchasedMaterialCount: Int {
selectedMaterialIds.count - selectedDeletableMaterialIds.count
}
/// 删除确认弹窗文案;选中已购素材时明确说明保护规则与实际删除数量。
var deleteSelectedConfirmationMessage: String {
let selectedCount = selectedMaterialIds.count
let purchasedCount = selectedPurchasedMaterialCount
guard purchasedCount > 0 else {
return "确定删除选中的 \(selectedCount) 张素材吗?"
}
let deletableCount = selectedCount - purchasedCount
return "已选择 \(selectedCount) 张素材,其中 \(purchasedCount) 张已购买。"
+ "不会删除已购买的项目,只会删除 \(deletableCount) 张未购买的项目。"
}
/// 切换素材 tab。
func selectTab(_ tab: Tab, api: any TravelAlbumServing) async {
guard selectedTab != tab else { return }
selectedTab = tab
isSelectionMode = false
selectedMaterialIds = []
clearMaterialSelection()
notifyStateChange()
await loadMaterials(reset: true, api: api)
}
@@ -223,10 +241,9 @@ final class TravelAlbumDetailViewModel {
/// 切换选择模式。
func toggleSelectionMode() {
guard selectedTab == .all else { return }
isSelectionMode.toggle()
if !isSelectionMode {
selectedMaterialIds = []
clearMaterialSelection()
}
notifyStateChange()
}
@@ -234,14 +251,12 @@ final class TravelAlbumDetailViewModel {
/// 切换素材选中状态。
func toggleMaterialSelection(_ material: TravelAlbumMaterial) {
guard isSelectionMode else { return }
guard material.status == 1 else {
onShowMessage?("仅未购买素材可删除")
return
}
if selectedMaterialIds.contains(material.id) {
selectedMaterialIds.remove(material.id)
selectedMaterialPurchaseStates.removeValue(forKey: material.id)
} else {
selectedMaterialIds.insert(material.id)
selectedMaterialPurchaseStates[material.id] = material.isPurchased
}
notifyStateChange()
}
@@ -249,15 +264,22 @@ final class TravelAlbumDetailViewModel {
/// AI 修图任务提交成功后退出选择模式并清空当前选择。
func completeAIRetouchSubmission() {
isSelectionMode = false
selectedMaterialIds = []
clearMaterialSelection()
notifyStateChange()
}
/// AI 修图提交成功后重置选择状态,并刷新相册摘要、数量与当前素材列表。
func refreshAfterAIRetouchSubmission(api: any TravelAlbumServing) async {
completeAIRetouchSubmission()
await refreshAll(api: api)
}
/// 预览页删除成功后移除对应素材,并同步全部/已购计数。
func removeMaterialAfterPreviewDeletion(id: Int) {
guard let index = materials.firstIndex(where: { $0.id == id }) else { return }
let material = materials.remove(at: index)
selectedMaterialIds.remove(id)
selectedMaterialPurchaseStates.removeValue(forKey: id)
allPhotoCount = max(0, allPhotoCount - 1)
if material.isPurchased {
purchasedPhotoCount = max(0, purchasedPhotoCount - 1)
@@ -267,16 +289,20 @@ final class TravelAlbumDetailViewModel {
/// 删除已选素材。
func deleteSelectedMaterials(api: any TravelAlbumServing) async {
let ids = selectedMaterialIds.sorted()
guard !ids.isEmpty else {
guard !selectedMaterialIds.isEmpty else {
onShowMessage?("请选择要删除的素材")
return
}
let ids = selectedDeletableMaterialIds
guard !ids.isEmpty else {
onShowMessage?("选中的素材均已购买,未删除任何项目")
return
}
do {
try await api.batchDeleteMaterials(ids: ids)
onShowMessage?("删除成功")
isSelectionMode = false
selectedMaterialIds = []
clearMaterialSelection()
await refreshAll(api: api)
} catch is CancellationError {
return
@@ -301,4 +327,18 @@ final class TravelAlbumDetailViewModel {
private func notifyStateChange() {
onStateChange?()
}
private func clearMaterialSelection() {
selectedMaterialIds = []
selectedMaterialPurchaseStates = [:]
}
private var selectedDeletableMaterialIds: [Int] {
selectedMaterialIds
.filter { id in
guard selectedMaterialPurchaseStates[id] == false else { return false }
return materials.first(where: { $0.id == id })?.isPurchased != true
}
.sorted()
}
}
@@ -52,7 +52,10 @@ final class TravelAlbumEntryViewModel {
}
/// 重新拉取相册列表。
func loadAlbums(api: any TravelAlbumServing) async {
func loadAlbums(
api: any TravelAlbumServing,
preservingContentOnFailure: Bool = false
) async {
isLoading = true
notifyStateChange()
defer {
@@ -67,12 +70,20 @@ final class TravelAlbumEntryViewModel {
} catch is CancellationError {
return
} catch {
if !preservingContentOnFailure {
albums = []
albumTotal = 0
}
onShowMessage?(error.localizedDescription)
}
}
/// 用户下拉刷新相册列表;失败时保留当前内容,避免页面瞬间清空。
func refreshAlbums(api: any TravelAlbumServing) async {
guard !isLoading else { return }
await loadAlbums(api: api, preservingContentOnFailure: true)
}
/// 打开创建相册弹窗。
func openCreateSheet(api: any TravelAlbumServing) async {
guard !isCreating else { return }
@@ -533,7 +533,7 @@ private final class AIJobDetailAlbumCard: UIView {
}
}
/// 任务内容卡,以浅蓝标签展示各输出目标数量并补充额度结算。
/// 任务内容卡,以三色紧凑标签展示各输出目标数量并补充额度结算。
private final class AIJobDetailContentCard: UIView {
private let outputStack = UIStackView()
private let quotaLabel = UILabel()
@@ -548,7 +548,8 @@ private final class AIJobDetailContentCard: UIView {
titleLabel.textColor = AIJobDetailStyle.textPrimary
outputStack.axis = .horizontal
outputStack.spacing = 10
outputStack.distribution = .fillEqually
outputStack.distribution = .fill
outputStack.alignment = .center
quotaLabel.font = .systemFont(ofSize: 12)
quotaLabel.textColor = AIJobDetailStyle.textSecondary
quotaLabel.numberOfLines = 0
@@ -565,6 +566,12 @@ private final class AIJobDetailContentCard: UIView {
func apply(_ detail: TravelAlbumAIJobDetail) {
outputStack.arrangedSubviews.forEach { $0.removeFromSuperview() }
detail.outputs.forEach { outputStack.addArrangedSubview(makeChip($0)) }
if let lastChip = outputStack.arrangedSubviews.last {
let spacer = UIView()
spacer.setContentHuggingPriority(.defaultLow, for: .horizontal)
outputStack.setCustomSpacing(0, after: lastChip)
outputStack.addArrangedSubview(spacer)
}
outputStack.isHidden = detail.outputs.isEmpty
let quota = detail.quotaSettlement
quotaLabel.text = "额度:预占 \(quota.reservedUnits) · 消耗 \(quota.consumedUnits) · 释放 \(quota.releasedUnits)"
@@ -573,32 +580,42 @@ private final class AIJobDetailContentCard: UIView {
}
private func makeChip(_ output: TravelAlbumAIJobOutput) -> UIView {
let container = UIView()
container.backgroundColor = AppColor.primaryLight
container.layer.cornerRadius = 9
let iconView = UIImageView(image: UIImage(systemName: output.type.symbolName))
iconView.tintColor = AppColor.primary
iconView.contentMode = .scaleAspectFit
let label = UILabel()
label.text = "\(output.type.shortTitle) \(output.count) 张"
AIJobDetailOutputChip(
text: "\(output.type.shortTitle) \(output.count)张",
backgroundColor: output.type.chipBackgroundColor,
accessibilityIdentifier: "aiRetouchJob.contentChip.\(output.type.chipIdentifier)"
)
}
}
/// 依据文字固有宽度展示的任务输出类型标签。
private final class AIJobDetailOutputChip: UIView {
private let label = UILabel()
init(text: String, backgroundColor: UIColor, accessibilityIdentifier: String) {
super.init(frame: .zero)
self.backgroundColor = backgroundColor
self.accessibilityIdentifier = accessibilityIdentifier
layer.cornerRadius = 9
setContentHuggingPriority(.required, for: .horizontal)
setContentCompressionResistancePriority(.required, for: .horizontal)
label.text = text
label.font = .systemFont(ofSize: 14, weight: .medium)
label.textColor = AIJobDetailStyle.textPrimary
label.adjustsFontSizeToFitWidth = true
label.minimumScaleFactor = 0.75
container.addSubview(iconView)
container.addSubview(label)
iconView.snp.makeConstraints { make in
make.leading.equalToSuperview().offset(10)
make.centerY.equalToSuperview()
make.size.equalTo(18)
}
label.setContentCompressionResistancePriority(.required, for: .horizontal)
addSubview(label)
label.snp.makeConstraints { make in
make.leading.equalTo(iconView.snp.trailing).offset(6)
make.trailing.equalToSuperview().inset(8)
make.leading.trailing.equalToSuperview().inset(12)
make.centerY.equalToSuperview()
}
container.snp.makeConstraints { $0.height.equalTo(44) }
return container
snp.makeConstraints { $0.height.equalTo(44) }
}
@available(*, unavailable)
required init?(coder: NSCoder) { fatalError("init(coder:) has not been implemented") }
override var intrinsicContentSize: CGSize {
CGSize(width: ceil(label.intrinsicContentSize.width) + 24, height: 44)
}
}
@@ -663,6 +680,9 @@ private final class AIJobDetailTargetRow: UIView {
private let templateLabel = UILabel()
private let statusIconView = UIImageView()
private let statusLabel = UILabel()
private let statusStack = UIStackView()
private let statusRow = UIStackView()
private let contentStack = UIStackView()
private let resultButton = UIButton(type: .system)
private let errorContainer = UIView()
private let errorIconView = UIImageView(image: UIImage(systemName: "exclamationmark.circle.fill"))
@@ -679,8 +699,11 @@ private final class AIJobDetailTargetRow: UIView {
titleLabel.font = .systemFont(ofSize: 15, weight: .medium)
titleLabel.textColor = AIJobDetailStyle.textPrimary
titleLabel.lineBreakMode = .byTruncatingMiddle
titleLabel.numberOfLines = 1
templateLabel.font = .systemFont(ofSize: 13)
templateLabel.textColor = AIJobDetailStyle.textSecondary
templateLabel.lineBreakMode = .byTruncatingTail
templateLabel.numberOfLines = 1
statusIconView.contentMode = .scaleAspectFit
statusLabel.font = .systemFont(ofSize: 14, weight: .medium)
resultButton.setTitle("查看结果", for: .normal)
@@ -693,49 +716,45 @@ private final class AIJobDetailTargetRow: UIView {
errorLabel.textColor = AppColor.danger
errorLabel.numberOfLines = 0
let statusStack = UIStackView(arrangedSubviews: [statusIconView, statusLabel])
statusStack.addArrangedSubview(statusIconView)
statusStack.addArrangedSubview(statusLabel)
statusStack.axis = .horizontal
statusStack.spacing = 6
statusStack.alignment = .center
statusStack.setContentCompressionResistancePriority(.required, for: .horizontal)
statusLabel.setContentCompressionResistancePriority(.required, for: .horizontal)
let statusSpacer = UIView()
statusRow.addArrangedSubview(statusStack)
statusRow.addArrangedSubview(statusSpacer)
statusRow.addArrangedSubview(resultButton)
statusRow.axis = .horizontal
statusRow.spacing = 8
statusRow.alignment = .center
contentStack.addArrangedSubview(titleLabel)
contentStack.addArrangedSubview(templateLabel)
contentStack.addArrangedSubview(statusRow)
contentStack.addArrangedSubview(errorContainer)
contentStack.axis = .vertical
contentStack.spacing = 0
contentStack.setCustomSpacing(7, after: titleLabel)
contentStack.setCustomSpacing(8, after: templateLabel)
contentStack.setCustomSpacing(12, after: statusRow)
statusIconView.snp.makeConstraints { $0.size.equalTo(18) }
addSubview(thumbnailView)
addSubview(titleLabel)
addSubview(templateLabel)
addSubview(statusStack)
addSubview(resultButton)
addSubview(errorContainer)
addSubview(contentStack)
errorContainer.addSubview(errorIconView)
errorContainer.addSubview(errorLabel)
thumbnailView.snp.makeConstraints { make in
make.leading.top.equalToSuperview().offset(4)
make.size.equalTo(80)
make.bottom.lessThanOrEqualToSuperview().inset(12)
}
titleLabel.snp.makeConstraints { make in
contentStack.snp.makeConstraints { make in
make.leading.equalTo(thumbnailView.snp.trailing).offset(14)
make.top.equalTo(thumbnailView).offset(18)
make.trailing.lessThanOrEqualTo(statusStack.snp.leading).offset(-8)
}
templateLabel.snp.makeConstraints { make in
make.leading.equalTo(titleLabel)
make.top.equalTo(titleLabel.snp.bottom).offset(7)
make.trailing.lessThanOrEqualTo(statusStack.snp.leading).offset(-8)
}
statusStack.snp.makeConstraints { make in
make.trailing.equalToSuperview().inset(4)
make.centerY.equalTo(thumbnailView).offset(-7)
}
resultButton.snp.makeConstraints { make in
make.trailing.equalToSuperview().inset(4)
make.top.equalTo(statusStack.snp.bottom).offset(4)
make.height.equalTo(28)
}
errorContainer.snp.makeConstraints { make in
make.top.equalTo(thumbnailView.snp.bottom).offset(10)
make.leading.equalTo(titleLabel)
make.trailing.equalToSuperview().inset(4)
make.bottom.equalToSuperview().inset(12)
make.height.greaterThanOrEqualTo(38)
}
errorIconView.snp.makeConstraints { make in
make.leading.equalToSuperview().offset(10)
@@ -780,17 +799,9 @@ private final class AIJobDetailTargetRow: UIView {
errorLabel.text = target.displayFailureMessage.map { "失败原因:\($0)" }
let showsError = target.displayFailureMessage != nil
errorContainer.isHidden = !showsError
errorContainer.snp.remakeConstraints { make in
make.top.equalTo(thumbnailView.snp.bottom).offset(showsError ? 10 : 0)
make.leading.equalTo(titleLabel)
make.trailing.equalToSuperview().inset(4)
make.bottom.equalToSuperview().inset(12)
if showsError {
make.height.greaterThanOrEqualTo(38)
} else {
make.height.equalTo(0)
}
}
titleLabel.accessibilityIdentifier = "aiRetouchJob.target.\(target.targetId).title"
templateLabel.accessibilityIdentifier = "aiRetouchJob.target.\(target.targetId).template"
statusStack.accessibilityIdentifier = "aiRetouchJob.target.\(target.targetId).status"
accessibilityLabel = [titleLabel.text, templateLabel.text, statusLabel.text, errorLabel.text]
.compactMap { $0 }.joined(separator: ",")
}
@@ -892,12 +903,21 @@ private extension TravelAlbumAIJobOutputType {
}
}
var symbolName: String {
var chipBackgroundColor: UIColor {
switch self {
case .refined: "wand.and.stars"
case .atmosphere: "sun.max"
case .cover: "bookmark"
case .unknown: "sparkles"
case .refined: UIColor(hex: 0xEAF3FF)
case .atmosphere: UIColor(hex: 0xFFF2DC)
case .cover: UIColor(hex: 0xF2ECFF)
case .unknown: UIColor(hex: 0xF1F5F9)
}
}
var chipIdentifier: String {
switch self {
case .refined: "refined"
case .atmosphere: "atmosphere"
case .cover: "cover"
case .unknown: "unknown"
}
}
}
@@ -284,8 +284,8 @@ final class TravelAlbumAIRetouchTemplateViewController: BaseViewController {
header.apply(
title: category.title,
badge: category == .cover
? .gift
: (self.viewModel.isOptional(category) ? .optional : nil)
? .requiredGift
: (self.viewModel.isOptional(category) ? .optional : .required)
)
case .mode:
return nil
@@ -845,11 +845,12 @@ final class TravelAlbumAIRetouchModeCell: UICollectionViewCell {
/// 模板分组标题右侧的业务标记。
fileprivate enum AIRetouchTemplateSectionBadge {
case required
case optional
case gift
case requiredGift
}
/// AI 修图模板分组标题,可附带选填或赠送标记。
/// AI 修图模板分组标题,展示必选、选填或封面赠送规则。
final class TravelAlbumAIRetouchSectionHeader: UICollectionReusableView {
static let reuseIdentifier = "TravelAlbumAIRetouchSectionHeader"
@@ -896,12 +897,16 @@ final class TravelAlbumAIRetouchSectionHeader: UICollectionReusableView {
titleLabel.text = title
badgeContainer.isHidden = badge == nil
switch badge {
case .required:
badgeLabel.text = "必选"
badgeLabel.textColor = AIRetouchTemplateStyle.danger
badgeContainer.backgroundColor = AIRetouchTemplateStyle.danger.withAlphaComponent(0.1)
case .optional:
badgeLabel.text = "选填"
badgeLabel.textColor = AIRetouchTemplateStyle.primary
badgeContainer.backgroundColor = AIRetouchTemplateStyle.primary.withAlphaComponent(0.1)
case .gift:
badgeLabel.text = "赠送 · 不占额度"
case .requiredGift:
badgeLabel.text = "必选 · 赠送 · 不占额度"
badgeLabel.textColor = AIRetouchTemplateStyle.gift
badgeContainer.backgroundColor = AIRetouchTemplateStyle.giftBackground
case nil:
@@ -31,6 +31,7 @@ final class TravelAlbumDetailViewController: BaseViewController {
private let aiRetouchButton = UIButton(type: .system)
private let deleteSelectedButton = UIButton(type: .system)
private let uploadButton = UIButton(type: .system)
private var refreshState = TravelAlbumReturnRefreshState()
init(
albumId: Int,
@@ -262,6 +263,18 @@ final class TravelAlbumDetailViewController: BaseViewController {
Task { await viewModel.refreshAll(api: api) }
}
override func viewDidAppear(_ animated: Bool) {
super.viewDidAppear(animated)
guard refreshState.beginRefreshIfNeeded() else { return }
Task {
await viewModel.refreshAll(api: api)
await MainActor.run {
self.applyViewModel()
self.refreshState.finishRefresh()
}
}
}
@MainActor
private func applyViewModel() {
if let album = viewModel.album {
@@ -280,14 +293,11 @@ final class TravelAlbumDetailViewController: BaseViewController {
sortButton.accessibilityValue = viewModel.sortOption.title
sortButton.menu = makeSortMenu()
let canSelectMaterials = viewModel.selectedTab == .all
selectButton.isHidden = false
selectButton.isEnabled = canSelectMaterials
selectButton.alpha = canSelectMaterials ? 1 : 0.45
selectButton.isEnabled = true
selectButton.alpha = 1
selectButton.setTitle(viewModel.isSelectionMode ? "完成" : "选择", for: .normal)
selectButton.accessibilityValue = canSelectMaterials
? (viewModel.isSelectionMode ? "选择模式已开启" : "选择模式已关闭")
: "已购照片不可删除"
selectButton.accessibilityValue = viewModel.isSelectionMode ? "选择模式已开启" : "选择模式已关闭"
let selectedCount = viewModel.selectedMaterialIds.count
let hasSelection = selectedCount > 0
@@ -306,7 +316,10 @@ final class TravelAlbumDetailViewController: BaseViewController {
if !snapshot.itemIdentifiers.isEmpty {
snapshot.reconfigureItems(snapshot.itemIdentifiers)
}
dataSource.apply(snapshot, animatingDifferences: true)
dataSource.apply(
snapshot,
animatingDifferences: !refreshState.suppressesSnapshotAnimations
)
if !viewModel.isRefreshing {
refreshControl.endRefreshing()
@@ -458,35 +471,24 @@ final class TravelAlbumDetailViewController: BaseViewController {
materialIds: materialIds
),
api: api,
onSubmitted: { [weak self] submission in
guard let self else { return }
self.viewModel.completeAIRetouchSubmission()
self.presentAIRetouchSubmitted(submission)
}
onSubmitted: { [weak self] _ in self?.handleAIRetouchSubmitted() }
)
present(controller, animated: true)
}
private func presentAIRetouchSubmitted(_ submission: TravelAlbumAIJobSubmission) {
let alert = UIAlertController(
title: "AI修图任务已提交",
message: "完成后将通过消息通知,你也可以随时查看处理进度。",
preferredStyle: .alert
)
alert.addAction(UIAlertAction(title: "稍后查看", style: .cancel))
alert.addAction(UIAlertAction(title: "查看任务", style: .default) { [weak self] _ in
self?.navigationController?.pushViewController(
TravelAlbumAIJobDetailViewController(batchId: submission.aiRetouchBatchId),
animated: true
)
})
present(alert, animated: true)
private func handleAIRetouchSubmitted() {
showToast("AI修图任务已提交,完成后将通过消息通知")
Task { await viewModel.refreshAfterAIRetouchSubmission(api: api) }
}
@objc private func deleteSelectedTapped() {
let count = viewModel.selectedMaterialIds.count
guard count > 0 else { return }
let alert = UIAlertController(title: "删除素材", message: "确定删除选中的 \(count) 张素材吗?", preferredStyle: .alert)
let alert = UIAlertController(
title: "删除素材",
message: viewModel.deleteSelectedConfirmationMessage,
preferredStyle: .alert
)
alert.addAction(UIAlertAction(title: "取消", style: .cancel))
alert.addAction(UIAlertAction(title: "删除", style: .destructive) { [weak self] _ in
guard let self else { return }
@@ -497,6 +499,7 @@ final class TravelAlbumDetailViewController: BaseViewController {
@objc private func uploadTapped() {
let album = viewModel.album
refreshState.markRefreshNeeded()
let controller = WiredCameraTransferViewController(
viewModel: WiredCameraTransferViewModel(
albumId: album?.id ?? viewModel.albumId,
@@ -533,7 +536,8 @@ final class TravelAlbumDetailViewController: BaseViewController {
},
onProjectDeleted: { materialId in
previewViewModel.removeMaterialAfterPreviewDeletion(id: materialId)
}
},
onAIRetouchSubmitted: { [weak self] in self?.handleAIRetouchSubmitted() }
)
present(controller, animated: true)
}
@@ -691,14 +695,16 @@ private final class TravelAlbumInfoCard: UIView {
}
}
/// 旅拍相册素材网格单元,展示正方形缩略图、文件名、大小和选择状态。
/// 旅拍相册素材网格单元,独立展示购买、修图和选择状态。
final class TravelAlbumMaterialCell: UICollectionViewCell {
static let reuseIdentifier = "TravelAlbumMaterialCell"
private let imageView = UIImageView()
private let checkImageView = UIImageView()
private let badgeView = UIView()
private let badgeLabel = UILabel()
private let purchaseBadgeView = UIView()
private let purchaseBadgeLabel = UILabel()
private let aiRetouchBadgeView = UIView()
private let aiRetouchBadgeLabel = UILabel()
private let nameLabel = UILabel()
private let sizeLabel = UILabel()
@@ -712,18 +718,18 @@ final class TravelAlbumMaterialCell: UICollectionViewCell {
checkImageView.backgroundColor = UIColor.black.withAlphaComponent(0.35)
checkImageView.layer.cornerRadius = 11
checkImageView.accessibilityIdentifier = "travelAlbum.materialSelectionCheck"
badgeView.layer.cornerRadius = 5
badgeView.clipsToBounds = true
badgeView.isHidden = true
badgeView.isAccessibilityElement = false
badgeView.accessibilityIdentifier = "travelAlbum.materialStatusBadge"
badgeLabel.font = .systemFont(ofSize: 10, weight: .semibold)
badgeLabel.textColor = .white
badgeLabel.textAlignment = .center
badgeLabel.isAccessibilityElement = false
badgeLabel.accessibilityIdentifier = "travelAlbum.materialStatusBadgeLabel"
badgeLabel.setContentHuggingPriority(.required, for: .horizontal)
badgeLabel.setContentCompressionResistancePriority(.required, for: .horizontal)
configureBadge(
purchaseBadgeView,
label: purchaseBadgeLabel,
viewIdentifier: "travelAlbum.materialPurchaseBadge",
labelIdentifier: "travelAlbum.materialPurchaseBadgeLabel"
)
configureBadge(
aiRetouchBadgeView,
label: aiRetouchBadgeLabel,
viewIdentifier: "travelAlbum.materialAIRetouchBadge",
labelIdentifier: "travelAlbum.materialAIRetouchBadgeLabel"
)
nameLabel.font = .systemFont(ofSize: 12, weight: .medium)
nameLabel.textColor = TravelAlbumDetailStyle.textPrimary
nameLabel.lineBreakMode = .byTruncatingMiddle
@@ -731,8 +737,8 @@ final class TravelAlbumMaterialCell: UICollectionViewCell {
sizeLabel.textColor = TravelAlbumDetailStyle.textSecondary
contentView.addSubview(imageView)
imageView.addSubview(badgeView)
badgeView.addSubview(badgeLabel)
imageView.addSubview(purchaseBadgeView)
imageView.addSubview(aiRetouchBadgeView)
imageView.addSubview(checkImageView)
contentView.addSubview(nameLabel)
contentView.addSubview(sizeLabel)
@@ -744,14 +750,13 @@ final class TravelAlbumMaterialCell: UICollectionViewCell {
make.top.trailing.equalToSuperview().inset(6)
make.size.equalTo(22)
}
badgeView.snp.makeConstraints { make in
aiRetouchBadgeView.snp.makeConstraints { make in
make.top.leading.equalToSuperview().inset(6)
make.trailing.lessThanOrEqualTo(checkImageView.snp.leading).offset(-4)
}
badgeLabel.snp.makeConstraints { make in
make.edges.equalToSuperview().inset(
UIEdgeInsets(top: 3, left: 6, bottom: 3, right: 6)
)
purchaseBadgeView.snp.makeConstraints { make in
make.leading.bottom.equalToSuperview().inset(6)
make.trailing.lessThanOrEqualToSuperview().inset(6)
}
nameLabel.snp.makeConstraints { make in
make.top.equalTo(imageView.snp.bottom).offset(6)
@@ -781,14 +786,54 @@ final class TravelAlbumMaterialCell: UICollectionViewCell {
checkImageView.isHidden = !selectionMode
checkImageView.image = UIImage(systemName: selected ? "checkmark.circle.fill" : "circle")
checkImageView.tintColor = selected ? TravelAlbumDetailStyle.primary : .white
let badge = material.badgePresentation
badgeView.isHidden = badge == nil
badgeLabel.text = badge?.text
badgeView.backgroundColor = badge.map { TravelAlbumDetailStyle.badgeColor(for: $0.kind) }
let purchaseBadge = material.purchaseBadgePresentation
let aiRetouchBadge = material.aiRetouchBadgePresentation
applyBadge(purchaseBadge, to: purchaseBadgeView, label: purchaseBadgeLabel)
applyBadge(aiRetouchBadge, to: aiRetouchBadgeView, label: aiRetouchBadgeLabel)
nameLabel.text = material.fileName.isEmpty ? "未命名照片" : material.fileName
sizeLabel.text = TravelAlbumDisplayFormatter.fileSizeText(material.fileSize)
let badgeAccessibilityText = badge.map { ",状态:\($0.text)" } ?? ""
accessibilityLabel = "\(nameLabel.text ?? "照片"),\(sizeLabel.text ?? "")\(badgeAccessibilityText)"
let purchaseAccessibilityText = purchaseBadge.map { ",购买状态:\($0.text)" } ?? ""
let aiRetouchAccessibilityText = aiRetouchBadge.map { ",修图状态:\($0.text)" } ?? ""
accessibilityLabel = "\(nameLabel.text ?? "照片"),\(sizeLabel.text ?? "")"
+ purchaseAccessibilityText
+ aiRetouchAccessibilityText
accessibilityValue = selectionMode ? (selected ? "已选择" : "未选择") : nil
}
private func configureBadge(
_ badgeView: UIView,
label: UILabel,
viewIdentifier: String,
labelIdentifier: String
) {
badgeView.layer.cornerRadius = 5
badgeView.clipsToBounds = true
badgeView.isHidden = true
badgeView.isAccessibilityElement = false
badgeView.accessibilityIdentifier = viewIdentifier
label.font = .systemFont(ofSize: 10, weight: .semibold)
label.textColor = .white
label.textAlignment = .center
label.lineBreakMode = .byTruncatingTail
label.isAccessibilityElement = false
label.accessibilityIdentifier = labelIdentifier
label.setContentHuggingPriority(.required, for: .horizontal)
label.setContentCompressionResistancePriority(.defaultHigh, for: .horizontal)
badgeView.addSubview(label)
label.snp.makeConstraints { make in
make.edges.equalToSuperview().inset(
UIEdgeInsets(top: 3, left: 6, bottom: 3, right: 6)
)
}
}
private func applyBadge(
_ presentation: TravelAlbumMaterialBadgePresentation?,
to badgeView: UIView,
label: UILabel
) {
badgeView.isHidden = presentation == nil
label.text = presentation?.text
badgeView.backgroundColor = presentation.map { TravelAlbumDetailStyle.badgeColor(for: $0.kind) }
}
}
@@ -7,6 +7,27 @@ import Kingfisher
import SnapKit
import UIKit
/// 相册页面的一次性返回刷新状态,避免初次展示或重复出现时多发列表请求。
struct TravelAlbumReturnRefreshState {
private var needsRefresh = false
private(set) var suppressesSnapshotAnimations = false
mutating func markRefreshNeeded() {
needsRefresh = true
}
mutating func beginRefreshIfNeeded() -> Bool {
guard needsRefresh else { return false }
needsRefresh = false
suppressesSnapshotAnimations = true
return true
}
mutating func finishRefresh() {
suppressesSnapshotAnimations = false
}
}
/// 新增相册入口页,对齐 Android `TravelAlbumEntryScreen`。
final class TravelAlbumEntryViewController: BaseViewController {
private let viewModel = TravelAlbumEntryViewModel()
@@ -15,8 +36,10 @@ final class TravelAlbumEntryViewController: BaseViewController {
private let heroCard = TravelAlbumHeroCard()
private let titleLabel = UILabel()
private let tableView = UITableView(frame: .zero, style: .plain)
private let refreshControl = UIRefreshControl()
private let emptyView = TravelAlbumEmptyView()
private var dataSource: UITableViewDiffableDataSource<Int, TravelAlbum>!
private var refreshState = TravelAlbumReturnRefreshState()
init(api: (any TravelAlbumServing)? = nil) {
self.api = api ?? NetworkServices.shared.travelAlbumAPI
@@ -43,6 +66,8 @@ final class TravelAlbumEntryViewController: BaseViewController {
tableView.separatorStyle = .none
tableView.rowHeight = UITableView.automaticDimension
tableView.estimatedRowHeight = 148
tableView.alwaysBounceVertical = true
tableView.refreshControl = refreshControl
tableView.delegate = self
tableView.register(TravelAlbumTaskCell.self, forCellReuseIdentifier: TravelAlbumTaskCell.reuseIdentifier)
@@ -62,6 +87,7 @@ final class TravelAlbumEntryViewController: BaseViewController {
view.addSubview(titleLabel)
view.addSubview(tableView)
view.addSubview(emptyView)
emptyView.isUserInteractionEnabled = false
}
override func setupConstraints() {
@@ -86,6 +112,7 @@ final class TravelAlbumEntryViewController: BaseViewController {
override func bindActions() {
heroCard.addTarget(self, action: #selector(createTapped), for: .touchUpInside)
refreshControl.addTarget(self, action: #selector(refreshPulled), for: .valueChanged)
viewModel.onStateChange = { [weak self] in
Task { @MainActor in self?.applyViewModel() }
}
@@ -99,24 +126,49 @@ final class TravelAlbumEntryViewController: BaseViewController {
Task { await viewModel.loadAlbums(api: api) }
}
override func viewDidAppear(_ animated: Bool) {
super.viewDidAppear(animated)
guard refreshState.beginRefreshIfNeeded() else { return }
Task {
await viewModel.loadAlbums(api: api)
await MainActor.run {
self.applyViewModel()
self.refreshState.finishRefresh()
}
}
}
@MainActor
private func applyViewModel() {
heroCard.isLoading = viewModel.isCreating
titleLabel.text = "我的任务(\(viewModel.albumTotal))"
titleLabel.isHidden = viewModel.albums.isEmpty
tableView.isHidden = viewModel.albums.isEmpty
emptyView.isHidden = !viewModel.albums.isEmpty || viewModel.isLoading
emptyView.isHidden = !viewModel.albums.isEmpty
|| (viewModel.isLoading && !refreshControl.isRefreshing)
var snapshot = NSDiffableDataSourceSnapshot<Int, TravelAlbum>()
snapshot.appendSections([0])
snapshot.appendItems(viewModel.albums)
dataSource.apply(snapshot, animatingDifferences: true)
if viewModel.isLoading && viewModel.albums.isEmpty {
dataSource.apply(
snapshot,
animatingDifferences: !refreshState.suppressesSnapshotAnimations
)
if viewModel.isLoading && viewModel.albums.isEmpty && !refreshControl.isRefreshing {
showLoading()
} else {
hideLoading()
}
}
@objc private func refreshPulled() {
Task {
await viewModel.refreshAlbums(api: api)
await MainActor.run {
self.refreshControl.endRefreshing()
self.applyViewModel()
}
}
}
@objc private func createTapped() {
Task {
await viewModel.openCreateSheet(api: api)
@@ -162,6 +214,7 @@ final class TravelAlbumEntryViewController: BaseViewController {
}
private func pushWiredTransfer(album: TravelAlbum) {
refreshState.markRefreshNeeded()
let controller = WiredCameraTransferViewController(
viewModel: WiredCameraTransferViewModel(
albumId: album.id,
@@ -177,7 +230,11 @@ extension TravelAlbumEntryViewController: UITableViewDelegate {
func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
tableView.deselectRow(at: indexPath, animated: true)
guard let album = dataSource.itemIdentifier(for: indexPath) else { return }
navigationController?.pushViewController(TravelAlbumDetailViewController(albumId: album.id), animated: true)
refreshState.markRefreshNeeded()
navigationController?.pushViewController(
TravelAlbumDetailViewController(albumId: album.id, api: api),
animated: true
)
}
}
@@ -25,6 +25,7 @@ final class TravelAlbumPhotoPreviewViewController: UIViewController {
private let loadMore: TravelAlbumPreviewLoadMore?
private let reload: TravelAlbumPreviewReload?
private let onProjectDeleted: ((Int) -> Void)?
private let onAIRetouchSubmitted: (() -> Void)?
private let allowsActions: Bool
private var nodes: [TravelAlbumPreviewNode] = []
private var currentNodeIndex = 0
@@ -73,7 +74,8 @@ final class TravelAlbumPhotoPreviewViewController: UIViewController {
aiRetouchAPI: (any TravelAlbumServing)? = nil,
loadMore: TravelAlbumPreviewLoadMore? = nil,
reload: TravelAlbumPreviewReload? = nil,
onProjectDeleted: ((Int) -> Void)? = nil
onProjectDeleted: ((Int) -> Void)? = nil,
onAIRetouchSubmitted: (() -> Void)? = nil
) {
self.projects = Self.deduplicated(projects)
self.totalCount = max(totalCount, projects.count)
@@ -85,10 +87,19 @@ final class TravelAlbumPhotoPreviewViewController: UIViewController {
self.loadMore = loadMore
self.reload = reload
self.onProjectDeleted = onProjectDeleted
self.onAIRetouchSubmitted = onAIRetouchSubmitted
self.allowsActions = allowsActions
super.init(nibName: nil, bundle: nil)
modalPresentationStyle = .fullScreen
rebuildNodes(keepingProjectIndex: max(0, min(startProjectIndex, projects.count - 1)), kind: startKind)
let resolvedProjectIndex = max(0, min(startProjectIndex, self.projects.count - 1))
let initialProject = self.projects.indices.contains(resolvedProjectIndex)
? self.projects[resolvedProjectIndex]
: nil
let resolvedStartKind = initialProject?.asset(for: startKind) == nil
? initialProject?.orderedAssets.first?.kind ?? .original
: startKind
selectedKind = resolvedStartKind
rebuildNodes(keepingProjectIndex: resolvedProjectIndex, kind: resolvedStartKind)
}
@available(*, unavailable)
@@ -162,6 +173,7 @@ final class TravelAlbumPhotoPreviewViewController: UIViewController {
sizeLabel.textColor = UIColor.white.withAlphaComponent(0.52)
sizeLabel.font = .systemFont(ofSize: 13, weight: .regular)
sizeLabel.textAlignment = .center
sizeLabel.accessibilityIdentifier = "travelAlbum.previewFileSizeLabel"
counterLabel.textColor = UIColor.white.withAlphaComponent(0.82)
counterLabel.font = .monospacedDigitSystemFont(ofSize: 15, weight: .medium)
@@ -420,7 +432,9 @@ final class TravelAlbumPhotoPreviewViewController: UIViewController {
guard let node = currentNode else { return }
let asset = currentAsset
titleLabel.text = asset?.fileName.isEmpty == false ? asset?.fileName : "未命名照片"
sizeLabel.text = TravelAlbumDisplayFormatter.fileSizeText(asset?.fileSize ?? 0)
let fileSize = asset?.fileSize ?? 0
sizeLabel.text = fileSize > 0 ? TravelAlbumDisplayFormatter.fileSizeText(fileSize) : nil
sizeLabel.isHidden = fileSize <= 0
counterLabel.text = "\(node.projectIndex + 1)/\(max(totalCount, projects.count))"
counterLabel.accessibilityLabel = "第 \(node.projectIndex + 1) 张,共 \(max(totalCount, projects.count)) 张"
highResolutionButton.isEnabled = asset != nil
@@ -726,34 +740,14 @@ final class TravelAlbumPhotoPreviewViewController: UIViewController {
workflow: workflow
),
api: aiRetouchAPI,
onSubmitted: { [weak self] submission in
onSubmitted: { [weak self] _ in
guard let self else { return }
self.reloadProjects(showGlobalLoading: false, forceRefreshImage: false)
self.presentAIRetouchSubmitted(submission)
self.dismiss(animated: true) { self.onAIRetouchSubmitted?() }
}
)
present(controller, animated: true)
}
private func presentAIRetouchSubmitted(_ submission: TravelAlbumAIJobSubmission) {
let alert = UIAlertController(
title: "AI修图任务已提交",
message: "完成后将通过消息通知。",
preferredStyle: .alert
)
alert.addAction(UIAlertAction(title: "知道了", style: .cancel))
alert.addAction(UIAlertAction(title: "查看任务", style: .default) { [weak self] _ in
guard let self, let navigationController = presentingViewController?.navigationController else { return }
dismiss(animated: true) {
navigationController.pushViewController(
TravelAlbumAIJobDetailViewController(batchId: submission.aiRetouchBatchId),
animated: true
)
}
})
present(alert, animated: true)
}
@objc private func deleteTapped() {
guard currentProject != nil, !isDeletingProject else { return }
let alert = UIAlertController(
@@ -999,6 +993,56 @@ extension TravelAlbumPhotoPreviewViewController: UICollectionViewDataSource, UIC
}
}
/// 深色图片预览中使用的高对比度延迟加载指示器。
@MainActor
private final class TravelAlbumPreviewLoadingIndicator: Indicator {
private let containerView = UIView()
private let activityIndicator = UIActivityIndicatorView(style: .medium)
private var revealTask: Task<Void, Never>?
var view: IndicatorView { containerView }
init() {
containerView.backgroundColor = UIColor(red: 39 / 255, green: 39 / 255, blue: 42 / 255, alpha: 0.94)
containerView.layer.cornerRadius = 22
containerView.isAccessibilityElement = true
containerView.accessibilityIdentifier = "travelAlbum.previewLoadingIndicator"
containerView.accessibilityLabel = "图片加载中"
activityIndicator.color = UIColor.white.withAlphaComponent(0.92)
activityIndicator.transform = CGAffineTransform(scaleX: 1.2, y: 1.2)
containerView.addSubview(activityIndicator)
activityIndicator.snp.makeConstraints { $0.center.equalToSuperview() }
}
func startAnimatingView() {
revealTask?.cancel()
containerView.alpha = 0
containerView.isHidden = true
revealTask = Task { @MainActor [weak self] in
try? await Task.sleep(for: .milliseconds(150))
guard !Task.isCancelled, let self else { return }
activityIndicator.startAnimating()
containerView.isHidden = false
UIView.animate(
withDuration: UIAccessibility.isReduceMotionEnabled ? 0 : 0.12,
animations: { self.containerView.alpha = 1 }
)
}
}
func stopAnimatingView() {
revealTask?.cancel()
revealTask = nil
activityIndicator.stopAnimating()
containerView.alpha = 0
containerView.isHidden = true
}
func sizeStrategy(in imageView: KFCrossPlatformImageView) -> IndicatorSizeStrategy {
.size(CGSize(width: 44, height: 44))
}
}
/// 预览图片 Cell,使用 UIScrollView 提供远程加载、双击和双指缩放。
private final class TravelAlbumPreviewImageCell: UICollectionViewCell, UIScrollViewDelegate {
static let reuseIdentifier = "TravelAlbumPreviewImageCell"
@@ -1071,7 +1115,7 @@ private final class TravelAlbumPreviewImageCell: UICollectionViewCell, UIScrollV
imageView.contentMode = .scaleAspectFit
imageView.backgroundColor = .black
imageView.kf.indicatorType = .activity
imageView.kf.indicatorType = .custom(indicator: TravelAlbumPreviewLoadingIndicator())
imageView.accessibilityIdentifier = "travelAlbum.previewImageView"
retryButton.setTitle("图片加载失败,点击重试", for: .normal)
@@ -25,7 +25,7 @@ final class TravelAlbumAIRetouchTemplateViewModelTests: XCTestCase {
XCTAssertTrue(viewModel.shouldShowInitialTips)
XCTAssertEqual(
viewModel.initialTipsText,
"Tips:氛围感修图为选填,可横向选择一种样式;选中后每张照片会额外生成1个独立结果,第一张照片仍另生成封面。"
"Tips:氛围感修图为选填;封面风格为必选,将使用第一张照片另生成封面。"
)
XCTAssertEqual(viewModel.visibleCategories, [.refined, .atmosphere, .cover])
XCTAssertEqual(viewModel.selectedRefinedTemplateId, 11)
@@ -44,6 +44,14 @@ final class TravelAlbumAIRetouchTemplateViewModelTests: XCTestCase {
)
await viewModel.loadTemplates(api: api)
XCTAssertEqual(
viewModel.initialTipsText,
"Tips:氛围感修图为选填,选中后每张照片会额外生成1个独立结果。"
)
viewModel.toggleTemplate(id: 11, category: .refined)
XCTAssertEqual(viewModel.selectedRefinedTemplateId, 11)
viewModel.toggleTemplate(id: 21, category: .atmosphere)
XCTAssertEqual(viewModel.selectedAtmosphereTemplateId, 21)
@@ -94,7 +102,7 @@ final class TravelAlbumAIRetouchTemplateViewModelTests: XCTestCase {
)
}
func testReretouchWorkflowShowsAndSubmitsOnlyRequiredCategories() async {
func testReretouchWorkflowAllowsAtmosphereOnlyFromOriginalTab() async {
let api = makeAPI()
let refined = TravelAlbumAIRetouchTemplateViewModel(
scenicId: 18,
@@ -125,8 +133,21 @@ final class TravelAlbumAIRetouchTemplateViewModelTests: XCTestCase {
)
await all.loadTemplates(api: api)
XCTAssertEqual(all.visibleCategories, [.refined, .atmosphere])
XCTAssertNil(all.selectedRefinedTemplateId)
XCTAssertNil(all.selectedAtmosphereTemplateId)
XCTAssertTrue(all.isOptional(.refined))
XCTAssertTrue(all.isOptional(.atmosphere))
XCTAssertEqual(all.requiredQuota, 0)
XCTAssertFalse(all.canSubmit)
XCTAssertEqual(all.validationMessage, "请至少选择一个修图模板")
all.toggleTemplate(id: 11, category: .refined)
XCTAssertEqual(all.selectedRefinedTemplateId, 11)
XCTAssertEqual(all.selectedAtmosphereTemplateId, 21)
all.toggleTemplate(id: 11, category: .refined)
XCTAssertNil(all.selectedRefinedTemplateId)
all.toggleTemplate(id: 21, category: .atmosphere)
XCTAssertEqual(all.requiredQuota, 1)
XCTAssertTrue(all.canSubmit)
await all.submit(api: api)
XCTAssertEqual(api.aiReretouchRequests, [
@@ -148,7 +169,7 @@ final class TravelAlbumAIRetouchTemplateViewModelTests: XCTestCase {
id: 9,
aiRetouchBatchId: 90,
type: .all,
refinedTemplateId: 11,
refinedTemplateId: nil,
atmosphereTemplateId: 21
),
])
@@ -172,12 +193,13 @@ final class TravelAlbumAIRetouchTemplateViewModelTests: XCTestCase {
XCTAssertTrue(api.aiReretouchRequests.isEmpty)
}
func testMissingRequiredTemplateDisablesMatchingWorkflow() async {
func testMissingOptionalAtmosphereStillAllowsSelectingRefinedInAllReretouchWorkflow() async {
let api = makeAPI()
api.aiRetouchTemplatesResponse = TravelAlbumAIRetouchTemplatesResponse(
refinedTemplates: [template(11, "清透")],
atmosphereTemplates: [],
coverTemplates: []
coverTemplates: [],
remainingQuota: 100
)
let viewModel = TravelAlbumAIRetouchTemplateViewModel(
scenicId: 18,
@@ -187,7 +209,43 @@ final class TravelAlbumAIRetouchTemplateViewModelTests: XCTestCase {
await viewModel.loadTemplates(api: api)
XCTAssertFalse(viewModel.canSubmit)
XCTAssertEqual(viewModel.validationMessage, "暂无可用的氛围感修图模板")
viewModel.toggleTemplate(id: 11, category: .refined)
XCTAssertTrue(viewModel.canSubmit)
XCTAssertNil(viewModel.validationMessage)
XCTAssertEqual(viewModel.requiredQuota, 1)
}
func testMissingRequiredRefinedAndCoverTemplatesDisableMatchingWorkflows() async {
let refinedAPI = makeAPI()
refinedAPI.aiRetouchTemplatesResponse = TravelAlbumAIRetouchTemplatesResponse(
refinedTemplates: [],
atmosphereTemplates: [template(21, "暖阳")],
coverTemplates: [template(31, "杂志")],
remainingQuota: 100
)
let reretouch = TravelAlbumAIRetouchTemplateViewModel(
scenicId: 18,
workflow: .reretouch(materialId: 7, batchId: 70, type: .refined)
)
await reretouch.loadTemplates(api: refinedAPI)
XCTAssertFalse(reretouch.canSubmit)
XCTAssertEqual(reretouch.validationMessage, "暂无可用的原图精修模板")
let coverAPI = makeAPI()
coverAPI.aiRetouchTemplatesResponse = TravelAlbumAIRetouchTemplatesResponse(
refinedTemplates: [template(11, "清透")],
atmosphereTemplates: [template(21, "暖阳")],
coverTemplates: [],
remainingQuota: 100
)
let batch = TravelAlbumAIRetouchTemplateViewModel(
albumId: 8,
scenicId: 18,
materialIds: [1, 2, 3, 4]
)
await batch.loadTemplates(api: coverAPI)
XCTAssertFalse(batch.canSubmit)
XCTAssertEqual(batch.validationMessage, "暂无可用的封面风格模板")
}
func testLoadFailureExposesRetryMessageAndKeepsSubmissionDisabled() async {
+18 -1
View File
@@ -243,6 +243,7 @@ final class TravelAlbumAPITests: XCTestCase {
jobSubmissionJSON(batchId: 51, albumId: 6),
jobSubmissionJSON(batchId: 52, albumId: 6),
jobSubmissionJSON(batchId: 53, albumId: 6),
jobSubmissionJSON(batchId: 54, albumId: 6),
])
let api = TravelAlbumAPI(client: APIClient(environment: .testing, session: session))
@@ -273,13 +274,22 @@ final class TravelAlbumAPITests: XCTestCase {
atmosphereTemplateId: 22
)
)
try await api.submitAIReretouch(
TravelAlbumAIReretouchRequest(
id: 14,
aiRetouchBatchId: 54,
type: .all,
refinedTemplateId: 21,
atmosphereTemplateId: nil
)
)
let bodies = try session.requests.map { request in
try JSONSerialization.jsonObject(with: XCTUnwrap(request.httpBody)) as? [String: Any]
}
XCTAssertEqual(session.requests.map { $0.url?.path }, Array(
repeating: "/api/yf-handset-app/photog/travel-album/ai-reretouch",
count: 3
count: 4
))
XCTAssertEqual(Set(bodies[0]?.keys.map { $0 } ?? []), ["id", "ai_retouch_batch_id", "type", "refined_template_id"])
XCTAssertEqual(Set(bodies[1]?.keys.map { $0 } ?? []), ["id", "ai_retouch_batch_id", "type", "atmosphere_template_id"])
@@ -290,9 +300,16 @@ final class TravelAlbumAPITests: XCTestCase {
"refined_template_id",
"atmosphere_template_id",
])
XCTAssertEqual(Set(bodies[3]?.keys.map { $0 } ?? []), [
"id",
"ai_retouch_batch_id",
"type",
"refined_template_id",
])
XCTAssertEqual(bodies[0]?["type"] as? Int, 1)
XCTAssertEqual(bodies[1]?["type"] as? Int, 2)
XCTAssertEqual(bodies[2]?["type"] as? Int, 3)
XCTAssertEqual(bodies[3]?["type"] as? Int, 3)
}
func testAIJobListBuildsCursorQueryAndDecodesUnknownStatusSafely() async throws {
@@ -14,7 +14,7 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
let api = TravelAlbumMockAPI()
let source = TravelAlbumAIJobSourceMaterial(
id: 691,
fileName: "IMG_8293.JPG",
fileName: "beauty_1757851944526_这个文件标题特别长用于验证布局不会挤压状态.jpg",
thumbnailURL: ""
)
let failedTarget = TravelAlbumAIJobTarget(
@@ -104,6 +104,42 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
XCTAssertTrue(labels.contains("失败原因:AI服务处理超时,请重新修图"))
XCTAssertFalse(labels.contains("去相册处理"))
XCTAssertFalse(labels.contains { $0.contains("VENDOR_TIMEOUT") })
let contentCard = try XCTUnwrap(controller.view.findSubview {
$0.accessibilityIdentifier == "aiRetouchJob.detail.contentCard"
})
let refinedChip = try XCTUnwrap(controller.view.findSubview {
$0.accessibilityIdentifier == "aiRetouchJob.contentChip.refined"
})
let atmosphereChip = try XCTUnwrap(controller.view.findSubview {
$0.accessibilityIdentifier == "aiRetouchJob.contentChip.atmosphere"
})
let coverChip = try XCTUnwrap(controller.view.findSubview {
$0.accessibilityIdentifier == "aiRetouchJob.contentChip.cover"
})
XCTAssertEqual(refinedChip.backgroundColor?.travelAlbumTestHexRGB, 0xEAF3FF)
XCTAssertEqual(atmosphereChip.backgroundColor?.travelAlbumTestHexRGB, 0xFFF2DC)
XCTAssertEqual(coverChip.backgroundColor?.travelAlbumTestHexRGB, 0xF2ECFF)
XCTAssertNil(refinedChip.findSubview { $0 is UIImageView })
XCTAssertNil(atmosphereChip.findSubview { $0 is UIImageView })
XCTAssertNil(coverChip.findSubview { $0 is UIImageView })
XCTAssertGreaterThan(atmosphereChip.bounds.width, refinedChip.bounds.width)
let coverFrame = coverChip.convert(coverChip.bounds, to: contentCard)
XCTAssertLessThan(coverFrame.maxX, contentCard.bounds.maxX - 16)
let titleLabel = try XCTUnwrap(controller.view.findSubview {
$0.accessibilityIdentifier == "aiRetouchJob.target.3.title"
})
let templateLabel = try XCTUnwrap(controller.view.findSubview {
$0.accessibilityIdentifier == "aiRetouchJob.target.3.template"
})
let statusView = try XCTUnwrap(controller.view.findSubview {
$0.accessibilityIdentifier == "aiRetouchJob.target.3.status"
})
let titleFrame = titleLabel.convert(titleLabel.bounds, to: controller.view)
let templateFrame = templateLabel.convert(templateLabel.bounds, to: controller.view)
let statusFrame = statusView.convert(statusView.bounds, to: controller.view)
XCTAssertLessThanOrEqual(titleFrame.maxX, controller.view.bounds.maxX - 18)
XCTAssertGreaterThan(statusFrame.minY, templateFrame.maxY)
XCTAssertLessThanOrEqual(statusFrame.maxX, controller.view.bounds.maxX - 18)
let detailScreenshot = UIGraphicsImageRenderer(bounds: window.bounds).image { context in
window.layer.render(in: context.cgContext)
}
@@ -315,6 +351,56 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
XCTAssertEqual(counterContainer.bounds.width - counterLabel.frame.maxX, 12, accuracy: 0.5)
}
func testPreviewAppliesRequestedInitialKindForCoverOnlyProject() throws {
let cover = TravelAlbumPreviewAsset(
id: "cover-897",
kind: .cover,
fileURL: "https://example.com/final.png",
coverURL: "https://example.com/final.png",
fileName: "AI封面",
fileSize: 0
)
let controller = TravelAlbumPhotoPreviewViewController(
projects: [TravelAlbumPreviewProject(originalMaterialId: 897, assets: [cover])],
totalCount: 1,
startProjectIndex: 0,
startKind: .cover,
allowsActions: false
)
controller.loadViewIfNeeded()
controller.view.frame = CGRect(x: 0, y: 0, width: 390, height: 844)
controller.view.layoutIfNeeded()
let highResolutionButton = try XCTUnwrap(
controller.view.findSubview {
$0.accessibilityIdentifier == "travelAlbum.previewHighResolutionButton"
} as? UIButton
)
let loadingIndicator = try XCTUnwrap(
controller.view.findSubview {
$0.accessibilityIdentifier == "travelAlbum.previewLoadingIndicator"
}
)
let activityIndicator = try XCTUnwrap(
loadingIndicator.findSubview { $0 is UIActivityIndicatorView } as? UIActivityIndicatorView
)
let fileSizeLabel = try XCTUnwrap(
controller.view.findSubview {
$0.accessibilityIdentifier == "travelAlbum.previewFileSizeLabel"
} as? UILabel
)
XCTAssertTrue(controller.view.allLabels().contains { $0.text == "AI封面" })
XCTAssertTrue(highResolutionButton.isEnabled)
XCTAssertTrue(fileSizeLabel.isHidden)
XCTAssertNil(fileSizeLabel.text)
XCTAssertEqual(fileSizeLabel.bounds.height, 0, accuracy: 0.5)
XCTAssertEqual(loadingIndicator.bounds.size, CGSize(width: 44, height: 44))
XCTAssertEqual(loadingIndicator.backgroundColor?.travelAlbumTestHexRGB, 0x27272A)
XCTAssertEqual(loadingIndicator.accessibilityLabel, "图片加载中")
XCTAssertEqual(activityIndicator.color.travelAlbumTestHexRGB, 0xFFFFFF)
}
func testPreviewSuccessfulDeletionShowsNextProjectAndPreventsDuplicateRequests() async throws {
let api = TravelAlbumMockAPI()
api.deleteMaterialDelayNanoseconds = 1_000_000
@@ -763,7 +849,7 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
XCTAssertEqual(selectionCountLabel.text, "已选择 4 张照片")
XCTAssertEqual(
tipsLabel.text,
"Tips:氛围感修图为选填,可横向选择一种样式;选中后每张照片会额外生成1个独立结果,第一张照片仍另生成封面。"
"Tips:氛围感修图为选填;封面风格为必选,将使用第一张照片另生成封面。"
)
XCTAssertFalse(tipsContainer.isHidden)
XCTAssertEqual(tipsContainer.backgroundColor?.travelAlbumTestHexRGB, 0xF4F8FF)
@@ -806,10 +892,9 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
} as? UICollectionView
)
let accessibleLabels = controller.view.allAccessibilityLabels()
XCTAssertTrue(accessibleLabels.contains("原图精修"))
XCTAssertFalse(accessibleLabels.contains("原图精修,必选"))
XCTAssertTrue(accessibleLabels.contains("原图精修,必选"))
XCTAssertTrue(accessibleLabels.contains("氛围感修图,选填"))
XCTAssertTrue(accessibleLabels.contains("封面风格,赠送 · 不占额度"))
XCTAssertTrue(accessibleLabels.contains("封面风格,必选 · 赠送 · 不占额度"))
collectionView.scrollToItem(
at: IndexPath(item: 0, section: 4),
@@ -833,6 +918,43 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
XCTAssertEqual(quotaLabel.accessibilityLabel, "剩余8张修图额度")
}
func testOriginalTabReretouchSheetShowsBothTemplateGroupsAsOptional() async throws {
let api = TravelAlbumMockAPI()
api.aiRetouchTemplatesResponse = TravelAlbumAIRetouchTemplatesResponse(
refinedTemplates: [TravelAlbumAIRetouchTemplate(id: 1, name: "清透", previewURL: "")],
atmosphereTemplates: [TravelAlbumAIRetouchTemplate(id: 2, name: "暖阳", previewURL: "")],
remainingQuota: 8
)
let viewModel = TravelAlbumAIRetouchTemplateViewModel(
scenicId: 18,
workflow: .reretouch(materialId: 7, batchId: 70, type: .all)
)
let controller = TravelAlbumAIRetouchTemplateViewController(
viewModel: viewModel,
api: api,
onSubmitted: { _ in }
)
controller.loadViewIfNeeded()
controller.view.frame = CGRect(x: 0, y: 0, width: 390, height: 844)
await waitUntil { !viewModel.isLoading && api.aiRetouchTemplateScenicIds.count == 1 }
controller.view.layoutIfNeeded()
let confirmButton = try XCTUnwrap(
controller.view.findSubview {
$0.accessibilityIdentifier == "travelAlbum.aiRetouchConfirmButton"
} as? UIButton
)
let accessibleLabels = controller.view.allAccessibilityLabels()
XCTAssertTrue(accessibleLabels.contains("原图精修,选填"))
XCTAssertTrue(accessibleLabels.contains("氛围感修图,选填"))
XCTAssertFalse(confirmButton.isEnabled)
viewModel.toggleTemplate(id: 2, category: .atmosphere)
await waitUntil { confirmButton.isEnabled }
XCTAssertNil(viewModel.selectedRefinedTemplateId)
XCTAssertEqual(viewModel.selectedAtmosphereTemplateId, 2)
}
func testAIRetouchTemplateCellExposesSelectedState() {
let cell = TravelAlbumAIRetouchTemplateCell(frame: .zero)
let template = TravelAlbumAIRetouchTemplate(id: 1, name: "清透", previewURL: "")
@@ -954,7 +1076,7 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
XCTAssertEqual(comparisonButton.bounds.size, CGSize(width: 48, height: 48))
}
func testPreviewAIRetouchUsesSelectedRetouchedTabWorkflowAndShowsTaskAction() async throws {
func testPreviewAIRetouchFromRetouchedTabAllowsBothOptionalTemplates() async throws {
UIView.setAnimationsEnabled(false)
defer { UIView.setAnimationsEnabled(true) }
let api = TravelAlbumMockAPI()
@@ -985,13 +1107,15 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
),
]
)
var submissionNotified = false
let controller = TravelAlbumPhotoPreviewViewController(
projects: [project],
totalCount: 1,
startProjectIndex: 0,
albumId: 8,
scenicIdProvider: { 18 },
aiRetouchAPI: api
aiRetouchAPI: api,
onAIRetouchSubmitted: { submissionNotified = true }
)
let window = UIWindow(frame: CGRect(x: 0, y: 0, width: 390, height: 844))
window.rootViewController = controller
@@ -1019,7 +1143,16 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
$0.accessibilityIdentifier == "travelAlbum.aiRetouchConfirmButton"
} as? UIButton
)
await waitUntil { api.aiRetouchTemplateScenicIds.count == 1 && confirmButton.isEnabled }
let collectionView = try XCTUnwrap(
sheet.view.findSubview {
$0.accessibilityIdentifier == "travelAlbum.aiRetouchTemplateCollection"
} as? UICollectionView
)
await waitUntil {
api.aiRetouchTemplateScenicIds.count == 1 &&
collectionView.numberOfSections > 2 &&
collectionView.numberOfItems(inSection: 2) == 1
}
sheet.view.frame = CGRect(x: 0, y: 0, width: 390, height: 844)
sheet.view.layoutIfNeeded()
@@ -1027,7 +1160,11 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
label.isHidden ? nil : label.text
}
XCTAssertTrue(visibleLabelTexts.contains("原图精修"))
XCTAssertFalse(visibleLabelTexts.contains { $0.contains("氛围感修图") })
XCTAssertTrue(visibleLabelTexts.contains("氛围感修图"))
let accessibleLabels = sheet.view.allAccessibilityLabels()
XCTAssertTrue(accessibleLabels.contains("原图精修,选填"))
XCTAssertTrue(accessibleLabels.contains("氛围感修图,选填"))
XCTAssertFalse(confirmButton.isEnabled)
let selectionCountLabel = try XCTUnwrap(
sheet.view.findSubview {
$0.accessibilityIdentifier == "travelAlbum.aiRetouchSelectionCountLabel"
@@ -1046,14 +1183,26 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
XCTAssertEqual(selectionCountLabel.text, "已选择 1 张照片")
XCTAssertTrue(tipsContainer.isHidden)
XCTAssertTrue(tipsLabel.isHidden)
collectionView.delegate?.collectionView?(
collectionView,
didSelectItemAt: IndexPath(item: 0, section: 2)
)
await waitUntil { confirmButton.isEnabled }
confirmButton.sendActions(for: .touchUpInside)
await waitUntil { api.aiReretouchRequests.count == 1 }
await waitUntil { controller.presentedViewController is UIAlertController }
await waitUntil { submissionNotified }
XCTAssertEqual(api.aiReretouchRequests.first?.type, .refined)
let successAlert = try XCTUnwrap(controller.presentedViewController as? UIAlertController)
XCTAssertEqual(successAlert.title, "AI修图任务已提交")
XCTAssertEqual(successAlert.actions.map(\.title), ["知道了", "查看任务"])
XCTAssertEqual(
api.aiReretouchRequests.first,
TravelAlbumAIReretouchRequest(
id: 7,
aiRetouchBatchId: 70,
type: .all,
refinedTemplateId: nil,
atmosphereTemplateId: 21
)
)
XCTAssertTrue(submissionNotified)
XCTAssertTrue(window.rootViewController === controller)
}
@@ -1150,9 +1299,8 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
XCTAssertFalse(controller.view.allLabels().contains { $0.text == "刷新失败" })
}
func testMaterialCellShowsSemanticStatusBadgeWithoutOverlappingSelectionCheck() throws {
func testMaterialCellShowsIndependentPurchaseAndAIRetouchBadgesWithoutOverlappingSelectionCheck() throws {
let cases: [(TravelAlbumMaterial, String, UInt)] = [
(TravelAlbumMaterial(isPurchased: true), "已购", 0x475569),
(TravelAlbumMaterial(aiRetouchStatus: 1), "待处理", 0xB45309),
(TravelAlbumMaterial(aiRetouchStatus: 2), "修图中", 0x1D4ED8),
(TravelAlbumMaterial(aiRetouchStatus: 3), "AI已修", 0x047857),
@@ -1171,12 +1319,12 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
let badgeView = try XCTUnwrap(
cell.findSubview {
$0.accessibilityIdentifier == "travelAlbum.materialStatusBadge"
$0.accessibilityIdentifier == "travelAlbum.materialAIRetouchBadge"
}
)
let badgeLabel = try XCTUnwrap(
cell.findSubview {
$0.accessibilityIdentifier == "travelAlbum.materialStatusBadgeLabel"
$0.accessibilityIdentifier == "travelAlbum.materialAIRetouchBadgeLabel"
} as? UILabel
)
let selectionCheck = try XCTUnwrap(
@@ -1189,22 +1337,61 @@ final class TravelAlbumDetailViewControllerTests: XCTestCase {
XCTAssertEqual(badgeLabel.text, expectedText)
XCTAssertEqual(badgeView.backgroundColor?.travelAlbumTestHexRGB, expectedColor)
XCTAssertLessThan(badgeView.frame.maxX, selectionCheck.frame.minX)
XCTAssertTrue(cell.accessibilityLabel?.contains("状态:\(expectedText)") == true)
XCTAssertTrue(cell.accessibilityLabel?.contains("修图状态:\(expectedText)") == true)
}
let combinedCell = TravelAlbumMaterialCell(frame: CGRect(x: 0, y: 0, width: 176, height: 220))
combinedCell.apply(
material: TravelAlbumMaterial(isPurchased: true, aiRetouchStatus: 2),
selectionMode: true,
selected: false
)
combinedCell.layoutIfNeeded()
let purchaseBadge = try XCTUnwrap(
combinedCell.findSubview {
$0.accessibilityIdentifier == "travelAlbum.materialPurchaseBadge"
}
)
let purchaseLabel = try XCTUnwrap(
combinedCell.findSubview {
$0.accessibilityIdentifier == "travelAlbum.materialPurchaseBadgeLabel"
} as? UILabel
)
let aiRetouchBadge = try XCTUnwrap(
combinedCell.findSubview {
$0.accessibilityIdentifier == "travelAlbum.materialAIRetouchBadge"
}
)
XCTAssertFalse(purchaseBadge.isHidden)
XCTAssertFalse(aiRetouchBadge.isHidden)
XCTAssertEqual(purchaseLabel.text, "已购")
XCTAssertEqual(purchaseBadge.backgroundColor?.travelAlbumTestHexRGB, 0x475569)
XCTAssertGreaterThan(purchaseBadge.frame.minY, aiRetouchBadge.frame.maxY)
XCTAssertEqual(purchaseBadge.frame.minX, 6, accuracy: 0.5)
XCTAssertEqual(purchaseBadge.frame.maxY, 170, accuracy: 0.5)
XCTAssertTrue(combinedCell.accessibilityLabel?.contains("购买状态:已购") == true)
XCTAssertTrue(combinedCell.accessibilityLabel?.contains("修图状态:修图中") == true)
let hiddenCell = TravelAlbumMaterialCell(frame: CGRect(x: 0, y: 0, width: 176, height: 220))
hiddenCell.apply(
material: TravelAlbumMaterial(aiRetouchStatus: 0, aiRetouchStatusName: "已上传"),
selectionMode: false,
selected: false
)
let hiddenBadge = try XCTUnwrap(
let hiddenPurchaseBadge = try XCTUnwrap(
hiddenCell.findSubview {
$0.accessibilityIdentifier == "travelAlbum.materialStatusBadge"
$0.accessibilityIdentifier == "travelAlbum.materialPurchaseBadge"
}
)
XCTAssertTrue(hiddenBadge.isHidden)
XCTAssertFalse(hiddenCell.accessibilityLabel?.contains("状态:") == true)
let hiddenAIRetouchBadge = try XCTUnwrap(
hiddenCell.findSubview {
$0.accessibilityIdentifier == "travelAlbum.materialAIRetouchBadge"
}
)
XCTAssertTrue(hiddenPurchaseBadge.isHidden)
XCTAssertTrue(hiddenAIRetouchBadge.isHidden)
XCTAssertFalse(hiddenCell.accessibilityLabel?.contains("购买状态:") == true)
XCTAssertFalse(hiddenCell.accessibilityLabel?.contains("修图状态:") == true)
}
private func waitUntil(_ condition: @escaping () -> Bool) async {
+30 -16
View File
@@ -268,7 +268,7 @@ final class TravelAlbumModelsTests: XCTestCase {
)
}
func testPreviewProjectMapsCurrentTabToMinimalAIRetouchWorkflow() {
func testPreviewProjectUsesAllReretouchWorkflowForEveryVisibleVariantTab() {
let originalOnly = makePreviewProject(id: 9, kinds: [.original])
let variants = TravelAlbumPreviewProject(
originalMaterialId: 10,
@@ -286,11 +286,11 @@ final class TravelAlbumModelsTests: XCTestCase {
)
XCTAssertEqual(
variants.aiRetouchWorkflow(albumId: 7, selectedKind: .retouched),
.reretouch(materialId: 10, batchId: 88, type: .refined)
.reretouch(materialId: 10, batchId: 88, type: .all)
)
XCTAssertEqual(
variants.aiRetouchWorkflow(albumId: 7, selectedKind: .atmosphere),
.reretouch(materialId: 10, batchId: 88, type: .atmosphere)
.reretouch(materialId: 10, batchId: 88, type: .all)
)
XCTAssertNil(variants.aiRetouchWorkflow(albumId: 7, selectedKind: .cover))
}
@@ -399,34 +399,48 @@ final class TravelAlbumModelsTests: XCTestCase {
XCTAssertEqual(material.aiAtmosphereURL, "")
}
func testTravelAlbumMaterialBadgePriorityAndFallbacks() {
XCTAssertNil(TravelAlbumMaterial(aiRetouchStatus: 0).badgePresentation)
func testTravelAlbumMaterialPurchaseAndAIRetouchBadgesAreIndependent() {
XCTAssertNil(TravelAlbumMaterial().purchaseBadgePresentation)
XCTAssertEqual(
TravelAlbumMaterial(isPurchased: true, aiRetouchStatus: 0, aiRetouchStatusName: "已上传")
.badgePresentation,
TravelAlbumMaterial(isPurchased: true).purchaseBadgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .purchased, text: "已购")
)
XCTAssertNil(TravelAlbumMaterial(isPurchased: true).aiRetouchBadgePresentation)
let purchasedAndProcessing = TravelAlbumMaterial(isPurchased: true, aiRetouchStatus: 2)
XCTAssertEqual(
purchasedAndProcessing.purchaseBadgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .purchased, text: "已购")
)
XCTAssertEqual(
TravelAlbumMaterial(aiRetouchStatus: 1, aiRetouchStatusName: " 排队中 ").badgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .pending, text: "排队中")
)
XCTAssertEqual(
TravelAlbumMaterial(aiRetouchStatus: 2, aiRetouchStatusName: " ").badgePresentation,
purchasedAndProcessing.aiRetouchBadgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .processing, text: "修图中")
)
XCTAssertEqual(
TravelAlbumMaterial(aiRetouchStatus: 3).badgePresentation,
TravelAlbumMaterial(aiRetouchStatus: 1, aiRetouchStatusName: " 排队中 ")
.aiRetouchBadgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .pending, text: "排队中")
)
XCTAssertEqual(
TravelAlbumMaterial(aiRetouchStatus: 2, aiRetouchStatusName: " ").aiRetouchBadgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .processing, text: "修图中")
)
XCTAssertEqual(
TravelAlbumMaterial(aiRetouchStatus: 3).aiRetouchBadgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .retouched, text: "AI已修")
)
XCTAssertEqual(
TravelAlbumMaterial(aiRetouchStatus: 3, aiRetouchStatusName: "AI封面").badgePresentation,
TravelAlbumMaterial(aiRetouchStatus: 3, aiRetouchStatusName: "AI封面")
.aiRetouchBadgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .cover, text: "AI封面")
)
XCTAssertEqual(
TravelAlbumMaterial(aiRetouchStatus: 4).badgePresentation,
TravelAlbumMaterial(aiRetouchStatus: 4).aiRetouchBadgePresentation,
TravelAlbumMaterialBadgePresentation(kind: .failed, text: "失败")
)
XCTAssertNil(TravelAlbumMaterial(aiRetouchStatus: 99, aiRetouchStatusName: "未知").badgePresentation)
XCTAssertNil(
TravelAlbumMaterial(aiRetouchStatus: 99, aiRetouchStatusName: "未知")
.aiRetouchBadgePresentation
)
}
func testDisplayFormatters() {
+105 -6
View File
@@ -99,6 +99,19 @@ final class TravelAlbumAIJobViewModelTests: XCTestCase {
@MainActor
/// 旅拍相册 ViewModel 测试。
final class TravelAlbumEntryViewModelTests: XCTestCase {
func testReturnRefreshStateSuppressesAnimationUntilRefreshFinishes() {
var state = TravelAlbumReturnRefreshState()
XCTAssertFalse(state.beginRefreshIfNeeded())
XCTAssertFalse(state.suppressesSnapshotAnimations)
state.markRefreshNeeded()
XCTAssertTrue(state.beginRefreshIfNeeded())
XCTAssertTrue(state.suppressesSnapshotAnimations)
state.finishRefresh()
XCTAssertFalse(state.suppressesSnapshotAnimations)
XCTAssertFalse(state.beginRefreshIfNeeded())
}
func testLoadAlbumsUpdatesListAndTotal() async {
let api = TravelAlbumMockAPI()
api.listResponse = TravelAlbumListResponse(total: 1, list: [TravelAlbum(id: 1, name: "A")])
@@ -110,6 +123,19 @@ final class TravelAlbumEntryViewModelTests: XCTestCase {
XCTAssertEqual(viewModel.albums.first?.name, "A")
}
func testPullRefreshFailureKeepsExistingAlbums() async {
let api = TravelAlbumMockAPI()
api.listResponse = TravelAlbumListResponse(total: 1, list: [TravelAlbum(id: 1, name: "A")])
let viewModel = TravelAlbumEntryViewModel(currentScenicIdProvider: { 1 })
await viewModel.loadAlbums(api: api)
api.listError = APIError.serverCode(500, "刷新失败")
await viewModel.refreshAlbums(api: api)
XCTAssertEqual(viewModel.albumTotal, 1)
XCTAssertEqual(viewModel.albums.map(\.id), [1])
}
func testPreShootRequiresSinglePrice() async {
let api = TravelAlbumMockAPI()
let viewModel = TravelAlbumEntryViewModel(currentScenicIdProvider: { 1 })
@@ -253,14 +279,22 @@ final class TravelAlbumDetailViewModelTests: XCTestCase {
XCTAssertEqual(TravelAlbumDisplayFormatter.creationTimeText(""), "--")
}
func testOnlyUnpurchasedMaterialCanBeSelected() {
func testAllMaterialsCanBeSelectedAndPurchasedTabSupportsSelectionMode() async {
let api = TravelAlbumMockAPI()
api.materialListResponses = [TravelAlbumListResponse(total: 1, list: [])]
let viewModel = TravelAlbumDetailViewModel(albumId: 2)
await viewModel.selectTab(.purchased, api: api)
viewModel.toggleSelectionMode()
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 1, status: 2))
XCTAssertTrue(viewModel.selectedMaterialIds.isEmpty)
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 1, isPurchased: true))
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 2, isPurchased: false))
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 1, status: 1))
XCTAssertEqual(viewModel.selectedMaterialIds, [1])
XCTAssertTrue(viewModel.isSelectionMode)
XCTAssertEqual(viewModel.selectedMaterialIds, [1, 2])
XCTAssertEqual(viewModel.selectedPurchasedMaterialCount, 1)
XCTAssertEqual(
viewModel.deleteSelectedConfirmationMessage,
"已选择 2 张素材,其中 1 张已购买。不会删除已购买的项目,只会删除 1 张未购买的项目。"
)
}
func testCompletingAIRetouchClearsSelectionModeAndSelectedMaterials() {
@@ -274,6 +308,33 @@ final class TravelAlbumDetailViewModelTests: XCTestCase {
XCTAssertTrue(viewModel.selectedMaterialIds.isEmpty)
}
func testAIRetouchSubmissionRefreshesAlbumCountsAndCurrentMaterialList() async {
let api = TravelAlbumMockAPI()
api.infoResponse = TravelAlbum(id: 2, name: "刷新后的相册")
api.materialListResponses = [
TravelAlbumListResponse(total: 3, list: []),
TravelAlbumListResponse(total: 1, list: []),
TravelAlbumListResponse(total: 3, list: [
TravelAlbumMaterial(id: 7),
TravelAlbumMaterial(id: 8),
TravelAlbumMaterial(id: 9),
]),
]
let viewModel = TravelAlbumDetailViewModel(albumId: 2)
viewModel.toggleSelectionMode()
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 7, status: 1))
await viewModel.refreshAfterAIRetouchSubmission(api: api)
XCTAssertFalse(viewModel.isSelectionMode)
XCTAssertTrue(viewModel.selectedMaterialIds.isEmpty)
XCTAssertEqual(viewModel.album?.name, "刷新后的相册")
XCTAssertEqual(viewModel.allPhotoCount, 3)
XCTAssertEqual(viewModel.purchasedPhotoCount, 1)
XCTAssertEqual(viewModel.materials.map(\.id), [7, 8, 9])
XCTAssertEqual(api.materialRequests.count, 3)
}
func testPreviewDeletionRemovesMaterialAndSynchronizesCounts() async {
let api = TravelAlbumMockAPI()
api.materialListResponses = [
@@ -352,6 +413,42 @@ final class TravelAlbumDetailViewModelTests: XCTestCase {
XCTAssertFalse(viewModel.isSelectionMode)
XCTAssertTrue(viewModel.selectedMaterialIds.isEmpty)
}
func testDeleteSelectedMaterialsFiltersPurchasedItemsFromBatchRequest() async {
let api = TravelAlbumMockAPI()
api.materialListResponses = [
TravelAlbumListResponse(total: 0, list: []),
TravelAlbumListResponse(total: 0, list: []),
TravelAlbumListResponse(total: 0, list: []),
]
let viewModel = TravelAlbumDetailViewModel(albumId: 5)
viewModel.toggleSelectionMode()
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 10, isPurchased: true))
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 11, isPurchased: false))
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 12, isPurchased: true))
await viewModel.deleteSelectedMaterials(api: api)
XCTAssertEqual(api.deletedMaterialIDBatches, [[11]])
XCTAssertFalse(viewModel.isSelectionMode)
XCTAssertTrue(viewModel.selectedMaterialIds.isEmpty)
}
func testDeleteSelectedMaterialsDoesNotRequestAPIWhenAllSelectedItemsArePurchased() async {
let api = TravelAlbumMockAPI()
let viewModel = TravelAlbumDetailViewModel(albumId: 5)
var message = ""
viewModel.onShowMessage = { message = $0 }
viewModel.toggleSelectionMode()
viewModel.toggleMaterialSelection(TravelAlbumMaterial(id: 10, isPurchased: true))
await viewModel.deleteSelectedMaterials(api: api)
XCTAssertTrue(api.deletedMaterialIDBatches.isEmpty)
XCTAssertEqual(message, "选中的素材均已购买,未删除任何项目")
XCTAssertTrue(viewModel.isSelectionMode)
XCTAssertEqual(viewModel.selectedMaterialIds, [10])
}
}
/// 有线传输 ViewModel 测试。
@@ -1186,6 +1283,7 @@ final class TravelAlbumMockAPI: TravelAlbumServing {
var availableOrdersResponse: [TravelAlbumAvailableOrder] = []
var listResponse = TravelAlbumListResponse<TravelAlbum>()
var listError: Error?
var createResponse = TravelAlbumCreateResponse(id: 0)
var infoResponse = TravelAlbum()
var materialListResponses: [TravelAlbumListResponse<TravelAlbumMaterial>] = []
@@ -1241,7 +1339,8 @@ final class TravelAlbumMockAPI: TravelAlbumServing {
}
func list(page: Int, pageSize: Int) async throws -> TravelAlbumListResponse<TravelAlbum> {
listResponse
if let listError { throw listError }
return listResponse
}
func info(id: Int) async throws -> TravelAlbum {