fix: auto-retry at 504

Author: Madoshakalaka

frontend/src/components/output.rs

@@ -8,31 +8,31 @@ use yew::HtmlResult;
 use yew::prelude::*;
 use yew::suspense::use_future_with;
 
-#[derive(Clone, PartialEq)]
-enum CompileResult {
-    Done(String),
-    ColdStartTimeout,
-}
-
 #[derive(Properties, PartialEq, Eq)]
 pub struct OutputContainerProps {
     pub value: Rc<str>,
 }
 
+async fn compile(code: &str) -> String {
+    let query = QueryParams::new();
+    query.append("code", code);
+    let url = format!("{}/run?{}", BACKEND_URL, query);
+
+    loop {
+        match Request::get(&url).send().await {
+            Ok(resp) if resp.status() == 504 => continue,
+            Ok(resp) => return resp.text().await.unwrap_or_default(),
+            Err(_) => continue,
+        }
+    }
+}
+
 #[component]
 pub fn OutputContainer(props: &OutputContainerProps) -> HtmlResult {
     let action_button_state = use_context::<ActionButtonStateContext>().unwrap();
 
     let result = use_future_with(Rc::clone(&props.value), |value| async move {
-        let query = QueryParams::new();
-        query.append("code", &value);
-        let url = format!("{}/run?{}", BACKEND_URL, query);
-
-        match Request::get(&url).send().await {
-            Ok(resp) if resp.status() == 504 => CompileResult::ColdStartTimeout,
-            Ok(resp) => CompileResult::Done(resp.text().await.unwrap_or_default()),
-            Err(_) => CompileResult::ColdStartTimeout,
-        }
+        compile(&value).await
     })?;
 
     {
@@ -42,22 +42,7 @@ pub fn OutputContainer(props: &OutputContainerProps) -> HtmlResult {
         });
     }
 
-    Ok(match &*result {
-        CompileResult::Done(html_content) => {
-            html! { <iframe srcdoc={html_content.clone()} class="w-full h-full" /> }
-        }
-        CompileResult::ColdStartTimeout => {
-            html! {
-                <div class="h-full bg-gray-600 flex items-center justify-center">
-                    <div class="text-gray-200 flex flex-col items-center gap-3 max-w-md text-center">
-                        <span class="text-xl font-semibold">{"Backend service is waking up"}</span>
-                        <span class="text-gray-400">{"The service was asleep and timed out on the first request. It should be ready now."}</span>
-                        <span class="text-gray-300">{"Hit Run again."}</span>
-                    </div>
-                </div>
-            }
-        }
-    })
+    Ok(html! { <iframe srcdoc={(*result).clone()} class="w-full h-full" /> })
 }
 
 #[component]