diff --git a/src/install.rs b/src/install.rs index 31fb073..a467827 100644 --- a/src/install.rs +++ b/src/install.rs @@ -13,6 +13,11 @@ pub fn install() -> Result<(), Box> { get_user_input("would you like to install the Tetanus on this computer?")?.contains("y"); if client { client_install()?; + println!( + " + + " + ) } else { println!("Ok then bye... not sure what you wanted here..."); } diff --git a/src/main.rs b/src/main.rs index ca71c31..2d0087a 100644 --- a/src/main.rs +++ b/src/main.rs @@ -112,6 +112,31 @@ async fn main() { } } if args.client { + let banner = " + XXXXXXXXXXXXXXXXXXXXXXXXXX + XXXXXXXXXXXXXXXXXXXXXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXXXX + XXX + X + + ++----------++ + ++----------++ + ||Welcome To|| + || TETANUS || + ++----------++ + ++----------++ "; if !client_config_path.exists() { eprintln!( "error: no client config path found at {}", @@ -126,6 +151,9 @@ async fn main() { exit(1); } appstate.name = name; + banner.lines().for_each(|line| { + appstate.output.push(line.to_string()); + }); appstate.initialize_modules(); println!("entering tui..."); let _res = run_tui(appstate, rx).unwrap();