This commit is contained in:
fmodf 2024-08-18 01:21:15 +02:00
parent 6a5555a603
commit 062292bb11
2 changed files with 118 additions and 21 deletions

View file

@ -37,9 +37,16 @@ final class ClientMartinMessagesManager {
// Content type // Content type
var contentType: MessageContentType = .text var contentType: MessageContentType = .text
if message.hints.contains(.noStore) { if let oob = message.oob {
contentType = .attachment(.init(
type: oob.attachmentType,
localName: nil,
thumbnailName: nil,
remotePath: oob
))
} else if message.hints.contains(.noStore) {
contentType = .typing contentType = .typing
// for now just skip it // skip for now
return return
} }

View file

@ -45,12 +45,53 @@ extension ConversationStore {
} }
func sendMedia(_ items: [GalleryItem]) async { func sendMedia(_ items: [GalleryItem]) async {
print("media!", items) for item in items {
// guard !ids.isEmpty else { return } Task {
// let items = galleryItems.filter { ids.contains($0.id) } var message = Message.blank
// for item in items { message.from = roster.bareJid
// await client.uploadMedia(item.url) message.to = roster.contactBareJid
// }
switch item.type {
case .photo:
guard let asset = PHAsset.fetchAssets(withLocalIdentifiers: [item.id], options: nil).firstObject else { return }
guard let photo = try? await PHImageManager.default().getPhoto(for: asset) else { return }
guard let data = photo.jpegData(compressionQuality: 1.0) else { return }
let localName = "\(message.id)_\(UUID().uuidString).jpg"
let localUrl = Const.fileFolder.appendingPathComponent(localName)
try? data.write(to: localUrl)
message.contentType = .attachment(
Attachment(
type: .image,
localName: localName,
thumbnailName: nil,
remotePath: nil
)
)
try? await message.save()
case .video:
guard let asset = PHAsset.fetchAssets(withLocalIdentifiers: [item.id], options: nil).firstObject else { return }
guard let video = try? await PHImageManager.default().getVideo(for: asset) else { return }
// swiftlint:disable:next force_cast
let assetURL = video as! AVURLAsset
let url = assetURL.url
let localName = "\(message.id)_\(UUID().uuidString).mov"
let localUrl = Const.fileFolder.appendingPathComponent(localName)
try? FileManager.default.copyItem(at: url, to: localUrl)
message.contentType = .attachment(
Attachment(
type: .video,
localName: localName,
thumbnailName: nil,
remotePath: nil
)
)
try? await message.save()
}
await upload(message)
}
}
} }
func sendCaptured(_ data: Data, _ type: GalleryMediaType) async { func sendCaptured(_ data: Data, _ type: GalleryMediaType) async {
@ -108,19 +149,38 @@ extension ConversationStore {
} }
func sendDocuments(_ data: [Data], _ extensions: [String]) async { func sendDocuments(_ data: [Data], _ extensions: [String]) async {
// do { for (index, data) in data.enumerated() {
// let newMessageId = UUID().uuidString Task {
// let fileId = UUID().uuidString let newMessageId = UUID().uuidString
// let localName = "\(newMessageId)_\(fileId).\(ext)" let fileId = UUID().uuidString
// try await FileStore.shared.storeForUploading(data, localName) let localName = "\(newMessageId)_\(fileId).\(extensions[index])"
// let localUrl = Const.fileFolder.appendingPathComponent(localName)
// } catch { do {
// print("error", error) try data.write(to: localUrl)
// } } catch {
print("documents!", data, extensions) print("FileProcessing: Error writing document: \(error)")
// return
// }
//
var message = Message.blank
message.from = roster.bareJid
message.to = roster.contactBareJid
message.contentType = .attachment(
Attachment(
type: localName.attachmentType,
localName: localName,
thumbnailName: nil,
remotePath: nil
)
)
do {
try await message.save()
await upload(message)
} catch {
print("FileProcessing: Error saving document: \(error)")
}
}
}
} }
func sendContact(_ jidStr: String) async { func sendContact(_ jidStr: String) async {
@ -159,6 +219,36 @@ extension ConversationStore {
try? await message.setStatus(.error) try? await message.setStatus(.error)
} }
} }
func downloadAttachment(_ message: Message) async {
guard case .attachment(let attachment) = message.contentType else {
return
}
guard let remotePath = attachment.remotePath, let remoteUrl = URL(string: remotePath) else {
return
}
do {
let localName = "\(message.id)_\(UUID().uuidString).\(remoteUrl.lastPathComponent)"
let localUrl = Const.fileFolder.appendingPathComponent(localName)
// Download the file
let (tempUrl, _) = try await URLSession.shared.download(from: remoteUrl)
try FileManager.default.moveItem(at: tempUrl, to: localUrl)
var message = message
message.contentType = .attachment(
Attachment(
type: attachment.type,
localName: localName,
thumbnailName: attachment.thumbnailName,
remotePath: remotePath
)
)
try await message.save()
} catch {
logIt(.error, "Can't download attachment: \(error)")
}
}
} }
extension ConversationStore { extension ConversationStore {