Browse Source

Update

master
Adrian Heine 8 years ago
parent
commit
27e6531119
  1. 29
      src/command_runner.rs
  2. 7
      src/schema.rs
  3. 94
      src/symbols/file.rs
  4. 8
      src/symbols/mod.rs
  5. 1
      src/symbols/systemd/mod.rs
  6. 124
      src/symbols/systemd/node_js_user_service.rs
  7. 16
      src/symbols/systemd/user_session.rs
  8. 24
      src/symbols/user.rs

29
src/command_runner.rs

@ -11,7 +11,34 @@ pub struct StdCommandRunner;
impl CommandRunner for StdCommandRunner { impl CommandRunner for StdCommandRunner {
fn run_with_args(&self, program: &str, args: &[&str]) -> IoResult<Output> { fn run_with_args(&self, program: &str, args: &[&str]) -> IoResult<Output> {
Command::new(program).args(args).output()
// FIXME: logger
println!("{} {:?}", program, args);
let res = Command::new(program).args(args).output();
println!("{:?}", res);
res
}
}
#[derive(Debug)]
pub struct UserCommandRunner<'a, C> where C: 'a + CommandRunner {
command_runner: &'a C,
user_name: &'a str
}
impl<'a, C> UserCommandRunner<'a, C> where C: 'a + CommandRunner {
pub fn new(user_name: &'a str, command_runner: &'a C) -> UserCommandRunner<'a, C> {
UserCommandRunner {
command_runner: command_runner,
user_name: user_name
}
}
}
impl<'a, C> CommandRunner for UserCommandRunner<'a, C> where C: 'a + CommandRunner {
fn run_with_args(&self, program: &str, args: &[&str]) -> IoResult<Output> {
let mut new_args = vec![self.user_name, "-s", "/usr/bin/env", "--", program];
new_args.extend_from_slice(args);
self.command_runner.run_with_args("su", &new_args)
} }
} }

7
src/schema.rs

