oxen-io/session-ios
1
0
Fork 0
mirror of https://github.com/oxen-io/session-ios.git synced 2023-12-13 21:30:14 +01:00
Code Issues Projects Releases Wiki Activity

onion routing attachment download for open groups

Browse source
This commit is contained in:
Ryan ZHAO 2020-10-14 14:07:18 +11:00
parent 76b6afe0e3
commit bdd178b227
1 changed files with 23 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
SignalServiceKit/src/Loki/API/FileServerAPI.swift
View file

@ -66,24 +66,38 @@ public final class FileServerAPI : DotNetAPI {
public static func downloadAttachment(_ downloadURL: String) -> Promise<Data> {
var error: NSError?
var url = downloadURL
if downloadURL.contains(fileStaticServer) {
url = downloadURL.replacingOccurrences(of: fileStaticServer, with: "\(server)/loki/v1")
let url = URL(string: downloadURL)!
var host = "https://\(url.host!)"
var actualDownloadURL = downloadURL
if fileStaticServer.contains(host) {
actualDownloadURL = downloadURL.replacingOccurrences(of: fileStaticServer, with: "\(server)/loki/v1")
host = server
} else {
actualDownloadURL = downloadURL.replacingOccurrences(of: host, with: "\(host)/loki/v1")
}
let request = AFHTTPRequestSerializer().request(withMethod: "GET", urlString: url, parameters: nil, error: &error)
let request = AFHTTPRequestSerializer().request(withMethod: "GET", urlString: actualDownloadURL, parameters: nil, error: &error)
if let error = error {
print("[Loki] Couldn't download attachment due to error: \(error).")
return Promise(error: error)
}
return OnionRequestAPI.sendOnionRequest(request, to: server, using: fileServerPublicKey, isJSONRequired: false).map2 { json in
guard let body = json["body"] as? JSON, let dataArray = body["data"] as? [UInt8] else {
print("[Loki] Couldn't download attachment.")
return Data()
return getServerPublicKey(for: host).then2 { serverPublicKey in
OnionRequestAPI.sendOnionRequest(request, to: host, using: serverPublicKey, isJSONRequired: false).map2 { json in
guard let body = json["body"] as? JSON, let dataArray = body["data"] as? [UInt8] else {
print("[Loki] Couldn't download attachment.")
return Data()
}
return Data(dataArray)
}
return Data(dataArray)
}
}
public static func getServerPublicKey(for host: String) -> Promise<String> {
if server.contains(host) {
return Promise.value(fileServerPublicKey)
}
return PublicChatAPI.getOpenGroupServerPublicKey(for: host)
}
// MARK: Open Group Server Public Key
public static func getPublicKey(for openGroupServer: String) -> Promise<String> {
let url = URL(string: "\(server)/loki/v1/getOpenGroupKey/\(URL(string: openGroupServer)!.host!)")!