flow.kt

  private fun generateIosFlow(
    declaration: KSDeclaration,
  ): FunSpec? {
    val returnType = when (declaration) {
      is KSFunctionDeclaration -> {
        declaration.returnType
      }
      is KSPropertyDeclaration -> {
        declaration.getter?.returnType
      }
      else -> error("Unexpected declaration=$declaration")
    } ?: return null
    if (!declaration.isPublic()) return null
    if (returnType.toString() != "Flow") return null
    val classDeclaration = declaration.closestClassDeclaration() ?: return null
    if (!classDeclaration.isPublic()) {
      return null
    }

    val resolvedReturnType = returnType.resolve()
    if (resolvedReturnType.declaration.qualifiedName?.asString() != "kotlinx.coroutines.flow.Flow") {
      return null
    }
    val flowType = resolvedReturnType.arguments.single().type?.resolve() ?: return null
    val nullableFlowType = flowType.isMarkedNullable
    val iosFlowName = if (nullableFlowType) {
      "NullableIOSFlow"
    } else {
      "IOSFlow"
    }

    val flowTypeName = try {
      flowType.makeNotNullable().toTypeName()
    } catch (e: NoSuchElementException) {
      logger.info("Can't parse type", symbol = declaration)

      return null
    }