Future Extension Flutter

future_extension.dart

extension FutureExtension on Future {
  Widget toBuild<T>({Widget Function(T data) onSuccess, Widget onError, dynamic data}) {
    return FutureBuilder<T>(
      future: this,
      initialData: data,
      builder: (BuildContext context, AsyncSnapshot snapshot) {
        switch (snapshot.connectionState) {
          case ConnectionState.waiting:
          case ConnectionState.active:
            return Center(child: CircularProgressIndicator());
          case ConnectionState.done:
            if (snapshot.hasData)
              return onSuccess(snapshot.data);
            else
              return onError ?? NotFoundLottie();
            break;
          default:
            return NotFoundLottie();
        }
      },
    );
  }
}

class SampleFuture extends StatelessWidget {
  @override
  Widget build(BuildContext context) {
    return Future.value(5).toBuild<int>(
        onSuccess: (data) {
          return Text(data.toString());
        },
        onError: Text("Not Found"));
  }
}