git-next/src/server/mod.rs

110 lines
3.6 KiB
Rust
Raw Normal View History

mod actors;
2024-04-07 13:47:39 +01:00
mod config;
2024-04-07 20:16:04 +01:00
use actix::prelude::*;
use std::path::PathBuf;
use tracing::{error, info};
2024-04-07 20:16:04 +01:00
use crate::{
filesystem::FileSystem,
server::config::{ForgeName, RepoName},
};
2024-04-07 16:09:16 +01:00
pub fn init(fs: FileSystem) {
let file_name = "git-next-server.toml";
let path = PathBuf::from(file_name);
if fs.file_exists(&path) {
eprintln!(
"The configuration file already exists at {} - not overwritting it.",
file_name
);
} else {
match fs.write_file(file_name, 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)
}
}
}
}
2024-04-07 16:09:16 +01:00
pub fn start(fs: FileSystem) {
let Ok(_) = init_logging() else {
eprintln!("Failed to initialize logging.");
return;
};
info!("Starting Server...");
let config = match config::Config::load(&fs) {
Ok(config) => config,
Err(err) => {
error!("Failed to load config file. Error: {}", err);
return;
}
};
info!("Config loaded");
2024-04-07 19:08:23 +01:00
info!("Loaded config");
2024-04-07 20:16:04 +01:00
let mut actors: Vec<(ForgeName, RepoName, actors::repo::RepoActor)> = vec![];
config.forges().for_each(|(forge_name, forge)| {
let forge_name = ForgeName(forge_name.clone());
let span = tracing::info_span!("Forge", %forge_name);
2024-04-07 19:08:23 +01:00
let _guard = span.enter();
2024-04-07 20:16:04 +01:00
info!("Forge: {}", forge_name);
2024-04-07 19:08:23 +01:00
info!("Forge Type: {}", forge.forge_type());
info!("Hostname: {}", forge.hostname());
info!("User: {}", forge.user());
2024-04-07 20:16:04 +01:00
forge.repos().for_each(|(repo_name, path)| {
let repo_name = RepoName(repo_name.clone());
let span = tracing::info_span!("Repo", %repo_name);
let _guard = span.enter();
info!("Repo: {} - {}", repo_name, path);
let actor = actors::repo::RepoActor {
2024-04-07 20:56:20 +01:00
details: config::RepoDetails {
2024-04-07 20:16:04 +01:00
name: repo_name.clone(),
path: config::RepoPath(path.clone()),
forge: (&forge_name, forge).into(),
},
};
actors.push((forge_name.clone(), repo_name.clone(), actor));
info!("Created actor for repo: {} - {}", repo_name, path);
});
});
info!("Sending StartRepo to all actors...");
actix::System::new().block_on(async {
let mut addresses: Vec<Addr<actors::repo::RepoActor>> = vec![];
actors.into_iter().for_each(|actor| {
let (forge_name, repo_name, actor) = actor;
let span = tracing::info_span!("Forge/Repo", %forge_name, %repo_name);
2024-04-07 19:08:23 +01:00
let _guard = span.enter();
2024-04-07 20:16:04 +01:00
info!("Starting actor");
let addr = actor.start();
info!("Sending StartRepo to actor");
addr.do_send(actors::repo::StartRepo);
addresses.push(addr);
info!("Sent StartRepo to actor");
2024-04-07 19:08:23 +01:00
});
2024-04-07 20:16:04 +01:00
let _ = actix_rt::signal::ctrl_c().await;
info!("Ctrl-C received, shutting down...");
drop(addresses);
2024-04-07 19:08:23 +01:00
});
}
pub fn init_logging() -> Result<(), tracing::subscriber::SetGlobalDefaultError> {
use tracing_subscriber::prelude::*;
let console_layer = console_subscriber::ConsoleLayer::builder().spawn();
let subscriber = tracing_subscriber::fmt::layer()
// NOTE: set RUSTLOG in ${root}/.cargo/config
.with_target(false)
.with_file(true)
.with_line_number(true);
tracing_subscriber::registry()
.with(console_layer)
.with(subscriber)
.init();
Ok(())
}