mirror of
https://github.com/Start9Labs/start-os.git
synced 2026-04-01 04:53:40 +00:00
29e8210782
* wip: enabling support for wireguard and firewall * wip * wip * wip * wip * wip * implement some things * fix warning * wip * alpha.23 * misc fixes * remove ufw since no longer required * remove debug info * add cli bindings * debugging * fixes * individualized acme and privacy settings for domains and bindings * sdk version bump * migration * misc fixes * refactor Host::update * debug info * refactor webserver * misc fixes * misc fixes * refactor port forwarding * recheck interfaces every 5 min if no dbus event * misc fixes and cleanup * misc fixes
103 lines
3.1 KiB
Rust
103 lines
3.1 KiB
Rust
use std::path::Path;
|
|
use std::sync::Arc;
|
|
|
|
use rpc_toolkit::yajrc::RpcError;
|
|
use rpc_toolkit::{
|
|
from_fn, from_fn_async, CallRemoteHandler, Context, Empty, HandlerExt, ParentHandler,
|
|
};
|
|
|
|
use crate::context::{CliContext, DiagnosticContext, RpcContext};
|
|
use crate::init::SYSTEM_REBUILD_PATH;
|
|
use crate::shutdown::Shutdown;
|
|
use crate::util::io::delete_file;
|
|
use crate::{Error, DATA_DIR};
|
|
|
|
pub fn diagnostic<C: Context>() -> ParentHandler<C> {
|
|
ParentHandler::new()
|
|
.subcommand(
|
|
"error",
|
|
from_fn(error)
|
|
.with_about("Display diagnostic error")
|
|
.with_call_remote::<CliContext>(),
|
|
)
|
|
.subcommand(
|
|
"logs",
|
|
crate::system::logs::<DiagnosticContext>().with_about("Display OS logs"),
|
|
)
|
|
.subcommand(
|
|
"logs",
|
|
from_fn_async(crate::logs::cli_logs::<DiagnosticContext, Empty>)
|
|
.no_display()
|
|
.with_about("Display OS logs"),
|
|
)
|
|
.subcommand(
|
|
"kernel-logs",
|
|
crate::system::kernel_logs::<DiagnosticContext>().with_about("Display kernel logs"),
|
|
)
|
|
.subcommand(
|
|
"kernel-logs",
|
|
from_fn_async(crate::logs::cli_logs::<DiagnosticContext, Empty>)
|
|
.no_display()
|
|
.with_about("Display kernal logs"),
|
|
)
|
|
.subcommand(
|
|
"restart",
|
|
from_fn(restart)
|
|
.no_display()
|
|
.with_about("Restart the server")
|
|
.with_call_remote::<CliContext>(),
|
|
)
|
|
.subcommand(
|
|
"disk",
|
|
disk::<C>().with_about("Command to remove disk from filesystem"),
|
|
)
|
|
.subcommand(
|
|
"rebuild",
|
|
from_fn_async(rebuild)
|
|
.no_display()
|
|
.with_about("Teardown and rebuild service containers")
|
|
.with_call_remote::<CliContext>(),
|
|
)
|
|
}
|
|
|
|
// #[command]
|
|
pub fn error(ctx: DiagnosticContext) -> Result<Arc<RpcError>, Error> {
|
|
Ok(ctx.error.clone())
|
|
}
|
|
|
|
pub fn restart(ctx: DiagnosticContext) -> Result<(), Error> {
|
|
ctx.shutdown
|
|
.send(Shutdown {
|
|
export_args: ctx
|
|
.disk_guid
|
|
.clone()
|
|
.map(|guid| (guid, Path::new(DATA_DIR).to_owned())),
|
|
restart: true,
|
|
})
|
|
.expect("receiver dropped");
|
|
Ok(())
|
|
}
|
|
pub async fn rebuild(ctx: DiagnosticContext) -> Result<(), Error> {
|
|
tokio::fs::write(SYSTEM_REBUILD_PATH, b"").await?;
|
|
restart(ctx)
|
|
}
|
|
|
|
pub fn disk<C: Context>() -> ParentHandler<C> {
|
|
ParentHandler::new()
|
|
.subcommand("forget", from_fn_async(forget_disk::<C>).no_cli())
|
|
.subcommand(
|
|
"forget",
|
|
CallRemoteHandler::<CliContext, _, _>::new(
|
|
from_fn_async(forget_disk::<RpcContext>).no_display(),
|
|
)
|
|
.no_display()
|
|
.with_about("Remove disk from filesystem"),
|
|
)
|
|
}
|
|
|
|
pub async fn forget_disk<C: Context>(_: C) -> Result<(), Error> {
|
|
delete_file("/media/startos/config/overlay/etc/hostname").await?;
|
|
delete_file("/media/startos/config/disk.guid").await?;
|
|
Ok(())
|
|
}
|