diff options
author | Jesse Luehrs <doy@tozt.net> | 2020-05-03 03:30:59 -0400 |
---|---|---|
committer | Jesse Luehrs <doy@tozt.net> | 2020-05-03 03:30:59 -0400 |
commit | 6ebf7d55e4c553870306a70092cfb677c17429b9 (patch) | |
tree | 2d20288e44fa627159d0a4e578f0fb506d16f921 /src/bin | |
parent | 36b5107239b75136ce74799d87a645e153d95948 (diff) | |
download | rbw-6ebf7d55e4c553870306a70092cfb677c17429b9.tar.gz rbw-6ebf7d55e4c553870306a70092cfb677c17429b9.zip |
simplify
Diffstat (limited to 'src/bin')
-rw-r--r-- | src/bin/rbw-agent/daemon.rs | 15 | ||||
-rw-r--r-- | src/bin/rbw-agent/sock.rs | 6 | ||||
-rw-r--r-- | src/bin/rbw/actions.rs | 3 | ||||
-rw-r--r-- | src/bin/rbw/sock.rs | 2 |
4 files changed, 7 insertions, 19 deletions
diff --git a/src/bin/rbw-agent/daemon.rs b/src/bin/rbw-agent/daemon.rs index f87d0ec..923a217 100644 --- a/src/bin/rbw-agent/daemon.rs +++ b/src/bin/rbw-agent/daemon.rs @@ -1,5 +1,3 @@ -use anyhow::Context as _; - pub struct StartupAck { writer: std::os::unix::io::RawFd, } @@ -20,25 +18,20 @@ impl Drop for StartupAck { } pub fn daemonize() -> anyhow::Result<StartupAck> { - let runtime_dir = rbw::dirs::runtime_dir(); - std::fs::create_dir_all(&runtime_dir) - .context("failed to create runtime directory")?; + rbw::dirs::make_all()?; - let data_dir = rbw::dirs::data_dir(); - std::fs::create_dir_all(&data_dir) - .context("failed to create data directory")?; let stdout = std::fs::OpenOptions::new() .append(true) .create(true) - .open(data_dir.join("agent.out"))?; + .open(rbw::dirs::agent_stdout_file())?; let stderr = std::fs::OpenOptions::new() .append(true) .create(true) - .open(data_dir.join("agent.err"))?; + .open(rbw::dirs::agent_stderr_file())?; let (r, w) = nix::unistd::pipe()?; let res = daemonize::Daemonize::new() - .pid_file(runtime_dir.join("pidfile")) + .pid_file(rbw::dirs::pid_file()) .stdout(stdout) .stderr(stderr) .exit_action(move || { diff --git a/src/bin/rbw-agent/sock.rs b/src/bin/rbw-agent/sock.rs index bb2d6fb..311176c 100644 --- a/src/bin/rbw-agent/sock.rs +++ b/src/bin/rbw-agent/sock.rs @@ -43,11 +43,7 @@ impl Sock { } pub fn listen() -> anyhow::Result<tokio::net::UnixListener> { - let runtime_dir = rbw::dirs::runtime_dir(); - std::fs::create_dir_all(&runtime_dir) - .context("failed to create runtime dir")?; - - let path = runtime_dir.join("socket"); + let path = rbw::dirs::socket_file(); // if the socket already doesn't exist, that's fine let _ = std::fs::remove_file(&path); let sock = tokio::net::UnixListener::bind(&path) diff --git a/src/bin/rbw/actions.rs b/src/bin/rbw/actions.rs index 6de2fc6..01f448f 100644 --- a/src/bin/rbw/actions.rs +++ b/src/bin/rbw/actions.rs @@ -20,8 +20,7 @@ pub fn lock() -> anyhow::Result<()> { pub fn quit() -> anyhow::Result<()> { match crate::sock::Sock::connect() { Ok(mut sock) => { - let runtime_dir = rbw::dirs::runtime_dir(); - let pidfile = runtime_dir.join("pidfile"); + let pidfile = rbw::dirs::pid_file(); let mut pid = String::new(); std::fs::File::open(pidfile)?.read_to_string(&mut pid)?; let pid = nix::unistd::Pid::from_raw(pid.parse()?); diff --git a/src/bin/rbw/sock.rs b/src/bin/rbw/sock.rs index 1ba13ec..b3ff1d6 100644 --- a/src/bin/rbw/sock.rs +++ b/src/bin/rbw/sock.rs @@ -6,7 +6,7 @@ pub struct Sock(std::os::unix::net::UnixStream); impl Sock { pub fn connect() -> std::io::Result<Self> { Ok(Self(std::os::unix::net::UnixStream::connect( - rbw::dirs::runtime_dir().join("socket"), + rbw::dirs::socket_file(), )?)) } |