Autoresize UICollectionView height para ajustarlo a su tamaño de contenido

Tengo un UICollectionView, un botón que crea una nueva celda en la vista de colección. Y quiero que UICollectionView ajuste su tamaño según su tamaño de contenido (cuando hay una o dos celdas, entonces UICollectionView es corto, si hay muchas celdas UICollectionView es lo suficientemente grande).

Sé cómo obtener el tamaño del contenido:

collectionView.collectionViewLayout.collectionViewContentSize

Pero no tengo idea de dónde usar este valor. Apreciaría si alguien me ayudara a averiguar cómo hacer que UICollectionView ajuste automáticamente su altura.

UPD: publiqué en GitHub un proyecto de demostración que describe el problema:https://github.com/avokin/PostViewer

Respuestas a la pregunta(4)

Su respuesta a la pregunta