Custom Dynamic Model in Swift

Custom Dynamic Model in Swift.swift

enum RowType: String, CaseIterable {
    case Profile
    case Info
    case Rewards
}

// big array model to store all detail
struct DetailInfoModel <T> {
    var headerTitle : String?
    var value        : T?
    var rowType      : RowType
}

// properties of each detail
struct ContactInfo {
    var profile : String?
    var name : String?
    var icone : String?
    var title : String?
}

DetailViewController.swift

import UIKit

class DetailViewController: UIViewController, UITableViewDelegate, UITableViewDataSource {
    
    let tableView = UITableView(frame: .zero, style: .grouped)
    
    let secondVM = SecondVM()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        view.backgroundColor = .white
        secondVM.getData()
        tableView.delegate = self
        tableView.dataSource = self
        tableView.register(ProfileCell.self, forCellReuseIdentifier: "ProfileCell")
        tableView.register(DetailCell.self, forCellReuseIdentifier: "DetailCell")
        
        tableView.translatesAutoresizingMaskIntoConstraints = false
        view.addSubview(tableView)
        
        NSLayoutConstraint.activate([
            tableView.topAnchor.constraint(equalTo: view.topAnchor),
            tableView.bottomAnchor.constraint(equalTo: view.bottomAnchor),
            tableView.leadingAnchor.constraint(equalTo: view.leadingAnchor),
            tableView.trailingAnchor.constraint(equalTo: view.trailingAnchor)
        ])
    }
    
    func numberOfSections(in tableView: UITableView) -> Int {
        return secondVM.detailData.count // Profile section + other sections
    }
    
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        if section == 0 {
            return 1
        } else {
            let details = secondVM.detailData[section].value as! [ContactInfo]
            return details.count
        }
    }
    
    func tableView(_ tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat {
        if section == 0 {
            return 0
        } else {
            return secondVM.detailData[section].headerTitle?.isEmpty == true ? 0 : 40
        }
    }
    
    func tableView(_ tableView: UITableView, viewForHeaderInSection section: Int) -> UIView? {
        guard section > 0 else { return nil }
        
        let headerView = UIView()
        headerView.backgroundColor = .clear
        
        let headerLabel = UILabel()
//        headerLabel.text = sections[section - 1].0
        headerLabel.text = secondVM.detailData[section].headerTitle
        headerLabel.font = UIFont.systemFont(ofSize: 18, weight: .bold)
        headerLabel.translatesAutoresizingMaskIntoConstraints = false
        headerView.addSubview(headerLabel)
        
        NSLayoutConstraint.activate([
            headerLabel.leadingAnchor.constraint(equalTo: headerView.leadingAnchor, constant: 16),
            headerLabel.trailingAnchor.constraint(equalTo: headerView.trailingAnchor, constant: -16),
            headerLabel.topAnchor.constraint(equalTo: headerView.topAnchor),
            headerLabel.bottomAnchor.constraint(equalTo: headerView.bottomAnchor)
        ])
        
        return headerView
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        if indexPath.section == 0 {
            let cell = tableView.dequeueReusableCell(withIdentifier: "ProfileCell", for: indexPath) as! ProfileCell
            let profile = secondVM.detailData[indexPath.section].value as! [ContactInfo]
            let value = profile[indexPath.row]
            
            cell.configure(name: value.name ?? "", jobTitle: value.title ?? "", imageName: value.profile ?? "", editText: value.icone ?? "")
            
            return cell
        } else {
            let cell = tableView.dequeueReusableCell(withIdentifier: "DetailCell", for: indexPath) as! DetailCell
            let details = secondVM.detailData[indexPath.section].value as! [ContactInfo]
            
            cell.configure(title: details[indexPath.row].title ?? "", subtitle: "", imageName: details[indexPath.row].icone ?? "" )
            return cell
        }
    }
    
    func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
        fatalError("Crash was triggered")
    }
    
}

class ProfileCell: UITableViewCell {
    let profileImageView = UIImageView()
    let nameLabel = UILabel()
    let jobTitleLabel = UILabel()
    let editButton = UIButton(type: .system)
    
