kochab/examples/user_management.rs

120 lines
3.6 KiB
Rust
Raw Normal View History

2020-11-16 06:13:16 +00:00
use anyhow::*;
use log::LevelFilter;
use northstar::{
GEMINI_PORT,
Document,
2020-11-16 06:13:16 +00:00
Request,
Response,
Server,
user_management::{
User,
UserManagementRoutes,
},
2020-11-16 06:13:16 +00:00
};
#[tokio::main]
/// An ultra-simple demonstration of authentication.
///
/// The user should be able to set a secret string that only they can see. They should be
/// able to change this at any time to any thing. Both the string and the user account
/// will persist across restarts.
///
/// This method sets up and starts the server
2020-11-16 06:13:16 +00:00
async fn main() -> Result<()> {
// Turn on logging
2020-11-16 06:13:16 +00:00
env_logger::builder()
.filter_module("northstar", LevelFilter::Debug)
.init();
Server::bind(("0.0.0.0", GEMINI_PORT))
// Add our main routes
.add_route("/", handle_main)
.add_route("/update", handle_update)
// Add routes for handling user authentication
2020-11-22 04:03:56 +00:00
.add_um_routes::<String>("/")
// Start the server
.serve()
2020-11-16 06:13:16 +00:00
.await
}
/// The landing page
///
/// Displays the user's current secret string, or prompts the user to sign in if they
/// haven't. Includes links to update your string (`/update`) or your account
/// (`/account`). Even though we haven't added an explicit handler for `/account`, this
/// route is managed by northstar.
async fn handle_main(request: Request) -> Result<Response> {
// Check to see if the user is signed in. If they are, we get a copy of their data
// (just a simple [`String`] for this demo, but this can be any serializable struct.
if let User::SignedIn(user) = request.user::<String>()? {
// If the user is signed in, render and return their page
let response = Document::new()
.add_text("Your personal secret string:")
.add_text(user.as_ref())
.add_blank_line()
.add_link("/update", "Change your string")
.add_link("/account", "Update your account")
.into();
Ok(response)
} else {
// If the user is not logged in, prompt them to go to /account
let response = Document::new()
.add_text("Please login to use this app.")
.add_blank_line()
.add_link("/account", "Login or create account")
.into();
Ok(response)
}
}
/// The update endpoint
2020-11-16 06:13:16 +00:00
///
/// Users can update their secret string here. Users who haven't logged in will be
/// promped to do so.
async fn handle_update(request: Request) -> Result<Response> {
// Check if the user is signed in again
if let User::SignedIn(mut user) = request.user::<String>()? {
// If the user is logged in, check to see if they provided any input. If they
// have, we can set that input as their new string, otherwise we ask them for it
if let Some(string) = request.input() {
// Update the users data
*user.as_mut() = string.to_owned();
// Render a response
let response = Document::new()
.add_text("String updated!")
.add_blank_line()
.add_link("/", "Back")
.into();
Ok(response)
} else {
// Ask the user for some input
Ok(Response::input_lossy("Enter your new string"))
}
} else {
// The user isn't logged in, so we should ask them too
let response = Document::new()
.add_text("Please login to use this app.")
.add_blank_line()
.add_link("/account", "Login or create account")
.into();
Ok(response)
}
2020-11-16 06:13:16 +00:00
}