/*
* iOS-specific ViewModel that lives in iosMain of common module
*/
class KSearchViewModel(
  exitDb: KExitDatabase
) : CoroutineScope {
  private val job = Job()
  override val coroutineContext: CoroutineContext = dispatcher() + job

  @InternalCoroutinesApi
  private val searchStateMachine = SearchStateMachine(this, exitDb)
  val cStateFlow = searchStateMachine.viewState.wrap()

  fun dispatchAction(action: Search.Action) {
    searchStateMachine.dispatchAction.invoke(action)
  }

  @UseExperimental(ExperimentalCoroutinesApi::class)
  fun clear() {
    job.cancel()
  }
}

fun <T> ConflatedBroadcastChannel<T>.wrap(): CFlow<T> = CFlow(asFlow())

fun <T> Flow<T>.wrap(): CFlow<T> = CFlow(this)

class CFlow<T>(private val origin: Flow<T>) : Flow<T> by origin {
  fun watch(block: (T) -> Unit): Closeable {
    val job = Job()

    onEach {
      block(it)
    }.launchIn(CoroutineScope(Dispatchers.Main + job))

    return object : Closeable {
      override fun close() {
        job.cancel()
      }
    }
  }
}