2020-11-16 06:13:16 +00:00
|
|
|
use anyhow::*;
|
|
|
|
use log::LevelFilter;
|
|
|
|
use northstar::{
|
|
|
|
GEMINI_PORT,
|
2020-11-22 06:29:24 +00:00
|
|
|
Document,
|
2020-11-16 06:13:16 +00:00
|
|
|
Request,
|
|
|
|
Response,
|
|
|
|
Server,
|
2020-11-22 06:29:24 +00:00
|
|
|
user_management::{
|
|
|
|
User,
|
|
|
|
UserManagementRoutes,
|
|
|
|
},
|
2020-11-16 06:13:16 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
#[tokio::main]
|
2020-11-22 06:29:24 +00:00
|
|
|
/// 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<()> {
|
2020-11-22 06:29:24 +00:00
|
|
|
// 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))
|
2020-11-22 06:29:24 +00:00
|
|
|
|
|
|
|
// 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>("/")
|
2020-11-22 06:29:24 +00:00
|
|
|
|
|
|
|
// Start the server
|
2020-11-21 02:53:00 +00:00
|
|
|
.serve()
|
2020-11-16 06:13:16 +00:00
|
|
|
.await
|
|
|
|
}
|
|
|
|
|
2020-11-22 06:29:24 +00:00
|
|
|
/// 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
|
|
|
///
|
2020-11-22 06:29:24 +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
|
|
|
}
|