lesavka/server/src/video.rs

// server/src/video.rs

use anyhow::Context;
use gstreamer as gst;
use gstreamer_app as gst_app;
use gst::prelude::*;
use lesavka_common::lesavka::VideoPacket;
use tokio_stream::wrappers::ReceiverStream;
use tonic::Status;

pub async fn spawn_camera(
    dev: &str,
    id: u32,
    max_bitrate_kbit: u32,
) -> anyhow::Result<ReceiverStream<Result<VideoPacket, Status>>> {
    gst::init().context("gst init")?;

    // v4l2src → H.264 already, we only parse & relay
    let desc = format!(
        "v4l2src device={dev} io-mode=auto ! queue ! h264parse config-interval=-1 ! \
         appsink name=sink emit-signals=true sync=false"
    );

    let pipeline = gst::parse::launch(&desc)?
        .downcast::<gst::Pipeline>()
        .expect("pipeline down-cast");

    let sink = pipeline
        .by_name("sink")
        .expect("appsink")
        .dynamic_cast::<gst_app::AppSink>()
        .expect("appsink downcast");

    let (tx, rx) = tokio::sync::mpsc::channel(256);

    sink.set_callbacks(
        gst_app::AppSinkCallbacks::builder()
            // .new_sample(move |sink| {
            //     let sample = sink.pull_sample().map_err(|_| gst::FlowError::Eos)?;
            //     let buffer = sample.buffer().ok_or(gst::FlowError::Error)?;

            //     let map = buffer.map_readable().map_err(|_| gst::FlowError::Error)?;
            //     let pkt = VideoPacket {
            //         id,
            //         pts: buffer.pts().nseconds() / 1_000, // → µs
            //         data: map.as_slice().to_vec(),
            //     };
            //     // ignore back‑pressure: drop oldest if channel is full
            //     let _ = tx.try_send(Ok(pkt));
            //     Ok(gst::FlowSuccess::Ok)
            // })
            // .build(),
            .new_sample(move |sink| {
                let sample = sink.pull_sample().map_err(|_| gst::FlowError::Eos)?;
                let buffer = sample.buffer().ok_or(gst::FlowError::Error)?;

                let pts_us = buffer
                    .pts()
                    .map(|t| t.nseconds() / 1_000)   // ns → µs
                    .unwrap_or(0);

                let map = buffer.map_readable().map_err(|_| gst::FlowError::Error)?;

                let pkt = VideoPacket {
                    id,
                    pts: pts_us,
                    data: map.as_slice().to_vec(),
                };
                let _ = tx.try_send(Ok(pkt));        // drop on overflow
                Ok(gst::FlowSuccess::Ok)
            })
            .build(),
    );

    pipeline.set_state(gst::State::Playing)?;

    Ok(ReceiverStream::new(rx))
}