streamFirebaseFunction function

Future<StreamSubscription<String>> streamFirebaseFunction(String functionName, Map<String, dynamic> data, dynamic onStreamReceived(String), { dynamic onStreamDone()?, Function? onStreamError, })

Implementation

Future<StreamSubscription<String>> streamFirebaseFunction(
  String functionName,
  Map<String, dynamic> data,
  Function(String) onStreamReceived, {
  Function()? onStreamDone,
  Function? onStreamError,
}) async {
  final Uri firebaseFunctionUrl = AppConfigBase.firebaseFunctionUri(functionName);

  // Get the current user's ID token
  String? token;
  try {
    token = await FirebaseAuth.instanceFor(app: AppConfigBase.firebaseApp)
        .currentUser
        ?.getIdToken();
  } catch (e) {
    loge('Error getting ID token: $e');
    onStreamError?.call(e);
    return Stream<String>.empty().listen((_) {});
  }

  if (token == null) {
    logd('User not authenticated.');
    onStreamError?.call('User not authenticated.');
    return Stream<String>.empty().listen((_) {});
  }

  final request = http.Request('POST', firebaseFunctionUrl)
    ..headers['Content-Type'] = 'application/json'
    ..headers['Authorization'] = 'Bearer $token'
    ..body = jsonEncode(data);

  final client = http.Client();
  http.StreamedResponse streamedResponse;
  try {
    streamedResponse = await client.send(request);
  } catch (e) {
    client.close();
    onStreamError?.call(e);
    return Stream<String>.empty().listen((_) {});
  }

  var cancelled = false;
  final controller = StreamController<String>(
    onCancel: () {
      cancelled = true;
      client.close();
    },
  );

  streamedResponse.stream.transform(utf8.decoder).listen(
    controller.add,
    onDone: () {
      try {
        if (!cancelled) onStreamDone?.call();
      } finally {
        client.close();
        controller.close();
      }
    },
    onError: (error) {
      try {
        if (!cancelled) onStreamError?.call(error);
      } finally {
        client.close();
        controller.close();
      }
    },
  );

  return controller.stream.listen(onStreamReceived);
}