use crate::imp::audio_stream as stream_imp;

#[cfg(not(target_arch = "wasm32"))]
pub mod native {
    use super::stream_imp;
    use crate::audio_frame::AudioFrame;
    use crate::audio_track::RtcAudioTrack;
    use futures::stream::Stream;
    use std::fmt::{Debug, Formatter};
    use std::pin::Pin;
    use std::task::{Context, Poll};

    pub struct NativeAudioStream {
        pub(crate) handle: stream_imp::NativeAudioStream,
    }

    impl Debug for NativeAudioStream {
        fn fmt(&self, f: &mut Formatter) -> std::fmt::Result {
            f.debug_struct("NativeAudioStream")
                .field("track", &self.track())
                .finish()
        }
    }

    impl NativeAudioStream {
        pub fn new(audio_track: RtcAudioTrack) -> Self {
            Self {
                handle: stream_imp::NativeAudioStream::new(audio_track),
            }
        }

        pub fn track(&self) -> RtcAudioTrack {
            self.handle.track()
        }

        pub fn close(&mut self) {
            self.handle.close()
        }
    }

    impl Stream for NativeAudioStream {
        type Item = AudioFrame;

        fn poll_next(self: Pin<&mut Self>, cx: &mut Context) -> Poll<Option<Self::Item>> {
            Pin::new(&mut self.get_mut().handle).poll_next(cx)
        }
    }
}