diff options
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/src/main.rs b/src/main.rs index 639ba2d..93ab9d6 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,3 +1,5 @@ +use poem::session::{CookieConfig, CookieSession}; +use poem::web::cookie::CookieKey; use poem::{delete, post, put, EndpointExt}; use poem::{get, listener::TcpListener, Route, Server}; @@ -10,11 +12,16 @@ async fn main() -> Result<(), std::io::Error> { let config = Config::from_file("./critch.toml").unwrap(); let state = Critch::new(config).await; + let cookie_config = CookieConfig::private(CookieKey::generate()); + let cookie_session = CookieSession::new(cookie_config); + let app = Route::new() + .at("/admin", get(routes::admin::get_dashboard)) .at( - "/admin", + "/admin/login", post(routes::admin::login).get(routes::admin::get_login_form), ) + .at("/admin/logout", post(routes::admin::logout)) .at("/", get(routes::artworks::get)) .at( "/artworks", @@ -40,7 +47,9 @@ async fn main() -> Result<(), std::io::Error> { "/artworks/:artwork/comments", post(routes::artworks::comments::post).delete(routes::artworks::comments::delete), ) - .data(state); + .catch_all_error(routes::error::error) + .data(state) + .with(cookie_session); Server::new(TcpListener::bind("0.0.0.0:3000")) .run(app) |