winsafe 0.0.27

#![allow(non_snake_case)]

use crate::co;
use crate::decl::*;
use crate::kernel::privs::*;
use crate::winspool::ffi;

/// [`AddPort`](https://learn.microsoft.com/en-us/windows/win32/printdocs/addport)
/// function.
pub fn AddPort(name: Option<&str>, hwnd: &HWND, monitor_name: &str) -> SysResult<()> {
	BoolRet(unsafe {
		ffi::AddPortW(
			WString::from_opt_str(name).as_mut_ptr(),
			hwnd.ptr(),
			WString::from_str(monitor_name).as_mut_ptr(),
		)
	})
	.to_invalidparm()
}

/// [`AddPrinterConnection`](https://learn.microsoft.com/en-us/windows/win32/printdocs/addprinterconnection)
/// function.
///
/// # Related functions
///
/// * [`DeletePrinterConnection`](crate::DeletePrinterConnection)
pub fn AddPrinterConnection(name: &str) -> SysResult<()> {
	BoolRet(unsafe { ffi::AddPrinterConnectionW(WString::from_str(name).as_mut_ptr()) })
		.to_invalidparm()
}

/// [`ConfigurePort`](https://learn.microsoft.com/en-us/windows/win32/printdocs/configureport)
/// function.
pub fn ConfigurePort(name: Option<&str>, hwnd: &HWND, port_name: &str) -> SysResult<()> {
	BoolRet(unsafe {
		ffi::ConfigurePortW(
			WString::from_opt_str(name).as_mut_ptr(),
			hwnd.ptr(),
			WString::from_str(port_name).as_mut_ptr(),
		)
	})
	.to_invalidparm()
}

/// [`DeleteMonitor`](https://learn.microsoft.com/en-us/windows/win32/printdocs/deletemonitor)
/// function.
pub fn DeleteMonitor(
	name: Option<&str>,
	environment: Option<&str>,
	monitor_name: &str,
) -> SysResult<()> {
	BoolRet(unsafe {
		ffi::DeleteMonitorW(
			WString::from_opt_str(name).as_mut_ptr(),
			WString::from_opt_str(environment).as_mut_ptr(),
			WString::from_str(monitor_name).as_mut_ptr(),
		)
	})
	.to_invalidparm()
}

/// [`DeletePrinterConnection`](https://learn.microsoft.com/en-us/windows/win32/printdocs/deleteprinterconnection)
/// function.
///
/// # Related functions
///
/// * [`AddPrinterConnection`](crate::AddPrinterConnection)
pub fn DeletePrinterConnection(name: &str) -> SysResult<()> {
	BoolRet(unsafe { ffi::DeletePrinterConnectionW(WString::from_str(name).as_mut_ptr()) })
		.to_invalidparm()
}

/// [`EnumPrinters`](https://learn.microsoft.com/en-us/windows/win32/printdocs/enumprinters)
/// function for Level 2.
///
/// **Note:** This function doesn't seem to work with `raw-dylib` Cargo feature.
///
/// # Examples
///
/// ```no_run
/// use winsafe::{self as w, prelude::*, co};
///
/// let printers = w::EnumPrinters2(
///     co::PRINTER_ENUM::LOCAL | co::PRINTER_ENUM::CONNECTIONS)?;
///
/// for p in printers {
///     println!("{}", p.pPrinterName().unwrap());
/// }
/// # w::SysResult::Ok(())
/// ```
///
/// # Related functions
///
/// * [`EnumPrinters4`](crate::EnumPrinters4)
/// * [`GetDefaultPrinter`](crate::GetDefaultPrinter)
#[must_use]
pub fn EnumPrinters2<'a, 'b, 'c, 'd, 'e, 'f, 'g, 'h, 'i, 'j, 'k, 'l, 'm>(
	flags: co::PRINTER_ENUM,
) -> SysResult<Vec<PRINTER_INFO_2<'a, 'b, 'c, 'd, 'e, 'f, 'g, 'h, 'i, 'j, 'k, 'l, 'm>>> {
	enum_printers::<PRINTER_INFO_2>(flags, 2)
}