    override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
        super.init(style: style, reuseIdentifier: reuseIdentifier)
        setupUI()
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    func setupUI() {
        profileImageView.layer.cornerRadius = 30
        profileImageView.clipsToBounds = true
        
        nameLabel.font = UIFont.boldSystemFont(ofSize: 18)
        jobTitleLabel.font = UIFont.systemFont(ofSize: 14)
        jobTitleLabel.textColor = .gray
        
        let textStackView = UIStackView(arrangedSubviews: [nameLabel, jobTitleLabel])
        textStackView.axis = .vertical
        textStackView.spacing = 5
        
        let mainStackView = UIStackView(arrangedSubviews: [profileImageView, textStackView, editButton])
        mainStackView.axis = .horizontal
        mainStackView.spacing = 10
        mainStackView.translatesAutoresizingMaskIntoConstraints = false
        contentView.addSubview(mainStackView)
        
        NSLayoutConstraint.activate([
            profileImageView.widthAnchor.constraint(equalToConstant: 60),
            profileImageView.heightAnchor.constraint(equalToConstant: 60),
            
            mainStackView.topAnchor.constraint(equalTo: contentView.topAnchor, constant: 10),
            mainStackView.leadingAnchor.constraint(equalTo: contentView.leadingAnchor, constant: 20),
            mainStackView.trailingAnchor.constraint(equalTo: contentView.trailingAnchor, constant: -20),
            mainStackView.bottomAnchor.constraint(equalTo: contentView.bottomAnchor, constant: -10)
        ])
    }
    
    func configure(name: String, jobTitle: String, imageName: String, editText: String) {
        profileImageView.image = UIImage(named: imageName) // Replace with actual image
        nameLabel.text = name
        jobTitleLabel.text = jobTitle
        editButton.setTitle(editText, for: .normal)
    }
}

class DetailCell: UITableViewCell {
    let iconImageView = UIImageView()
    let titleLabel = UILabel()
    let subtitleLabel = UILabel()
    
    override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
        super.init(style: style, reuseIdentifier: reuseIdentifier)
        setupUI()
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    func setupUI() {
        titleLabel.font = UIFont.systemFont(ofSize: 16)
        subtitleLabel.font = UIFont.systemFont(ofSize: 14)
        subtitleLabel.textColor = .gray
        
        let textStackView = UIStackView(arrangedSubviews: [titleLabel, subtitleLabel])
        textStackView.axis = .vertical
        textStackView.spacing = 5
        
        let mainStackView = UIStackView(arrangedSubviews: [iconImageView, textStackView])
        mainStackView.axis = .horizontal
        mainStackView.spacing = 10
        mainStackView.translatesAutoresizingMaskIntoConstraints = false
        contentView.addSubview(mainStackView)
        
        NSLayoutConstraint.activate([
            iconImageView.widthAnchor.constraint(equalToConstant: 24),
            iconImageView.heightAnchor.constraint(equalToConstant: 24),
            
            mainStackView.topAnchor.constraint(equalTo: contentView.topAnchor, constant: 10),
            mainStackView.leadingAnchor.constraint(equalTo: contentView.leadingAnchor, constant: 20),
            mainStackView.trailingAnchor.constraint(equalTo: contentView.trailingAnchor, constant: -20),
            mainStackView.bottomAnchor.constraint(equalTo: contentView.bottomAnchor, constant: -10)
        ])
    }
    
    func configure(title: String, subtitle: String, imageName: String) {
        iconImageView.image = UIImage(systemName: imageName) // Use system icons
        iconImageView.tintColor = .blue
        titleLabel.text = title
        subtitleLabel.text = subtitle
    }
}

DetailViewModel.swift

class DetailViewModel {
    
    var detailData : [DetailInfoModel<Any>] = []
    
    var contactData : [ContactInfo] = []
    
    func getData() {
        
        
        let profileSection = [
            ContactInfo(profile: "MyPic",name: "Yuth Fight",icone: "Edit", title: "Product Designer"),
        ]
        
        
        contactData = [
            ContactInfo(icone: "person.fill", title: "Username"),
            ContactInfo(icone: "heart.fill", title: "Health")
        ]
        
        let rewards = [
            ContactInfo(icone: "creditcard.fill", title: "My Reward"),
            ContactInfo(icone: "map.fill", title: "Prizes")
        ]
        
        let appVersion = [
            ContactInfo(icone: "creditcard.fill", title: "My Reward"),
        ]
        
        detailData = [
            DetailInfoModel(headerTitle: "",value: profileSection,rowType: .Profile),
            DetailInfoModel(headerTitle: "",value: contactData,rowType: .Info),
            DetailInfoModel(headerTitle: "Rewards",value: rewards,rowType: .Rewards),
        ]
    }
    
}

** The result: