mirror of
https://gitlab.com/vyla-entertainment/home.git
synced 2026-11-13 21:57:21 -05:00
548 lines
22 KiB
Rust
548 lines
22 KiB
Rust
#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
|
|
|
|
use std::path::PathBuf;
|
|
use std::process::{Child, Command, Stdio};
|
|
use std::sync::Mutex;
|
|
use tauri::{Emitter, Manager, State, WebviewUrl, WebviewWindowBuilder};
|
|
|
|
fn debug_log(msg: &str) {
|
|
let log_dir = if let Ok(app_data) = std::env::var("APPDATA") {
|
|
std::path::PathBuf::from(app_data).join("vyla-home").join("vyla-home-files").join("logs")
|
|
} else if let Ok(user_profile) = std::env::var("USERPROFILE") {
|
|
std::path::PathBuf::from(user_profile).join(".vyla-home").join("logs")
|
|
} else {
|
|
std::path::PathBuf::from("logs")
|
|
};
|
|
|
|
let _ = std::fs::create_dir_all(&log_dir);
|
|
let log_path = log_dir.join("rust-debug.log");
|
|
|
|
if let Ok(mut file) = std::fs::OpenOptions::new().create(true).append(true).open(&log_path) {
|
|
use std::io::Write;
|
|
let _ = writeln!(file, "[RUST] {}", msg);
|
|
}
|
|
}
|
|
|
|
struct NodeServices(Mutex<Option<Child>>);
|
|
|
|
#[derive(serde::Deserialize, serde::Serialize, Clone)]
|
|
struct BackendMessage {
|
|
event: String,
|
|
data: serde_json::Value,
|
|
}
|
|
|
|
fn clean_path(path: &std::path::Path) -> String {
|
|
let s = path.to_string_lossy().to_string();
|
|
if s.starts_with("\\\\?\\") {
|
|
s[4..].to_string()
|
|
} else {
|
|
s
|
|
}
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
fn kill_process_tree(pid: u32) {
|
|
use std::os::windows::process::CommandExt;
|
|
const CREATE_NO_WINDOW: u32 = 0x08000000;
|
|
let result = Command::new("taskkill")
|
|
.args(["/PID", &pid.to_string(), "/T", "/F"])
|
|
.creation_flags(CREATE_NO_WINDOW)
|
|
.output();
|
|
match result {
|
|
Ok(output) => debug_log(&format!(
|
|
"[KILL-TREE] taskkill on PID {} -> status: {}, stdout: {}, stderr: {}",
|
|
pid,
|
|
output.status,
|
|
String::from_utf8_lossy(&output.stdout).trim(),
|
|
String::from_utf8_lossy(&output.stderr).trim()
|
|
)),
|
|
Err(e) => debug_log(&format!("[KILL-TREE] taskkill failed to spawn for PID {}: {}", pid, e)),
|
|
}
|
|
}
|
|
|
|
#[cfg(not(windows))]
|
|
fn kill_process_tree(pid: u32) {
|
|
unsafe {
|
|
libc_kill_pgid(pid);
|
|
}
|
|
}
|
|
|
|
#[cfg(not(windows))]
|
|
unsafe fn libc_kill_pgid(pid: u32) {
|
|
let _ = Command::new("kill")
|
|
.args(["-9", &format!("-{}", pid)])
|
|
.output();
|
|
}
|
|
|
|
fn find_node_binary(app: &tauri::App) -> Option<String> {
|
|
if let Ok(resource_dir) = app.path().resource_dir() {
|
|
let candidates = [
|
|
resource_dir.join("bin").join("node.exe"),
|
|
resource_dir.join("node.exe"),
|
|
resource_dir.join("_up_").join("bin").join("node.exe"),
|
|
resource_dir.join("_up_").join("node.exe"),
|
|
];
|
|
for candidate in candidates {
|
|
if candidate.exists() {
|
|
return Some(clean_path(&candidate));
|
|
}
|
|
}
|
|
}
|
|
|
|
if let Ok(exe_path) = std::env::current_exe() {
|
|
if let Some(parent) = exe_path.parent() {
|
|
let candidates = [
|
|
parent.join("bin").join("node.exe"),
|
|
parent.join("node.exe"),
|
|
parent.join("resources").join("bin").join("node.exe"),
|
|
parent.join("resources").join("node.exe"),
|
|
];
|
|
for candidate in candidates {
|
|
if candidate.exists() {
|
|
return Some(clean_path(&candidate));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if let Ok(path_var) = std::env::var("PATH") {
|
|
for p in std::env::split_paths(&path_var) {
|
|
let candidate = p.join("node.exe");
|
|
if candidate.exists() {
|
|
return Some(clean_path(&candidate));
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
{
|
|
let default_paths = [
|
|
r"C:\Program Files\nodejs\node.exe",
|
|
r"C:\Program Files (x86)\nodejs\node.exe",
|
|
];
|
|
for path in default_paths {
|
|
let candidate = std::path::Path::new(path);
|
|
if candidate.exists() {
|
|
return Some(clean_path(candidate));
|
|
}
|
|
}
|
|
|
|
if let Ok(local_app_data) = std::env::var("LOCALAPPDATA") {
|
|
let candidate = std::path::PathBuf::from(local_app_data)
|
|
.join("Programs")
|
|
.join("nodejs")
|
|
.join("node.exe");
|
|
if candidate.exists() {
|
|
return Some(clean_path(&candidate));
|
|
}
|
|
}
|
|
}
|
|
|
|
None
|
|
}
|
|
|
|
fn resolve_script_and_root(app: &tauri::App) -> (String, String) {
|
|
if let Ok(resource_dir) = app.path().resource_dir() {
|
|
let candidates = vec![
|
|
resource_dir.join("main.js"),
|
|
resource_dir.join("_up_").join("main.js"),
|
|
];
|
|
|
|
for candidate in candidates {
|
|
if candidate.exists() {
|
|
let root_dir = candidate.parent().unwrap();
|
|
return (clean_path(&candidate), clean_path(root_dir));
|
|
}
|
|
}
|
|
}
|
|
|
|
let mut dir = std::env::current_dir().unwrap_or_else(|_| PathBuf::from("."));
|
|
if dir.ends_with("src-tauri") {
|
|
dir.pop();
|
|
}
|
|
let candidate = dir.join("main.js");
|
|
(clean_path(&candidate), clean_path(&dir))
|
|
}
|
|
|
|
const LINK_HANDLER_SCRIPT: &str = r#"
|
|
function doInvoke(cmd, args) {
|
|
if (window.__TAURI__ && window.__TAURI__.core) {
|
|
window.__TAURI__.core.invoke(cmd, args || {});
|
|
}
|
|
}
|
|
|
|
function minimizeWindow() { doInvoke('win_minimize'); }
|
|
function maximizeWindow() { doInvoke('win_maximize'); }
|
|
function closeWindow() { doInvoke('win_close'); }
|
|
function dragWindow() { doInvoke('win_drag'); }
|
|
|
|
window.open = function(url, target, features) {
|
|
doInvoke('open_new_window', { url: url });
|
|
return null;
|
|
};
|
|
|
|
document.addEventListener('click', (e) => {
|
|
const a = e.target.closest('a');
|
|
if (!a || !a.href) return;
|
|
|
|
if (a.target === '_blank' || e.ctrlKey || e.metaKey || e.button === 1) {
|
|
e.preventDefault();
|
|
e.stopPropagation();
|
|
doInvoke('open_new_window', { url: a.href });
|
|
}
|
|
}, true);
|
|
|
|
function injectTitlebar() {
|
|
if (document.getElementById('tauri-custom-titlebar')) {
|
|
return;
|
|
}
|
|
if (!document.body) {
|
|
return;
|
|
}
|
|
|
|
const style = document.createElement('style');
|
|
style.innerHTML = `
|
|
#tauri-custom-titlebar {
|
|
position: fixed !important;
|
|
top: 0 !important;
|
|
left: 0 !important;
|
|
right: 0 !important;
|
|
height: 32px !important;
|
|
max-height: 32px !important;
|
|
display: flex !important;
|
|
justify-content: space-between !important;
|
|
align-items: center !important;
|
|
z-index: 2147483647 !important;
|
|
background: transparent !important;
|
|
pointer-events: none !important;
|
|
box-sizing: border-box !important;
|
|
margin: 0 !important;
|
|
padding: 0 !important;
|
|
}
|
|
.tauri-drag-region {
|
|
flex: 1 !important;
|
|
height: 32px !important;
|
|
margin-left: 100px !important;
|
|
cursor: default !important;
|
|
pointer-events: auto !important;
|
|
}
|
|
.tauri-window-controls {
|
|
display: flex !important;
|
|
height: 32px !important;
|
|
margin: 0 !important;
|
|
padding: 0 !important;
|
|
pointer-events: auto !important;
|
|
z-index: 2147483648 !important;
|
|
}
|
|
button.tauri-btn {
|
|
width: 46px !important;
|
|
min-width: 46px !important;
|
|
max-width: 46px !important;
|
|
height: 32px !important;
|
|
min-height: 32px !important;
|
|
max-height: 32px !important;
|
|
background: transparent !important;
|
|
border: none !important;
|
|
color: #ffffff !important;
|
|
display: flex !important;
|
|
align-items: center !important;
|
|
justify-content: center !important;
|
|
cursor: pointer !important;
|
|
outline: none !important;
|
|
padding: 0 !important;
|
|
margin: 0 !important;
|
|
border-radius: 0 !important;
|
|
box-shadow: none !important;
|
|
appearance: none !important;
|
|
opacity: 1 !important;
|
|
transform: none !important;
|
|
pointer-events: auto !important;
|
|
user-select: none !important;
|
|
}
|
|
button.tauri-btn:hover { background: rgba(255, 255, 255, 0.2) !important; }
|
|
button#tauri-btn-close:hover { background: #e81123 !important; color: #ffffff !important; }
|
|
button.tauri-btn svg { width: 10px !important; height: 10px !important; fill: currentColor !important; pointer-events: none !important; display: block !important; }
|
|
`;
|
|
document.head.appendChild(style);
|
|
|
|
const tb = document.createElement('div');
|
|
tb.id = 'tauri-custom-titlebar';
|
|
tb.innerHTML = `
|
|
<div class="tauri-drag-region"></div>
|
|
<div class="tauri-window-controls">
|
|
<button class="tauri-btn" id="tauri-btn-min" title="Minimize"><svg viewBox="0 0 10 10"><rect x="1" y="4" width="8" height="1"/></svg></button>
|
|
<button class="tauri-btn" id="tauri-btn-max" title="Maximize"><svg viewBox="0 0 10 10"><path d="M1,1v8h8V1H1z M8,8H2V2h6V8z"/></svg></button>
|
|
<button class="tauri-btn" id="tauri-btn-close" title="Close"><svg viewBox="0 0 10 10"><path d="M10,1.4L8.6,0L5,3.6L1.4,0L0,1.4L3.6,5L0,8.6L1.4,10L5,6.4L8.6,10L10,8.6L6.4,5L10,1.4z"/></svg></button>
|
|
</div>
|
|
`;
|
|
|
|
document.body.appendChild(tb);
|
|
|
|
const btnMin = tb.querySelector('#tauri-btn-min');
|
|
const btnMax = tb.querySelector('#tauri-btn-max');
|
|
const btnClose = tb.querySelector('#tauri-btn-close');
|
|
const dragEl = tb.querySelector('.tauri-drag-region');
|
|
|
|
if (btnMin) btnMin.onclick = (e) => { e.preventDefault(); e.stopPropagation(); minimizeWindow(); };
|
|
if (btnMax) btnMax.onclick = (e) => { e.preventDefault(); e.stopPropagation(); maximizeWindow(); };
|
|
if (btnClose) btnClose.onclick = (e) => { e.preventDefault(); e.stopPropagation(); closeWindow(); };
|
|
if (dragEl) {
|
|
dragEl.onmousedown = (e) => {
|
|
if (e.buttons === 1) {
|
|
e.preventDefault();
|
|
dragWindow();
|
|
}
|
|
};
|
|
}
|
|
}
|
|
|
|
injectTitlebar();
|
|
setInterval(injectTitlebar, 1000);
|
|
|
|
if (document.readyState === 'loading') {
|
|
document.addEventListener('DOMContentLoaded', injectTitlebar);
|
|
} else {
|
|
injectTitlebar();
|
|
}
|
|
|
|
window.addEventListener('load', injectTitlebar);
|
|
"#;
|
|
|
|
#[tauri::command]
|
|
fn win_minimize(window: tauri::Window) -> Result<(), String> {
|
|
window.minimize().map_err(|e| e.to_string())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn win_maximize(window: tauri::Window) -> Result<(), String> {
|
|
if let Ok(is_max) = window.is_maximized() {
|
|
if is_max {
|
|
window.unmaximize().map_err(|e| e.to_string())
|
|
} else {
|
|
window.maximize().map_err(|e| e.to_string())
|
|
}
|
|
} else {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn win_close(window: tauri::Window) -> Result<(), String> {
|
|
window.close().map_err(|e| e.to_string())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn win_drag(window: tauri::Window) -> Result<(), String> {
|
|
window.start_dragging().map_err(|e| e.to_string())
|
|
}
|
|
|
|
#[tauri::command]
|
|
fn open_new_window(app: tauri::AppHandle, url: String) -> Result<(), String> {
|
|
debug_log(&format!("[SAME-TAB] Redirecting main window to: {}", url));
|
|
|
|
let final_url = if url.starts_with("http") || url.starts_with("tauri://") {
|
|
url.clone()
|
|
} else if url.starts_with('/') {
|
|
format!("http://localhost:4890{}", url)
|
|
} else {
|
|
format!("http://localhost:4890/{}", url)
|
|
};
|
|
|
|
if let Some(window) = app.get_webview_window("main") {
|
|
window
|
|
.eval(&format!("window.location.href = '{}';", final_url.replace('\'', "\\'")))
|
|
.map_err(|e| e.to_string())?;
|
|
Ok(())
|
|
} else {
|
|
Err("main window not found".to_string())
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
debug_log("=== Vyla Home Executable Started ===");
|
|
|
|
tauri::Builder::default()
|
|
.plugin(tauri_plugin_shell::init())
|
|
.plugin(tauri_plugin_process::init())
|
|
.plugin(tauri_plugin_opener::init())
|
|
.invoke_handler(tauri::generate_handler![
|
|
win_minimize,
|
|
win_maximize,
|
|
win_close,
|
|
win_drag,
|
|
open_new_window
|
|
])
|
|
.manage(NodeServices(Mutex::new(None)))
|
|
.setup(|app| {
|
|
debug_log("Tauri setup started");
|
|
let app_data = app.path().app_data_dir().unwrap_or_default();
|
|
let app_data_str = clean_path(&app_data);
|
|
|
|
let window = WebviewWindowBuilder::new(app, "main", WebviewUrl::App("index.html".into()))
|
|
.title("Vyla Home")
|
|
.inner_size(1280.0, 800.0)
|
|
.decorations(false)
|
|
.transparent(true)
|
|
.resizable(true)
|
|
.devtools(true)
|
|
.initialization_script("window.__VYLA_WINDOW_LABEL = 'main';")
|
|
.initialization_script(LINK_HANDLER_SCRIPT)
|
|
.build()?;
|
|
|
|
let (main_js, root_dir) = resolve_script_and_root(app);
|
|
let node_bin = match find_node_binary(app) {
|
|
Some(bin) => bin,
|
|
None => {
|
|
let _ = window.emit(
|
|
"error",
|
|
serde_json::json!({
|
|
"message": "Node.js runtime not found. Please ensure Node.js is installed or bundled correctly."
|
|
}),
|
|
);
|
|
return Ok(());
|
|
}
|
|
};
|
|
|
|
let mut cmd = Command::new(&node_bin);
|
|
cmd.arg(&main_js)
|
|
.current_dir(&root_dir)
|
|
.env("APPDATA_PATH", &app_data_str)
|
|
.env("NODE_EXE", &node_bin)
|
|
.stdout(Stdio::piped())
|
|
.stderr(Stdio::piped());
|
|
|
|
#[cfg(windows)]
|
|
{
|
|
use std::os::windows::process::CommandExt;
|
|
const CREATE_NO_WINDOW: u32 = 0x08000000;
|
|
cmd.creation_flags(CREATE_NO_WINDOW);
|
|
}
|
|
|
|
match cmd.spawn() {
|
|
Ok(mut child) => {
|
|
debug_log(&format!("Node child process spawned successfully, PID: {}", child.id()));
|
|
if let Some(stderr) = child.stderr.take() {
|
|
std::thread::spawn(move || {
|
|
use std::io::{BufRead, BufReader};
|
|
let reader = BufReader::new(stderr);
|
|
for line in reader.lines().flatten() {
|
|
debug_log(&format!("[NODE STDERR] {}", line));
|
|
}
|
|
});
|
|
}
|
|
|
|
if let Some(stdout) = child.stdout.take() {
|
|
let w = window.clone();
|
|
std::thread::spawn(move || {
|
|
use std::io::{BufRead, BufReader};
|
|
let reader = BufReader::new(stdout);
|
|
for line in reader.lines().flatten() {
|
|
debug_log(&format!("[NODE STDOUT] {}", line));
|
|
if let Ok(msg) = serde_json::from_str::<BackendMessage>(&line) {
|
|
debug_log(&format!("[MAIN-DEBUG] Received event: {}", msg.event));
|
|
debug_log(&format!("[MAIN-DEBUG] Event data: {}", serde_json::to_string(&msg.data).unwrap_or_default()));
|
|
if msg.event == "tauri-cmd" {
|
|
let window_label = msg.data.get("window").and_then(|val| val.as_str()).unwrap_or("main");
|
|
debug_log(&format!("[MAIN-DEBUG] Finding window by label: {}", window_label));
|
|
|
|
if let Some(cmd_window) = w.app_handle().get_webview_window(window_label) {
|
|
if let Some(cmd) = msg.data.get("cmd").and_then(|c| c.as_str()) {
|
|
debug_log(&format!("[TAURI-CMD] Executing action: {}", cmd));
|
|
match cmd {
|
|
"win_minimize" => {
|
|
let res = cmd_window.minimize();
|
|
debug_log(&format!("[TAURI-CMD] Minimize result: {:?}", res));
|
|
},
|
|
"win_maximize" => {
|
|
let is_max = cmd_window.is_maximized().unwrap_or(false);
|
|
debug_log(&format!("[TAURI-CMD] Currently maximized: {}", is_max));
|
|
let res = if is_max { cmd_window.unmaximize() } else { cmd_window.maximize() };
|
|
debug_log(&format!("[TAURI-CMD] Maximize/Unmaximize result: {:?}", res));
|
|
},
|
|
"win_close" => {
|
|
let res = cmd_window.close();
|
|
debug_log(&format!("[TAURI-CMD] Close result: {:?}", res));
|
|
},
|
|
"win_drag" => {
|
|
let res = cmd_window.start_dragging();
|
|
debug_log(&format!("[TAURI-CMD] Drag result: {:?}", res));
|
|
},
|
|
"open_new_window" => {
|
|
debug_log("[TAURI-CMD-DEBUG] open_new_window command received");
|
|
debug_log(&format!("[TAURI-CMD-DEBUG] Full message data: {}", serde_json::to_string(&msg.data).unwrap_or_default()));
|
|
if let Some(url) = msg.data.get("url").and_then(|u| u.as_str()) {
|
|
debug_log(&format!("[TAURI-CMD-DEBUG] URL extracted: {}", url));
|
|
debug_log(&format!("[TAURI-CMD-DEBUG] URL length: {}", url.len()));
|
|
debug_log("[TAURI-CMD-DEBUG] Attempting to dispatch to main thread");
|
|
let app_handle = w.app_handle().clone();
|
|
let closure_handle = app_handle.clone();
|
|
let url_owned = url.to_string();
|
|
let dispatch_result = app_handle.run_on_main_thread(move || {
|
|
debug_log("[TAURI-CMD-DEBUG] Now on main thread, calling open_new_window");
|
|
match open_new_window(closure_handle, url_owned) {
|
|
Ok(_) => {
|
|
debug_log("[TAURI-CMD-DEBUG] open_new_window succeeded");
|
|
},
|
|
Err(e) => {
|
|
debug_log(&format!("[TAURI-CMD-DEBUG] open_new_window error: {}", e));
|
|
},
|
|
}
|
|
});
|
|
match dispatch_result {
|
|
Ok(_) => debug_log("[TAURI-CMD-DEBUG] Main thread dispatch succeeded"),
|
|
Err(e) => debug_log(&format!("[TAURI-CMD-DEBUG] Main thread dispatch failed: {}", e)),
|
|
}
|
|
} else {
|
|
debug_log("[TAURI-CMD-DEBUG] ERROR: No URL found in message data");
|
|
debug_log(&format!("[TAURI-CMD-DEBUG] Available keys: {:?}", msg.data.as_object().map(|o| o.keys().collect::<Vec<_>>())));
|
|
}
|
|
},
|
|
_ => {}
|
|
}
|
|
}
|
|
} else {
|
|
debug_log(&format!("[TAURI-CMD] Window '{}' NOT FOUND!", window_label));
|
|
}
|
|
} else {
|
|
let _ = w.emit(&msg.event, msg.data.clone());
|
|
if msg.event == "ready" {
|
|
if let Some(url) = msg.data.get("url").and_then(|u| u.as_str()) {
|
|
let _ = w.eval(&format!("window.location.href = '{}';", url));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
let state: State<NodeServices> = app.state();
|
|
if let Ok(mut lock) = state.0.lock() {
|
|
*lock = Some(child);
|
|
};
|
|
}
|
|
Err(e) => {
|
|
debug_log(&format!("Failed to spawn node: {}", e));
|
|
let _ = window.emit("error", serde_json::json!({ "message": format!("Failed to launch backend runtime: {}", e) }));
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
})
|
|
.on_window_event(|window, event| {
|
|
if let tauri::WindowEvent::Destroyed = event {
|
|
if window.label() == "main" {
|
|
let state: State<NodeServices> = window.state();
|
|
if let Ok(mut lock) = state.0.lock() {
|
|
if let Some(child) = lock.take() {
|
|
let pid = child.id();
|
|
debug_log(&format!("[SHUTDOWN] Main window destroyed, killing process tree for PID {}", pid));
|
|
kill_process_tree(pid);
|
|
}
|
|
};
|
|
}
|
|
}
|
|
})
|
|
.run(tauri::generate_context!())
|
|
.expect("error while running tauri application");
|
|
} |