/// [`EnumPrinters`](https://learn.microsoft.com/en-us/windows/win32/printdocs/enumprinters)
/// function for Level 4.
///
/// **Note:** This function doesn't seem to work with `raw-dylib` Cargo feature.
///
/// # Examples
///
/// ```no_run
/// use winsafe::{self as w, prelude::*, co};
///
/// let printers = w::EnumPrinters4(
///     co::PRINTER_ENUM::LOCAL | co::PRINTER_ENUM::CONNECTIONS)?;
///
/// for p in printers {
///     println!("{}", p.pPrinterName().unwrap());
/// }
/// # w::SysResult::Ok(())
/// ```
///
/// # Related functions
///
/// * [`EnumPrinters2`](crate::EnumPrinters2)
/// * [`GetDefaultPrinter`](crate::GetDefaultPrinter)
#[must_use]
pub fn EnumPrinters4<'a, 'b>(flags: co::PRINTER_ENUM) -> SysResult<Vec<PRINTER_INFO_4<'a, 'b>>> {
	enum_printers::<PRINTER_INFO_4>(flags, 4)
}

#[must_use]
fn enum_printers<T>(flags: co::PRINTER_ENUM, lvl: u32) -> SysResult<Vec<T>>
where
	T: Default + Clone,
{
	let (mut sz_buf, mut count) = (0u32, 0u32);
	unsafe {
		ffi::EnumPrintersW(
			flags.raw(),
			std::ptr::null_mut(),
			lvl,
			std::ptr::null_mut(),
			0,
			&mut sz_buf,
			&mut count,
		);
	}

	let sz_unit = std::mem::size_of::<T>() as u32;
	let num_elems = ((sz_buf + sz_unit) - ((sz_buf + sz_unit) % sz_unit)) / sz_unit;
	let mut buf = vec![T::default(); num_elems as _];

	BoolRet(unsafe {
		ffi::EnumPrintersW(
			flags.raw(),
			std::ptr::null_mut(),
			lvl,
			buf.as_mut_ptr() as _,
			(buf.len() * std::mem::size_of::<T>()) as _,
			&mut sz_buf,
			&mut count,
		)
	})
	.to_invalidparm()
	.map(|_| {
		buf.truncate(count as _);
		buf
	})
}

/// [`GetDefaultPrinter`](https://learn.microsoft.com/en-us/windows/win32/printdocs/getdefaultprinter)
/// function.
///
/// **Note:** This function doesn't seem to work with `raw-dylib` Cargo feature.
///
/// * [`EnumPrinters2`](crate::EnumPrinters2)
/// * [`EnumPrinters4`](crate::EnumPrinters4)
#[must_use]
pub fn GetDefaultPrinter() -> SysResult<String> {
	let mut sz = 0u32;
	unsafe {
		ffi::GetDefaultPrinterW(std::ptr::null_mut(), &mut sz);
	}
	let get_size_err = GetLastError();
	if get_size_err != co::ERROR::INSUFFICIENT_BUFFER {
		return Err(get_size_err);
	}

	let mut name_buf = WString::new_alloc_buf(sz as _);
	BoolRet(unsafe { ffi::GetDefaultPrinterW(name_buf.as_mut_ptr(), &mut sz) })
		.to_invalidparm()
		.map(|_| name_buf.to_string())
}

/// [`SetDefaultPrinter`](https://learn.microsoft.com/en-us/windows/win32/printdocs/setdefaultprinter)
/// function.
pub fn SetDefaultPrinter(printer: Option<&str>) -> SysResult<()> {
	BoolRet(unsafe { ffi::SetDefaultPrinterW(WString::from_opt_str(printer).as_ptr()) })
		.to_invalidparm()
}