schematics/src/symbols/owner.rs

use std::error::Error;

use std::fmt;
use std::fs;
use std::os::unix::fs::MetadataExt;
use std::path::Path;

use users::get_user_by_name;

use command_runner::CommandRunner;
use resources::Resource;
use symbols::{Action, OwnedSymbolAction, Symbol, SymbolAction, SymbolRunner};

pub struct Owner<'a, C: CommandRunner, D: AsRef<Path>, U: AsRef<str>> {
  path: D,
  user_name: U,
  command_runner: &'a C,
}

impl<'a, C: CommandRunner, D: AsRef<Path>, U: AsRef<str>> Owner<'a, C, D, U> {
  pub fn new(path: D, user_name: U, command_runner: &'a C) -> Self {
    Owner {
      path,
      user_name,
      command_runner,
    }
  }
}

impl<'a, C: CommandRunner, D: AsRef<Path>, U: AsRef<str>> Symbol for Owner<'a, C, D, U> {
  fn target_reached(&self) -> Result<bool, Box<dyn Error>> {
    if !Path::new(self.path.as_ref()).exists() {
      return Ok(false);
    }
    let actual_uid = fs::metadata(self.path.as_ref()).unwrap().uid();
    let target_uid = get_user_by_name(self.user_name.as_ref()).unwrap().uid();
    Ok(actual_uid == target_uid)
  }

  fn execute(&self) -> Result<(), Box<dyn Error>> {
    let uname: &str = self.user_name.as_ref();
    self
      .command_runner
      .run_successfully("chown", args!["-R", uname, self.path.as_ref(),])
  }

  fn get_prerequisites(&self) -> Vec<Resource> {
    vec![Resource::new("user", self.user_name.as_ref())]
  }

  fn as_action<'b>(&'b self, runner: &'b dyn SymbolRunner) -> Box<dyn Action + 'b> {
    Box::new(SymbolAction::new(runner, self))
  }

  fn into_action<'b>(self: Box<Self>, runner: &'b dyn SymbolRunner) -> Box<dyn Action + 'b>
  where
    Self: 'b,
  {
    Box::new(OwnedSymbolAction::new(runner, *self))
  }
}

impl<'a, C: CommandRunner, D: AsRef<Path>, U: AsRef<str>> fmt::Display for Owner<'a, C, D, U> {
  fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
    write!(
      f,
      "Owner {} for {}",
      self.user_name.as_ref(),
      self.path.as_ref().display()
    )
  }
}