xlmnxp/blue-recorder
1
0
Fork 0
mirror of https://github.com/xlmnxp/blue-recorder.git synced 2024-11-24 01:24:07 +03:00
Code Issues Packages Projects Releases Wiki Activity

Compare commits

base: xlmnxp:2bc71a8168159643e2bbd7b211518a9d88504a8c
Branches Tags
xlmnxp:master
..
compare: xlmnxp:bc5f0ba614fa489be9e52f3cf91519c3753e5f41
Branches Tags
xlmnxp:master

No commits in common. "2bc71a8168159643e2bbd7b211518a9d88504a8c" and "bc5f0ba614fa489be9e52f3cf91519c3753e5f41" have entirely different histories.
2bc71a8168 ... bc5f0ba614

1 changed files with 25 additions and 58 deletions
Show Stats Expand all files Collapse all files

75
src/wayland_record.rs
View File

@ -5,7 +5,7 @@ use zbus::{
dbus_proxy,
export::futures_util::TryStreamExt,
zvariant::{ObjectPath, OwnedObjectPath, Structure, Value},
Connection, MessageStream, MessageType, Result,
Connection, MessageStream, MessageType, Result
};
#[derive(Clone, Copy)]
@ -20,7 +20,7 @@ pub enum RecordTypes {
pub enum CursorModeTypes {
Default,
Hidden,
Show,
Show
}
#[dbus_proxy(
@ -54,12 +54,8 @@ pub struct WaylandRecorder {
impl WaylandRecorder {
pub async fn new() -> Self {
let connection = Connection::session()
.await
.expect("failed to connect to session bus");
let screen_cast_proxy = ScreenCastProxy::new(&connection)
.await
.expect("failed to create dbus proxy for screen-cast");
let connection = Connection::session().await.expect("failed to connect to session bus");
let screen_cast_proxy = ScreenCastProxy::new(&connection).await.expect("failed to create dbus proxy for screen-cast");
gst::init().expect("failed to initialize gstreamer");
WaylandRecorder {
@ -71,37 +67,24 @@ impl WaylandRecorder {
}
}
pub async fn start(
&mut self,
filename: String,
record_type: RecordTypes,
cursor_mode_type: CursorModeTypes,
) -> bool {
self.screen_cast_proxy
.create_session(HashMap::from([
("handle_token", Value::from("blue_recorder_1")),
("session_handle_token", Value::from("blue_recorder_1")),
]))
.await
.expect("failed to create session");
pub async fn start(&mut self, filename: String, record_type: RecordTypes, cursor_mode_type: CursorModeTypes) -> bool {
self.screen_cast_proxy.create_session(HashMap::from([
("handle_token", Value::from("blue_recorder_1")),
("session_handle_token", Value::from("blue_recorder_1")),
]))
.await.expect("failed to create session");
let mut message_stream = MessageStream::from(self.connection.clone());
self.filename = filename.clone();
while let Some(msg) = message_stream
.try_next()
.await
.expect("failed to get message")
{
while let Some(msg) = message_stream.try_next().await.expect("failed to get message") {
match msg.message_type() {
MessageType::Signal => {
let (response_num, response) = msg
.body::<(u32, HashMap<&str, Value>)>()
.expect("failed to get body");
let (response_num, response) = msg.body::<(u32, HashMap<&str, Value>)>().expect("failed to get body");
if response_num > 0 {
return false;
return false;
}
if response.len() == 0 {
@ -113,17 +96,14 @@ impl WaylandRecorder {
self.screen_cast_proxy.clone(),
response.clone(),
record_type,
cursor_mode_type,
cursor_mode_type
)
.await
.expect("failed to handle session");
.await.expect("failed to handle session");
continue;
}
if response.contains_key("streams") {
self.record_screen_cast(response.clone())
.await
.expect("failed to record screen cast");
self.record_screen_cast(response.clone()).await.expect("failed to record screen cast");
break;
}
}
@ -144,21 +124,8 @@ impl WaylandRecorder {
}
if self.session_path.len() > 0 {
println!(
"Closing session...: {:?}",
self.session_path.replace("request", "session")
);
self.connection
.clone()
.call_method(
Some("org.freedesktop.portal.Desktop"),
self.session_path.clone().replace("request", "session"),
Some("org.freedesktop.portal.Session"),
"Close",
&(),
)
.await
.expect("failed to close session");
println!("Closing session...: {:?}", self.session_path.replace("request", "session"));
self.connection.clone().call_method(Some("org.freedesktop.portal.Desktop"), self.session_path.clone().replace("request", "session"), Some("org.freedesktop.portal.Session"), "Close", &()).await.expect("failed to close session");
self.session_path = String::new();
}
}
@ -168,7 +135,7 @@ impl WaylandRecorder {
screen_cast_proxy: ScreenCastProxy<'_>,
response: HashMap<&str, Value<'_>>,
record_type: RecordTypes,
cursor_mode_type: CursorModeTypes,
cursor_mode_type: CursorModeTypes
) -> Result<()> {
let response_session_handle = response
.get("session_handle")
@ -224,13 +191,13 @@ impl WaylandRecorder {
.clone()
.downcast::<Vec<Value>>()
.expect("cannot down cast streams to vec array")
.first()
.get(0)
.expect("cannot get first object from streams array")
.clone()
.downcast::<Structure>()
.expect("cannot down cast first object to structure")
.fields()
.first()
.get(0)
.expect("cannot get first field from structure")
.clone()
.downcast::<u32>()