81 lines
2.9 KiB
Rust
81 lines
2.9 KiB
Rust
use dirs;
|
|
use std::fs;
|
|
use std::io::{self, Write};
|
|
use std::path::{Path, PathBuf};
|
|
use std::process::Command;
|
|
use std::str;
|
|
|
|
pub fn bg_from_video(video: &Path) -> PathBuf {
|
|
let video = PathBuf::from(video);
|
|
println!("{:?}", video);
|
|
println!("{:?}", video.file_name());
|
|
let mut data_dir = dirs::data_local_dir().unwrap();
|
|
data_dir.push("librepresenter");
|
|
data_dir.push("thumbnails");
|
|
if !data_dir.exists() {
|
|
fs::create_dir(&data_dir)
|
|
.expect("Could not create thumbnails dir");
|
|
}
|
|
let mut screenshot = data_dir.clone();
|
|
screenshot.push(video.file_name().unwrap());
|
|
screenshot.set_extension("png");
|
|
if !screenshot.exists() {
|
|
let output_duration = Command::new("ffprobe")
|
|
.args(&["-i", &video.to_string_lossy()])
|
|
.output()
|
|
.expect("failed to execute ffprobe");
|
|
io::stderr().write_all(&output_duration.stderr).unwrap();
|
|
let mut at_second = 2;
|
|
let mut log = str::from_utf8(&output_duration.stderr)
|
|
.expect("Using non UTF-8 characters")
|
|
.to_string();
|
|
println!("{log}");
|
|
if let Some(duration_index) = log.find("Duration") {
|
|
let mut duration = log.split_off(duration_index + 10);
|
|
duration.truncate(11);
|
|
println!("rust-duration-is: {duration}");
|
|
let mut hours = String::from("");
|
|
let mut minutes = String::from("");
|
|
let mut seconds = String::from("");
|
|
for (i, c) in duration.chars().enumerate() {
|
|
if i <= 1 {
|
|
hours.push(c);
|
|
} else if i > 2 && i <= 4 {
|
|
minutes.push(c);
|
|
} else if i > 5 && i <= 7 {
|
|
seconds.push(c);
|
|
}
|
|
}
|
|
let hours: i32 = hours.parse().unwrap_or_default();
|
|
let mut minutes: i32 =
|
|
minutes.parse().unwrap_or_default();
|
|
let mut seconds: i32 =
|
|
seconds.parse().unwrap_or_default();
|
|
minutes += hours * 60;
|
|
seconds += minutes * 60;
|
|
at_second = seconds / 5;
|
|
println!(
|
|
"hours: {}, minutes: {}, seconds: {}, at_second: {}",
|
|
hours, minutes, seconds, at_second
|
|
);
|
|
}
|
|
let _output = Command::new("ffmpeg")
|
|
.args(&[
|
|
"-i",
|
|
&video.to_string_lossy(),
|
|
"-ss",
|
|
&at_second.to_string(),
|
|
"-vframes",
|
|
"1",
|
|
"-y",
|
|
&screenshot.to_string_lossy(),
|
|
])
|
|
.output()
|
|
.expect("failed to execute ffmpeg");
|
|
// io::stdout().write_all(&output.stdout).unwrap();
|
|
// io::stderr().write_all(&output.stderr).unwrap();
|
|
} else {
|
|
println!("Screenshot already exists");
|
|
}
|
|
screenshot
|
|
}
|