user management, desktop and network backend mostly finished

axtloss/rework-partitioning
Amy 3 years ago
parent 73f0801b0d
commit 1015f96727
No known key found for this signature in database
GPG Key ID: C642EA4B2F4096BE

@ -4,29 +4,71 @@ pub fn choose_pkgs(desktop_setup: &str) {
println!("Installing {}", desktop_setup); println!("Installing {}", desktop_setup);
match desktop_setup { match desktop_setup {
"onyx" => { "onyx" => {
install(vec!("onyx", "lightdm", "lightdm-gtk-greeter", "lightdm-gtk-greeter-settings")); install(vec![
"onyx",
"lightdm",
"lightdm-gtk-greeter",
"lightdm-gtk-greeter-settings",
]);
} }
"gnome" => { "gnome" => {
install(vec!("gnome","gnome-tweaks","chrome-gnome-shell","gdm")); install(vec!["gnome", "gnome-tweaks", "chrome-gnome-shell", "gdm"]);
}, }
"kde" => { "kde" => {
install(vec!("kde", "plasma", "plasma-wayland-session", "kde-applications", "sddm")); install(vec![
}, "kde",
"plasma",
"plasma-wayland-session",
"kde-applications",
"sddm",
]);
}
"budgie" => { "budgie" => {
install(vec!("budgie-desktop", "gnome", "lightdm", "lightdm-gtk-greeter", "lightdm-gtk-greeter-settings")); install(vec![
}, "budgie-desktop",
"gnome",
"lightdm",
"lightdm-gtk-greeter",
"lightdm-gtk-greeter-settings",
]);
}
"cinnamon" => { "cinnamon" => {
install(vec!("cinnamon", "lightdm", "lightdm-gtk-greeter", "lightdm-gtk-greeter-settings", "metacity", "gnome-shell")); install(vec![
}, "cinnamon",
"lightdm",
"lightdm-gtk-greeter",
"lightdm-gtk-greeter-settings",
"metacity",
"gnome-shell",
]);
}
"mate" => { "mate" => {
install(vec!("mate", "lightdm", "lightdm-gtk-greeter", "lightdm-gtk-greeter-settings", "mate-extra")); install(vec![
}, "mate",
"lightdm",
"lightdm-gtk-greeter",
"lightdm-gtk-greeter-settings",
"mate-extra",
]);
}
"xfce" => { "xfce" => {
install(vec!("xfce4", "lightdm", "lightdm-gtk-greeter", "lightdm-gtk-greeter-settings", "xfce4-goodies")); install(vec![
}, "xfce4",
"lightdm",
"lightdm-gtk-greeter",
"lightdm-gtk-greeter-settings",
"xfce4-goodies",
]);
}
"enlightenment" => { "enlightenment" => {
install(vec!("enlightenment", "lightdm", "lightdm-gtk-greeter", "lightdm-gtk-greeter-settings", "terminology")); install(vec![
}, "enlightenment",
"lightdm",
"lightdm-gtk-greeter",
"lightdm-gtk-greeter-settings",
"terminology",
]);
}
_ => { _ => {
crash("Unknown desktop setup".to_string(), 1); crash("Unknown desktop setup".to_string(), 1);

@ -1,4 +1,4 @@
pub mod partition; pub mod desktops;
pub mod network; pub mod network;
pub mod partition;
pub mod users; pub mod users;
pub mod desktops;

@ -1,11 +1,35 @@
use crate::internal::*;
pub fn set_hostname(hostname: &str) { pub fn set_hostname(hostname: &str) {
println!("Setting hostname to {}", hostname); println!("Setting hostname to {}", hostname);
files::create_file("/etc/hostname");
let return_val = files::append_file("/etc/hostname", hostname);
match return_val {
Ok(_) => {
info(format!("Set hostname to {}", hostname));
}
Err(e) => {
crash(
format!("Failed to set hostname to {}, Error: {}", hostname, e),
1,
);
}
}
} }
pub fn enable_ipv6(ipv6: bool) { pub fn enable_ipv6(ipv6: bool) {
if ipv6 { if ipv6 {
println!("enabling ipv6"); println!("Enabling IPv6");
let return_val = files::append_file("/etc/hosts", "::1 localhost");
match return_val {
Ok(_) => {
info("Enabled IPv6".to_string());
}
Err(e) => {
crash(format!("Failed to enable IPv6, Error: {}", e), 1);
}
}
} else { } else {
println!("disabling ipv6"); println!("Not enabling ipv6");
} }
} }

@ -1,13 +1,87 @@
use crate::internal::*;
use std::process::Command;
pub fn new_user(username: &str, hasroot: bool, password: &str) { pub fn new_user(username: &str, hasroot: bool, password: &str) {
println!("Creating new user '{}'", username); let return_val = Command::new("useradd")
.arg("-m")
.arg("-s")
.arg("/bin/bash")
.arg(username)
.output();
match return_val {
Ok(_) => {
info(format!("Created user {}", username));
}
Err(e) => {
crash(
format!("Failed to create user {}, Error: {}", username, e),
1,
);
}
}
if hasroot { if hasroot {
println!("User '{}' will have root privileges", username); let return_val = Command::new("usermod")
} else { .arg("-a")
println!("User '{}' will not have root privileges", username); .arg("-G")
.arg("wheel")
.arg(username)
.output();
match return_val {
Ok(_) => {
info(format!("Added user {} to group wheel", username));
}
Err(e) => {
crash(format!("Failed to add user {}, Error: {}", username, e), 1);
}
}
}
let return_val = Command::new("arch-chroot")
.arg("/mnt")
.arg("usermod")
.arg("--password")
.arg("$(echo")
.arg(format!("${{{}}}", password))
.arg("|")
.arg("openssl")
.arg("passwd")
.arg("-1")
.arg("-stdin)")
.arg(username)
.output();
match return_val {
Ok(_) => {
info(format!("Set password for user {}", username));
}
Err(e) => {
crash(
format!("Failed to set password for user {}, Error: {}", username, e),
1,
);
}
} }
println!("Setting password for user '{}' to '{}'", username, password);
} }
pub fn root_pass(root_pass: &str) { pub fn root_pass(root_pass: &str) {
println!("Setting root password to '{}'", root_pass); println!("Setting root password to '{}'", root_pass);
let return_val = Command::new("arch-chroot")
.arg("/mnt")
.arg("usermod")
.arg("--password")
.arg("$(echo")
.arg(format!("${{{}}}", root_pass))
.arg("|")
.arg("openssl")
.arg("passwd")
.arg("-1")
.arg("-stdin)")
.arg("root")
.output();
match return_val {
Ok(_) => {
info("Set root password".to_string());
}
Err(e) => {
crash(format!("Failed to set root password, Error: {}", e), 1);
}
}
} }

@ -0,0 +1,66 @@
use crate::internal::*;
use std::fs::{File, OpenOptions};
use std::io::prelude::*;
pub fn create_file(path: &str) {
let return_val = File::create(path);
match return_val {
Ok(_file) => {
info(format!("Created file {}", path));
}
Err(e) => {
crash(format!("Failed to create file {}, Error: {}", path, e), 1);
}
}
}
pub fn append_file(path: &str, content: &str) -> std::io::Result<()> {
info(format!("Appending {} to file {}", content, path));
let mut file = OpenOptions::new().append(true).open(path)?;
file.write(content.as_bytes())?;
Ok(())
}
pub fn delete_file(path: &str) {
let return_val = std::fs::remove_file(path);
match return_val {
Ok(_) => {
info(format!("Deleted file {}", path));
}
Err(e) => {
crash(format!("Failed to delete file {}, Error: {}", path, e), 1);
}
}
}
pub fn create_directory(path: &str) {
let return_val = std::fs::create_dir(path);
match return_val {
Ok(_) => {
info(format!("Created directory {}", path));
}
Err(e) => {
crash(
format!("Failed to create directory {}, Error: {}", path, e),
1,
);
}
}
}
pub fn overwrite_file(path: &str, content: &str) {
delete_file(path);
create_file(path);
let return_val = append_file(path, content);
match return_val {
Ok(_) => {
info(format!("Overwrote file {}", path));
}
Err(e) => {
crash(
format!("Failed to overwrite file {}, Error: {}", path, e),
1,
);
}
}
}

@ -1,5 +1,9 @@
use std::process::Command; use std::process::Command;
pub fn install(pkgs: Vec<&str>) { pub fn install(pkgs: Vec<&str>) {
Command::new("pacman").arg("-S").args(pkgs).output().expect("Failed to install packages"); Command::new("pacstrap")
.arg("/mnt")
.args(pkgs)
.output()
.expect("Failed to install packages");
} }

@ -1,3 +1,4 @@
pub mod files;
pub mod install; pub mod install;
pub mod strings; pub mod strings;
@ -17,10 +18,6 @@ pub fn log(a: String) {
strings::log(a); strings::log(a);
} }
pub fn prompt(a: String, b: bool) -> bool {
strings ::prompt(a, b)
}
#[macro_export] #[macro_export]
macro_rules! uwu { macro_rules! uwu {
($x:expr) => {{ ($x:expr) => {{

@ -1,9 +1,9 @@
use std::{env, io}; use crate::uwu;
use std::io::Write; use std::io::Write;
use std::process::exit; use std::process::exit;
use std::str::FromStr; use std::str::FromStr;
use std::time::UNIX_EPOCH; use std::time::UNIX_EPOCH;
use crate::uwu; use std::{env, io};
pub fn info(a: String) { pub fn info(a: String) {
let a = if env::var("JADE_UWU").unwrap_or_else(|_| "".to_string()) == "true" { let a = if env::var("JADE_UWU").unwrap_or_else(|_| "".to_string()) == "true" {
uwu!(&a) uwu!(&a)
@ -38,26 +38,3 @@ pub fn log(a: String) {
a a
); );
} }
pub fn prompt(a: String, b: bool) -> bool {
let default = ["[Y/n]", "[y/N]"];
let i = if b { 0 } else { 1 };
let a = if env::var("JADE_UWU").unwrap_or_else(|_| "".to_string()) == "true" {
uwu!(&a)
} else {
a
};
print!(
"\x1b[2;22;35m?\x1b[0m \x1b[1;37m{}\x1b[0m \x1b[2;22;37m{}\x1b[0m: ",
a, default[i]
);
let mut yn: String = String::new();
io::stdout().flush().ok();
let _ = std::io::stdin().read_line(&mut yn);
if yn.trim().to_lowercase() == "n" || yn.trim().to_lowercase() == "no" {
false
} else if yn.trim().to_lowercase() == "y" || yn.trim().to_lowercase() == "yes" {
true
} else {
b
}
}

@ -100,9 +100,11 @@ fn main() {
), ),
).get_matches(); ).get_matches();
if let Some(app) = app.subcommand_matches("partition") { if let Some(app) = app.subcommand_matches("partition") {
partition::partition(app.value_of("device").unwrap(), app.value_of("mode").unwrap()); partition::partition(
app.value_of("device").unwrap(),
app.value_of("mode").unwrap(),
);
} else if let Some(app) = app.subcommand_matches("locale") { } else if let Some(app) = app.subcommand_matches("locale") {
let kbrlayout = app.value_of("keyboard").unwrap(); let kbrlayout = app.value_of("keyboard").unwrap();
let timezn = app.value_of("timezone").unwrap(); let timezn = app.value_of("timezone").unwrap();
@ -115,7 +117,11 @@ fn main() {
network::set_hostname(app.value_of("hostname").unwrap()) network::set_hostname(app.value_of("hostname").unwrap())
} else if let Some(app) = app.subcommand_matches("users") { } else if let Some(app) = app.subcommand_matches("users") {
if let Some(app) = app.subcommand_matches("newUser") { if let Some(app) = app.subcommand_matches("newUser") {
users::new_user(app.value_of("username").unwrap(), app.value_of("hasroot").unwrap().parse::<bool>().unwrap(), app.value_of("password").unwrap()); users::new_user(
app.value_of("username").unwrap(),
app.value_of("hasroot").unwrap().parse::<bool>().unwrap(),
app.value_of("password").unwrap(),
);
} else if let Some(app) = app.subcommand_matches("rootPass") { } else if let Some(app) = app.subcommand_matches("rootPass") {
let rootpass = app.value_of("rootPass").unwrap(); let rootpass = app.value_of("rootPass").unwrap();
println!("{}", rootpass); println!("{}", rootpass);

Loading…
Cancel
Save