Signal-iOS/Signal/Calls/CallLinkManager.swift
2024-06-17 12:19:00 -05:00

108 lines
4.5 KiB
Swift

//
// Copyright 2024 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//
import Foundation
import LibSignalClient
import SignalRingRTC
import SignalServiceKit
protocol CallLinkManager {
func createCallLink(rootKey: CallLinkRootKey) async throws -> CallLinkState
}
class CallLinkManagerImpl: CallLinkManager {
private let networkManager: NetworkManager
private let serverParams: GenericServerPublicParams
private let sfuClient: SignalRingRTC.SFUClient
// Even though we never use this, we need to retain it to ensure
// `sfuClient` continues to work properly.
private let sfuClientHttpClient: AnyObject
private let tsAccountManager: any TSAccountManager
init(
networkManager: NetworkManager,
serverParams: GenericServerPublicParams,
tsAccountManager: any TSAccountManager
) {
self.networkManager = networkManager
self.serverParams = serverParams
let httpClient = CallHTTPClient()
self.sfuClient = SignalRingRTC.SFUClient(httpClient: httpClient.ringRtcHttpClient)
self.sfuClientHttpClient = httpClient
self.tsAccountManager = tsAccountManager
}
// MARK: - Create Call Link
private struct CallLinkCreateAuthResponse: Decodable {
var credential: Data
}
private func fetchCreateCredential(for roomId: Data, localAci: Aci) async throws -> CreateCallLinkCredential {
let credentialRequestContext = CreateCallLinkCredentialRequestContext.forRoomId(roomId)
let httpRequest = TSRequest(
url: URL(string: "v1/call-link/create-auth")!,
method: "POST",
parameters: [
"createCallLinkCredentialRequest": credentialRequestContext.getRequest().serialize().asData.base64EncodedString()
]
)
let httpResult = try await self.networkManager.makePromise(request: httpRequest, canUseWebSocket: true).awaitable()
guard httpResult.responseStatusCode == 200, let responseBodyData = httpResult.responseBodyData else {
throw OWSGenericError("Couldn't handle successful result from the server.")
}
let httpResponse = try JSONDecoder().decode(CallLinkCreateAuthResponse.self, from: responseBodyData)
let credentialResponse = try CreateCallLinkCredentialResponse(contents: [UInt8](httpResponse.credential))
return try credentialRequestContext.receive(credentialResponse, userId: localAci, params: self.serverParams)
}
func createCallLink(rootKey: CallLinkRootKey) async throws -> CallLinkState {
let roomId = rootKey.deriveRoomId()
let localIdentifiers = self.tsAccountManager.localIdentifiersWithMaybeSneakyTransaction!
let sfuUrl = DebugFlags.callingUseTestSFU.get() ? TSConstants.sfuTestURL : TSConstants.sfuURL
let secretParams = CallLinkSecretParams.deriveFromRootKey(rootKey.bytes)
let createCredential = try await fetchCreateCredential(for: roomId, localAci: localIdentifiers.aci)
let createCredentialPresentation = createCredential.present(roomId: roomId, userId: localIdentifiers.aci, serverParams: self.serverParams, callLinkParams: secretParams)
let publicParams = secretParams.getPublicParams()
let adminPasskey = CallLinkRootKey.generateAdminPasskey()
return CallLinkState(try await self.sfuClient.createCallLink(
sfuUrl: sfuUrl,
createCredentialPresentation: createCredentialPresentation.serialize(),
linkRootKey: rootKey,
adminPasskey: adminPasskey,
callLinkPublicParams: publicParams.serialize()
).unwrap())
}
func readCallLink(
_ rootKey: CallLinkRootKey,
authCredential: SignalServiceKit.CallLinkAuthCredential
) async throws -> CallLinkState {
let sfuUrl = DebugFlags.callingUseTestSFU.get() ? TSConstants.sfuTestURL : TSConstants.sfuURL
let secretParams = CallLinkSecretParams.deriveFromRootKey(rootKey.bytes)
let authCredentialPresentation = authCredential.present(callLinkParams: secretParams)
return CallLinkState(try await self.sfuClient.readCallLink(
sfuUrl: sfuUrl,
authCredentialPresentation: authCredentialPresentation.serialize(),
linkRootKey: rootKey
).unwrap())
}
}
private struct SFUError: Error {
let rawValue: UInt16
}
private extension SFUResult {
func unwrap() throws -> Value {
switch self {
case .success(let value):
return value
case .failure(let errorCode):
throw SFUError(rawValue: errorCode)
}
}
}