4 files changed, 193 insertions(+), 0 deletions(-)

A .gitignore
A Cargo.lock
A Cargo.toml
A src/lib.rs

@@ 1,3 @@
+/target
+**/*.rs.bk
+.cargo-ok

@@ 1,45 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+[[package]]
+name = "bitflags"
+version = "1.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cf1de2fe8c75bc145a2f577add951f8134889b4795d47466a54a5c846d691693"
+
+[[package]]
+name = "cc"
+version = "1.0.67"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e3c69b077ad434294d3ce9f1f6143a2a4b89a8a2d54ef813d85003a4fd1137fd"
+
+[[package]]
+name = "cfg-if"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
+
+[[package]]
+name = "file-owner"
+version = "0.1.0"
+dependencies = [
+ "libc",
+ "nix",
+]
+
+[[package]]
+name = "libc"
+version = "0.2.88"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "03b07a082330a35e43f63177cc01689da34fbffa0105e1246cf0311472cac73a"
+
+[[package]]
+name = "nix"
+version = "0.20.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fa9b4819da1bc61c0ea48b63b7bc8604064dd43013e7cc325df098d49cd7c18a"
+dependencies = [
+ "bitflags",
+ "cc",
+ "cfg-if",
+ "libc",
+]

@@ 1,11 @@
+[package]
+name = "file-owner"
+version = "0.1.0"
+authors = ["Jakub Pastuszek <jpastuszek@protonmail.com>"]
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+nix = "0.20.0"
+libc = "0.2.88"

@@ 1,134 @@
+use nix::unistd::chown;
+use nix::unistd::{Gid, Uid, Group as NixGroup, User};
+use libc;
+use std::path::Path;
+use std::fmt::{self, Display};
+use std::error::Error;
+use std::convert::{TryFrom, TryInto, Infallible};
+
+#[derive(Debug)]
+pub enum FileOwnerError {
+    NixError(nix::Error),
+    UserNotFound(String),
+    GroupNotFound(String),
+}
+
+impl Display for FileOwnerError {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        match self {
+            FileOwnerError::NixError(_) => write!(f, "nix error"),
+            FileOwnerError::UserNotFound(name) => write!(f, "user name {:?} not found", name),
+			FileOwnerError::GroupNotFound(name) => write!(f, "group name {:?} not found", name),
+        }
+    }
+}
+
+impl Error for FileOwnerError {
+    fn source(&self) -> Option<&(dyn Error + 'static)> {
+        match self {
+            FileOwnerError::NixError(err) => Some(err),
+            FileOwnerError::UserNotFound(_) => None,
+			FileOwnerError::GroupNotFound(_) => None,
+        }
+    }
+}
+
+impl From<nix::Error> for FileOwnerError {
+    fn from(err: nix::Error) -> FileOwnerError {
+        FileOwnerError::NixError(err)
+    }
+}
+
+impl From<Infallible> for FileOwnerError {
+    fn from(_err: Infallible) -> FileOwnerError {
+        unreachable!()
+    }
+}
+
+#[derive(Debug, Copy, Clone, Eq, PartialEq, Hash)]
+pub struct Owner(Uid);
+
+impl Owner {
+    pub fn from_uid(uid: libc::uid_t) -> Owner {
+        Owner(Uid::from_raw(uid))
+    }
+
+    pub fn from_name(user: &str) -> Result<Owner, FileOwnerError> {
+        Ok(Owner(User::from_name(user)?.ok_or_else(|| FileOwnerError::UserNotFound(user.to_owned()))?.uid))
+    }
+}
+
+impl From<libc::uid_t> for Owner {
+    fn from(uid: libc::uid_t) -> Owner {
+        Owner::from_uid(uid)
+    }
+}
+
+impl<'s> TryFrom<&'s str> for Owner {
+    type Error = FileOwnerError;
+
+    fn try_from(name: &'s str) -> Result<Owner, Self::Error> {
+        Owner::from_name(name)
+    }
+}
+
+#[derive(Debug, Copy, Clone, Eq, PartialEq, Hash)]
+pub struct Group(Gid);
+
+impl Group {
+    pub fn from_gid(gid: libc::gid_t) -> Group {
+        Group(Gid::from_raw(gid))
+    }
+
+    pub fn from_name(group: &str) -> Result<Group, FileOwnerError> {
+        Ok(Group(NixGroup::from_name(group)?.ok_or_else(|| FileOwnerError::GroupNotFound(group.to_owned()))?.gid))
+    }
+}
+
+impl From<libc::gid_t> for Group {
+    fn from(gid: libc::gid_t) -> Group {
+        Group::from_gid(gid)
+    }
+}
+
+impl<'s> TryFrom<&'s str> for Group {
+    type Error = FileOwnerError;
+
+    fn try_from(name: &'s str) -> Result<Group, Self::Error> {
+        Group::from_name(name)
+    }
+}
+
+pub fn set_owner<E: Into<FileOwnerError>>(path: impl AsRef<Path>, owner: impl TryInto<Owner, Error = E>) -> Result<(), FileOwnerError> {
+    Ok(chown(path.as_ref().into(), Some(owner.try_into().map_err(Into::into)?.0), None)?)
+}
+
+pub fn set_group<E: Into<FileOwnerError>>(path: impl AsRef<Path>, group: impl TryInto<Group, Error = E>) -> Result<(), FileOwnerError> {
+    Ok(chown(path.as_ref().into(), None, Some(group.try_into().map_err(Into::into)?.0))?)
+}
+
+pub fn set_owner_group<E1: Into<FileOwnerError>, E2: Into<FileOwnerError>>(path: impl AsRef<Path>, owner: impl TryInto<Owner, Error = E1>, group: impl TryInto<Group, Error = E2>) -> Result<(), FileOwnerError> {
+    Ok(chown(path.as_ref().into(), Some(owner.try_into().map_err(Into::into)?.0), Some(group.try_into().map_err(Into::into)?.0))?)
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_calling() {
+        let foo = Path::new("/tmp/foo");
+        std::fs::write(foo, "test").unwrap();
+
+        set_owner(foo, "nobody").unwrap();
+        set_owner(foo, 99).unwrap();
+
+        set_group(foo, "nogroup").unwrap();
+        set_group(foo, 99).unwrap();
+
+        set_owner_group(foo, "nobody", "nogroup").unwrap();
+        set_owner_group(foo, 99, 99).unwrap();
+        set_owner_group(foo, 99, "nogroup").unwrap();
+        set_owner_group(foo, "nobody", 99).unwrap();
+    }
+}