schematics/src/symbols/systemd/node_js_user_service.rs

use std::error::Error;
use std::os::unix::fs::PermissionsExt;
use std::fmt;
use std::io;
use std::fs;
use std::process::Output;
use std::path::Path;
use std::thread::sleep;
use std::time::Duration;
use std::ops::Deref;

use command_runner::CommandRunner;
use symbols::Symbol;
use symbols::file::File as FileSymbol;
use resources::Resource;

#[derive(Debug)]
pub enum NodeJsSystemdUserServiceError<E: Error> {
  ActivationFailed(io::Result<Output>),
  ExecError(E),
  GenericError
}

impl From<io::Error> for NodeJsSystemdUserServiceError<io::Error> {
  fn from(err: io::Error) -> NodeJsSystemdUserServiceError<io::Error> {
    NodeJsSystemdUserServiceError::ExecError(err)
  }
}

impl<E: Error> Error for NodeJsSystemdUserServiceError<E> {
  fn description(&self) -> &str {
    match self {
      &NodeJsSystemdUserServiceError::ExecError(ref e) => e.description(),
      &NodeJsSystemdUserServiceError::GenericError => "Generic error",
      &NodeJsSystemdUserServiceError::ActivationFailed(_) => "Activation of service failed"
    }
  }
  fn cause(&self) -> Option<&Error> {
    match self {
      &NodeJsSystemdUserServiceError::ExecError(ref e) => Some(e),
      _ => None
    }
  }
}

impl<E: Error> fmt::Display for NodeJsSystemdUserServiceError<E> {
  fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
    try!(write!(f, "{}", self.description()));
    if let &NodeJsSystemdUserServiceError::ActivationFailed(Ok(ref log)) = self {
      try!(write!(f, ": {:?}", log));
    };
    Ok(())
  }
}

pub struct NodeJsSystemdUserService<'a, P, C> where P: AsRef<str> + fmt::Display, C: Deref<Target=str> {
  name: &'a str,
  home: &'a str,
  path: P,
  command_runner: &'a CommandRunner,
  file: FileSymbol<C, Cow<'a, str>>
}

use std::borrow::Cow;

impl<'a> NodeJsSystemdUserService<'a, Cow<'a, str>, String> {
  pub fn new(home: &'a str, name: &'a str, path: &'a str, command_runner: &'a CommandRunner) -> Self {
    let file_path: Cow<str> = Cow::from(String::from(home.trim_right()) + "/.config/systemd/user/" + name + ".service");

    let content = format!("[Service]
ExecStart=/usr/bin/nodejs {}
Restart=always
Environment=NODE_ENV=production
Environment=PORT={}/var/service.socket

[Install]
WantedBy=default.target
", path, home);
    NodeJsSystemdUserService {
      name: name,
      home: home,
      path: file_path.clone(),
      command_runner: command_runner,
      file: FileSymbol::new(file_path, content)
    }
  }
}

impl<'a, P, C> NodeJsSystemdUserService<'a, P, C> where P: AsRef<str> + fmt::Display, C: Deref<Target=str> {
  fn check_if_service(&self) -> Result<bool, Box<Error>> {
    loop {
      // Check if service is registered
      let active_state = try!(self.command_runner.run_with_args("systemctl", &["--user", "show", "--property", "ActiveState", self.name]));
      if !active_state.status.success() {
        return Ok(false);
      }
      // Check if service is running
      match String::from_utf8(active_state.stdout).unwrap().trim_right() {
        "ActiveState=activating" => sleep(Duration::from_millis(500)),
        "ActiveState=active" => return Ok(true),
        "ActiveState=failed" => return Err(Box::new(NodeJsSystemdUserServiceError::ActivationFailed(self.command_runner.run_with_args("journalctl", &["--user", &format!("--user-unit={}", self.name)])) as NodeJsSystemdUserServiceError<io::Error>)),
        _ => return Ok(false)
      }
    }
  }
}

impl<'a, P, C> Symbol for NodeJsSystemdUserService<'a, P, C> where P: AsRef<str> + fmt::Display, C: Deref<Target=str> {
  fn target_reached(&self) -> Result<bool, Box<Error>> {
    match self.file.target_reached() {
      Ok(false) => return Ok(false),
      Ok(true) => {},
      Err(e) => return Err(Box::new(NodeJsSystemdUserServiceError::GenericError as NodeJsSystemdUserServiceError<io::Error>))
    }
/*
    if !(try!(self.file.target_reached())) {
      return Ok(false)
    }
*/
    self.check_if_service()
  }

  fn execute(&self) -> Result<(), Box<Error>> {
    try!(self.command_runner.run_with_args("mkdir", &["-p", &format!("{}/var", self.home)]));
    try!(self.command_runner.run_with_args("rm", &[&format!("{}/var/service.socket", self.home)]));

    try!(self.command_runner.run_with_args("mkdir", &["-p", Path::new(self.path.as_ref()).parent().unwrap().to_str().unwrap()]));
    // FIXME: Permissions
    //    try!(create_dir_all(Path::new(&path).parent().unwrap()));
    match self.file.execute() {
      Ok(_) => {},
      Err(e) => return Err(Box::new(NodeJsSystemdUserServiceError::GenericError as NodeJsSystemdUserServiceError<io::Error>))
    }
    try!(self.command_runner.run_with_args("systemctl", &["--user", "enable", self.name]));
    try!(self.command_runner.run_with_args("systemctl", &["--user", "start", self.name]));

    if !(try!(self.check_if_service())) { return Err(Box::new(NodeJsSystemdUserServiceError::GenericError as NodeJsSystemdUserServiceError<io::Error>)); }

    let file_name = format!("{}/var/service.socket", self.home);
//    try!(self.command_runner.run_with_args("chmod", &["666", &file_name]));

    sleep(Duration::from_millis(500));
    let metadata = try!(fs::metadata(file_name.clone()));
    let mut perms = metadata.permissions();
    perms.set_mode(0o666);
    try!(fs::set_permissions(file_name, perms));

    Ok(())
  }

  fn get_prerequisites(&self) -> Vec<Box<Resource>> {
    self.file.get_prerequisites()
  }
}

impl<'a, P, C> fmt::Display for NodeJsSystemdUserService<'a, P, C> where P: AsRef<str> + fmt::Display, C: Deref<Target=str> {
  fn fmt(&self, f: &mut fmt::Formatter) -> Result<(),fmt::Error>{
    write!(f, "Systemd Node.js user service unit for {}", self.path)
  }
}