From 468b9b97f91240a5568f62c68688d93352e8abb9 Mon Sep 17 00:00:00 2001
From: Martin McKeaveney <martin@shogunsystems.co.uk>
Date: Mon, 12 Apr 2021 10:47:48 +0100
Subject: [PATCH] builder login

---
 hosting/envoy.dev.yaml.hbs                    |  2 +-
 .../src/components/login/LoginForm.svelte     | 29 ++++++++++++++++---
 .../builder/src/pages/builder/_layout.svelte  |  1 -
 packages/builder/src/stores/backend/auth.js   | 20 +++++++++----
 packages/worker/src/api/controllers/auth.js   |  9 +++++-
 5 files changed, 49 insertions(+), 12 deletions(-)

diff --git a/hosting/envoy.dev.yaml.hbs b/hosting/envoy.dev.yaml.hbs
index 40cfc551ef..a4e2a97118 100644
--- a/hosting/envoy.dev.yaml.hbs
+++ b/hosting/envoy.dev.yaml.hbs
@@ -26,7 +26,7 @@ static_resources:
                   cluster: redis-service
                   prefix_rewrite: "/"
 
-              - match: { prefix: "/api/admin" }
+              - match: { prefix: "/api/admin/" }
                 route:
                   cluster: worker-dev
 
diff --git a/packages/builder/src/components/login/LoginForm.svelte b/packages/builder/src/components/login/LoginForm.svelte
index 216891e35f..5265d2e6d6 100644
--- a/packages/builder/src/components/login/LoginForm.svelte
+++ b/packages/builder/src/components/login/LoginForm.svelte
@@ -1,15 +1,35 @@
 <script>
   import { Button, Label, Input, TextArea, Spacer } from "@budibase/bbui"
+  import { notifier } from "builderStore/store/notifications"
   import { auth } from "stores/backend"
 
   let username = ""
   let password = ""
 
   async function login() {
-    const json = await auth.login({
-      username,
-      password,
-    })
+    try {
+      const json = await auth.login({
+        username,
+        password,
+      })
+      notifier.success("Logged in successfully.")
+    } catch (err) {
+      console.error(err)
+      notifier.danger(`Error logging in: ${err}`)
+    }
+  }
+
+  async function createTestUser() {
+    try {
+      const json = await auth.createUser({
+        email: "test@test.com",
+        password: "test",
+        roles: {},
+      })
+      notifier.success("Test user created")
+    } catch (err) {
+      console.error(err)
+    }
   }
 </script>
 
@@ -22,6 +42,7 @@
   <Input outline type="password" on:change bind:value={password} />
   <Spacer large />
   <Button primary on:click={login}>Login</Button>
+  <Button secondary on:click={createTestUser}>Create Test User</Button>
 </form>
 
 <style>
diff --git a/packages/builder/src/pages/builder/_layout.svelte b/packages/builder/src/pages/builder/_layout.svelte
index 2b98536ab0..c02f9b414a 100644
--- a/packages/builder/src/pages/builder/_layout.svelte
+++ b/packages/builder/src/pages/builder/_layout.svelte
@@ -50,7 +50,6 @@
       <slot />
     </div>
   </div>
-
 {:else}
   <section class="login">
     <LoginForm />
diff --git a/packages/builder/src/stores/backend/auth.js b/packages/builder/src/stores/backend/auth.js
index 2beb878471..6b83b098cb 100644
--- a/packages/builder/src/stores/backend/auth.js
+++ b/packages/builder/src/stores/backend/auth.js
@@ -2,19 +2,29 @@ import { writable, get } from "svelte/store"
 import api from "../../builderStore/api"
 
 export function createAuthStore() {
-  const { subscribe } = writable({})
+  const { subscribe, set } = writable({})
+
+  const user = localStorage.getItem("auth:user")
+  if (user) set({ user: JSON.parse(user) })
 
   return {
     subscribe,
-    login: async () => {
-      const response = await api.post(`/api/admin/auth/authenticate`)
+    login: async creds => {
+      const response = await api.post(`/api/admin/auth`, creds)
       const json = await response.json()
-      set({ user: json })
+      if (json.user) {
+        localStorage.setItem("auth:user", JSON.stringify(json.user))
+        set({ user: json.user })
+      }
     },
     logout: async () => {
       const response = await api.post(`/api/auth/logout`)
       const json = await response.json()
-      set({ user: null })
+      set({ user: false })
+    },
+    createUser: async user => {
+      const response = await api.post(`/api/admin/users`, user)
+      const json = await response.json()
     },
   }
 }
diff --git a/packages/worker/src/api/controllers/auth.js b/packages/worker/src/api/controllers/auth.js
index 4b11634d57..3fa6aef749 100644
--- a/packages/worker/src/api/controllers/auth.js
+++ b/packages/worker/src/api/controllers/auth.js
@@ -9,6 +9,11 @@ exports.authenticate = async (ctx, next) => {
     const expires = new Date()
     expires.setDate(expires.getDate() + 1)
 
+    if (!user) {
+      ctx.body = { success: false, user }
+      return
+    }
+
     ctx.cookies.set(Cookies.Auth, user.token, {
       expires,
       path: "/",
@@ -16,7 +21,9 @@ exports.authenticate = async (ctx, next) => {
       overwrite: true,
     })
 
-    ctx.body = { success: true }
+    delete user.token
+
+    ctx.body = { success: true, user }
   })(ctx, next)
 }