429 lines
15 KiB
Rust
429 lines
15 KiB
Rust
use crate::application_browser::ApplicationBrowser;
|
|
use crate::data_types::*;
|
|
use crate::utils;
|
|
use crate::utils::PacmanWrapper;
|
|
use gtk::{glib, Builder};
|
|
use once_cell::sync::Lazy;
|
|
use std::fmt::Write as _;
|
|
use std::path::Path;
|
|
use std::sync::Mutex;
|
|
|
|
use gtk::prelude::*;
|
|
|
|
use std::str;
|
|
use subprocess::{Exec, Redirection};
|
|
|
|
static mut g_local_units: Lazy<Mutex<SystemdUnits>> = Lazy::new(|| Mutex::new(SystemdUnits::new()));
|
|
static mut g_global_units: Lazy<Mutex<SystemdUnits>> =
|
|
Lazy::new(|| Mutex::new(SystemdUnits::new()));
|
|
|
|
fn create_fixes_section() -> gtk::Box {
|
|
let topbox = gtk::Box::new(gtk::Orientation::Vertical, 2);
|
|
let button_box_f = gtk::Box::new(gtk::Orientation::Horizontal, 10);
|
|
let button_box_s = gtk::Box::new(gtk::Orientation::Horizontal, 10);
|
|
let label = gtk::Label::new(None);
|
|
label.set_line_wrap(true);
|
|
label.set_justify(gtk::Justification::Center);
|
|
label.set_text("Fixes");
|
|
|
|
let removelock_btn = gtk::Button::with_label("Remove db lock");
|
|
let reinstall_btn = gtk::Button::with_label("Reinstall all packages");
|
|
let refreshkeyring_btn = gtk::Button::with_label("Refresh keyrings");
|
|
let update_system_btn = gtk::Button::with_label("System update");
|
|
let remove_orphans_btn = gtk::Button::with_label("Remove orphans");
|
|
let clear_pkgcache_btn = gtk::Button::with_label("Clear package cache");
|
|
|
|
removelock_btn.connect_clicked(move |_| {
|
|
if Path::new("/var/lib/pacman/db.lck").exists() {
|
|
let _ = Exec::cmd("/sbin/pkexec")
|
|
.arg("bash")
|
|
.arg("-c")
|
|
.arg("rm /var/lib/pacman/db.lck")
|
|
.join()
|
|
.unwrap();
|
|
if !Path::new("/var/lib/pacman/db.lck").exists() {
|
|
let dialog = gtk::MessageDialog::builder()
|
|
.message_type(gtk::MessageType::Info)
|
|
.text("Pacman db lock was removed!")
|
|
.build();
|
|
dialog.show();
|
|
}
|
|
}
|
|
});
|
|
reinstall_btn.connect_clicked(move |_| {
|
|
let _ = utils::run_cmd_terminal(String::from("pacman -S $(pacman -Qnq)"), true);
|
|
});
|
|
refreshkeyring_btn.connect_clicked(on_refreshkeyring_btn_clicked);
|
|
update_system_btn.connect_clicked(on_update_system_btn_clicked);
|
|
remove_orphans_btn.connect_clicked(move |_| {
|
|
let _ = utils::run_cmd_terminal(String::from("pacman -Rns $(pacman -Qtdq)"), true);
|
|
});
|
|
clear_pkgcache_btn.connect_clicked(on_clear_pkgcache_btn_clicked);
|
|
|
|
topbox.pack_start(&label, true, false, 1);
|
|
button_box_f.pack_start(&update_system_btn, true, true, 2);
|
|
button_box_f.pack_start(&reinstall_btn, true, true, 2);
|
|
button_box_f.pack_end(&refreshkeyring_btn, true, true, 2);
|
|
button_box_s.pack_start(&removelock_btn, true, true, 2);
|
|
button_box_s.pack_start(&clear_pkgcache_btn, true, true, 2);
|
|
button_box_s.pack_end(&remove_orphans_btn, true, true, 2);
|
|
button_box_f.set_halign(gtk::Align::Fill);
|
|
button_box_s.set_halign(gtk::Align::Fill);
|
|
topbox.pack_end(&button_box_s, true, true, 5);
|
|
topbox.pack_end(&button_box_f, true, true, 5);
|
|
|
|
topbox.set_hexpand(true);
|
|
topbox
|
|
}
|
|
|
|
fn create_options_section() -> gtk::Box {
|
|
let topbox = gtk::Box::new(gtk::Orientation::Vertical, 2);
|
|
let box_collection = gtk::Box::new(gtk::Orientation::Horizontal, 10);
|
|
let label = gtk::Label::new(None);
|
|
label.set_line_wrap(true);
|
|
label.set_justify(gtk::Justification::Center);
|
|
label.set_text("Tweaks");
|
|
|
|
let psd_btn = gtk::CheckButton::with_label("Profile-sync-daemon enable");
|
|
let systemd_oomd_btn = gtk::CheckButton::with_label("Systemd-oomd enabled");
|
|
let apparmor_btn = gtk::CheckButton::with_label("Apparmor enabled");
|
|
let ananicy_cpp_btn = gtk::CheckButton::with_label("Ananicy Cpp enabled");
|
|
|
|
unsafe {
|
|
psd_btn.set_data("actionData", "psd.service");
|
|
psd_btn.set_data("actionType", "user_service");
|
|
systemd_oomd_btn.set_data("actionData", "systemd-oomd.service");
|
|
systemd_oomd_btn.set_data("actionType", "service");
|
|
apparmor_btn.set_data("actionData", "apparmor.service");
|
|
apparmor_btn.set_data("actionType", "service");
|
|
ananicy_cpp_btn.set_data("actionData", "ananicy-cpp.service");
|
|
ananicy_cpp_btn.set_data("actionType", "service");
|
|
}
|
|
|
|
for btn in &[&psd_btn, &systemd_oomd_btn, &apparmor_btn, &ananicy_cpp_btn] {
|
|
unsafe {
|
|
let data: &str = *btn.data("actionData").unwrap().as_ptr();
|
|
if g_local_units.lock().unwrap().enabled_units.contains(&String::from(data))
|
|
|| g_global_units.lock().unwrap().enabled_units.contains(&String::from(data))
|
|
{
|
|
btn.set_active(true);
|
|
}
|
|
}
|
|
}
|
|
|
|
psd_btn.connect_clicked(on_servbtn_clicked);
|
|
systemd_oomd_btn.connect_clicked(on_servbtn_clicked);
|
|
apparmor_btn.connect_clicked(on_servbtn_clicked);
|
|
ananicy_cpp_btn.connect_clicked(on_servbtn_clicked);
|
|
|
|
topbox.pack_start(&label, true, false, 1);
|
|
box_collection.pack_start(&psd_btn, true, false, 2);
|
|
box_collection.pack_start(&systemd_oomd_btn, true, false, 2);
|
|
box_collection.pack_start(&apparmor_btn, true, false, 2);
|
|
box_collection.pack_start(&ananicy_cpp_btn, true, false, 2);
|
|
box_collection.set_halign(gtk::Align::Fill);
|
|
topbox.pack_end(&box_collection, true, false, 1);
|
|
|
|
topbox.set_hexpand(true);
|
|
topbox
|
|
}
|
|
|
|
fn create_apps_section() -> gtk::Box {
|
|
let topbox = gtk::Box::new(gtk::Orientation::Vertical, 2);
|
|
let box_collection = gtk::Box::new(gtk::Orientation::Horizontal, 10);
|
|
let label = gtk::Label::new(None);
|
|
label.set_line_wrap(true);
|
|
label.set_justify(gtk::Justification::Center);
|
|
label.set_text("Applications");
|
|
|
|
let cachyos_pi = gtk::Button::with_label("CachyOS PackageInstaller");
|
|
let cachyos_km = gtk::Button::with_label("CachyOS Kernel Manager");
|
|
|
|
cachyos_pi.connect_clicked(on_appbtn_clicked);
|
|
cachyos_km.connect_clicked(on_appbtn_clicked);
|
|
|
|
box_collection.pack_start(&cachyos_pi, true, true, 2);
|
|
box_collection.pack_start(&cachyos_km, true, true, 2);
|
|
|
|
topbox.pack_start(&label, true, true, 5);
|
|
|
|
box_collection.set_halign(gtk::Align::Fill);
|
|
topbox.pack_end(&box_collection, true, true, 0);
|
|
|
|
topbox.set_hexpand(true);
|
|
topbox
|
|
}
|
|
|
|
fn load_enabled_units() {
|
|
unsafe {
|
|
g_local_units.lock().unwrap().loaded_units.clear();
|
|
g_local_units.lock().unwrap().enabled_units.clear();
|
|
|
|
let mut exec_out = Exec::shell("systemctl list-unit-files -q --no-pager | tr -s \" \"")
|
|
.stdout(Redirection::Pipe)
|
|
.capture()
|
|
.unwrap()
|
|
.stdout_str();
|
|
exec_out.pop();
|
|
|
|
let service_list = exec_out.split('\n');
|
|
|
|
for service in service_list {
|
|
let out: Vec<&str> = service.split(' ').collect();
|
|
g_local_units.lock().unwrap().loaded_units.push(String::from(out[0]));
|
|
if out[1] == "enabled" {
|
|
g_local_units.lock().unwrap().enabled_units.push(String::from(out[0]));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn load_global_enabled_units() {
|
|
unsafe {
|
|
g_global_units.lock().unwrap().loaded_units.clear();
|
|
g_global_units.lock().unwrap().enabled_units.clear();
|
|
|
|
let mut exec_out =
|
|
Exec::shell("systemctl --global list-unit-files -q --no-pager | tr -s \" \"")
|
|
.stdout(Redirection::Pipe)
|
|
.capture()
|
|
.unwrap()
|
|
.stdout_str();
|
|
exec_out.pop();
|
|
|
|
let service_list = exec_out.split('\n');
|
|
for service in service_list {
|
|
let out: Vec<&str> = service.split(' ').collect();
|
|
g_global_units.lock().unwrap().loaded_units.push(String::from(out[0]));
|
|
if out[1] == "enabled" {
|
|
g_global_units.lock().unwrap().enabled_units.push(String::from(out[0]));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn create_tweaks_page(builder: &Builder) {
|
|
let install: gtk::Button = builder.object("tweaksBrowser").unwrap();
|
|
install.set_visible(true);
|
|
|
|
load_enabled_units();
|
|
load_global_enabled_units();
|
|
|
|
let viewport = gtk::Viewport::new(gtk::Adjustment::NONE, gtk::Adjustment::NONE);
|
|
let image = gtk::Image::from_icon_name(Some("go-previous"), gtk::IconSize::Button);
|
|
let back_btn = gtk::Button::new();
|
|
back_btn.set_image(Some(&image));
|
|
back_btn.set_widget_name("home");
|
|
|
|
back_btn.connect_clicked(glib::clone!(@weak builder => move |button| {
|
|
let name = button.widget_name();
|
|
let stack: gtk::Stack = builder.object("stack").unwrap();
|
|
stack.set_visible_child_name(&format!("{}page", name));
|
|
}));
|
|
|
|
let options_section_box = create_options_section();
|
|
let fixes_section_box = create_fixes_section();
|
|
let apps_section_box = create_apps_section();
|
|
|
|
let grid = gtk::Grid::new();
|
|
grid.set_hexpand(true);
|
|
grid.set_margin_start(10);
|
|
grid.set_margin_end(10);
|
|
grid.set_margin_top(5);
|
|
grid.set_margin_bottom(5);
|
|
grid.attach(&back_btn, 0, 1, 1, 1);
|
|
let box_collection = gtk::Box::new(gtk::Orientation::Vertical, 5);
|
|
|
|
box_collection.pack_start(&options_section_box, false, false, 10);
|
|
box_collection.pack_start(&fixes_section_box, false, false, 10);
|
|
box_collection.pack_end(&apps_section_box, false, false, 10);
|
|
|
|
box_collection.set_valign(gtk::Align::Center);
|
|
box_collection.set_halign(gtk::Align::Center);
|
|
grid.attach(&box_collection, 1, 2, 5, 1);
|
|
viewport.add(&grid);
|
|
viewport.show_all();
|
|
|
|
let stack: gtk::Stack = builder.object("stack").unwrap();
|
|
let child_name = "tweaksBrowserpage";
|
|
stack.add_named(&viewport, child_name);
|
|
}
|
|
|
|
pub fn create_appbrowser_page(builder: &Builder) {
|
|
let install: gtk::Button = builder.object("appBrowser").unwrap();
|
|
install.set_visible(true);
|
|
|
|
let viewport = gtk::Viewport::new(gtk::Adjustment::NONE, gtk::Adjustment::NONE);
|
|
let image = gtk::Image::from_icon_name(Some("go-previous"), gtk::IconSize::Button);
|
|
let back_btn = gtk::Button::new();
|
|
back_btn.set_image(Some(&image));
|
|
back_btn.set_widget_name("home");
|
|
|
|
back_btn.connect_clicked(glib::clone!(@weak builder => move |button| {
|
|
let name = button.widget_name();
|
|
let stack: gtk::Stack = builder.object("stack").unwrap();
|
|
stack.set_visible_child_name(&format!("{}page", name));
|
|
}));
|
|
|
|
let grid = gtk::Grid::new();
|
|
grid.set_hexpand(true);
|
|
grid.set_margin_start(10);
|
|
grid.set_margin_end(10);
|
|
grid.set_margin_top(5);
|
|
grid.set_margin_bottom(5);
|
|
grid.attach(&back_btn, 0, 1, 1, 1);
|
|
|
|
let app_browser_ref = ApplicationBrowser::default_impl().lock().unwrap();
|
|
let app_browser_box = app_browser_ref.get_page();
|
|
grid.attach(app_browser_box, 0, 2, 1, 1);
|
|
|
|
// Add grid to the viewport
|
|
// NOTE: we might eliminate that?
|
|
viewport.add(&grid);
|
|
viewport.show_all();
|
|
|
|
let stack: gtk::Stack = builder.object("stack").unwrap();
|
|
let child_name = "appBrowserpage";
|
|
stack.add_named(&viewport, child_name);
|
|
}
|
|
|
|
fn on_servbtn_clicked(button: >k::CheckButton) {
|
|
// Get action data/type.
|
|
let action_type: &str;
|
|
let action_data: &str;
|
|
unsafe {
|
|
action_type = *button.data("actionType").unwrap().as_ptr();
|
|
action_data = *button.data("actionData").unwrap().as_ptr();
|
|
}
|
|
|
|
let (user_only, pkexec_only) =
|
|
if action_type == "user_service" { ("--user", "--user $(logname)") } else { ("", "") };
|
|
|
|
let cmd: String;
|
|
unsafe {
|
|
let local_units = &g_local_units.lock().unwrap().enabled_units;
|
|
cmd = if !local_units.contains(&String::from(action_data)) {
|
|
format!(
|
|
"/sbin/pkexec {} bash -c \"systemctl {} enable --now --force {}\"",
|
|
pkexec_only, user_only, action_data
|
|
)
|
|
} else {
|
|
format!(
|
|
"/sbin/pkexec {} bash -c \"systemctl {} disable --now {}\"",
|
|
pkexec_only, user_only, action_data
|
|
)
|
|
};
|
|
}
|
|
|
|
// Spawn child process in separate thread.
|
|
std::thread::spawn(move || {
|
|
Exec::shell(cmd).join().unwrap();
|
|
|
|
if action_type == "user_service" {
|
|
load_global_enabled_units();
|
|
} else {
|
|
load_enabled_units();
|
|
}
|
|
});
|
|
}
|
|
|
|
fn on_refreshkeyring_btn_clicked(_: >k::Button) {
|
|
let pacman = pacmanconf::Config::with_opts(None, Some("/etc/pacman.conf"), Some("/")).unwrap();
|
|
let alpm = alpm_utils::alpm_with_conf(&pacman).unwrap();
|
|
// pacman -Qq | grep keyring
|
|
let needles = alpm
|
|
.localdb()
|
|
.search([".*-keyring"].iter())
|
|
.unwrap()
|
|
.into_iter()
|
|
.filter(|pkg| pkg.name() != "gnome-keyring")
|
|
.map(|pkg| {
|
|
let mut pkgname = String::from(pkg.name());
|
|
pkgname.remove_matches("-keyring");
|
|
format!("{} ", pkgname)
|
|
})
|
|
.collect::<String>();
|
|
|
|
let _ = utils::run_cmd_terminal(
|
|
format!("pacman-key --init && pacman-key --populate {}", needles),
|
|
true,
|
|
);
|
|
}
|
|
|
|
fn on_update_system_btn_clicked(_: >k::Button) {
|
|
let (cmd, escalate) = match utils::get_pacman_wrapper() {
|
|
PacmanWrapper::Pak => ("pak -Syu", false),
|
|
PacmanWrapper::Yay => ("yay -Syu", false),
|
|
PacmanWrapper::Paru => ("paru --removemake -Syu", false),
|
|
_ => ("pacman -Syu", true),
|
|
};
|
|
let _ = utils::run_cmd_terminal(String::from(cmd), escalate);
|
|
}
|
|
|
|
fn on_clear_pkgcache_btn_clicked(_: >k::Button) {
|
|
let (cmd, escalate) = match utils::get_pacman_wrapper() {
|
|
PacmanWrapper::Pak => ("pak -Sc", false),
|
|
PacmanWrapper::Yay => ("yay -Sc", false),
|
|
PacmanWrapper::Paru => ("paru -Sc", false),
|
|
_ => ("pacman -Sc", true),
|
|
};
|
|
let _ = utils::run_cmd_terminal(String::from(cmd), escalate);
|
|
}
|
|
|
|
fn on_appbtn_clicked(button: >k::Button) {
|
|
// Get button label.
|
|
let name = button.label().unwrap();
|
|
let (binname, is_sudo) = if name == "CachyOS PackageInstaller" {
|
|
("cachyos-pi-bin", true)
|
|
} else if name == "CachyOS Kernel Manager" {
|
|
("cachyos-kernel-manager", false)
|
|
} else {
|
|
("", false)
|
|
};
|
|
|
|
// Check if executable exists.
|
|
let exit_status = Exec::cmd("which").arg(binname).join().unwrap();
|
|
if !exit_status.success() {
|
|
return;
|
|
}
|
|
|
|
let mut envs = String::new();
|
|
for env in glib::listenv() {
|
|
if env == "PATH" {
|
|
envs += "PATH=/sbin:/bin:/usr/local/sbin:/usr/local/bin:/usr/bin:/usr/sbin ";
|
|
continue;
|
|
}
|
|
let _ = write!(
|
|
envs,
|
|
"{}=\"{}\" ",
|
|
env.to_str().unwrap(),
|
|
glib::getenv(&env).unwrap().to_str().unwrap()
|
|
);
|
|
}
|
|
|
|
// Get executable path.
|
|
let mut exe_path =
|
|
Exec::cmd("which").arg(binname).stdout(Redirection::Pipe).capture().unwrap().stdout_str();
|
|
exe_path.pop();
|
|
let bash_cmd = format!("{} {}", &envs, &exe_path);
|
|
|
|
// Create context channel.
|
|
let (tx, rx) = glib::MainContext::channel(glib::PRIORITY_DEFAULT);
|
|
|
|
// Spawn child process in separate thread.
|
|
std::thread::spawn(move || {
|
|
let exit_status = if is_sudo {
|
|
Exec::cmd("/sbin/pkexec").arg("bash").arg("-c").arg(bash_cmd).join().unwrap()
|
|
} else {
|
|
Exec::shell(bash_cmd).join().unwrap()
|
|
};
|
|
tx.send(format!("Exit status successfully? = {:?}", exit_status.success()))
|
|
.expect("Couldn't send data to channel");
|
|
});
|
|
|
|
rx.attach(None, move |text| {
|
|
println!("{}", text);
|
|
glib::Continue(true)
|
|
});
|
|
}
|