added cookies and login

Author: sachinbhutani

Cargo.lock

@@ -1,11 +1,14 @@
 [package]
 name = "svelte-on-rust"
-version = "0.1.0"
+version = "0.1.2"
 authors = ["Sachin Bhutani <[email protected]>"]
 edition = "2018"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
 rocket = "0.4.2"
+serde = "1.0"
+serde_json = "1.0"
+serde_derive = "1.0"
 rocket_contrib = "0.4.2"

cargo.toml

@@ -7,6 +7,21 @@
 	import About from './pages/About.svelte';
 	import Common from './pages/Common.svelte';
 	import Catcher from './pages/Catcher.svelte';
+	import Login from './pages/Login.svelte';
+	import Logout from './pages/Logout.svelte';
+	import Secure from './pages/Secure.svelte';
+	import {user} from './store.js';
+	import {onMount} from 'svelte';
+	onMount( async() => {
+        const res = await fetch('/api/checkuser',{credentials: 'same-origin'});
+        let stateResponse = await res.json();
+        if (stateResponse.user_id !== '_') {
+            user.set(stateResponse.user_id);
+		}else 
+		{
+			user.set('');
+		}
+    });
 </script>
 
 <main>
@@ -16,6 +31,9 @@
     <Route path="#about"><About></About></Route>
 	<Route path="#app/common" let:router><Common></Common></Route>
 	<Route path="#:name" let:router><Catcher></Catcher></Route> 
+	<Route exact path="#login"><Login></Login></Route>
+	<Route exact path="#logout"><Logout></Logout></Route>
+	<Route exact path="#secure"><Secure></Secure></Route>
   </Router> 
   <Footer></Footer>
 </main>

client/App.svelte

@@ -1,5 +1,7 @@
 <script>
+    import {user} from './store.js';
     import {Link} from 'yrv';
+
     let isActive = false;
     function toggleMenu(){
         isActive = !isActive;
@@ -39,8 +41,32 @@
                     </div>
             </div>
         <Link class="navbar-item" href="#app/common?p1=test&p2=3"> Common Route </Link>
+        {#if $user}<Link class="navbar-item" href="#secure"> Secure Route </Link>{/if}
         <Link class="navbar-item" href="#catch-it"> Catch All </Link>
         </div>
     </div>
 
+    <div class="navbar-end">
+    {#if !$user}
+        <div class="navbar-item">
+        <div class="buttons">
+          <Link class="button is-light" href="#login">
+            Log in
+          </Link>
+        </div>
+        </div>
+    {:else}
+        <div class="navbar-item">
+            Hi {$user}
+        </div>
+        <div class="navbar-item">
+        <div class="buttons">
+          <Link class="button is-danger" href="#logout">
+            Log Out
+          </Link>
+        </div>
+        </div>
+    {/if}
+    </div>
+
 </nav>

client/Nav.svelte

@@ -9,6 +9,7 @@
 	<link rel='icon' type='image/png' href='/favicon.png'>
 	<link rel='stylesheet' href='/bulma.min.css'>
 	<link rel='stylesheet' href='/build/bundle.css'>
+	<script defer src="https://use.fontawesome.com/releases/v5.3.1/js/all.js"></script>
 
 	<script defer src='/build/bundle.js'></script>
 </head>

public/index.html

@@ -3,9 +3,13 @@
 
 #[macro_use] extern crate rocket;
 #[macro_use] extern crate rocket_contrib;
+#[macro_use] extern crate serde_derive;
 
+// use rocket::request::Form;
+use rocket::http::{Cookie, Cookies};
 use rocket_contrib::serve::StaticFiles;
-use rocket_contrib::json::JsonValue;
+use rocket_contrib::json::{JsonValue,Json};
+
 
 // for base route api
 #[get("/")]
@@ -22,6 +26,18 @@ fn json_message() ->  JsonValue{
     })
 }
 
+// check if user is logged in 
+#[get("/checkuser")]
+fn check_user(mut cookies: Cookies) -> JsonValue {
+    let user_id: String;
+    match cookies.get_private("session_id") {
+        Some(cookie) => user_id = cookie.value().to_string(),
+        None => user_id = '_'.to_string(),
+    }
+    json!({"user_id": user_id})
+}
+
+// common client and server route
 #[get("/common", rank = 1)]
 fn common() ->  JsonValue{
     json!({
@@ -31,9 +47,50 @@ fn common() ->  JsonValue{
     })
 }
 
+// User struct for login
+#[derive(Serialize, Deserialize)]
+struct User {
+    username: String,
+    password: String
+}
+
+#[post("/login", format = "json", data = "<user>")]
+fn login_user(mut cookies: Cookies<'_>, user: Json<User>) -> JsonValue{
+    // should be replaced with databased logic and encrypted passwords
+    if user.username == user.password {
+        cookies.add_private(Cookie::new("session_id", user.into_inner().username));
+        json!({ "result" : "success",
+                "messge" : "login successfull"
+        })
+    }else {
+        json!({ "result" : "error", 
+                "message": "Invalid Username/Password"
+        })
+    }
+}
+
+#[get("/logout")]
+fn logout_user(mut cookies: Cookies<'_>) -> JsonValue{
+    cookies.remove_private(Cookie::named("session_id"));
+    json!({ "result" : "success",
+            "messge" : "logout successfull"
+    })
+}
+
+//secured area content 
+#[get("/secure")]
+fn secure_content(mut cookies: Cookies<'_>) -> JsonValue{
+    // fectch private date for the user based on session_id 
+    match cookies.get_private("session_id"){
+        Some(_cookie)=> json!({ "result": "success", "message": "this is your ultra secret private data"}),
+        None => json!( {"result" : "error", "message": "no cookies for you"}),
+    }
+}
+
 fn mount_rocket() -> rocket::Rocket{
     rocket::ignite()
-    .mount("/api",routes![api,json_message])
+    .mount("/api",routes![api,json_message,check_user,secure_content])
+    .mount("/auth",routes![login_user,logout_user])
     .mount("/app",routes![common])
     .mount("/", StaticFiles::from(concat!(env!("CARGO_MANIFEST_DIR"), "/public")))