mirror of
https://github.com/YGGverse/Yoda.git
synced 2026-04-01 00:55:28 +00:00
660 lines
33 KiB
Rust
660 lines
33 KiB
Rust
mod content;
|
|
mod database;
|
|
mod input;
|
|
mod meta;
|
|
mod navigation;
|
|
mod widget;
|
|
|
|
use content::Content;
|
|
use database::Database;
|
|
use input::Input;
|
|
use navigation::Navigation;
|
|
use widget::Widget;
|
|
|
|
use meta::{Meta, Status};
|
|
|
|
use gtk::{
|
|
gdk_pixbuf::Pixbuf,
|
|
gio::{
|
|
Cancellable, SimpleAction, SocketClient, SocketClientEvent, SocketProtocol,
|
|
TlsCertificateFlags,
|
|
},
|
|
glib::{
|
|
gformat, uuid_string_random, Bytes, GString, Priority, Regex, RegexCompileFlags,
|
|
RegexMatchFlags, Uri, UriFlags,
|
|
},
|
|
prelude::{
|
|
ActionExt, IOStreamExt, InputStreamExt, OutputStreamExt, SocketClientExt,
|
|
StaticVariantType, ToVariant,
|
|
},
|
|
Box,
|
|
};
|
|
use sqlite::Transaction;
|
|
use std::{cell::RefCell, sync::Arc};
|
|
|
|
pub struct Page {
|
|
id: GString,
|
|
// Actions
|
|
action_page_open: Arc<SimpleAction>,
|
|
action_tab_page_navigation_reload: Arc<SimpleAction>,
|
|
action_update: Arc<SimpleAction>,
|
|
// Components
|
|
navigation: Arc<Navigation>,
|
|
content: Arc<Content>,
|
|
input: Arc<Input>,
|
|
// Extras
|
|
meta: Arc<RefCell<Meta>>,
|
|
// GTK
|
|
widget: Arc<Widget>,
|
|
}
|
|
|
|
impl Page {
|
|
// Construct
|
|
pub fn new_arc(
|
|
id: GString,
|
|
action_tab_open: Arc<SimpleAction>,
|
|
action_tab_page_navigation_base: Arc<SimpleAction>,
|
|
action_tab_page_navigation_history_back: Arc<SimpleAction>,
|
|
action_tab_page_navigation_history_forward: Arc<SimpleAction>,
|
|
action_tab_page_navigation_reload: Arc<SimpleAction>,
|
|
action_update: Arc<SimpleAction>,
|
|
) -> Arc<Self> {
|
|
// Init local actions
|
|
let action_page_open = Arc::new(SimpleAction::new(
|
|
&uuid_string_random(),
|
|
Some(&String::static_variant_type()),
|
|
));
|
|
|
|
// Init components
|
|
let content = Arc::new(Content::new(
|
|
action_tab_open.clone(),
|
|
action_page_open.clone(),
|
|
));
|
|
|
|
let navigation = Navigation::new_arc(
|
|
action_tab_page_navigation_base.clone(),
|
|
action_tab_page_navigation_history_back.clone(),
|
|
action_tab_page_navigation_history_forward.clone(),
|
|
action_tab_page_navigation_reload.clone(),
|
|
action_update.clone(),
|
|
);
|
|
|
|
let input = Input::new_arc();
|
|
|
|
let widget = Widget::new_arc(
|
|
&id,
|
|
action_page_open.clone(),
|
|
navigation.gobject(),
|
|
content.gobject(),
|
|
input.gobject(),
|
|
);
|
|
|
|
// Init async mutable Meta object
|
|
let meta = Arc::new(RefCell::new(Meta::new()));
|
|
|
|
// Init events
|
|
action_page_open.connect_activate({
|
|
let navigation = navigation.clone();
|
|
let action_tab_page_navigation_reload = action_tab_page_navigation_reload.clone();
|
|
move |_, request| {
|
|
// Update request
|
|
navigation.set_request_text(
|
|
request
|
|
.expect("Parameter required for `page.open` action")
|
|
.get::<String>()
|
|
.expect("Parameter does not match `String`")
|
|
.as_str(),
|
|
);
|
|
|
|
// Reload page
|
|
action_tab_page_navigation_reload.activate(None);
|
|
}
|
|
});
|
|
|
|
// Return activated structure
|
|
Arc::new(Self {
|
|
id,
|
|
// Actions
|
|
action_page_open,
|
|
action_tab_page_navigation_reload,
|
|
action_update,
|
|
// Components
|
|
content,
|
|
navigation,
|
|
input,
|
|
// Extras
|
|
meta,
|
|
// GTK
|
|
widget,
|
|
})
|
|
}
|
|
|
|
// Actions
|
|
pub fn navigation_request_grab_focus(&self) {
|
|
self.navigation.request_grab_focus();
|
|
}
|
|
|
|
pub fn navigation_base(&self) {
|
|
if let Some(url) = self.navigation.base_url() {
|
|
// Update with history record
|
|
self.action_page_open.activate(Some(&url.to_variant()));
|
|
}
|
|
}
|
|
|
|
pub fn navigation_history_back(&self) {
|
|
if let Some(request) = self.navigation.history_back(true) {
|
|
// Update
|
|
self.navigation.set_request_text(&request);
|
|
|
|
// Reload page
|
|
self.action_tab_page_navigation_reload.activate(None);
|
|
}
|
|
}
|
|
|
|
pub fn navigation_history_forward(&self) {
|
|
if let Some(request) = self.navigation.history_forward(true) {
|
|
// Update
|
|
self.navigation.set_request_text(&request);
|
|
|
|
// Reload page
|
|
self.action_tab_page_navigation_reload.activate(None);
|
|
}
|
|
}
|
|
|
|
pub fn navigation_reload(&self) {
|
|
// Reset widgets
|
|
self.input.unset();
|
|
|
|
// Init shared objects to not spawn a lot
|
|
let request_text = self.navigation.request_text();
|
|
let id = self.id.to_variant();
|
|
|
|
// Update
|
|
self.meta.borrow_mut().status = Some(Status::Reload);
|
|
self.meta.borrow_mut().title = Some(gformat!("Loading.."));
|
|
self.meta.borrow_mut().description = None;
|
|
|
|
self.action_update.activate(Some(&id));
|
|
|
|
// Route by request
|
|
match Uri::parse(&request_text, UriFlags::NONE) {
|
|
Ok(uri) => {
|
|
// Route by scheme
|
|
match uri.scheme().as_str() {
|
|
"file" => todo!(),
|
|
"gemini" => self.load_gemini(uri), // @TODO
|
|
scheme => {
|
|
// Define common data
|
|
let status = Status::Failure;
|
|
let title = gformat!("Oops");
|
|
let description = gformat!("Protocol `{scheme}` not supported");
|
|
|
|
// Update widget
|
|
self.content
|
|
.set_status_failure(title.as_str(), description.as_str());
|
|
|
|
// Update meta
|
|
self.meta.borrow_mut().status = Some(status);
|
|
self.meta.borrow_mut().title = Some(title);
|
|
self.meta.borrow_mut().description = Some(description);
|
|
|
|
// Update window
|
|
self.action_update.activate(Some(&id));
|
|
}
|
|
}
|
|
}
|
|
Err(_) => {
|
|
// Try interpret URI manually
|
|
if Regex::match_simple(
|
|
r"^[^\/\s]+\.[\w]{2,}",
|
|
request_text.clone(),
|
|
RegexCompileFlags::DEFAULT,
|
|
RegexMatchFlags::DEFAULT,
|
|
) {
|
|
// Seems request contain some host, try append default scheme
|
|
let request_text = gformat!("gemini://{request_text}");
|
|
// Make sure new request conversable to valid URI
|
|
match Uri::parse(&request_text, UriFlags::NONE) {
|
|
Ok(_) => {
|
|
// Update
|
|
self.navigation.set_request_text(&request_text);
|
|
|
|
// Reload page
|
|
self.action_tab_page_navigation_reload.activate(None);
|
|
}
|
|
Err(_) => {
|
|
// @TODO any action here?
|
|
}
|
|
}
|
|
} else {
|
|
// Plain text given, make search request to default provider
|
|
let request_text = gformat!(
|
|
"gemini://tlgs.one/search?{}",
|
|
Uri::escape_string(&request_text, None, false)
|
|
);
|
|
|
|
// Update
|
|
self.navigation.set_request_text(&request_text);
|
|
|
|
// Reload page
|
|
self.action_tab_page_navigation_reload.activate(None);
|
|
}
|
|
}
|
|
}; // Uri::parse
|
|
}
|
|
|
|
pub fn update(&self) {
|
|
// Update components
|
|
self.navigation.update(self.progress_fraction());
|
|
// @TODO self.content.update();
|
|
}
|
|
|
|
pub fn clean(
|
|
&self,
|
|
transaction: &Transaction,
|
|
app_browser_window_tab_item_id: &i64,
|
|
) -> Result<(), String> {
|
|
match Database::records(transaction, app_browser_window_tab_item_id) {
|
|
Ok(records) => {
|
|
for record in records {
|
|
match Database::delete(transaction, &record.id) {
|
|
Ok(_) => {
|
|
// Delegate clean action to the item childs
|
|
self.navigation.clean(transaction, &record.id)?;
|
|
}
|
|
Err(e) => return Err(e.to_string()),
|
|
}
|
|
}
|
|
}
|
|
Err(e) => return Err(e.to_string()),
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn restore(
|
|
&self,
|
|
transaction: &Transaction,
|
|
app_browser_window_tab_item_id: &i64,
|
|
) -> Result<(), String> {
|
|
match Database::records(transaction, app_browser_window_tab_item_id) {
|
|
Ok(records) => {
|
|
for record in records {
|
|
// Delegate restore action to the item childs
|
|
self.navigation.restore(transaction, &record.id)?;
|
|
}
|
|
}
|
|
Err(e) => return Err(e.to_string()),
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn save(
|
|
&self,
|
|
transaction: &Transaction,
|
|
app_browser_window_tab_item_id: &i64,
|
|
) -> Result<(), String> {
|
|
match Database::add(transaction, app_browser_window_tab_item_id) {
|
|
Ok(_) => {
|
|
let id = Database::last_insert_id(transaction);
|
|
|
|
// Delegate save action to childs
|
|
self.navigation.save(transaction, &id)?;
|
|
}
|
|
Err(e) => return Err(e.to_string()),
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
// Setters
|
|
pub fn set_navigation_request_text(&self, value: &str) {
|
|
self.navigation.set_request_text(value);
|
|
}
|
|
|
|
// Getters
|
|
pub fn progress_fraction(&self) -> Option<f64> {
|
|
// Interpret status to progress fraction
|
|
match self.meta.borrow().status {
|
|
Some(Status::Reload) => Some(0.0),
|
|
Some(Status::Resolving) => Some(0.1),
|
|
Some(Status::Resolved) => Some(0.2),
|
|
Some(Status::Connecting) => Some(0.3),
|
|
Some(Status::Connected) => Some(0.4),
|
|
Some(Status::ProxyNegotiating) => Some(0.5),
|
|
Some(Status::ProxyNegotiated) => Some(0.6),
|
|
Some(Status::TlsHandshaking) => Some(0.7),
|
|
Some(Status::TlsHandshaked) => Some(0.8),
|
|
Some(Status::Complete) => Some(0.9),
|
|
Some(Status::Failure | Status::Redirect | Status::Success | Status::Input) => Some(1.0),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn is_loading(&self) -> bool {
|
|
match self.progress_fraction() {
|
|
Some(progress_fraction) => progress_fraction < 1.0,
|
|
None => false,
|
|
}
|
|
}
|
|
|
|
pub fn meta_title(&self) -> Option<GString> {
|
|
self.meta.borrow().title.clone()
|
|
}
|
|
|
|
pub fn gobject(&self) -> &Box {
|
|
&self.widget.gobject()
|
|
}
|
|
|
|
// Tools
|
|
pub fn migrate(tx: &Transaction) -> Result<(), String> {
|
|
// Migrate self components
|
|
if let Err(e) = Database::init(tx) {
|
|
return Err(e.to_string());
|
|
}
|
|
|
|
// Delegate migration to childs
|
|
Navigation::migrate(tx)?;
|
|
|
|
// Success
|
|
Ok(())
|
|
}
|
|
|
|
// Private helpers @TODO
|
|
fn load_gemini(&self, uri: Uri) {
|
|
// Use local namespaces
|
|
use gemini::client::{
|
|
buffer::{Buffer, Error as BufferError},
|
|
response::{
|
|
header::{Mime as ClientMime, Status as ClientStatus},
|
|
Header,
|
|
},
|
|
};
|
|
|
|
// Init shared objects (async)
|
|
let action_page_open = self.action_page_open.clone();
|
|
let action_update = self.action_update.clone();
|
|
let content = self.content.clone();
|
|
let id = self.id.to_variant();
|
|
let input = self.input.clone();
|
|
let meta = self.meta.clone();
|
|
let navigation = self.navigation.clone();
|
|
let url = uri.clone().to_str();
|
|
|
|
// Init socket
|
|
let client = SocketClient::new();
|
|
|
|
client.set_protocol(SocketProtocol::Tcp);
|
|
client.set_tls_validation_flags(TlsCertificateFlags::INSECURE);
|
|
client.set_tls(true);
|
|
|
|
// Listen for connection status updates
|
|
client.connect_event({
|
|
let action_update = action_update.clone();
|
|
let id = id.clone();
|
|
let meta = meta.clone();
|
|
move |_, event, _, _| {
|
|
meta.borrow_mut().status = Some(match event {
|
|
SocketClientEvent::Resolving => Status::Resolving,
|
|
SocketClientEvent::Resolved => Status::Resolved,
|
|
SocketClientEvent::Connecting => Status::Connecting,
|
|
SocketClientEvent::Connected => Status::Connected,
|
|
SocketClientEvent::ProxyNegotiating => Status::ProxyNegotiating,
|
|
SocketClientEvent::ProxyNegotiated => Status::ProxyNegotiated,
|
|
SocketClientEvent::TlsHandshaking => Status::TlsHandshaking,
|
|
SocketClientEvent::TlsHandshaked => Status::TlsHandshaked,
|
|
SocketClientEvent::Complete => Status::Complete,
|
|
_ => todo!(), // notice on API change
|
|
});
|
|
action_update.activate(Some(&id));
|
|
}
|
|
});
|
|
|
|
// Create connection
|
|
client.clone().connect_to_uri_async(
|
|
url.clone().as_str(),
|
|
1965,
|
|
None::<&Cancellable>,
|
|
move |connect| match connect {
|
|
Ok(connection) => {
|
|
// Send request
|
|
connection.output_stream().write_bytes_async(
|
|
&Bytes::from(gformat!("{url}\r\n").as_bytes()),
|
|
Priority::DEFAULT,
|
|
None::<&Cancellable>,
|
|
move |request| match request {
|
|
Ok(_) => {
|
|
// Read header from response
|
|
connection.input_stream().read_bytes_async(
|
|
1024,
|
|
Priority::DEFAULT,
|
|
None::<&Cancellable>,
|
|
move |response| match response {
|
|
Ok(bytes) => {
|
|
// Read header from response
|
|
match Header::from_response(
|
|
&bytes
|
|
) {
|
|
Ok(header) => {
|
|
// Route by status
|
|
match header.status() {
|
|
ClientStatus::Input | ClientStatus::SensitiveInput => {
|
|
// Format response
|
|
let status = Status::Input;
|
|
let title = gformat!("Input expected");
|
|
let description = match header.meta() {
|
|
Some(meta) => match meta.to_gstring() {
|
|
Ok(value) => value,
|
|
Err(_) => title.clone(),
|
|
},
|
|
None => title.clone(),
|
|
};
|
|
|
|
// Make input form
|
|
match header.status() {
|
|
ClientStatus::SensitiveInput =>
|
|
input.set_new_sensitive(
|
|
action_page_open,
|
|
uri,
|
|
Some(&description),
|
|
Some(1024),
|
|
),
|
|
_ =>
|
|
input.set_new_response(
|
|
action_page_open,
|
|
uri,
|
|
Some(&description),
|
|
Some(1024),
|
|
),
|
|
}
|
|
|
|
// Update meta
|
|
meta.borrow_mut().status = Some(status);
|
|
meta.borrow_mut().description = Some(description);
|
|
meta.borrow_mut().title = Some(title);
|
|
|
|
// Update page
|
|
action_update.activate(Some(&id));
|
|
},
|
|
ClientStatus::Success => {
|
|
// Route by MIME
|
|
match header.mime() {
|
|
Some(ClientMime::TextGemini) => {
|
|
// Read entire input stream to buffer
|
|
Buffer::from_connection_async(
|
|
connection,
|
|
move |result|{
|
|
match result {
|
|
Ok(buffer) => {
|
|
// Update page meta
|
|
meta.borrow_mut().status = Some(Status::Success);
|
|
meta.borrow_mut().title = content.set_text_gemini(
|
|
&uri,
|
|
&match GString::from_utf8(buffer.to_utf8()) {
|
|
Ok(gemtext) => gemtext,
|
|
Err(_) => todo!()
|
|
}
|
|
);
|
|
|
|
// Add new history record
|
|
let request = uri.to_str();
|
|
|
|
match navigation.history_current() {
|
|
Some(current) => {
|
|
if current != request {
|
|
navigation.history_add(request);
|
|
}
|
|
}
|
|
None => navigation.history_add(request),
|
|
}
|
|
|
|
// Update window components
|
|
action_update.activate(Some(&id));
|
|
}
|
|
Err((reason, message)) => {
|
|
// Define common data
|
|
let status = Status::Failure;
|
|
let title = gformat!("Oops");
|
|
let description = match reason {
|
|
BufferError::InputStream => match message {
|
|
Some(error) => gformat!("{error}"),
|
|
None => gformat!("Undefined connection error")
|
|
} ,
|
|
BufferError::Overflow => gformat!("Buffer overflow"),
|
|
};
|
|
|
|
// Update widget
|
|
content.set_status_failure(
|
|
title.as_str(),
|
|
description.as_str(),
|
|
);
|
|
|
|
// Update meta
|
|
meta.borrow_mut().status = Some(status);
|
|
meta.borrow_mut().title = Some(title);
|
|
meta.borrow_mut().description = Some(description);
|
|
|
|
// Update window
|
|
action_update.activate(Some(&id));
|
|
},
|
|
}
|
|
}
|
|
);
|
|
},
|
|
Some(
|
|
ClientMime::ImagePng | ClientMime::ImageGif |
|
|
ClientMime::ImageJpeg | ClientMime::ImageWebp
|
|
) => {
|
|
match Pixbuf::from_stream(
|
|
&connection.input_stream(),
|
|
None::<&Cancellable>,
|
|
) {
|
|
Ok(buffer) => {
|
|
// Update page meta
|
|
meta.borrow_mut().status = Some(Status::Success);
|
|
meta.borrow_mut().title = Some(gformat!("Image"));
|
|
|
|
// Update page content
|
|
content.set_image(&buffer);
|
|
|
|
// Add history record
|
|
let request = uri.to_str();
|
|
match navigation.history_current() {
|
|
Some(current) => {
|
|
if current != request {
|
|
navigation.history_add(request);
|
|
}
|
|
}
|
|
None => navigation.history_add(request),
|
|
}
|
|
|
|
// Update window components
|
|
action_update.activate(Some(&id));
|
|
}
|
|
Err(reason) => { // Pixbuf::from_stream
|
|
// Define common data
|
|
let status = Status::Failure;
|
|
let title = gformat!("Oops");
|
|
let description = gformat!("{}", reason.message());
|
|
|
|
// Update widget
|
|
content.set_status_failure(title.as_str(), description.as_str());
|
|
|
|
// Update meta
|
|
meta.borrow_mut().status = Some(status);
|
|
meta.borrow_mut().title = Some(title);
|
|
meta.borrow_mut().description = Some(description);
|
|
}
|
|
}
|
|
},
|
|
// @TODO stream extensions
|
|
_ => {
|
|
// Define common data
|
|
let status = Status::Failure;
|
|
let title = gformat!("Oops");
|
|
let description =
|
|
gformat!("Content type not supported");
|
|
|
|
// Update widget
|
|
content.set_status_failure(
|
|
title.as_str(),
|
|
description.as_str(),
|
|
);
|
|
|
|
// Update meta
|
|
meta.borrow_mut().status = Some(status);
|
|
meta.borrow_mut().title = Some(title);
|
|
meta.borrow_mut().description = Some(description);
|
|
|
|
// Update window
|
|
action_update.activate(Some(&id));
|
|
},
|
|
}
|
|
},
|
|
ClientStatus::Redirect => {
|
|
// Update meta
|
|
meta.borrow_mut().status = Some(Status::Redirect);
|
|
meta.borrow_mut().title = Some(gformat!("Redirect"));
|
|
|
|
// Build gemtext message for manual redirection @TODO use template?
|
|
match header.meta() {
|
|
Some(meta) => {
|
|
let _ = content.set_text_gemini(
|
|
&uri,
|
|
&match meta.to_gstring() {
|
|
Ok(url) => gformat!(
|
|
"# Redirect\n\nAuto-follow disabled, click on link below to continue\n\n=> {url}"
|
|
),
|
|
Err(_) => gformat!(
|
|
"# Redirect\n\nProvider request redirect but have not provided any target."
|
|
)
|
|
}
|
|
);
|
|
},
|
|
None => content.set_status_failure(
|
|
&"Oops",
|
|
&"Could not parse redirect meta"
|
|
),
|
|
}
|
|
|
|
action_update.activate(Some(&id));
|
|
},
|
|
}
|
|
},
|
|
Err(_) => todo!() // ResponseHeader::from_response
|
|
}
|
|
}
|
|
Err(_) => todo!(), // InputStream::read_bytes_async
|
|
},
|
|
);
|
|
}
|
|
Err(_) => todo!(), // OutputStream::write_bytes_async
|
|
},
|
|
);
|
|
}
|
|
Err(_) => todo!(), // SocketClient::connect_to_uri_async
|
|
},
|
|
);
|
|
}
|
|
}
|