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; #[derive(Debug)] pub enum NodeJsSystemdUserServiceError { ActivationFailed(io::Result), ExecError(E), GenericError } impl From for NodeJsSystemdUserServiceError { fn from(err: io::Error) -> NodeJsSystemdUserServiceError { NodeJsSystemdUserServiceError::ExecError(err) } } impl Error for NodeJsSystemdUserServiceError { 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 fmt::Display for NodeJsSystemdUserServiceError { 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 + fmt::Display, C: Deref { name: &'a str, home: &'a str, path: P, command_runner: &'a CommandRunner, file: FileSymbol> } 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 = 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 + fmt::Display, C: Deref { fn check_if_service(&self) -> Result> { 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)), _ => return Ok(false) } } } } impl<'a, P, C> Symbol for NodeJsSystemdUserService<'a, P, C> where P: AsRef + fmt::Display, C: Deref { fn target_reached(&self) -> Result> { match self.file.target_reached() { Ok(false) => return Ok(false), Ok(true) => {}, Err(e) => return Err(Box::new(NodeJsSystemdUserServiceError::GenericError as NodeJsSystemdUserServiceError)) } /* if !(try!(self.file.target_reached())) { return Ok(false) } */ self.check_if_service() } fn execute(&self) -> Result<(), Box> { 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)) } 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)); } 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(()) } } impl<'a, P, C> fmt::Display for NodeJsSystemdUserService<'a, P, C> where P: AsRef + fmt::Display, C: Deref { fn fmt(&self, f: &mut fmt::Formatter) -> Result<(),fmt::Error>{ write!(f, "Systemd Node.js user service unit for {}", self.path) } }