use error::to_std_io_result;
use gio_sys;
use glib::object::IsA;
use glib::translate::*;
use glib::Priority;
use glib_sys;
use gobject_sys;
use std::io;
use std::mem;
use std::ptr;
use Cancellable;
use Error;
use OutputStream;
use OutputStreamExt;
#[cfg(feature = "futures")]
use futures::future;
pub trait OutputStreamExtManual: Sized + OutputStreamExt {
fn write_async<
B: AsRef<[u8]> + Send + 'static,
Q: FnOnce(Result<(B, usize), (B, Error)>) + Send + 'static,
>(
&self,
buffer: B,
io_priority: Priority,
cancellable: Option<&Cancellable>,
callback: Q,
);
fn write_all(
&self,
buffer: &[u8],
cancellable: Option<&Cancellable>,
) -> Result<(usize, Option<Error>), Error>;
#[cfg(any(feature = "v2_44", feature = "dox"))]
fn write_all_async<
B: AsRef<[u8]> + Send + 'static,
Q: FnOnce(Result<(B, usize, Option<Error>), (B, Error)>) + Send + 'static,
>(
&self,
buffer: B,
io_priority: Priority,
cancellable: Option<&Cancellable>,
callback: Q,
);
#[cfg(feature = "futures")]
fn write_async_future<'a, B: AsRef<[u8]> + Send + 'static>(
&self,
buffer: B,
io_priority: Priority,
) -> Box<dyn future::Future<Output = Result<(B, usize), (B, Error)>> + std::marker::Unpin>;
#[cfg(feature = "futures")]
#[cfg(any(feature = "v2_44", feature = "dox"))]
fn write_all_async_future<'a, B: AsRef<[u8]> + Send + 'static>(
&self,
buffer: B,
io_priority: Priority,
) -> Box<
dyn future::Future<Output = Result<(B, usize, Option<Error>), (B, Error)>>
+ std::marker::Unpin,
>;
fn into_write(self) -> OutputStreamWrite<Self> {
OutputStreamWrite(self)
}
}
impl<O: IsA<OutputStream>> OutputStreamExtManual for O {
fn write_async<
B: AsRef<[u8]> + Send + 'static,
Q: FnOnce(Result<(B, usize), (B, Error)>) + Send + 'static,
>(
&self,
buffer: B,
io_priority: Priority,
cancellable: Option<&Cancellable>,
callback: Q,
) {
let cancellable = cancellable.to_glib_none();
let user_data: Box<Option<(Q, B)>> = Box::new(Some((callback, buffer)));
let (count, buffer_ptr) = {
let buffer = &(*user_data).as_ref().unwrap().1;
let slice = buffer.as_ref();
(slice.len(), slice.as_ptr())
};
unsafe extern "C" fn write_async_trampoline<
B: AsRef<[u8]> + Send + 'static,
Q: FnOnce(Result<(B, usize), (B, Error)>) + Send + 'static,
>(
_source_object: *mut gobject_sys::GObject,
res: *mut gio_sys::GAsyncResult,
user_data: glib_sys::gpointer,
) {
let mut user_data: Box<Option<(Q, B)>> = Box::from_raw(user_data as *mut _);
let (callback, buffer) = user_data.take().unwrap();
let mut error = ptr::null_mut();
let ret =
gio_sys::g_output_stream_write_finish(_source_object as *mut _, res, &mut error);
let result = if error.is_null() {
Ok((buffer, ret as usize))
} else {
Err((buffer, from_glib_full(error)))
};
callback(result);
}
let callback = write_async_trampoline::<B, Q>;
unsafe {
gio_sys::g_output_stream_write_async(
self.as_ref().to_glib_none().0,
mut_override(buffer_ptr),
count,
io_priority.to_glib(),
cancellable.0,
Some(callback),
Box::into_raw(user_data) as *mut _,
);
}
}
fn write_all(
&self,
buffer: &[u8],
cancellable: Option<&Cancellable>,
) -> Result<(usize, Option<Error>), Error> {
let cancellable = cancellable.to_glib_none();
let count = buffer.len() as usize;
unsafe {
let mut bytes_written = mem::uninitialized();
let mut error = ptr::null_mut();
let _ = gio_sys::g_output_stream_write_all(
self.as_ref().to_glib_none().0,
buffer.to_glib_none().0,
count,
&mut bytes_written,
cancellable.0,
&mut error,
);
if error.is_null() {
Ok((bytes_written, None))
} else if bytes_written != 0 {
Ok((bytes_written, Some(from_glib_full(error))))
} else {
Err(from_glib_full(error))
}
}
}
#[cfg(any(feature = "v2_44", feature = "dox"))]
fn write_all_async<
'a,
B: AsRef<[u8]> + Send + 'static,
Q: FnOnce(Result<(B, usize, Option<Error>), (B, Error)>) + Send + 'static,
>(
&self,
buffer: B,
io_priority: Priority,
cancellable: Option<&Cancellable>,
callback: Q,
) {
let cancellable = cancellable.to_glib_none();
let user_data: Box<Option<(Q, B)>> = Box::new(Some((callback, buffer)));
let (count, buffer_ptr) = {
let buffer = &(*user_data).as_ref().unwrap().1;
let slice = buffer.as_ref();
(slice.len(), slice.as_ptr())
};
unsafe extern "C" fn write_all_async_trampoline<
B: AsRef<[u8]> + Send + 'static,
Q: FnOnce(Result<(B, usize, Option<Error>), (B, Error)>) + Send + 'static,
>(
_source_object: *mut gobject_sys::GObject,
res: *mut gio_sys::GAsyncResult,
user_data: glib_sys::gpointer,
) {
let mut user_data: Box<Option<(Q, B)>> = Box::from_raw(user_data as *mut _);
let (callback, buffer) = user_data.take().unwrap();
let mut error = ptr::null_mut();
let mut bytes_written = mem::uninitialized();
let _ = gio_sys::g_output_stream_write_all_finish(
_source_object as *mut _,
res,
&mut bytes_written,
&mut error,
);
let result = if error.is_null() {
Ok((buffer, bytes_written, None))
} else if bytes_written != 0 {
Ok((buffer, bytes_written, from_glib_full(error)))
} else {
Err((buffer, from_glib_full(error)))
};
callback(result);
}
let callback = write_all_async_trampoline::<B, Q>;
unsafe {
gio_sys::g_output_stream_write_all_async(
self.as_ref().to_glib_none().0,
mut_override(buffer_ptr),
count,
io_priority.to_glib(),
cancellable.0,
Some(callback),
Box::into_raw(user_data) as *mut _,
);
}
}
#[cfg(feature = "futures")]
fn write_async_future<'a, B: AsRef<[u8]> + Send + 'static>(
&self,
buffer: B,
io_priority: Priority,
) -> Box<dyn future::Future<Output = Result<(B, usize), (B, Error)>> + std::marker::Unpin> {
use GioFuture;
GioFuture::new(self, move |obj, send| {
use fragile::Fragile;
let cancellable = Cancellable::new();
let send = Fragile::new(send);
obj.write_async(buffer, io_priority, Some(&cancellable), move |res| {
let _ = send.into_inner().send(res);
});
cancellable
})
}
#[cfg(feature = "futures")]
#[cfg(any(feature = "v2_44", feature = "dox"))]
fn write_all_async_future<'a, B: AsRef<[u8]> + Send + 'static>(
&self,
buffer: B,
io_priority: Priority,
) -> Box<
dyn future::Future<Output = Result<(B, usize, Option<Error>), (B, Error)>>
+ std::marker::Unpin,
> {
use GioFuture;
GioFuture::new(self, move |obj, send| {
use fragile::Fragile;
let cancellable = Cancellable::new();
let send = Fragile::new(send);
obj.write_all_async(buffer, io_priority, Some(&cancellable), move |res| {
let _ = send.into_inner().send(res);
});
cancellable
})
}
}
#[derive(Debug, PartialEq, Eq, Hash, Clone)]
pub struct OutputStreamWrite<T: OutputStreamExt>(T);
impl<T: OutputStreamExt> OutputStreamWrite<T> {
pub fn into_output_stream(self) -> T {
self.0
}
pub fn output_stream(&self) -> &T {
&self.0
}
}
impl<T: OutputStreamExt> io::Write for OutputStreamWrite<T> {
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
let result = self
.0
.write(buf, ::NONE_CANCELLABLE)
.map(|size| size as usize);
to_std_io_result(result)
}
fn flush(&mut self) -> io::Result<()> {
let gio_result = self.0.flush(::NONE_CANCELLABLE);
to_std_io_result(gio_result)
}
}
#[cfg(test)]
mod tests {
use glib::*;
use std::io::Write;
use test_util::run_async;
use *;
#[test]
fn splice_async() {
let ret = run_async(|tx, l| {
let input = MemoryInputStream::new();
let b = Bytes::from_owned(vec![1, 2, 3]);
input.add_bytes(&b);
let strm = MemoryOutputStream::new_resizable();
strm.splice_async(
&input,
OutputStreamSpliceFlags::CLOSE_SOURCE,
PRIORITY_DEFAULT_IDLE,
::NONE_CANCELLABLE,
move |ret| {
tx.send(ret).unwrap();
l.quit();
},
);
});
assert_eq!(ret.unwrap(), 3);
}
#[test]
fn write_async() {
let ret = run_async(|tx, l| {
let strm = MemoryOutputStream::new_resizable();
let buf = vec![1, 2, 3];
strm.write_async(buf, PRIORITY_DEFAULT_IDLE, ::NONE_CANCELLABLE, move |ret| {
tx.send(ret).unwrap();
l.quit();
});
});
let (buf, size) = ret.unwrap();
assert_eq!(buf, vec![1, 2, 3]);
assert_eq!(size, 3);
}
#[test]
#[cfg(any(feature = "v2_44", feature = "dox"))]
fn write_all_async() {
let ret = run_async(|tx, l| {
let strm = MemoryOutputStream::new_resizable();
let buf = vec![1, 2, 3];
strm.write_all_async(buf, PRIORITY_DEFAULT_IDLE, ::NONE_CANCELLABLE, move |ret| {
tx.send(ret).unwrap();
l.quit();
});
});
let (buf, size, err) = ret.unwrap();
assert_eq!(buf, vec![1, 2, 3]);
assert_eq!(size, 3);
assert!(err.is_none());
}
#[test]
fn write_bytes_async() {
let ret = run_async(|tx, l| {
let strm = MemoryOutputStream::new_resizable();
let b = Bytes::from_owned(vec![1, 2, 3]);
strm.write_bytes_async(&b, PRIORITY_DEFAULT_IDLE, ::NONE_CANCELLABLE, move |ret| {
tx.send(ret).unwrap();
l.quit();
});
});
assert_eq!(ret.unwrap(), 3);
}
#[test]
fn std_io_write() {
let b = Bytes::from_owned(vec![1, 2, 3]);
let mut write = MemoryOutputStream::new_resizable().into_write();
let ret = write.write(&b);
let stream = write.into_output_stream();
stream.close(::NONE_CANCELLABLE).unwrap();
assert_eq!(ret.unwrap(), 3);
assert_eq!(stream.steal_as_bytes().unwrap(), [1, 2, 3].as_ref());
}
#[test]
fn into_output_stream() {
let stream = MemoryOutputStream::new_resizable();
let stream_clone = stream.clone();
let stream = stream.into_write().into_output_stream();
assert_eq!(stream, stream_clone);
}
}