2019-08-19 04:07:09 +02:00
|
|
|
import PromiseKit
|
|
|
|
|
|
|
|
@objc(LKGroupChatAPI)
|
|
|
|
public final class LokiGroupChatAPI : NSObject {
|
|
|
|
internal static let storage = OWSPrimaryStorage.shared()
|
|
|
|
|
|
|
|
@objc public static let serverURL = "https://chat.lokinet.org"
|
2019-08-20 07:34:59 +02:00
|
|
|
private static let batchCount = 8
|
2019-08-19 05:41:23 +02:00
|
|
|
@objc public static let publicChatMessageType = "network.loki.messenger.publicChat"
|
|
|
|
@objc public static let publicChatID = 1
|
2019-08-22 05:14:35 +02:00
|
|
|
private static let tokenCollection = "LokiGroupChatTokenCollection"
|
|
|
|
|
2019-08-23 08:14:19 +02:00
|
|
|
internal static var userDisplayName: String { return SSKEnvironment.shared.contactsManager.displayName(forPhoneIdentifier: userHexEncodedPublicKey) ?? "Anonymous" }
|
|
|
|
private static var userKeyPair: ECKeyPair { return OWSIdentityManager.shared().identityKeyPair()! }
|
|
|
|
private static var userHexEncodedPublicKey: String { return userKeyPair.hexEncodedPublicKey }
|
2019-08-22 04:34:24 +02:00
|
|
|
|
2019-08-20 07:34:59 +02:00
|
|
|
public enum Error : Swift.Error {
|
2019-08-22 04:34:24 +02:00
|
|
|
case tokenParsingFailed, tokenDecryptionFailed, messageParsingFailed
|
2019-08-20 07:54:12 +02:00
|
|
|
}
|
|
|
|
|
2019-08-23 08:14:19 +02:00
|
|
|
private static func getTokenFromServer() -> Promise<String> {
|
2019-08-20 07:54:12 +02:00
|
|
|
print("[Loki] Getting group chat auth token.")
|
2019-08-22 04:34:24 +02:00
|
|
|
let url = URL(string: "\(serverURL)/loki/v1/get_challenge?pubKey=\(userHexEncodedPublicKey)")!
|
|
|
|
let request = TSRequest(url: url)
|
2019-08-20 07:54:12 +02:00
|
|
|
return TSNetworkManager.shared().makePromise(request: request).map { $0.responseObject }.map { rawResponse in
|
2019-08-23 08:14:19 +02:00
|
|
|
guard let json = rawResponse as? JSON, let base64EncodedChallenge = json["cipherText64"] as? String, let base64EncodedServerPublicKey = json["serverPubKey64"] as? String,
|
|
|
|
let challenge = Data(base64Encoded: base64EncodedChallenge), var serverPublicKey = Data(base64Encoded: base64EncodedServerPublicKey) else {
|
2019-08-22 07:33:19 +02:00
|
|
|
throw Error.tokenParsingFailed
|
|
|
|
}
|
2019-08-23 08:14:19 +02:00
|
|
|
// Discard the "05" prefix if needed
|
|
|
|
if (serverPublicKey.count == 33) {
|
|
|
|
let hexEncodedServerPublicKey = serverPublicKey.hexadecimalString
|
|
|
|
serverPublicKey = Data.data(fromHex: hexEncodedServerPublicKey.substring(from: 2))!
|
2019-08-23 02:27:22 +02:00
|
|
|
}
|
2019-08-23 08:14:19 +02:00
|
|
|
// The challenge is prefixed by the 16 bit IV
|
|
|
|
guard let tokenAsData = try? DiffieHellman.decrypt(challenge, publicKey: serverPublicKey, privateKey: userKeyPair.privateKey),
|
|
|
|
let token = String(bytes: tokenAsData, encoding: .utf8), token.count > 0 else {
|
2019-08-22 07:33:19 +02:00
|
|
|
throw Error.tokenDecryptionFailed
|
|
|
|
}
|
|
|
|
return token
|
2019-08-20 07:54:12 +02:00
|
|
|
}
|
2019-08-20 07:34:59 +02:00
|
|
|
}
|
|
|
|
|
2019-08-22 05:14:35 +02:00
|
|
|
internal static func submitToken(_ token: String) -> Promise<String> {
|
2019-08-22 04:34:24 +02:00
|
|
|
print("[Loki] Submitting group chat auth token.")
|
|
|
|
let url = URL(string: "\(serverURL)/loki/v1/submit_challenge")!
|
|
|
|
let parameters = [ "pubKey" : userHexEncodedPublicKey, "token" : token ]
|
|
|
|
let request = TSRequest(url: url, method: "POST", parameters: parameters)
|
2019-08-22 05:14:35 +02:00
|
|
|
return TSNetworkManager.shared().makePromise(request: request).map { _ in token }
|
|
|
|
}
|
|
|
|
|
|
|
|
internal static func getToken() -> Promise<String> {
|
2019-08-23 08:14:19 +02:00
|
|
|
if let token = storage.dbReadConnection.string(forKey: serverURL, inCollection: tokenCollection), token.count > 0 {
|
|
|
|
return Promise.value(token)
|
|
|
|
} else {
|
|
|
|
return getTokenFromServer().then { submitToken($0) }.map { token -> String in
|
2019-08-22 05:14:35 +02:00
|
|
|
storage.dbReadWriteConnection.setObject(token, forKey: serverURL, inCollection: tokenCollection)
|
|
|
|
return token
|
|
|
|
}
|
|
|
|
}
|
2019-08-22 04:34:24 +02:00
|
|
|
}
|
|
|
|
|
2019-08-20 07:34:59 +02:00
|
|
|
public static func getMessages(for group: UInt) -> Promise<[LokiGroupMessage]> {
|
|
|
|
print("[Loki] Getting messages for group chat with ID: \(group).")
|
|
|
|
let queryParameters = "include_annotations=1&count=-\(batchCount)"
|
|
|
|
let url = URL(string: "\(serverURL)/channels/\(group)/messages?\(queryParameters)")!
|
2019-08-19 04:07:09 +02:00
|
|
|
let request = TSRequest(url: url)
|
|
|
|
return TSNetworkManager.shared().makePromise(request: request).map { $0.responseObject }.map { rawResponse in
|
|
|
|
guard let json = rawResponse as? JSON, let rawMessages = json["data"] as? [JSON] else {
|
2019-08-20 07:34:59 +02:00
|
|
|
print("[Loki] Couldn't parse messages for group chat with ID: \(group) from: \(rawResponse).")
|
|
|
|
throw Error.messageParsingFailed
|
2019-08-19 04:07:09 +02:00
|
|
|
}
|
|
|
|
return rawMessages.flatMap { message in
|
|
|
|
guard let annotations = message["annotations"] as? [JSON], let annotation = annotations.first, let value = annotation["value"] as? JSON,
|
2019-08-23 08:14:19 +02:00
|
|
|
let serverID = message["id"] as? UInt, let body = message["text"] as? String, let hexEncodedPublicKey = value["source"] as? String, let displayName = value["from"] as? String,
|
|
|
|
let timestamp = value["timestamp"] as? UInt64 else {
|
2019-08-20 07:34:59 +02:00
|
|
|
print("[Loki] Couldn't parse message for group chat with ID: \(group) from: \(message).")
|
2019-08-19 04:07:09 +02:00
|
|
|
return nil
|
|
|
|
}
|
2019-08-21 03:50:23 +02:00
|
|
|
guard hexEncodedPublicKey != userHexEncodedPublicKey else { return nil }
|
2019-08-19 04:07:09 +02:00
|
|
|
return LokiGroupMessage(serverID: serverID, hexEncodedPublicKey: hexEncodedPublicKey, displayName: displayName, body: body, type: publicChatMessageType, timestamp: timestamp)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-08-20 07:34:59 +02:00
|
|
|
public static func sendMessage(_ message: LokiGroupMessage, to group: UInt) -> Promise<LokiGroupMessage> {
|
2019-08-22 05:14:35 +02:00
|
|
|
return getToken().then { token -> Promise<LokiGroupMessage> in
|
|
|
|
print("[Loki] Sending message to group chat with ID: \(group).")
|
|
|
|
let url = URL(string: "\(serverURL)/channels/\(group)/messages")!
|
|
|
|
let parameters = message.toJSON()
|
|
|
|
let request = TSRequest(url: url, method: "POST", parameters: parameters)
|
|
|
|
request.allHTTPHeaderFields = [ "Content-Type" : "application/json", "Authorization" : "Bearer \(token)" ]
|
|
|
|
let displayName = userDisplayName
|
|
|
|
return TSNetworkManager.shared().makePromise(request: request).map { $0.responseObject }.map { rawResponse in
|
2019-08-23 08:14:19 +02:00
|
|
|
// ISO8601DateFormatter doesn't support milliseconds before iOS 11
|
2019-08-23 02:27:22 +02:00
|
|
|
let dateFormatter = DateFormatter()
|
|
|
|
dateFormatter.dateFormat = "yyyy-MM-dd'T'HH:mm:ss.SSSZ"
|
2019-08-23 08:14:19 +02:00
|
|
|
guard let json = rawResponse as? JSON, let message = json["data"] as? JSON, let serverID = message["id"] as? UInt, let body = message["text"] as? String,
|
|
|
|
let dateAsString = message["created_at"] as? String, let date = dateFormatter.date(from: dateAsString) else {
|
2019-08-22 05:14:35 +02:00
|
|
|
print("[Loki] Couldn't parse messages for group chat with ID: \(group) from: \(rawResponse).")
|
|
|
|
throw Error.messageParsingFailed
|
|
|
|
}
|
|
|
|
let timestamp = UInt64(date.timeIntervalSince1970) * 1000
|
|
|
|
return LokiGroupMessage(serverID: serverID, hexEncodedPublicKey: userHexEncodedPublicKey, displayName: displayName, body: body, type: publicChatMessageType, timestamp: timestamp)
|
2019-08-19 04:07:09 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-08-20 07:34:59 +02:00
|
|
|
@objc(getMessagesForGroup:)
|
|
|
|
public static func objc_getMessages(for group: UInt) -> AnyPromise {
|
|
|
|
return AnyPromise.from(getMessages(for: group))
|
2019-08-19 04:07:09 +02:00
|
|
|
}
|
|
|
|
|
2019-08-20 07:34:59 +02:00
|
|
|
@objc(sendMessage:toGroup:)
|
|
|
|
public static func objc_sendMessage(_ message: LokiGroupMessage, to group: UInt) -> AnyPromise {
|
|
|
|
return AnyPromise.from(sendMessage(message, to: group))
|
2019-08-19 04:07:09 +02:00
|
|
|
}
|
|
|
|
}
|