route.rs

use crate::view::home::Home;
use yew::prelude::*;
use yew_router::prelude::*;

#[derive(Clone, Routable, PartialEq)]
pub enum Route {
    #[at("/:token_address")]
    Collection { token_address: String },
    #[at("/")]
    Home,
    #[not_found]
    #[at("/404")]
    NotFound,
}

pub fn switch(route: Route) -> Html {
    match route {
        Route::Home => html! { <Home />},
        Route::Collection { token_address } => html! { <p class="text-white">{ token_address }</p> },
        Route::NotFound => html! { <p class="text-white">{ "Not found" }</p> },
    }
}