@ -1,6 +1,5 @@
use std::error::Error; use std::error::Error;
use std::fmt; use std::fmt;
use std::io::Write;
use loggers::Logger; use loggers::Logger;
use symbols::Symbol; use symbols::Symbol;
@ -42,14 +41,14 @@ impl<E: Error> From<E> for SymbolRunError<E> {
} }
pub trait SymbolRunner { pub trait SymbolRunner {
fn run_symbol<S: Symbol + fmt::Display>(&self, logger: &mut Logger, symbol: &S) -> Result<(), SymbolRunError<S::Error>>
fn run_symbol<S: ?Sized + Symbol + fmt::Display>(&self, logger: &mut Logger, symbol: &S) -> Result<(), SymbolRunError<S::Error>>
where S::Error: Error; where S::Error: Error;
} }
pub struct InitializingSymbolRunner; pub struct InitializingSymbolRunner;
impl SymbolRunner for InitializingSymbolRunner { impl SymbolRunner for InitializingSymbolRunner {
fn run_symbol<S: Symbol + fmt::Display>(&self, logger: &mut Logger, symbol: &S) -> Result<(), SymbolRunError<S::Error>>
fn run_symbol<S: ?Sized + Symbol + fmt::Display>(&self, logger: &mut Logger, symbol: &S) -> Result<(), SymbolRunError<S::Error>>
where S::Error: Error where S::Error: Error
{ {
let target_reached = try!(symbol.target_reached()); let target_reached = try!(symbol.target_reached());
@ -72,7 +71,7 @@ impl SymbolRunner for InitializingSymbolRunner {
pub struct DrySymbolRunner; pub struct DrySymbolRunner;
impl SymbolRunner for DrySymbolRunner { impl SymbolRunner for DrySymbolRunner {
fn run_symbol<S: Symbol + fmt::Display>(&self, logger: &mut Logger, symbol: &S) -> Result<(), SymbolRunError<S::Error>>
fn run_symbol<S: ?Sized + Symbol + fmt::Display>(&self, logger: &mut Logger, symbol: &S) -> Result<(), SymbolRunError<S::Error>>
where S::Error: Error where S::Error: Error
{ {
let target_reached = try!(symbol.target_reached()); let target_reached = try!(symbol.target_reached());

94
src/symbols/file.rs

@ -0,0 +1,94 @@
use std::error::Error;
use std::fmt;
use std::io;
use std::ops::Deref;
use symbols::Symbol;
#[derive(Debug)]
pub enum FileError<E: Error> {
ExecError(E),
GenericError
}
impl From<io::Error> for FileError<io::Error> {
fn from(err: io::Error) -> FileError<io::Error> {
FileError::ExecError(err)
}
}
impl<E: Error> Error for FileError<E> {
fn description(&self) -> &str {
match self {
&FileError::ExecError(ref e) => e.description(),
&FileError::GenericError => "Generic error"
}
}
fn cause(&self) -> Option<&Error> {
match self {
&FileError::ExecError(ref e) => Some(e),
_ => None
}
}
}
impl<E: Error> fmt::Display for FileError<E> {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
write!(f, "{}", self.description())
}
}
use std::convert::AsRef;
pub struct File<C, D> where C: Deref<Target=str>, D: AsRef<str> + fmt::Display {
path: D,
content: C
}
impl<C, D> File<C, D> where C: Deref<Target=str>, D: AsRef<str> + fmt::Display {
pub fn new(path: D, content: C) -> Self {
File {
path: path,
content: content
}
}
}
use std::fs::File as FsFile;
use std::io::{Read, Write};
impl<C, D> Symbol for File<C, D> where C: Deref<Target=str>, D: AsRef<str> + fmt::Display {
type Error = FileError<io::Error>;
fn target_reached(&self) -> Result<bool, Self::Error> {
let file = FsFile::open(self.path.as_ref());
// Check if file exists
if let Err(e) = file {
return if e.kind() == io::ErrorKind::NotFound {
Ok(false)
} else {
Err(e.into())
};
}
// Check if content is the same
let file_content = file.unwrap().bytes();
let content_equal = try!(self.content.bytes().zip(file_content).fold(
Ok(true),
|state, (target_byte, file_byte_option)| state.and_then(|s| file_byte_option.map(|file_byte| s && file_byte == target_byte))
));
return Ok(content_equal)
}
fn execute(&self) -> Result<(), Self::Error> {
//try!(self.command_runner.run_with_args("mkdir", &["-p", Path::new(&path).parent().unwrap().to_str().unwrap()]));
// FIXME: Permissions
// try!(create_dir_all(Path::new(&path).parent().unwrap()));
try!(try!(FsFile::create(self.path.as_ref())).write_all(self.content.as_bytes()));
Ok(())
}
}
impl<C, D> fmt::Display for File<C, D> where C: Deref<Target=str>, D: AsRef<str> + fmt::Display {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(),fmt::Error>{
write!(f, "File {}", self.path)
}
}

8
src/symbols/mod.rs

@ -1,9 +1,13 @@
pub trait Symbol {
type Error;
use std::error::Error;
use std::fmt::Display;
pub trait Symbol: Display {
type Error: Error;
fn target_reached(&self) -> Result<bool, Self::Error>; fn target_reached(&self) -> Result<bool, Self::Error>;
fn execute(&self) -> Result<(), Self::Error>; fn execute(&self) -> Result<(), Self::Error>;
} }
pub mod file;
pub mod git; pub mod git;
pub mod user; pub mod user;
pub mod systemd; pub mod systemd;

1
src/symbols/systemd/mod.rs

@ -1 +1,2 @@
pub mod user_session; pub mod user_session;
pub mod node_js_user_service;

124
src/symbols/systemd/node_js_user_service.rs

@ -0,0 +1,124 @@
use std::error::Error;
use std::fmt;
use std::io;
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<E: Error> {
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"
}
}
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> {
write!(f, "{}", self.description())
}
}
pub struct NodeJsSystemdUserService<'a, P, C> where P: AsRef<str> + fmt::Display, C: Deref<Target=str> {
name: &'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(name: &'a str, path: &'a str, command_runner: &'a CommandRunner) -> Self {
let home = String::from_utf8(command_runner.run_with_args("sh", &["-c", "echo \"$HOME\""]).unwrap().stdout).unwrap();
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,
path: file_path.clone(),
command_runner: command_runner,
file: FileSymbol::new(file_path, content)
}
}
}
impl<'a, P, C> Symbol for NodeJsSystemdUserService<'a, P, C> where P: AsRef<str> + fmt::Display, C: Deref<Target=str> {
type Error = NodeJsSystemdUserServiceError<io::Error>;
fn target_reached(&self) -> Result<bool, Self::Error> {
match self.file.target_reached() {
Ok(false) => return Ok(false),
Ok(true) => {},
Err(e) => return Err(NodeJsSystemdUserServiceError::GenericError)
}
/*
if !(try!(self.file.target_reached())) {
return Ok(false)
}
*/
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),
_ => return Ok(false)
}
}
}
fn execute(&self) -> Result<(), Self::Error> {
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(NodeJsSystemdUserServiceError::GenericError)
}
try!(self.command_runner.run_with_args("systemctl", &["--user", "enable", self.name]));
try!(self.command_runner.run_with_args("systemctl", &["--user", "start", self.name]));
Ok(())
}
}
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)
}
}

