refactor send_command

- refactor the combinators,
- make it take a `&T: Serialize` instead of a Value, and
  allow sending the raw string via `send_raw_command`.

Signed-off-by: Wolfgang Bumiller <w.bumiller@proxmox.com>

src/server/command_socket.rs

@@ -2,11 +2,12 @@ use anyhow::{bail, format_err, Error};
 
 use std::collections::HashMap;
 use std::os::unix::io::AsRawFd;
-use std::path::PathBuf;
+use std::path::{PathBuf, Path};
 use std::sync::Arc;
 
 use futures::*;
 use tokio::net::UnixListener;
+use serde::Serialize;
 use serde_json::Value;
 use nix::sys::socket;
 
@@ -102,25 +103,31 @@ where
 }
 
 
-pub async fn send_command<P>(
+pub async fn send_command<P, T>(path: P, params: &T) -> Result<Value, Error>
-    path: P,
+where
-    params: Value
+    P: AsRef<Path>,
-) -> Result<Value, Error>
+    T: ?Sized + Serialize,
-    where P: Into<PathBuf>,
 {
-    let path: PathBuf = path.into();
+    let mut command_string = serde_json::to_string(params)?;
-
-    tokio::net::UnixStream::connect(path)
-        .map_err(move |err| format_err!("control socket connect failed - {}", err))
-        .and_then(move |mut conn| {
-
-            let mut command_string = params.to_string();
     command_string.push('\n');
+    send_raw_command(path.as_ref(), &command_string).await
+}
 
-            async move {
+pub async fn send_raw_command<P>(path: P, command_string: &str) -> Result<Value, Error>
+where
+    P: AsRef<Path>,
+{
     use tokio::io::{AsyncBufReadExt, AsyncWriteExt};
 
+    let mut conn = tokio::net::UnixStream::connect(path)
+        .map_err(move |err| format_err!("control socket connect failed - {}", err))
+        .await?;
+
     conn.write_all(command_string.as_bytes()).await?;
+    if !command_string.as_bytes().ends_with(b"\n") {
+        conn.write_all(b"\n").await?;
+    }
+
     AsyncWriteExt::shutdown(&mut conn).await?;
     let mut rx = tokio::io::BufReader::new(conn);
     let mut data = String::new();
@@ -137,8 +144,6 @@ pub async fn send_command<P>(
     } else {
         bail!("unable to parse response: {}", data);
     }
-            }
-        }).await
 }
 
 /// A callback for a specific commando socket.