starred/yaak-mountain-loop
1
0
Fork 0
mirror of https://github.com/mountain-loop/yaak.git synced 2026-03-19 07:53:54 +01:00
Code Issues 8 Packages Projects Releases 10 Wiki Activity

Fix migrations for build and iframe rendering

Browse Source
This commit is contained in:
Gregory Schier
2023-03-01 10:31:50 -08:00
parent aace2580da
commit 915e0e8613
3 changed files with 21 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src-tauri/src/main.rs
View File

@@ -23,7 +23,6 @@ use tauri::{AppHandle, State, Wry};
use tauri::{CustomMenuItem, Manager, SystemTray, SystemTrayEvent, SystemTrayMenu, WindowEvent};
use tokio::sync::Mutex;
use crate::models::{update_response, HttpResponse};
use window_ext::WindowExt;
mod models;
@@ -42,11 +41,16 @@ pub struct CustomResponse {
pub status_reason: Option<&'static str>,
}
async fn migrate_db(db_instance: &Mutex<Pool<Sqlite>>) -> Result<(), String> {
async fn migrate_db(
app_handle: AppHandle<Wry>,
db_instance: &Mutex<Pool<Sqlite>>,
) -> Result<(), String> {
let pool = &*db_instance.lock().await;
let m = Migrator::new(Path::new("./migrations"))
.await
.expect("Failed to load migrations");
let p = app_handle
.path_resolver()
.resolve_resource("migrations")
.expect("failed to resolve resource");
let m = Migrator::new(p).await.expect("Failed to load migrations");
m.run(pool).await.expect("Failed to run migrations");
println!("Migrations ran");
Ok(())
@@ -148,7 +152,7 @@ async fn send_request(
response.url = v.url().to_string();
response.body = v.text().await.expect("Failed to get body");
response.elapsed = start.elapsed().as_millis() as i64;
response = update_response(response, pool)
response = models::update_response(response, pool)
.await
.expect("Failed to update response");
app_handle.emit_all("updated_response", &response).unwrap();
@@ -159,13 +163,13 @@ async fn send_request(
}
async fn response_err(
mut response: HttpResponse,
mut response: models::HttpResponse,
error: String,
app_handle: AppHandle<Wry>,
pool: &Pool<Sqlite>,
) -> Result<String, String> {
response.error = Some(error.clone());
response = update_response(response, pool)
response = models::update_response(response, pool)
.await
.expect("Failed to update response");
app_handle.emit_all("updated_response", &response).unwrap();
@@ -339,7 +343,9 @@ fn main() {
.await
.expect("Failed to connect to database");
let m = Mutex::new(pool);
migrate_db(&m).await.expect("Failed to migrate database");
migrate_db(app.handle(), &m)
.await
.expect("Failed to migrate database");
app.manage(m);
Ok(())
})

2
src-tauri/tauri.toml
View File

@@ -36,7 +36,7 @@ icon = [
]
identifier = "co.schier.twosomnia"
longDescription = ""
resources = [ "plugins/*" ]
resources = [ "plugins/*", "migrations/*" ]
shortDescription = ""
targets = "all"

9
src-web/components/ResponsePane.tsx
View File

@@ -34,13 +34,14 @@ export function ResponsePane({ requestId, error }: Props) {
[response],
);
const contentForIframe: string = useMemo(() => {
if (response == null) return '';
const contentForIframe: string | null = useMemo(() => {
if (!contentType.includes('html')) return null;
if (response == null) return null;
if (response.body.includes('<head>')) {
return response.body.replace(/<head>/gi, `<head><base href="${response.url}"/>`);
}
return response.body;
}, [response?.id]);
}, [response?.body, contentType]);
return (
<VStack className="w-full">
@@ -97,7 +98,7 @@ export function ResponsePane({ requestId, error }: Props) {
)}
</HStack>
</HStack>
{viewMode === 'pretty' && contentType.includes('html') ? (
{viewMode === 'pretty' && contentForIframe !== null ? (
<iframe
title="Response preview"
srcDoc={contentForIframe}