16
src/symbols/systemd/user_session.rs

@ -2,24 +2,28 @@ use std::error::Error;
use std::fmt; use std::fmt;
use std::io::Error as IoError; use std::io::Error as IoError;
use std::path::PathBuf; use std::path::PathBuf;
use std::str::from_utf8;
use command_runner::CommandRunner; use command_runner::CommandRunner;
use symbols::Symbol; use symbols::Symbol;
#[derive(Debug)] #[derive(Debug)]
pub enum SystemdUserSessionError<E: Error> { pub enum SystemdUserSessionError<E: Error> {
ExecError(E)
ExecError(E),
GenericError
} }
impl<E: Error> Error for SystemdUserSessionError<E> { impl<E: Error> Error for SystemdUserSessionError<E> {
fn description(&self) -> &str { fn description(&self) -> &str {
match self { match self {
&SystemdUserSessionError::ExecError(ref e) => e.description()
&SystemdUserSessionError::ExecError(ref e) => e.description(),
&SystemdUserSessionError::GenericError => "Generic error"
} }
} }
fn cause(&self) -> Option<&Error> { fn cause(&self) -> Option<&Error> {
match self { match self {
&SystemdUserSessionError::ExecError(ref e) => Some(e)
&SystemdUserSessionError::ExecError(ref e) => Some(e),
_ => None
} }
} }
} }
@ -50,11 +54,15 @@ impl<'a> Symbol for SystemdUserSession<'a> {
let mut path = PathBuf::from("/var/lib/systemd/linger"); let mut path = PathBuf::from("/var/lib/systemd/linger");
path.push(self.user_name); path.push(self.user_name);
Ok(path.exists()) Ok(path.exists())
// Could also do `loginctl show-user ${self.user_name} | grep -F 'Linger=yes`
} }
fn execute(&self) -> Result<(), Self::Error> { fn execute(&self) -> Result<(), Self::Error> {
match self.command_runner.run_with_args("loginctl", &["enable-linger", self.user_name]) { match self.command_runner.run_with_args("loginctl", &["enable-linger", self.user_name]) {
Ok(_) => Ok(()),
Ok(output) => { println!("{:?} {:?}", from_utf8(&output.stdout).unwrap(), from_utf8(&output.stderr).unwrap() ); match output.status.code() {
Some(0) => Ok(()),
_ => Err(SystemdUserSessionError::GenericError)
} },
Err(e) => Err(SystemdUserSessionError::ExecError(e)) Err(e) => Err(SystemdUserSessionError::ExecError(e))
} }
} }

24
src/symbols/user.rs

@ -126,13 +126,29 @@ impl<'a> SystemUserAdder<'a> {
impl<'a> UserAdder for SystemUserAdder<'a> { impl<'a> UserAdder for SystemUserAdder<'a> {
type SubE = IoError; type SubE = IoError;
fn add_user(&self, user_name: &str) -> Result<(), UserAdderError<IoError>> { fn add_user(&self, user_name: &str) -> Result<(), UserAdderError<IoError>> {
let output = self.command_runner.run_with_args("adduser", &["--system", "--disabled-login", "--disabled-password", user_name]);
let output = self.command_runner.run_with_args(
"adduser",
&[
// "-m", // Necessary for Fedora, not accepted in Debian
"--system",
user_name
]);
match output { match output {
Ok(output) => match output.status.code() { Ok(output) => match output.status.code() {
Some(0) => Ok(()), Some(0) => Ok(()),
Some(1) => Err(UserAdderError::AlreadyExists),
Some(_) => Err(UserAdderError::UnknownError),
None => Err(UserAdderError::UnknownError),
Some(1) =>
{
println!("{:?}", output);
Err(UserAdderError::AlreadyExists)},
Some(_) =>
{
println!("{:?}", output);
Err(UserAdderError::UnknownError)
},
None => {
println!("{:?}", output);
Err(UserAdderError::UnknownError)
},
}, },
Err(e) => Err(UserAdderError::ImplError(e)) Err(e) => Err(UserAdderError::ImplError(e))
} }

Loading…
Cancel
Save