Stream4

import 'package:flutter/material.dart';
import 'stream.dart';
import 'number.dart';
import 'dart:async';
import 'dart:math';

void main() {
  runApp(MyApp());
}

class MyApp extends StatelessWidget {
// This widget is the root of your application.
  @override
  Widget build(BuildContext context) {
    return MaterialApp(
        title: 'Stream',
        theme: ThemeData(
          primarySwatch: Colors.deepPurple,
          visualDensity: VisualDensity.adaptivePlatformDensity,
        ),
        // home: StreamHomePage(),
        home: StreamNumber());
  }
}

class StreamHomePage extends StatefulWidget {
  @override
  _StreamHomePageState createState() => _StreamHomePageState();
}

class _StreamHomePageState extends State<StreamHomePage> {
  Color? bgColor;
  ColorStream? colorStream;

  @override
  void initState() {
    colorStream = ColorStream();
    changeColor();
    super.initState();
  }

  changeColor() async {
    // await for (var eventColor in colorStream!.getColors()) {
    //   setState(() {
    //     bgColor = eventColor;
    //   });
    // }
    colorStream?.getColors().listen((eventColor) {
      setState(() {
        bgColor = eventColor;
      });
    });
  }

  @override
  Widget build(BuildContext context) {
    return Scaffold(
        appBar: AppBar(
          title: Text('Stream'),
        ),
        body: Container(
          decoration: BoxDecoration(color: bgColor),
        ));
  }
}

class StreamNumber extends StatefulWidget {
  const StreamNumber({super.key});

  @override
  State<StreamNumber> createState() => _StreamNumberState();
}

class _StreamNumberState extends State<StreamNumber> {
  int lastNumber = 0;
  StreamTransformer? transformer;
  NumberStream? numberStream;
  StreamController? numberStreamController;
  StreamSubscription? subscription;
  @override
  void initState() {
    numberStream = NumberStream();
    numberStreamController = numberStream?.controller;
    Stream? stream = numberStreamController?.stream;

    // stream?.listen((event) {
    //   setState(() {
    //     lastNumber = event;
    //   });
    // });
    // transformer = StreamTransformer<dynamic, dynamic>.fromHandlers(
    //     handleData: (value, sink) {
    //       sink.add(value * 10);
    //     },
    //     handleError: (error, trace, sink) {
    //       sink.add(-1);
    //     },
    //     handleDone: (sink) => sink.close());
    // stream?.transform(transformer!).listen((event) {
    //   setState(() {
    //     lastNumber = event;
    //   });
    // }).onError((error) {
    //   setState(() {
    //     lastNumber = -1;
    //   });
    // });
    subscription = stream?.listen((event) {
      setState(() {
        lastNumber = event;
      });
      subscription?.onDone(() {
        print('OnDone was called');
      });
    });

    super.initState();
  }

  @override
  Widget build(BuildContext context) {
    return Scaffold(
      body: Container(
        width: double.infinity,
        child: Column(
          mainAxisAlignment: MainAxisAlignment.spaceEvenly,
          crossAxisAlignment: CrossAxisAlignment.center,
          children: [
            Text(lastNumber.toString()),
            ElevatedButton(
              onPressed: () => addRandomNumber(),
              child: Text('New Random Number'),
            ),
            ElevatedButton(
              onPressed: () => stopStream(),
              child: Text('Stop Stream'),
            )
          ],
        ),
      ),
    );
  }

  void addRandomNumber() {
    Random random = Random();
    int myNum = random.nextInt(10);
    if (!numberStreamController!.isClosed) {
      numberStream?.addNumberToSink(myNum);
    } else {
      setState(() {
        lastNumber = -1;
      });
    }
  }

  void stopStream() {
    numberStreamController?.close();
  }
}

subscription = stream?.listen((event) {
      setState(() {
        lastNumber = event;
      });
      subscription?.onDone(() {
        print('OnDone was called');
      });
    });