You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
2.6 KiB
Rust
89 lines
2.6 KiB
Rust
use mediarepo_core::bromine::prelude::*;
|
|
use mediarepo_core::error::{RepoError, RepoResult};
|
|
use mediarepo_core::mediarepo_api::types::misc::InfoResponse;
|
|
use mediarepo_core::settings::Settings;
|
|
use mediarepo_core::type_keys::{RepoPathKey, SettingsKey};
|
|
use mediarepo_model::repo::Repo;
|
|
use mediarepo_model::type_keys::RepoKey;
|
|
use std::net::{IpAddr, SocketAddr};
|
|
use std::path::PathBuf;
|
|
use std::sync::Arc;
|
|
use tokio::net::TcpListener;
|
|
use tokio::task::JoinHandle;
|
|
|
|
mod from_model;
|
|
mod namespaces;
|
|
mod utils;
|
|
|
|
#[tracing::instrument(skip(settings, repo))]
|
|
pub fn start_tcp_server(
|
|
ip: IpAddr,
|
|
port_range: (u16, u16),
|
|
repo_path: PathBuf,
|
|
settings: Settings,
|
|
repo: Repo,
|
|
) -> RepoResult<(String, JoinHandle<()>)> {
|
|
let port = port_check::free_local_port_in_range(port_range.0, port_range.1)
|
|
.ok_or_else(|| RepoError::PortUnavailable)?;
|
|
let address = SocketAddr::new(ip, port);
|
|
let address_string = address.to_string();
|
|
|
|
let join_handle = tokio::task::Builder::new()
|
|
.name("mediarepo_tcp::listen")
|
|
.spawn(async move {
|
|
get_builder::<TcpListener>(address)
|
|
.insert::<RepoKey>(Arc::new(repo))
|
|
.insert::<SettingsKey>(settings)
|
|
.insert::<RepoPathKey>(repo_path)
|
|
.build_server()
|
|
.await
|
|
.expect("Failed to start tcp server")
|
|
});
|
|
|
|
Ok((address_string, join_handle))
|
|
}
|
|
|
|
#[cfg(unix)]
|
|
#[tracing::instrument(skip(settings, repo))]
|
|
pub fn create_unix_socket(
|
|
path: std::path::PathBuf,
|
|
repo_path: PathBuf,
|
|
settings: Settings,
|
|
repo: Repo,
|
|
) -> RepoResult<JoinHandle<()>> {
|
|
use std::fs;
|
|
use tokio::net::UnixListener;
|
|
|
|
if path.exists() {
|
|
fs::remove_file(&path)?;
|
|
}
|
|
let join_handle = tokio::task::Builder::new()
|
|
.name("mediarepo_unix_socket::listen")
|
|
.spawn(async move {
|
|
get_builder::<UnixListener>(path)
|
|
.insert::<RepoKey>(Arc::new(repo))
|
|
.insert::<SettingsKey>(settings)
|
|
.insert::<RepoPathKey>(repo_path)
|
|
.build_server()
|
|
.await
|
|
.expect("Failed to create unix domain socket");
|
|
});
|
|
|
|
Ok(join_handle)
|
|
}
|
|
|
|
fn get_builder<L: AsyncStreamProtocolListener>(address: L::AddressType) -> IPCBuilder<L> {
|
|
namespaces::build_namespaces(IPCBuilder::new().address(address)).on("info", callback!(info))
|
|
}
|
|
|
|
#[tracing::instrument(skip_all)]
|
|
async fn info(ctx: &Context, _: Event) -> IPCResult<()> {
|
|
let response = InfoResponse::new(
|
|
env!("CARGO_PKG_NAME").to_string(),
|
|
env!("CARGO_PKG_VERSION").to_string(),
|
|
);
|
|
ctx.emit("info", response).await?;
|
|
|
|
Ok(())
|
|
}
|