Swiftで動的な要素を含んだViewで縦スクロールする方法

Question

###前提・実現したいこと
RxSwiftとMVVMを使ってiOSアプリの開発を行っております。只今、アプリのホーム的な画面を作成しております。イメージとしては、クックパッドの起動時に表示される画面のような、縦スクロールができ、行ごとに動的に要素が変更されるものを作ろうとしています。

この画面はおそらくですが、TableViewを使い、Cellごとに異なるCustomCellで描画しているのではないかと考えています。ただ、そこから先の実装方法がわかりません。

###わからないこと
下記コードのように、ViewModelのitemsをTableViewにbindして、TableViewを描画すると思っています。

// ViewController

viewModel.items.bind(to: tableView.rx.items) { tableView, cell, element in
   // ここでelementに応じてCustomCellをつくってreturnする？
}

この際、itemsの型は構造体の配列になるかと思うのですが、各Cell毎に全く異なる要素を表示するため、同一の構造体にすることができません。無理やる方法としては、下記の様な構造体を用意し、itemsを[Item]として、上記コードのクロージャ内でtypeで分岐して、bodyをキャストして扱う方法はありそうな気がしています。

struct Item {
  let type: ItemType
  let body: Any
}

しかし、このAnyにダウンキャストしている時点で、Swift的にはあまり良いコードではないと考えております。どのように実装すればよいのでしょうか？ご教示いただければ幸いです。

Answer 1

色々方法はありそうですがプロトコルを使えば

protocol CellProvider {
    func makeCell(for tableView: UITableView) -> UITableViewCell
}


struct BodyA: CellProvider {
    func makeCell(for tableView: UITableView) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "CellA")!
        // セルの設定
        return cell
    }
}
struct BodyB: CellProvider { ... }  // 表示内容ごとに作る

とすれば itemsを[CellPrvider]とすることで

viewModel.items.bind(to: tableView.rx.items) { tableView, row, element in
   
    return element.makeCell(for: tableView)
}

でいけるのではないでしょうか。

すでにデータがある場合は extensionでプロトコルを採用すればよいです。

動作確認はしてないです。

Comments

[seiya_orz]

ご回答ありがとうございました。私も動作を確認してはいませんが、この方法で実装するイメージが付きました。