LibraryViewModel.swift

LibraryViewModel.swift

class LibraryViewModel: CollectionViewModel<NoteCell> {
    init(collectionView: UICollectionView) {
        super.init(collectionView: collectionView, cellReuseIdentifier: "NoteCell")
    }
}

// MARK: - UICollectionViewDelegate Implementation

extension LibraryViewModel: UICollectionViewDelegate {
    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        remove([items.value[indexPath.item]])
    }
}