forked from InfoshockTech/xcrab
Akshat Deshpande
a3a4b21981
Reviewed-on: InfoshockTech/xcrab#13 Co-authored-by: Akshat Deshpande <akshatd18@outlook.com> Co-committed-by: Akshat Deshpande <akshatd18@outlook.com>
68 lines
1.9 KiB
Rust
68 lines
1.9 KiB
Rust
// Copyright (C) 2022 Infoshock Tech
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
#![warn(clippy::pedantic)]
|
|
|
|
mod config;
|
|
|
|
use std::error::Error;
|
|
use std::fmt::{Debug, Display, Formatter, Result as FmtResult};
|
|
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
|
use tokio::net::UnixStream;
|
|
|
|
type Result<T> = std::result::Result<T, Box<dyn Error>>;
|
|
|
|
struct CustomError(String);
|
|
|
|
impl Debug for CustomError {
|
|
fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
|
|
f.write_str(&self.0)
|
|
}
|
|
}
|
|
|
|
impl Display for CustomError {
|
|
fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
|
|
f.write_str(&self.0)
|
|
}
|
|
}
|
|
|
|
impl Error for CustomError {}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
let msg = std::env::args().skip(1).collect::<Vec<String>>().join(" ");
|
|
|
|
let conf = config::load_file();
|
|
|
|
let path = conf.msg.socket_path;
|
|
|
|
let stream = UnixStream::connect(path).await?;
|
|
|
|
let (mut read, mut write) = stream.into_split();
|
|
|
|
write.write_all(msg.as_bytes()).await?;
|
|
drop(write); // Shutdown the writer half so that the write actually goes through
|
|
// "Don't cross the streams!""
|
|
|
|
let mut buf = String::new();
|
|
|
|
read.read_to_string(&mut buf).await?;
|
|
if !buf.is_empty() {
|
|
return Err(CustomError(buf).into());
|
|
}
|
|
|
|
Ok(())
|
|
}
|