Refactor

gistfile1.txt

private fun loggedInMenuUi(menu: Menu, rootView: View, checkoutCartStream: Stream<Bla>, startActivity: () -> Unit) {
  val alertMenuItem = menu.findItem(R.id.activity_main_alerts_menu_item)

  val rootView = (alertMenuItem.actionView as FrameLayout).apply {
    setOnClickListener {
      startActivity()
    }
  }

  val redCircle: FrameLayout = rootView.findViewById(R.id.view_alert_red_circle)

  val countTextView: TextView = rootView.findViewById(R.id.view_alert_count_textview)
  
  suspend fun updateItemCount(itemsInCart: List<Item>) {
      withContext(Dispatchers.Main) {
          val itemsInCartCount = itemsInCart.size
          if (itemsInCartCount == 0) {
              redCircle.visibility = View.GONE
          } else {
              redCircle.visibility = View.VISIBLE
              countTextView.text = itemsInCartCount.toString()
          }
      }
  }
  launch {
    checkoutCart
        .itemsInCartStream()
        .consumeEach { itemsInCart ->
            updateItemCount(itemsInCart)
        }
  }
}

Thanks, used here: https://github.com/handstandsam/ShoppingApp/pull/14/files