session-ios/SessionMessagingKit/Messages/Control Messages/MessageRequestResponse.swift

// Copyright © 2022 Rangeproof Pty Ltd. All rights reserved.

import Foundation
import GRDB
import SessionUtilitiesKit

public final class MessageRequestResponse: ControlMessage {
    private enum CodingKeys: String, CodingKey {
        case isApproved
        case profile
    }
    
    public var isApproved: Bool
    public var profile: VisibleMessage.VMProfile?
    
    // MARK: - Initialization
    
    public init(
        isApproved: Bool,
        profile: VisibleMessage.VMProfile? = nil,   // Added when sending via the `MessageWithProfile` protocol
        sentTimestampMs: UInt64? = nil
    ) {
        self.isApproved = isApproved
        self.profile = profile
        
        super.init(
            sentTimestampMs: sentTimestampMs
        )
    }
    
    // MARK: - Codable
    
    required init(from decoder: Decoder) throws {
        let container: KeyedDecodingContainer<CodingKeys> = try decoder.container(keyedBy: CodingKeys.self)
        
        isApproved = try container.decode(Bool.self, forKey: .isApproved)
        profile = try? container.decode(VisibleMessage.VMProfile.self, forKey: .profile)
        
        try super.init(from: decoder)
    }
    
    public override func encode(to encoder: Encoder) throws {
        try super.encode(to: encoder)
        
        var container: KeyedEncodingContainer<CodingKeys> = encoder.container(keyedBy: CodingKeys.self)
        
        try container.encodeIfPresent(isApproved, forKey: .isApproved)
        try container.encodeIfPresent(profile, forKey: .profile)
    }
    
    // MARK: - Proto Conversion

    public override class func fromProto(_ proto: SNProtoContent, sender: String, using dependencies: Dependencies) -> MessageRequestResponse? {
        guard let messageRequestResponseProto = proto.messageRequestResponse else { return nil }
        
        return MessageRequestResponse(
            isApproved: messageRequestResponseProto.isApproved,
            profile: VisibleMessage.VMProfile.fromProto(messageRequestResponseProto)
        )
    }

    public override func toProto(_ db: Database, threadId: String) -> SNProtoContent? {
        let messageRequestResponseProto: SNProtoMessageRequestResponse.SNProtoMessageRequestResponseBuilder
        
        // Profile
        if let profile = profile, let profileProto: SNProtoMessageRequestResponse = profile.toProto(isApproved: isApproved) {
            messageRequestResponseProto = profileProto.asBuilder()
        }
        else {
            messageRequestResponseProto = SNProtoMessageRequestResponse.builder(isApproved: isApproved)
        }
        
        let contentProto = SNProtoContent.builder()
        
        do {
            contentProto.setMessageRequestResponse(try messageRequestResponseProto.build())
            return try contentProto.build()
        } catch {
            SNLog("Couldn't construct unsend request proto from: \(self).")
            return nil
        }
    }
    
    // MARK: - Description
    
    public var description: String {
        """
        MessageRequestResponse(
            isApproved: \(isApproved),
            profile: \(profile?.description ?? "null")
        )
        """
    }
}