Minor Bugfixes

frontend/src/lib.rs

@@ -1,20 +1,29 @@
+mod pages;
+use crate::pages::*;
 use yew::prelude::*;
+use yew_router::prelude::*;
 
-#[function_component]
-pub fn App() -> Html {
-    let counter = use_state(|| 0);
-    let onclick = {
-        let counter = counter.clone();
-        move |_| {
-            let value = *counter + 1;
-            counter.set(value);
-        }
-    };
+#[derive(Clone, PartialEq, Routable)]
+enum Route {
+    #[at("/")]
+    Home,
+    #[not_found]
+    #[at("/404")]
+    NotFound,
+}
 
-    html! {
-        <div>
-            <button {onclick}>{ "+1" }</button>
-            <p>{ *counter }</p>
-        </div>
+fn switch(route: Route) -> Html {
+    match route {
+        Route::Home => html! { <basic_pages::Home/>},
+        Route::NotFound => html! { <basic_pages::Home/> },
+    }
+}
+
+#[component(App)]
+pub fn app() -> Html {
+    html! {
+        <BrowserRouter>
+            <Switch<Route> render={switch} />
+        </BrowserRouter>
     }
 }

frontend/src/pages/basic_pages.rs

@@ -0,0 +1,30 @@
+use yew::prelude::*;
+
+#[component(Home)]
+pub fn home_component() -> Html {
+    let counter = use_state(|| 0);
+    let onclick = {
+        let counter = counter.clone();
+        move |_| {
+            let value = *counter + 1;
+            counter.set(value);
+        }
+    };
+
+    html! {
+        <div>
+            <button {onclick}>{ "+1" }</button>
+            <p>{ *counter }</p>
+        </div>
+    }
+}
+
+#[component(NotFound)]
+pub fn not_found_component() -> Html {
+    let message = "404 Not found";
+    html! {
+        <div>
+            <h1>{&message}</h1>
+        </div>
+    }
+}

frontend/src/pages/mod.rs

@@ -0,0 +1 @@
+pub mod basic_pages;