fix: 素材上传

This commit is contained in:
Junhui Chen 2025-09-06 17:17:31 +08:00
parent 6eea4cf717
commit 2b300d29fe
2 changed files with 107 additions and 49 deletions

View File

@ -283,6 +283,10 @@ struct UserInfo: View {
// MARK: -
extension UserInfo {
private struct MaterialSubmitResponse: Codable {
let code: Int
let data: [String]?
}
private struct GenerateFileInfo: Codable {
let id: String
let fileName: String?
@ -315,44 +319,68 @@ extension UserInfo {
}
private func generateFirstBlindBox(with fileId: String) {
let params: [String: Any] = [
"box_type": "First",
"material_ids": [fileId]
]
// 1)
let materialPayload: [[String: String]] = [[
"file_id": fileId,
"preview_file_id": fileId
]]
NetworkService.shared.postWithToken(
path: "/blind_box/generate",
parameters: params
) { (result: Result<GenerateResponse, NetworkError>) in
path: "/material",
parameters: materialPayload
) { (matResult: Result<MaterialSubmitResponse, NetworkError>) in
DispatchQueue.main.async {
switch result {
case .success(let response):
guard response.code == 0, let urlStr = response.data?.resultFile?.url, let url = URL(string: urlStr) else {
self.errorMessage = "Create first blind box failed: invalid response"
switch matResult {
case .success(let matResp):
guard matResp.code == 0, let materialIds = matResp.data, let materialId = materialIds.first else {
self.errorMessage = "Submit material failed"
self.showError = true
return
}
//
URLSession.shared.dataTask(with: url) { data, _, error in
if let error = error {
DispatchQueue.main.async {
self.errorMessage = "Load result image failed: \(error.localizedDescription)"
self.showError = true
}
return
}
guard let data = data, let image = UIImage(data: data) else {
DispatchQueue.main.async {
self.errorMessage = "Invalid image data"
self.showError = true
}
return
}
// 2) material_id
let params: [String: Any] = [
"box_type": "First",
"material_ids": [materialId]
]
NetworkService.shared.postWithToken(
path: "/blind_box/generate",
parameters: params
) { (result: Result<GenerateResponse, NetworkError>) in
DispatchQueue.main.async {
Router.shared.navigate(to: .blindOutcome(media: .image(image), time: nil, description: nil))
switch result {
case .success(let response):
guard response.code == 0, let urlStr = response.data?.resultFile?.url, let url = URL(string: urlStr) else {
self.errorMessage = "Create first blind box failed: invalid response"
self.showError = true
return
}
//
URLSession.shared.dataTask(with: url) { data, _, error in
if let error = error {
DispatchQueue.main.async {
self.errorMessage = "Load result image failed: \(error.localizedDescription)"
self.showError = true
}
return
}
guard let data = data, let image = UIImage(data: data) else {
DispatchQueue.main.async {
self.errorMessage = "Invalid image data"
self.showError = true
}
return
}
DispatchQueue.main.async {
Router.shared.navigate(to: .blindOutcome(media: .image(image), time: nil, description: nil))
}
}.resume()
case .failure(let error):
self.errorMessage = "Create first blind box failed: \(error.localizedDescription)"
self.showError = true
}
}
}.resume()
}
case .failure(let error):
self.errorMessage = "Create first blind box failed: \(error.localizedDescription)"
self.errorMessage = "Submit material failed: \(error.localizedDescription)"
self.showError = true
}
}

View File

@ -49,12 +49,19 @@ private struct QueryResponse: Codable {
let data: GenerateData?
}
private struct MaterialSubmitResponse: Codable {
let code: Int
let data: [String]?
}
extension MediaUploadView {
private func startPollingSecondBox(id: Int64) {
pollingTimer?.invalidate()
// 2
pollingTimer = Timer.scheduledTimer(withTimeInterval: 2.0, repeats: true) { _ in
querySecondBox(id: id)
DispatchQueue.main.async {
querySecondBox(id: id)
}
}
//
querySecondBox(id: id)
@ -422,31 +429,54 @@ struct MediaUploadView: View {
print("⚠️ 没有可用的文件ID")
return
}
// file_id
let materialIds: [String] = uploadResults.map { $0.value.fileId }
//
let params: [String: Any] = [
"box_type": "Second",
"material_ids": materialIds
]
// 1) ID material_ids
let materialPayload: [[String: String]] = uploadResults.map { (_, result) in
[
"file_id": result.fileId,
"preview_file_id": result.thumbnailId ?? result.fileId
]
}
isGeneratingSecond = true
NetworkService.shared.postWithToken(
path: "/blind_box/generate",
parameters: params
) { (result: Result<GenerateResponse, NetworkError>) in
path: "/material",
parameters: materialPayload
) { (matResult: Result<MaterialSubmitResponse, NetworkError>) in
DispatchQueue.main.async {
switch result {
case .success(let response):
guard response.code == 0, let id = response.data?.id else {
print("创建第二个盲盒失败:响应无效")
switch matResult {
case .success(let matResp):
guard matResp.code == 0, let materialIds = matResp.data, !materialIds.isEmpty else {
print("素材提交失败:响应无效或为空")
isGeneratingSecond = false
return
}
generatedSecondBoxId = id
//
startPollingSecondBox(id: id)
// 2) material_ids
let params: [String: Any] = [
"box_type": "Second",
"material_ids": materialIds
]
NetworkService.shared.postWithToken(
path: "/blind_box/generate",
parameters: params
) { (result: Result<GenerateResponse, NetworkError>) in
DispatchQueue.main.async {
switch result {
case .success(let response):
guard response.code == 0, let id = response.data?.id else {
print("❌ 创建第二个盲盒失败:响应无效")
isGeneratingSecond = false
return
}
generatedSecondBoxId = id
//
startPollingSecondBox(id: id)
case .failure(let error):
print("❌ 创建第二个盲盒失败: \(error.localizedDescription)")
isGeneratingSecond = false
}
}
}
case .failure(let error):
print("❌ 创建第二个盲盒失败: \(error.localizedDescription)")
print("素材提交失败: \(error.localizedDescription)")
isGeneratingSecond = false
}
}