1
0
mirror of https://github.com/Tim-Paik/neutauri.git synced 2024-10-12 23:29:41 +00:00

update runner

This commit is contained in:
2022-01-29 22:06:03 +08:00
parent 775ad8024d
commit acfb6b6c6b
4 changed files with 499 additions and 37 deletions

View File

@ -5,7 +5,7 @@ use std::{
io::{self, Read, Result, Seek, SeekFrom},
path::{self, Component, Path, PathBuf},
};
use wry::application::dpi::{Position, Size};
use wry::application::dpi::Position;
const MAGIC_NUMBER_START: &[u8; 9] = b"NEUTFSv01";
const MAGIC_NUMBER_END: &[u8; 9] = b"NEUTFSEnd";
@ -31,30 +31,55 @@ struct Dir {
dirs: Vec<(String, Dir)>,
}
#[derive(Serialize, Deserialize, Debug)]
struct Config {
window_attr: WindowAttr,
webview_attr: WebViewAttr,
}
#[derive(Serialize, Deserialize, Debug)]
pub struct Data {
config: Config,
pub window_attr: WindowAttr,
pub webview_attr: WebViewAttr,
fs: Dir,
}
#[derive(Serialize, Deserialize, Clone, Debug)]
pub enum WindowSize {
Large,
Medium,
Small,
Fixed(f64, f64),
}
#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct Config {
pub source: PathBuf,
pub target: PathBuf,
pub inner_size: Option<WindowSize>,
pub min_inner_size: Option<WindowSize>,
pub max_inner_size: Option<WindowSize>,
pub resizable: bool,
pub fullscreen: bool,
pub title: String,
pub maximized: bool,
pub visible: bool,
pub transparent: bool,
pub decorations: bool,
pub always_on_top: bool,
pub window_icon: Option<PathBuf>,
pub spa: bool,
pub url: Option<String>,
pub html: Option<PathBuf>,
pub initialization_script: Option<String>,
}
#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct Icon {
rgba: Vec<u8>,
width: u32,
height: u32,
pub rgba: Vec<u8>,
pub width: u32,
pub height: u32,
}
#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct WindowAttr {
pub inner_size: Option<Size>,
pub min_inner_size: Option<Size>,
pub max_inner_size: Option<Size>,
pub inner_size: Option<WindowSize>,
pub min_inner_size: Option<WindowSize>,
pub max_inner_size: Option<WindowSize>,
pub position: Option<Position>,
pub resizable: bool,
pub fullscreen: bool,
@ -71,15 +96,15 @@ pub struct WindowAttr {
pub struct WebViewAttr {
pub visible: bool,
pub transparent: bool,
pub spa: bool,
pub url: Option<String>,
pub html: Option<String>,
pub initialization_scripts: Vec<String>,
pub initialization_script: Option<String>,
}
impl File {
pub fn decompressed_data(&mut self) -> Result<Vec<u8>> {
let mut data = Vec::with_capacity(self.data.len());
//brotli::BrotliDecompress(&mut self.data, &mut data);
let mut r = brotli::Decompressor::new(self.data.as_slice(), 4096);
r.read_to_end(&mut data)?;
Ok(data)
@ -164,10 +189,8 @@ impl Data {
dir.fill_with(source, source, &mut length)?;
Ok(Self {
fs: dir,
config: Config {
window_attr,
webview_attr,
},
window_attr,
webview_attr,
})
}
@ -197,11 +220,20 @@ impl Data {
Ok(target)
}
pub fn pack<P: AsRef<path::Path>>(source: P, target: P, config: P) -> Result<()> {
let config: Config = toml::from_str(fs::read_to_string(config)?.as_str())?;
pub fn pack<P: AsRef<path::Path>>(config_path: P) -> Result<()> {
let config_path: &Path = config_path.as_ref().clone();
let config: Config = toml::from_str(fs::read_to_string(config_path)?.as_str())?;
let source = &match config_path.parent() {
Some(path) => path.join(&config.source).canonicalize()?,
None => config.source.canonicalize()?,
};
let target = &match config_path.parent() {
Some(path) => path.join(&config.target).canonicalize()?,
None => config.target.canonicalize()?,
};
fs::write(
target,
Self::build_from_dir(source, config.window_attr, config.webview_attr)?,
Self::build_from_dir(source, config.window_attr()?, config.webview_attr()?)?,
)?;
Ok(())
}
@ -242,7 +274,6 @@ impl Data {
// 此时指针指向 Data 前
base.take(u64::from_be_bytes(data_length_data))
.read_to_end(&mut data)?;
fs::write("test2", &data)?;
let serialize_options = bincode::DefaultOptions::new()
.with_fixint_encoding()
.allow_trailing_bytes()
@ -287,12 +318,91 @@ impl Data {
}
}
impl Default for Config {
fn default() -> Self {
Self {
source: PathBuf::from("."),
target: PathBuf::from("app.neu"),
inner_size: Some(WindowSize::Medium),
min_inner_size: None,
max_inner_size: None,
resizable: true,
fullscreen: false,
title: "".to_string(),
maximized: false,
visible: true,
transparent: false,
decorations: true,
always_on_top: false,
window_icon: None,
spa: false,
url: Some("/index.html".to_string()),
html: None,
initialization_script: Some("".to_string()),
}
}
}
impl Config {
pub fn window_attr(&self) -> Result<WindowAttr> {
Ok(WindowAttr {
inner_size: self.inner_size.clone(),
min_inner_size: self.min_inner_size.clone(),
max_inner_size: self.max_inner_size.clone(),
position: None,
resizable: self.resizable,
fullscreen: self.fullscreen,
title: self.title.clone(),
maximized: self.maximized,
visible: self.visible,
transparent: self.transparent,
decorations: self.decorations,
always_on_top: self.always_on_top,
window_icon: match &self.window_icon {
Some(path) => Some(load_icon(&path.as_path())?),
None => None,
},
})
}
pub fn webview_attr(&self) -> Result<WebViewAttr> {
Ok(WebViewAttr {
visible: self.visible,
transparent: self.transparent,
spa: self.spa,
url: self.url.clone(),
html: match &self.html {
Some(path) => fs::read_to_string(path.as_path()).ok(),
None => None,
},
initialization_script: self.initialization_script.clone(),
})
}
}
pub fn load<P: AsRef<path::Path> + Copy>(path: P) -> Result<Data> {
Data::new(path)
}
pub fn pack<P: AsRef<path::Path>>(source: P, target: P, config: P) -> Result<()> {
Data::pack(source, target, config)
pub fn pack<P: AsRef<path::Path>>(config: P) -> Result<()> {
Data::pack(config)
}
fn load_icon(path: &Path) -> Result<Icon> {
let (icon_rgba, icon_width, icon_height) = {
let image = match image::open(path) {
Ok(img) => img,
Err(e) => return Err(io::Error::new(io::ErrorKind::InvalidData, e)),
}
.into_rgba8();
let (width, height) = image.dimensions();
let rgba = image.into_raw();
(rgba, width, height)
};
Ok(Icon {
rgba: icon_rgba,
width: icon_width,
height: icon_height,
})
}
fn normalize_path(path: &Path) -> PathBuf {

View File

@ -1,10 +1,10 @@
use wry::{
application::{
event::{Event, StartCause, WindowEvent},
event_loop::{ControlFlow, EventLoop},
window::WindowBuilder,
event_loop::{self, ControlFlow, EventLoop},
window::{Fullscreen, Icon, Window, WindowBuilder},
},
webview::WebViewBuilder,
webview::{RpcRequest, WebViewBuilder},
};
mod data;
@ -13,7 +13,7 @@ const PROTOCOL_PREFIX: &str = "https://{PROTOCOL}.";
#[cfg(not(windows))]
const PROTOCOL_PREFIX: &str = "{PROTOCOL}://";
const PROTOCOL: &str = "neutauri";
const PROTOCOL: &str = "neu";
fn custom_protocol_uri<T: Into<String>>(protocol: T, path: T) -> String {
PROTOCOL_PREFIX.replacen("{PROTOCOL}", &protocol.into(), 1) + &path.into()
@ -32,26 +32,135 @@ fn custom_protocol_uri_to_path<T: Into<String>>(protocol: T, uri: T) -> wry::Res
}
fn main() -> wry::Result<()> {
let res = data::load(std::env::current_exe()?.as_path())?;
data::pack("config.toml")?;
let res = match data::load(std::env::current_exe()?.as_path()) {
Ok(data) => data,
Err(_) => data::load("app.neu")?,
};
let event_loop = EventLoop::new();
let window = WindowBuilder::new().build(&event_loop)?;
let _webview = WebViewBuilder::new(window)?
.with_transparent(false)
let window_builder = WindowBuilder::new()
.with_always_on_top(res.window_attr.always_on_top)
.with_decorations(res.window_attr.decorations)
.with_resizable(res.window_attr.resizable)
.with_title(res.window_attr.title.clone())
.with_maximized(res.window_attr.maximized)
.with_transparent(res.window_attr.transparent)
.with_visible(res.window_attr.visible);
let window_builder = match res.window_attr.fullscreen {
true => window_builder.with_fullscreen(Some(Fullscreen::Borderless(None))),
false => window_builder,
};
let window_builder = match res.window_attr.window_icon {
Some(ref icon) => window_builder.with_window_icon(Some(Icon::from_rgba(
icon.rgba.clone(),
icon.width,
icon.height,
)?)),
None => window_builder,
};
let monitor = event_loop
.primary_monitor()
.unwrap_or_else(|| event_loop.available_monitors().next().unwrap());
dbg!(
monitor.size(),
monitor.name(),
monitor.position(),
monitor.scale_factor(),
monitor.video_modes().collect::<Vec<_>>()
);
let window_builder = match res.window_attr.inner_size {
Some(size) => match size {
data::WindowSize::Large => todo!(),
data::WindowSize::Medium => todo!(),
data::WindowSize::Small => todo!(),
data::WindowSize::Fixed(width, height) => todo!(),
},
None => window_builder,
};
let window_builder = match res.window_attr.max_inner_size {
Some(size) => match size {
data::WindowSize::Large => todo!(),
data::WindowSize::Medium => todo!(),
data::WindowSize::Small => todo!(),
data::WindowSize::Fixed(width, height) => todo!(),
},
None => window_builder,
};
let window_builder = match res.window_attr.min_inner_size {
Some(size) => match size {
data::WindowSize::Large => todo!(),
data::WindowSize::Medium => todo!(),
data::WindowSize::Small => todo!(),
data::WindowSize::Fixed(width, height) => todo!(),
},
None => window_builder,
};
let window = window_builder.build(&event_loop)?;
let webview_builder = WebViewBuilder::new(window)?;
let url = res.webview_attr.url.clone();
let webview_builder = match url {
Some(url) => {
if url.starts_with("/") {
webview_builder.with_url(&custom_protocol_uri(PROTOCOL, &url))?
} else {
webview_builder.with_url(&url)?
}
}
None => webview_builder.with_url(&custom_protocol_uri(PROTOCOL, "/index.html"))?,
};
let html = res.webview_attr.html.clone();
let webview_builder = match html {
Some(html) => webview_builder.with_html(&html)?,
None => webview_builder,
};
let initialization_script = res.webview_attr.initialization_script.clone();
let webview_builder = match initialization_script {
Some(script) => webview_builder.with_initialization_script(&script),
None => webview_builder,
};
let webview_builder = match res.window_attr.visible {
true => webview_builder.with_visible(true),
false => webview_builder
.with_visible(false)
.with_initialization_script(
r#"window.addEventListener('load', function(event) { rpc.call('show_window'); });"#,
),
};
let _webview = webview_builder
.with_visible(res.window_attr.visible)
.with_transparent(res.window_attr.transparent)
.with_custom_protocol(PROTOCOL.to_string(), move |request| {
let path = custom_protocol_uri_to_path(PROTOCOL, request.uri())?;
let mut file = res.open(path)?;
let mut file = match res.open(path) {
Ok(file) => file,
Err(e) => {
if e.kind() == std::io::ErrorKind::NotFound && res.webview_attr.spa {
res.open("index.html")?
} else {
return Err(wry::Error::Io(e));
}
}
};
wry::http::ResponseBuilder::new()
.mimetype(&file.mimetype())
.body(file.decompressed_data()?)
})
.with_url(&custom_protocol_uri(PROTOCOL, "/index.html"))?
.with_rpc_handler(|window: &Window, req: RpcRequest| {
match req.method.as_str() {
"show_window" => window.set_visible(true),
"ping" => println!("recived a ping"),
_ => (),
};
None
})
.build()?;
event_loop.run(move |event, _, control_flow| {
*control_flow = ControlFlow::Wait;
match event {
Event::NewEvents(StartCause::Init) => println!("Wry has started!"),
Event::WindowEvent {
event: WindowEvent::CloseRequested,
..