From 5f89940ab621ea29d44bb568de18b5c1e3cb3ad2 Mon Sep 17 00:00:00 2001
From: LouisLam <louislam@users.noreply.github.com>
Date: Sun, 11 Jul 2021 13:47:57 +0800
Subject: [PATCH] add setup page

---
 db/kuma.db              | Bin 0 -> 57344 bytes
 package.json            |   2 +-
 server/server.js        |  68 +++++++++++++++++++++++++---
 src/main.js             |   9 +++-
 src/mixins/socket.js    |   4 ++
 src/pages/Dashboard.vue |   4 +-
 src/pages/Setup.vue     |  95 ++++++++++++++++++++++++++++++++++++++++
 7 files changed, 173 insertions(+), 9 deletions(-)
 create mode 100644 db/kuma.db
 create mode 100644 src/pages/Setup.vue

diff --git a/db/kuma.db b/db/kuma.db
new file mode 100644
index 0000000000000000000000000000000000000000..9eeb517ea0d2eec3cf8bc4fcc1184ac10549d1cd
GIT binary patch
literal 57344
zcmeI&e{0)D7{GBWPOQYKoxli&We{yFG@{*bGRzodFf~zEGuui1V>ATMb7LhDi|y2w
z(`Af-n~lASzLLF$y@9=f{X6LX-%j>PmXdgDHw>nH4RPeV)1$l3b0=9?rAIm2v9zOh
zr&)Kjk0Q$vMTvZ-X_3ee_ac!<Qv7+Z7h+@9`yl=+f$cGyBF2ZMuM^34B8gZzk~~ko
zpNJ-2F1{E4B>r+?NBvWMp=M)$i7H+QAb<b@Z-795BcY~@jM6{0TMg^mX1irO?M|)L
zcI+eju<qFHR&8uea=U06WmC%*cFf1x!T6o4iTG~e<_6V`L|hwH*^TO5ZBP>KLZhlb
zycJ8Wt}6DRJH4*eso4#A$V9j!va+fL4qDwf@QU@~Vl1_`rt}7p?B-d!<J4QuiB<17
zPprCgD;&*lFpTFn6A?#MdL<r9y}PDdgs|*}-+D0|)Uy)f9rV&~C!z-G^ym2nHI>OI
zKYr@epPX7=RS@T<vba22*|?uGN4mk?B)r>!Y@uv^ZWgt@Vm5CS_q8v~eXX!t)(Vwe
zPG-Y9NU*urL@SyP%%WM?HcS4YTpW=zcWgUm&J>em+bC@tJEmJw+1qiajVu|HZ@*VJ
z9wt1JS3Z{B@64;I^>yX)_P{e3Cb+#2_6t`1i<?m??`8{P_q<sUf?>s0y=hgoFOA~%
zL!-E{mC5LG@Q3Zzk$rq)RsGPh&&6K%cXxB9QSd_w*9S%=SJpOV3~zd0gVXyT$5eOe
zJU0h}?plsxw~pmMH9S;V`Z}YO$N$E<2n-;d)<<2f6ta&hrfhd!KkWq-L`zxYT%3g}
zbo~Q0mNHhA$Y51<pPq{EzqNYLX%9ZvWM6BWa%2ATtGU5Ie|{^tTIAe~g$F82UuU5H
zfMkEe>D<it1NrYc3oa!$q1V?@$s8`su?xv=IaWs?kJhy}3;ijxm6jvubwsnW`M5kT
z1G4;?Wgnk7qln(z+e)XsP$%uK69x&)Z}$}I?=52<cEq{XYSbFyU34OwH%nzBzb9+O
zY<kx2h?7%rwdnn3R86g{D3`<2p)2J6&4yDugd1E%g6FH?>Fpnvsk{I6q4?R?Leu%I
z-tAuRsT{>CPXxzqTHm!>f$8r~C;j$pEOl>X-28X@S=)5q=fqc$dH3tc-b^fY=Z>;J
zT<F6E)3u)Vtk$6=*N^ux8bZ5Lx#5ogu4t-Xs*37fEq|&GV)_><e`6*T=fE%jKPzi=
zVI6<o^CsTpFHK#i?)17<t&}%%IdQS{XM)Ufvnx*B!Br%ZS=HQYoI9(pwmuW*-m7b&
z(d!JpYx(8=PX3xr>j^i(%kCR51Q0*~0R#|0009ILKmY**5O`As-24BzrC%aTe=PmJ
z^xM*}Vv`pF2q1s}0tg_000IagfB*srOpn0wTy$LtK1H^$I2&D6<a0|ii&61GKF9F3
zIul)4a3AVOD$%tW|8a`%`~3btJqe{J2q1s}0tg_000IagfB*srOiCc`2Il$yN%0aF
z0R#|0009ILKmY**5I_KdsSx1z|EWj=oj?Eq1Q0*~0R#|0009ILKwzQ*-2YFMm*5B>
zfB*srAb<b@2q1s}0tifn0MGwVMH=V?0tg_000IagfB*srAb<b@6BXe1|B3Pv903Fn
zKmY**5I_I{1Q0*~fvFJS{(macKqnAD009ILKmY**5I_I{1Q3|00MGwVl$YQLAb<b@
z2q1s}0tg_000Ib1g#h>eQ;`NbfdB#sAb<b@2q1s}0tg_0z(fVO|DPx?!4W_J0R#|0
q009ILKmY**5SR)9?*FGE4Ritl1Q0*~0R#|0009ILKmdV>3j7V9Q4KNx

literal 0
HcmV?d00001

diff --git a/package.json b/package.json
index c45883d92..5a8fdc03c 100644
--- a/package.json
+++ b/package.json
@@ -5,7 +5,7 @@
         "dev": "vite --host",
         "start-server": "node server/server.js",
         "update": "",
-        "build": "npm install && vite build",
+        "build": "vite build",
         "vite-preview-dist": "vite preview --host"
     },
     "dependencies": {
diff --git a/server/server.js b/server/server.js
index a958a247a..7b340372e 100644
--- a/server/server.js
+++ b/server/server.js
@@ -9,6 +9,7 @@ const {R} = require("redbean-node");
 const passwordHash = require('password-hash');
 const jwt = require('jsonwebtoken');
 const Monitor = require("./model/monitor");
+const fs = require("fs");
 const {getSettings} = require("./util-server");
 const {Notification} = require("./notification")
 
@@ -17,14 +18,9 @@ app.use(express.json())
 let totalClient = 0;
 let jwtSecret = null;
 let monitorList = {};
+let needSetup = false;
 
 (async () => {
-    R.setup('sqlite', {
-        filename: './data/kuma.db'
-    });
-    R.freeze(true)
-    await R.autoloadModels("./server/model");
-
     await initDatabase();
 
     app.use('/', express.static("dist"));
@@ -44,6 +40,11 @@ let monitorList = {};
         console.log('a user connected');
         totalClient++;
 
+        if (needSetup) {
+            console.log("Redirect to setup page")
+            socket.emit("setup")
+        }
+
         socket.on('disconnect', () => {
             console.log('user disconnected');
             totalClient--;
@@ -113,6 +114,40 @@ let monitorList = {};
             socket.leave(socket.userID)
             socket.userID = null;
             callback();
+        });
+
+        socket.on("needSetup", async (callback) => {
+            callback(needSetup);
+        });
+
+        socket.on("setup", async (username, password, callback) => {
+            try {
+                if ((await R.count("user")) !== 0) {
+                    throw new Error("Uptime Kuma has been setup. If you want to setup again, please delete the database.")
+                }
+
+                let user = R.dispense("user")
+                user.username = username;
+                user.password = passwordHash.generate(password)
+                await R.store(user)
+
+                needSetup = false;
+
+                callback({
+                    ok: true,
+                    msg: "Added Successfully."
+                });
+
+            } catch (e) {
+                callback({
+                    ok: false,
+                    msg: e.message
+                });
+            }
+
+
+
+
         });
 
         // Auth Only API
@@ -402,6 +437,7 @@ let monitorList = {};
 
     server.listen(3001, () => {
         console.log('Listening on 3001');
+
         startMonitors();
     });
 
@@ -489,6 +525,21 @@ function checkLogin(socket) {
 }
 
 async function initDatabase() {
+    const path = './data/kuma.db';
+
+    if (! fs.existsSync(path)) {
+        console.log("Copy Database")
+        fs.copyFileSync("./db/kuma.db", path);
+    }
+
+    console.log("Connect to Database")
+
+    R.setup('sqlite', {
+        filename: path
+    });
+    R.freeze(true)
+    await R.autoloadModels("./server/model");
+
     let jwtSecretBean = await R.findOne("setting", " `key` = ? ", [
         "jwtSecret"
     ]);
@@ -504,6 +555,11 @@ async function initDatabase() {
         console.log("Load JWT secret from database.")
     }
 
+    if ((await R.count("user")) === 0) {
+        console.log("No user, need setup")
+        needSetup = true;
+    }
+
     jwtSecret = jwtSecretBean.value;
 }
 
diff --git a/src/main.js b/src/main.js
index 112512e5e..a44bef87c 100644
--- a/src/main.js
+++ b/src/main.js
@@ -14,6 +14,7 @@ import EditMonitor from "./pages/EditMonitor.vue";
 import Toast from "vue-toastification";
 import "vue-toastification/dist/index.css";
 import "bootstrap"
+import Setup from "./pages/Setup.vue";
 
 const routes = [
     {
@@ -56,8 +57,14 @@ const routes = [
                     },
                 ],
             },
+
         ],
-    }
+
+    },
+    {
+        path: '/setup',
+        component: Setup,
+    },
 ]
 
 const router = createRouter({
diff --git a/src/mixins/socket.js b/src/mixins/socket.js
index 9c864776c..38d58136c 100644
--- a/src/mixins/socket.js
+++ b/src/mixins/socket.js
@@ -33,6 +33,10 @@ export default {
             transports: ['websocket']
         });
 
+        socket.on('setup', (monitorID, data) => {
+            this.$router.push("/setup")
+        });
+
         socket.on('monitorList', (data) => {
             this.monitorList = data;
         });
diff --git a/src/pages/Dashboard.vue b/src/pages/Dashboard.vue
index 9814e2561..d96e14b5e 100644
--- a/src/pages/Dashboard.vue
+++ b/src/pages/Dashboard.vue
@@ -9,7 +9,9 @@
 
                 <div class="shadow-box list mb-4">
 
-                    <span v-if="$root.monitorList.length === 0">No Monitors, please <router-link to="/add">add one</router-link>.</span>
+                    <div class="text-center mt-3" v-if="Object.keys($root.monitorList).length === 0">
+                        No Monitors, please <router-link to="/add">add one</router-link>.
+                    </div>
 
                     <router-link :to="monitorURL(item.id)" class="item" :class="{ 'disabled': ! item.active }" v-for="item in sortedMonitorList">
 
diff --git a/src/pages/Setup.vue b/src/pages/Setup.vue
new file mode 100644
index 000000000..78ff9e633
--- /dev/null
+++ b/src/pages/Setup.vue
@@ -0,0 +1,95 @@
+<template>
+    <div class="form-container">
+        <div class="form">
+            <form @submit.prevent="submit">
+
+                <div>
+                    <object width="64" data="/icon.svg"></object>
+                    <div style="font-size: 28px; font-weight: bold; margin-top: 5px;">Uptime Kuma</div>
+                </div>
+
+                <p class="mt-3">Create your admin account</p>
+
+                <div class="form-floating">
+                    <input type="text" class="form-control" id="floatingInput" placeholder="Username" v-model="username" required>
+                    <label for="floatingInput">Username</label>
+                </div>
+
+                <div class="form-floating mt-3">
+                    <input type="password" class="form-control" id="floatingPassword" placeholder="Password" v-model="password" required>
+                    <label for="floatingPassword">Password</label>
+                </div>
+
+                <div class="form-floating mt-3">
+                    <input type="password" class="form-control" id="repeat" placeholder="Repeat Password" v-model="repeatPassword" required>
+                    <label for="repeat">Repeat Password</label>
+                </div>
+
+                <button class="w-100 btn btn-primary mt-3" type="submit" :disabled="processing">Create</button>
+
+            </form>
+        </div>
+    </div>
+</template>
+
+<script>
+import { useToast } from 'vue-toastification'
+const toast = useToast()
+
+export default {
+    data() {
+        return {
+            processing: false,
+            username: "",
+            password: "",
+            repeatPassword: "",
+        }
+    },
+    mounted() {
+        this.$root.getSocket().emit("needSetup", (needSetup) => {
+            if (! needSetup) {
+                this.$router.push("/")
+            }
+        });
+    },
+    methods: {
+        submit() {
+            this.processing = true;
+
+            if (this.password !== this.repeatPassword) {
+                toast.error("Repeat password do not match.")
+                this.processing = false;
+                return;
+            }
+
+            this.$root.getSocket().emit("setup", this.username, this.password, (res) => {
+                this.processing = false;
+                this.$root.toastRes(res)
+
+                if (res.ok) {
+                    this.$router.push("/")
+                }
+            })
+        }
+    }
+}
+</script>
+
+<style scoped>
+
+.form-container {
+    display: flex;
+    align-items: center;
+    padding-top: 40px;
+    padding-bottom: 40px;
+}
+
+.form {
+
+    width: 100%;
+    max-width: 330px;
+    padding: 15px;
+    margin: auto;
+    text-align: center;
+}
+</style>