//
//  AlbumTracksDataSource.swift
//  Persephone
//
//  Created by Daniel Barber on 2019/5/19.
//  Copyright © 2019 Dan Barber. All rights reserved.
//

import UIKit

class AlbumTracksDataSource: NSObject, UITableViewDataSource {
  var albumSongs: [AlbumSongItem] = []

  func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
    return albumSongs.count
  }
  
  func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    guard let albumSongCell = tableView.dequeueReusableCell(withIdentifier: "albumSongCell") as? AlbumSongCell
      else { return AlbumSongCell() }
    
    albumSongCell.setSongItem(songItem: albumSongs[indexPath.row])

    return albumSongCell
  }

  func setAlbumSongs(_ songs: [Song]) {
    var disc: String? = ""

    songs.forEach { song in
      if song.disc != disc && song.disc != "0" {
        disc = song.disc
        albumSongs.append(AlbumSongItem(disc: song.disc))
      }

      albumSongs.append(AlbumSongItem(song: song))
    }
  }
}