twitch-hls-client 1.0.2

Minimal CLI client for watching low latency Twitch streams
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

use std::{
    io::{self, Write},
    process::{Child, ChildStdin, Command, Stdio},
};

use anyhow::{Context, Result};
use log::{debug, error, info};
use url::Url;

use crate::args::PlayerArgs;

pub struct Player {
    stdin: ChildStdin,
    process: Child,
    no_kill: bool,
}

impl Drop for Player {
    fn drop(&mut self) {
        if !self.no_kill {
            if let Err(e) = self.process.kill() {
                error!("Failed to kill player: {e}");
            }
        }
    }
}

impl Write for Player {
    fn write(&mut self, _buf: &[u8]) -> io::Result<usize> {
        unimplemented!();
    }

    fn flush(&mut self) -> io::Result<()> {
        debug!("Finished writing segment");
        Ok(())
    }

    fn write_all(&mut self, buf: &[u8]) -> io::Result<()> {
        self.stdin.write_all(buf)
    }
}

impl Player {
    pub fn spawn(args: &PlayerArgs) -> Result<Self> {
        info!("Opening player: {} {}", args.path, args.args);
        let mut command = Command::new(&args.path);
        command
            .args(args.args.split_whitespace())
            .stdin(Stdio::piped());

        if args.quiet {
            command.stdout(Stdio::null()).stderr(Stdio::null());
        }

        let mut process = command.spawn().context("Failed to open player")?;
        let stdin = process
            .stdin
            .take()
            .context("Failed to open player stdin")?;

        Ok(Self {
            stdin,
            process,
            no_kill: args.no_kill,
        })
    }

    pub fn passthrough(args: &PlayerArgs, url: &Url) -> Result<()> {
        info!("Passing through playlist URL to player");
        let mut args = args.to_owned();
        args.args = args
            .args
            .split_whitespace()
            .map(|s| {
                if s == "-" {
                    url.to_string()
                } else {
                    s.to_owned()
                }
            })
            .collect::<Vec<String>>()
            .join(" ");

        let mut player = Self::spawn(&args)?;
        player
            .process
            .wait()
            .context("Failed to wait for player process")?;

        Ok(())
    }
}