element-ios/RiotSwiftUI/Modules/UserSessions/UserSessionDetails/UserSessionDetailsViewModel.swift

//
// Copyright 2021 New Vector Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

import Foundation

typealias UserSessionDetailsViewModelType = StateStoreViewModel<UserSessionDetailsViewState, UserSessionDetailsViewAction>

class UserSessionDetailsViewModel: UserSessionDetailsViewModelType, UserSessionDetailsViewModelProtocol {
    var completion: ((UserSessionDetailsViewModelResult) -> Void)?
    
    init(session: UserSessionInfo) {
        super.init(initialViewState: UserSessionDetailsViewState(sections: []))
        updateViewState(session: session)
    }
    
    // MARK: - Public
    
    // MARK: - Private
    
    private func updateViewState(session: UserSessionInfo) {
        var sections = [UserSessionDetailsSectionViewData]()
        
        sections.append(sessionSection(session: session))
        
        if let deviceSection = deviceSection(session: session) {
            sections.append(deviceSection)
        }
        
        state = UserSessionDetailsViewState(sections: sections)
    }
    
    private func sessionSection(session: UserSessionInfo) -> UserSessionDetailsSectionViewData {
        var sessionItems = [UserSessionDetailsSectionItemViewData]()
        
        if let sessionName = session.name {
            sessionItems.append(UserSessionDetailsSectionItemViewData(title: VectorL10n.userSessionDetailsSessionName,
                                                                      value: sessionName))
        }
        
        sessionItems.append(UserSessionDetailsSectionItemViewData(title: VectorL10n.keyVerificationManuallyVerifyDeviceIdTitle,
                                                                  value: session.id))
        
        return UserSessionDetailsSectionViewData(header: VectorL10n.userSessionDetailsSessionSectionHeader.uppercased(),
                                                 footer: VectorL10n.userSessionDetailsSessionSectionFooter,
                                                 items: sessionItems)
    }
    
    private func deviceSection(session: UserSessionInfo) -> UserSessionDetailsSectionViewData? {
        var deviceSectionItems = [UserSessionDetailsSectionItemViewData]()
        if let lastSeenIP = session.lastSeenIP {
            deviceSectionItems.append(UserSessionDetailsSectionItemViewData(title: VectorL10n.userSessionDetailsDeviceIpAddress,
                                                                            value: lastSeenIP))
        }
        if deviceSectionItems.count > 0 {
            return UserSessionDetailsSectionViewData(header: VectorL10n.userSessionDetailsDeviceSectionHeader.uppercased(),
                                                     footer: nil,
                                                     items: deviceSectionItems)
        }
        return nil
    }
}