git-next/src/server/mod.rs

197 lines
5.8 KiB
Rust
Raw Normal View History

mod actors;
2024-04-07 13:47:39 +01:00
mod config;
pub mod gitforge;
pub mod types;
2024-04-07 13:47:39 +01:00
2024-04-07 20:16:04 +01:00
use actix::prelude::*;
2024-04-09 10:44:01 +01:00
use kxio::network::Network;
2024-04-07 20:16:04 +01:00
use std::path::PathBuf;
use tracing::{error, info, level_filters::LevelFilter};
2024-04-07 20:16:04 +01:00
use crate::{
2024-04-28 08:05:09 +01:00
fs::FileSystem,
server::{
actors::webhook,
2024-04-23 07:09:30 +01:00
config::{ForgeConfig, ForgeName, RepoAlias, ServerStorage, Webhook},
},
2024-04-07 20:16:04 +01:00
};
use self::{actors::repo::RepoActor, config::ServerRepoConfig};
#[derive(Debug, derive_more::Display, derive_more::From)]
pub enum Error {
#[display("Failed to create data directories")]
FailedToCreateDataDirectory(kxio::fs::Error),
#[display("The forge data path is not a directory: {path:?}")]
ForgeDirIsNotDirectory { path: PathBuf },
}
type Result<T> = core::result::Result<T, Error>;
2024-04-07 16:09:16 +01:00
pub fn init(fs: FileSystem) {
let file_name = "git-next-server.toml";
2024-04-28 08:05:09 +01:00
let pathbuf = PathBuf::from(file_name);
let Ok(exists) = fs.path_exists(&pathbuf) else {
eprintln!("Could not check if file exist: {}", file_name);
return;
};
if exists {
eprintln!(
"The configuration file already exists at {} - not overwritting it.",
file_name
);
} else {
2024-04-28 08:05:09 +01:00
match fs.file_write(&pathbuf, include_str!("../../server-default.toml")) {
Ok(_) => println!("Created a default configuration file at {}", file_name),
Err(e) => {
eprintln!("Failed to write to the configuration file: {}", e)
}
}
}
}
pub async fn start(fs: FileSystem, net: Network) {
init_logging();
info!("Starting Server...");
let server_config = match config::ServerConfig::load(&fs) {
Ok(server_config) => server_config,
Err(err) => {
error!("Failed to load config file. Error: {}", err);
return;
}
};
// create data dir if missing
let dir = server_config.storage().path();
if !dir.exists() {
if let Err(err) = fs.dir_create(dir) {
error!(?err, ?dir, "Failed to create server storage directory");
return;
}
}
let webhook_router = webhook::WebhookRouter::new().start();
let webhook = server_config.webhook();
2024-04-23 07:09:30 +01:00
let server_storage = server_config.storage();
if let Err(err) = create_forge_data_directories(&server_config, &fs, &dir) {
error!(?err, "Failure creating forge data directories");
return;
}
server_config
.forges()
2024-04-23 07:13:16 +01:00
.flat_map(|(forge_name, forge_config)| {
create_forge_repos(forge_config, forge_name, server_storage, webhook, &net)
2024-04-23 07:09:30 +01:00
})
.map(start_actor)
.map(|(alias, addr)| webhook::AddWebhookRecipient(alias, addr.recipient()))
.for_each(|msg| webhook_router.do_send(msg));
let webhook_server = webhook::WebhookActor::new(webhook_router.recipient()).start();
let _ = actix_rt::signal::ctrl_c().await;
info!("Ctrl-C received, shutting down...");
drop(webhook_server);
}
fn create_forge_data_directories(
server_config: &config::ServerConfig,
fs: &FileSystem,
server_dir: &&std::path::Path,
) -> Result<()> {
for (forge_name, _forge_config) in server_config.forges() {
let forge_dir: PathBuf = (&forge_name).into();
let path = server_dir.join(&forge_dir);
if fs.path_exists(&path)? {
if !fs.path_is_dir(&path)? {
return Err(Error::ForgeDirIsNotDirectory { path });
}
} else {
fs.dir_create_all(&path)?;
}
}
Ok(())
}
fn create_forge_repos(
2024-04-23 07:13:16 +01:00
forge_config: &ForgeConfig,
forge_name: ForgeName,
2024-04-23 07:09:30 +01:00
server_storage: &ServerStorage,
webhook: &Webhook,
2024-04-09 10:44:01 +01:00
net: &Network,
) -> Vec<(ForgeName, RepoAlias, RepoActor)> {
2024-04-23 07:13:16 +01:00
let span = tracing::info_span!("Forge", %forge_name, %forge_config);
let _guard = span.enter();
info!("Creating Forge");
2024-04-23 07:13:16 +01:00
forge_config
.repos()
2024-04-23 07:09:30 +01:00
.map(create_actor(
forge_name,
2024-04-23 07:13:16 +01:00
forge_config.clone(),
2024-04-23 07:09:30 +01:00
server_storage,
webhook,
net,
))
.collect::<Vec<_>>()
}
fn create_actor(
forge_name: ForgeName,
2024-04-23 07:13:16 +01:00
forge_config: config::ForgeConfig,
2024-04-23 07:09:30 +01:00
server_storage: &ServerStorage,
webhook: &Webhook,
2024-04-09 10:44:01 +01:00
net: &Network,
) -> impl Fn((RepoAlias, &ServerRepoConfig)) -> (ForgeName, RepoAlias, RepoActor) {
2024-04-23 07:09:30 +01:00
let server_storage = server_storage.clone();
let webhook = webhook.clone();
2024-04-09 10:44:01 +01:00
let net = net.clone();
move |(repo_name, server_repo_config)| {
let span = tracing::info_span!("Repo", %repo_name, %server_repo_config);
let _guard = span.enter();
info!("Creating Repo");
2024-04-09 10:44:01 +01:00
let actor = actors::repo::RepoActor::new(
2024-04-23 07:09:30 +01:00
config::RepoDetails::new(
&repo_name,
server_repo_config,
&forge_name,
2024-04-23 07:13:16 +01:00
&forge_config,
2024-04-23 07:09:30 +01:00
&server_storage,
),
webhook.clone(),
2024-04-09 10:44:01 +01:00
net.clone(),
);
info!("Created Repo");
(forge_name.clone(), repo_name, actor)
}
}
fn start_actor(
actor: (ForgeName, RepoAlias, actors::repo::RepoActor),
) -> (RepoAlias, Addr<actors::repo::RepoActor>) {
let (forge_name, repo_alias, actor) = actor;
let span = tracing::info_span!("Forge/Repo", %forge_name, %repo_alias);
let _guard = span.enter();
info!("Starting");
let addr = actor.start();
2024-04-23 07:09:30 +01:00
addr.do_send(actors::repo::CloneRepo);
info!("Started");
(repo_alias, addr)
}
pub fn init_logging() {
use tracing_subscriber::prelude::*;
let subscriber = tracing_subscriber::fmt::layer()
// NOTE: set RUSTLOG in ${root}/.cargo/config
.with_target(false)
.with_file(true)
.with_line_number(true)
.with_filter(LevelFilter::INFO);
tracing_subscriber::registry()
.with(console_subscriber::ConsoleLayer::builder().spawn())
.with(subscriber)
.init();
}