350 lines
11 KiB
Rust
350 lines
11 KiB
Rust
#![cfg_attr(
|
|
all(not(debug_assertions), target_os = "windows"),
|
|
windows_subsystem = "windows"
|
|
)]
|
|
#[cfg(target_os = "macos")]
|
|
use tauri::MenuItem;
|
|
|
|
use tauri::{
|
|
CustomMenuItem, Menu, Submenu, WindowBuilder, App, Config,
|
|
Window, WindowUrl, WindowMenuEvent, window::PlatformWebview,
|
|
SystemTrayMenu, SystemTray, SystemTrayEvent, Manager
|
|
};
|
|
|
|
// use tauri::Config;
|
|
|
|
pub mod pake {
|
|
use serde::Deserialize;
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub struct WindowConfig {
|
|
pub url: String,
|
|
pub transparent: bool,
|
|
pub fullscreen: bool,
|
|
pub width: f64,
|
|
pub height: f64,
|
|
pub resizable: bool,
|
|
pub url_type: String
|
|
}
|
|
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub struct UserAgent {
|
|
pub macos: String,
|
|
pub linux: String,
|
|
pub windows: String,
|
|
}
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub struct PakeConfig {
|
|
pub windows: Vec<WindowConfig>,
|
|
pub user_agent: UserAgent,
|
|
}
|
|
}
|
|
|
|
use pake::PakeConfig;
|
|
|
|
pub fn get_pake_config() -> (PakeConfig, Config){
|
|
let pake_config_path = include_str!("../pake.json");
|
|
let pake_config: PakeConfig = serde_json::from_str(pake_config_path)
|
|
.expect("failed to parse pake config");
|
|
// println!("{:#?}", config);
|
|
let tauri_config_path = include_str!("../tauri.conf.json");
|
|
let tauri_config: Config = serde_json::from_str(tauri_config_path)
|
|
.expect("failed to parse tauri config");
|
|
(pake_config, tauri_config)
|
|
}
|
|
|
|
|
|
pub fn get_menu() -> Menu {
|
|
// first menu
|
|
let hide = CustomMenuItem::new("hide", "Hide");
|
|
let close = CustomMenuItem::new("close", "Close");
|
|
let quit = CustomMenuItem::new("quit", "Quit");
|
|
#[cfg(target_os = "macos")]
|
|
let first_menu = Menu::new()
|
|
.add_native_item(MenuItem::EnterFullScreen)
|
|
.add_native_item(MenuItem::Minimize)
|
|
.add_native_item(MenuItem::Separator)
|
|
.add_native_item(MenuItem::Copy)
|
|
.add_native_item(MenuItem::Cut)
|
|
.add_native_item(MenuItem::Paste)
|
|
.add_native_item(MenuItem::Undo)
|
|
.add_native_item(MenuItem::Redo)
|
|
.add_native_item(MenuItem::SelectAll)
|
|
.add_native_item(MenuItem::Separator)
|
|
.add_item(debug)
|
|
.add_item(hide)
|
|
.add_item(close)
|
|
.add_item(quit);
|
|
#[cfg(any(target_os = "linux", target_os = "windows"))]
|
|
let first_menu = Menu::new()
|
|
.add_item(hide)
|
|
.add_item(close)
|
|
.add_item(quit);
|
|
let first_menu = Submenu::new("File", first_menu);
|
|
// Hot Key
|
|
// let top = CustomMenuItem::new("top", "Top (↑)");
|
|
// let buttom = CustomMenuItem::new("buttom", "Bottom (↓)");
|
|
// let previous = CustomMenuItem::new("previous", "Previous (←)");
|
|
// let next = CustomMenuItem::new("next", "next (→)");
|
|
// let refresh = CustomMenuItem::new("refresh", "Refresh");
|
|
let zoom_out = CustomMenuItem::new("zoom_out", "Zoom Out (125%)");
|
|
let zoom_in = CustomMenuItem::new("zoom_in", "Zoom In (75%)");
|
|
let zoom_reset = CustomMenuItem::new("reset", "Zoom Reset");
|
|
let hot_key = Menu::new()
|
|
// .add_item(top)
|
|
// .add_item(buttom)
|
|
// .add_item(previous)
|
|
// .add_item(next)
|
|
// .add_item(refresh)
|
|
.add_item(zoom_in)
|
|
.add_item(zoom_out)
|
|
.add_item(zoom_reset);
|
|
let hot_key_menu = Submenu::new("Hot Key", hot_key);
|
|
|
|
// Help
|
|
// let instructions = CustomMenuItem::new("instruction", "Instruction");
|
|
// let about = CustomMenuItem::new("about", "About");
|
|
// let help = Menu::new()
|
|
// .add_item(instructions)
|
|
// .add_item(about);
|
|
// let help_menu = Submenu::new("Help", help);
|
|
let menu = Menu::new()
|
|
.add_submenu(first_menu)
|
|
.add_submenu(hot_key_menu);
|
|
// .add_submenu(help_menu);
|
|
menu
|
|
|
|
|
|
}
|
|
|
|
|
|
pub fn get_data_dir(tauri_config: Config) -> std::path::PathBuf {
|
|
let package_name = tauri_config.package.product_name.unwrap();
|
|
let home_dir = match home::home_dir() {
|
|
Some(path1) => path1,
|
|
None => panic!("Error, can't found you home dir!!"),
|
|
};
|
|
#[cfg(target_os = "windows")]
|
|
let data_dir = home_dir.join("AppData").join("Roaming").join(package_name);
|
|
#[cfg(target_os = "linux")]
|
|
let data_dir = home_dir.join(".config").join(package_name);
|
|
if !data_dir.exists() {
|
|
std::fs::create_dir(&data_dir)
|
|
.unwrap_or_else(|_| panic!("can't create dir {}", data_dir.display()));
|
|
}
|
|
data_dir
|
|
}
|
|
|
|
|
|
pub fn get_window(app: & mut App, config: PakeConfig, data_dir: std::path::PathBuf) -> Window {
|
|
let window_config = config.windows.first().unwrap();
|
|
let user_agent = config.user_agent;
|
|
let url = match window_config.url_type.as_str() {
|
|
"web" => WindowUrl::External(window_config.url.parse().unwrap()),
|
|
"local" => WindowUrl::App(std::path::PathBuf::from(&window_config.url)),
|
|
_ => panic!("url type only can be web or local"),
|
|
};
|
|
#[cfg(target_os = "macos")]
|
|
let window = WindowBuilder::new(
|
|
app,
|
|
"pake",
|
|
url
|
|
)
|
|
.title("")
|
|
.user_agent(user_agent.macos.as_str())
|
|
.resizable(window_config.resizable)
|
|
.fullscreen(window_config.fullscreen)
|
|
.transparent(window_config.transparent)
|
|
.inner_size(window_config.width, window_config.height)
|
|
.initialization_script(include_str!("pake.js"));
|
|
|
|
#[cfg(any(target_os = "linux", target_os = "windows"))]
|
|
let window = {
|
|
#[cfg(target_os = "linux")]
|
|
let user_agent = user_agent.linux.as_str();
|
|
#[cfg(target_os = "windows")]
|
|
let user_agent = user_agent.windows.as_str();
|
|
WindowBuilder::new(
|
|
app,
|
|
"pake",
|
|
url
|
|
)
|
|
.title("")
|
|
.data_directory(data_dir)
|
|
.resizable(window_config.resizable)
|
|
.fullscreen(window_config.fullscreen)
|
|
.user_agent(user_agent)
|
|
.inner_size(window_config.width, window_config.height)
|
|
.initialization_script(include_str!("pake.js"))
|
|
};
|
|
window.build().unwrap()
|
|
|
|
}
|
|
|
|
pub fn set_zoom(webview: PlatformWebview, zoom_value: f64) {
|
|
#[cfg(target_os = "linux")]
|
|
{
|
|
// see https://docs.rs/webkit2gtk/0.18.2/webkit2gtk/struct.WebView.html
|
|
// and https://docs.rs/webkit2gtk/0.18.2/webkit2gtk/trait.WebViewExt.html
|
|
use webkit2gtk::traits::WebViewExt;
|
|
webview.inner().set_zoom_level(zoom_value);
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
unsafe {
|
|
// see https://docs.rs/webview2-com/0.19.1/webview2_com/Microsoft/Web/WebView2/Win32/struct.ICoreWebView2Controller.html
|
|
webview.controller().SetZoomFactor(zoom_value).unwrap();
|
|
}
|
|
|
|
#[cfg(target_os = "macos")]
|
|
unsafe {
|
|
let () = msg_send![webview.inner(), setPageZoom: zoom_value];
|
|
let () = msg_send![webview.controller(), removeAllUserScripts];
|
|
let bg_color: cocoa::base::id = msg_send![class!(NSColor), colorWithDeviceRed:0.5 green:0.2 blue:0.4 alpha:1.];
|
|
let () = msg_send![webview.ns_window(), setBackgroundColor: bg_color];
|
|
}
|
|
}
|
|
|
|
pub fn set_zoom_out(webview: PlatformWebview) {
|
|
set_zoom(webview, 1.25);
|
|
}
|
|
|
|
|
|
pub fn set_zoom_in(webview: PlatformWebview) {
|
|
set_zoom(webview, 0.75);
|
|
}
|
|
|
|
|
|
pub fn zoom_reset(webview: PlatformWebview) {
|
|
set_zoom(webview, 1.0);
|
|
}
|
|
|
|
pub fn menu_event_handle(event: WindowMenuEvent) {
|
|
match event.menu_item_id() {
|
|
"hide" => event.window().hide().expect("can't hide window"),
|
|
"close" => event.window().close().expect("can't close window"),
|
|
"quit" => std::process::exit(0),
|
|
"zoom_out" => {
|
|
event.window()
|
|
.with_webview(set_zoom_out)
|
|
.expect("can't set zoom out");
|
|
},
|
|
"zoom_in" => {
|
|
event.window()
|
|
.with_webview(set_zoom_in)
|
|
.expect("can't set zoom in");
|
|
},
|
|
"reset" => {
|
|
event.window()
|
|
.with_webview(zoom_reset)
|
|
.expect("can't reset zoom");
|
|
},
|
|
_ => {},
|
|
}
|
|
}
|
|
|
|
|
|
pub fn get_system_tray() -> SystemTray {
|
|
let hide = CustomMenuItem::new("hide".to_string(), "Hide");
|
|
let show = CustomMenuItem::new("show".to_string(), "Show");
|
|
let quit = CustomMenuItem::new("quit".to_string(), "Quit");
|
|
let about = CustomMenuItem::new("about".to_string(), "About");
|
|
let tray_menu = SystemTrayMenu::new()
|
|
.add_item(hide)
|
|
.add_item(show)
|
|
.add_item(quit)
|
|
.add_item(about);
|
|
SystemTray::new().with_menu(tray_menu)
|
|
}
|
|
|
|
|
|
pub fn system_tray_handle(app: &tauri::AppHandle, event: tauri::SystemTrayEvent) {
|
|
if let SystemTrayEvent::MenuItemClick { tray_id: _, id, .. } = event {
|
|
match id.as_str() {
|
|
"hide" => {
|
|
app.get_window("pake").unwrap().hide().unwrap();
|
|
},
|
|
"show" => {
|
|
app.get_window("pake").unwrap().show().unwrap();
|
|
},
|
|
"quit" => {
|
|
std::process::exit(0);
|
|
},
|
|
"about" => {
|
|
let _about_window = WindowBuilder::new(
|
|
app,
|
|
"about",
|
|
WindowUrl::App(std::path::PathBuf::from("about_pake.html"))
|
|
)
|
|
.resizable(true)
|
|
.title("About")
|
|
.inner_size(100.0, 100.0)
|
|
.build()
|
|
.expect("can't open about!")
|
|
;
|
|
}
|
|
_ => {},
|
|
}
|
|
};
|
|
}
|
|
|
|
|
|
pub fn run_app() {
|
|
let system_tray = get_system_tray();
|
|
#[cfg(target_os = "macos")]
|
|
{
|
|
let (pake_config, _) = get_pake_config();
|
|
let menu = get_menu();
|
|
tauri::Builder::default()
|
|
.menu(menu)
|
|
.on_menu_event(menu_event_handle)
|
|
.system_tray(system_tray)
|
|
.on_system_tray_event(system_tray_handle)
|
|
.plugin(tauri_plugin_window_state::Builder::default().build())
|
|
.invoke_handler(tauri::generate_handler![])
|
|
.setup(|app| {
|
|
let _window = get_window(app, pake_config, std::path::PathBuf::new());
|
|
#[cfg(feature = "devtools")]
|
|
{
|
|
app.get_window("pake").unwrap().open_devtools();
|
|
}
|
|
Ok(())
|
|
})
|
|
.run(tauri::generate_context!())
|
|
.expect("error while running tauri application");
|
|
}
|
|
#[cfg(any(target_os = "linux", target_os = "windows"))]
|
|
{
|
|
let (pake_config, tauri_config) = get_pake_config();
|
|
let data_dir = get_data_dir(tauri_config);
|
|
// let menu = get_menu();
|
|
tauri::Builder::default()
|
|
// .menu(menu)
|
|
// .on_menu_event(menu_event_handle)
|
|
.system_tray(system_tray)
|
|
.on_system_tray_event(system_tray_handle)
|
|
.plugin(tauri_plugin_window_state::Builder::default().build())
|
|
.invoke_handler(tauri::generate_handler![])
|
|
.setup(|app| {
|
|
let _window = get_window(app, pake_config, data_dir);
|
|
#[cfg(feature = "devtools")]
|
|
{
|
|
app.get_window("pake").unwrap().open_devtools();
|
|
}
|
|
Ok(())
|
|
})
|
|
.run(tauri::generate_context!())
|
|
.expect("error while running tauri application");
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
fn main() {
|
|
run_app()
|
|
|
|
}
|