This commit is contained in:
Adrian Heine 2019-09-19 13:48:44 +02:00
parent 6190b0e373
commit 7d6329a409
4 changed files with 137 additions and 9 deletions

77
src/symbols/cron.rs Normal file
View file

@ -0,0 +1,77 @@
use std::error::Error;
use std::fmt;
use std::ops::Deref;
use command_runner::CommandRunner;
use resources::Resource;
use symbols::{Action, OwnedSymbolAction, Symbol, SymbolAction, SymbolRunner};
pub struct Cron<'r, C, U, R: 'r + CommandRunner>
where
C: Deref<Target = str>,
U: Deref<Target = str>,
{
user: U,
content: C,
command_runner: &'r R,
}
impl<'r, U, R: 'r + CommandRunner> Cron<'r, String, U, R>
where
U: Deref<Target = str>,
{
pub fn new<C: Deref<Target = str>>(user: U, content: C, command_runner: &'r R) -> Self {
Cron {
user,
content: String::from(&*content) + "\n",
command_runner,
}
}
}
impl<'r, C, U, R: 'r + CommandRunner> Symbol for Cron<'r, C, U, R>
where
C: Deref<Target = str>,
U: Deref<Target = str>,
{
fn target_reached(&self) -> Result<bool, Box<dyn Error>> {
let tab = try!(self
.command_runner
.get_output("crontab", &["-l", "-u", &self.user]));
return Ok(tab == self.content.bytes().collect::<Vec<u8>>());
}
fn execute(&self) -> Result<(), Box<dyn Error>> {
self.command_runner.run_with_args_and_stdin(
"crontab",
&["-u", &self.user, "-"],
&self.content,
)?;
Ok(())
}
fn get_prerequisites(&self) -> Vec<Resource> {
vec![]
}
fn as_action<'a>(&'a self, runner: &'a dyn SymbolRunner) -> Box<dyn Action + 'a> {
Box::new(SymbolAction::new(runner, self))
}
fn into_action<'a>(self: Box<Self>, runner: &'a dyn SymbolRunner) -> Box<dyn Action + 'a>
where
Self: 'a,
{
Box::new(OwnedSymbolAction::new(runner, *self))
}
}
impl<'r, C, U, R: 'r + CommandRunner> fmt::Display for Cron<'r, C, U, R>
where
C: Deref<Target = str>,
U: Deref<Target = str>,
{
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
write!(f, "Cron {}", &*self.user)
}
}

View file

@ -5,6 +5,7 @@ use std::path::Path;
use command_runner::{CommandRunner, SetuidCommandRunner};
use storage::{SimpleStorage, Storage};
use symbols::acme::{AcmeCert, AcmeCertChain};
use symbols::cron::Cron;
use symbols::file::File;
use symbols::git::checkout::GitCheckout;
use symbols::hook::Hook;
@ -368,4 +369,12 @@ env[PATH] = /usr/local/bin:/usr/bin:/bin
) -> Box<dyn Action + 'a> {
Box::new(File::new(path, content)).into_action(self.symbol_runner)
}
pub fn get_cron<'a, T: 'a + Deref<Target = str>, U: 'a + Deref<Target = str>>(
&'a self,
user: U,
content: T,
) -> Box<dyn Action + 'a> {
Box::new(Cron::new(user, content, self.command_runner)).into_action(self.symbol_runner)
}
}

View file

@ -68,6 +68,7 @@ impl<'a, S: Symbol + 'a> Action for OwnedSymbolAction<'a, S> {
}
pub mod acme;
pub mod cron;
pub mod dir;
pub mod factory;
pub mod file;