From d2084f869be776f6b6da8bea7e6191c1cca22d68 Mon Sep 17 00:00:00 2001
From: steven <imrealleonardo@gmail.com>
Date: Sun, 12 Dec 2021 12:14:40 +0800
Subject: [PATCH] update initial default database

---
 resources/initial_db.sql |  53 ++++++++++++++++++---------------------
 resources/memos.db       | Bin 45056 -> 45056 bytes
 store/db.go              |  20 +++++++++++++--
 3 files changed, 42 insertions(+), 31 deletions(-)

diff --git a/resources/initial_db.sql b/resources/initial_db.sql
index 3bc24828..ce85a4a5 100644
--- a/resources/initial_db.sql
+++ b/resources/initial_db.sql
@@ -1,6 +1,26 @@
-/*
- * Re-create tables and insert initial data
- */
+DROP TABLE IF EXISTS `queries`;
+CREATE TABLE `queries`  (
+  `id` TEXT NOT NULL PRIMARY KEY,
+  `user_id` TEXT NOT NULL,
+  `title` TEXT NOT NULL,
+  `querystring` TEXT NOT NULL,
+  `created_at` TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `pinned_at` TEXT DEFAULT '',
+  FOREIGN KEY(`user_id`) REFERENCES `users`(`id`)
+);
+
+DROP TABLE IF EXISTS `memos`;
+CREATE TABLE `memos`  (
+  `id` TEXT NOT NULL PRIMARY KEY,
+  `content` TEXT NOT NULL,
+  `user_id` TEXT NOT NULL,
+  `created_at` TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `deleted_at` TEXT DEFAULT '',
+  FOREIGN KEY(`user_id`) REFERENCES `users`(`id`)
+);
+
 DROP TABLE IF EXISTS `users`;
 CREATE TABLE `users` (
 	`id` TEXT NOT NULL PRIMARY KEY,
@@ -15,34 +35,9 @@ CREATE TABLE `users` (
 INSERT INTO `users`
   (`id`, `username`, `password`)
 VALUES
-  ('0', 'admin', '123456'),
   ('1', 'guest', '123456');
 
-DROP TABLE IF EXISTS `memos`;
-CREATE TABLE `memos`  (
-  `id` TEXT NOT NULL PRIMARY KEY,
-  `content` TEXT NOT NULL,
-  `user_id` TEXT NOT NULL,
-  `created_at` TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
-  `updated_at` TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
-  `deleted_at` TEXT DEFAULT '',
-  FOREIGN KEY(`user_id`) REFERENCES `users`(`id`)
-);
-
 INSERT INTO `memos`
-  (`id`, `content`, `user_id`, )
+  (`id`, `content`, `user_id`)
 VALUES
-  ('0', '👋 Welcome to memos', '0'),
   ('1', '👋 Welcome to memos', '1');
-
-DROP TABLE IF EXISTS `queries`;
-CREATE TABLE `queries`  (
-  `id` TEXT NOT NULL PRIMARY KEY,
-  `user_id` TEXT NOT NULL,
-  `title` TEXT NOT NULL,
-  `querystring` TEXT NOT NULL,
-  `created_at` TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
-  `updated_at` TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
-  `pinned_at` TEXT DEFAULT '',
-  FOREIGN KEY(`user_id`) REFERENCES `users`(`id`)
-);
diff --git a/resources/memos.db b/resources/memos.db
index 5f0e267e6150493bf7aa4bac986b790f9ff8e526..5c78111ea8347dfa425033617d352eed1770a28f 100644
GIT binary patch
delta 879
zcmZp8z|`=7X@a!i3I+xSZXjj{Viq9wn5bi{xPn11IDwZ>oq>(Jf`Lz;e;&_z-ap(G
z{8zblaP8nd$$N4$qreJw7Oo9UksBL5n7H`VnXJXd#Ti@NC!gn&<xVIqPAw`<P|)Dw
zoP3c-p`H^gke8U7nxGKk8WE!4=MTi8K0Z2JoCyVq#l_|MMJZUM(lbjkN|WM|G`hID
zIfnX#D5$IJa49Gxlvl*(7o_IJXTo$N%X20q7o{eaq^3;v<2B}LGSOro)J{y%<I-&M
zQinLQxFj(tCp9-UH@_H6us8>~I)=C^gg83+xGE%o#3rBQ(H2Y2&nrpID?tmp$?tg-
zCcE<%NT;OcqyhsVJ`tt@5k8Yo^O`bgYE0h6B_j{EBHj>8vGSi~-~grrhNc`<c5!KG
z##a5w3weYYnKp0b5n|**F~E#fjZNH9dh$LliOF(&0^H0%ZD75cT0nC(*?2+b2msT{
zFaAjm9GjU1e(+B+U}0nA-^IYc3n;vSKfr~No!^|_*qGPw!~BWe3gM|a$@#gd3MKgp
zVAmKL85kMr8X5tSf`N&Zp|O>rDHb6vpgoNIzZv*{15N$N&(Ffl&uIv<oss_t1OE}A
z;4c0MD@G1}aY<fYV`Ibg($wM-LnC7oQ!^~4VG#m(1PDML<OE_+h5^|L@+=$B^d~_7
Vf&2w`0V~)AK$-vYn*{_I1puRf<<|fJ

literal 45056
zcmeI*&2QRf90zb4@=62lwgc*Nc|<h@q_NlrhuG7qk<{s;l7s@;)*c29JV}_q5HM<U
zjaF6FPCM<ixBp;=9d_AmJ5=3nJM2EG*S$zR?J_@4oP?xqlupsokT2xnfuDK&JkQ7W
z^8mp~@m8+xP_o|YG%bgugv){`3hxpk2*Lw5U2@Z?n}*yZd%2tOi@0~^nEZk8*4uxD
z$G;Fp#^S<QJd&CCJR(GX3VqM6;YZ;|$CSk)2tWV=5P$##&RAflGd>!PM#cN{j<wdH
zP1<aA`AulPU}j4uDP`w!CW-M_jKo8P#Ok#eDVZOZ$nuJtR&%*GS+v@+9cnv%i+Ww!
zDL<!FhguG;m93)%$#rufyP7MJ`PD+fTrQPMi%Vv)lwHd6+IsEUe=1j_4ccEHTUoxa
zxRfp2CO6I7B+h!5V9|w@g1LBO*^BOVAwde}g6TGB-Yk+Bi*{WVUWlwLyN%A7F52=6
z$<CJ+-GKYEn!nB<_hX?%=y!4QjJNyCT7BZxk?1>7QK;KBdat|Ha9_lht)A23kIUS<
ztnh3sGA(}hvUUUQ@)y`yzcd=1oD}b0<-^TZk9O*`>t#cShMbBQtq?xyRO+?L;Zdjm
zxMc@y#XCliJdH1M>Q00DEnt;wcb!h%-tb#o-5;KgEIodBsvHbYFIKKyw{1Qe?iKS!
z*S^6hzTeM7V-b&bD{<zO2qlseBhl>S^Nu5rwybz}h9hgj7~5ir&~b+!p9IPZ&)7T2
zs{cklwdl_5D|~YLWQFf7{IKUYBZu~s{b86*L;lkYD`i_v>Nh*Jt#0>jtK&PVZPcC3
z-dee@=-#w6HN|$U*xh^OR-4-Uvl|Z{>@`Q~JASW;9qctPRPNdNE0j33ZGaU)00Izz
z00bZa0SG_<0uX=z1Rwwb2tWV=5P$##AOHafKmY;|fB*y_aApEwR`X>+kn(5N1XK(G
z2tWV=5P$##AOHafKmY;|IFAC~o~K5KSYz4y|KHA2tI<{nKmY;|fB*y_009U<00Izz
z00caN_(>$#@BahRLqYmcdML$CQUcF`00bZa0SG_<0uX=z1Rwwb2n?ve#{bYSJuoIk
zqv3EWC0n&--R3u{tSZy8KCR9YIb+DWk=A@dhUgZ(!ip*zJ=%2?HI>$8`k$y7>9isD
zhZOg5#?Vx+XF)d~cN5D4ZW?wI`<wr`KmY;|fB*y_009U<00Izz00bZ~00Qj${{iW#
zAU&1-k$xV44xtMWfB*y_009U<00Izz00bZafx#Cz+5ht!4oadN7z#?`%HWfrhY)}O
z1Rwwb2tWV=5P$##AOHaf?4JLBO_08pu0?*Gm^!$^1_BU(00bZa0SG_<0uZ>M0^x6j
zzb{(X18-ck9uH1`KYmS0rBYOx(Y0zSldP$-l}yj5G&x(<YDrmFGun(*mDd%;Jy~Ai
zr^+kpv?3ElF;v}9)n`L#pHL+*s+(?2;f=LUKEGXUZ{C^9Se?wRt>T<^vv`$!KpWLo
zlalvnqtPPHChgEae)#gwuRi-!?mItU(F{d3)ZH3(Pxs#oeI$<RSyn^FEZXmHYlZb5
zo!f3N<nwm!&g>1}8VFneZl}DJrTO~x{wmW(S~oOp_xt~k1?h?O`0#abb_VnW0uX=z
z1Rwwb2tWV=5P$##&X<5RB@|A*#lM{T8c%<PiIaSg5qv$Ui~gTO1g-@2asLkx{sn+a
B1U~=(

diff --git a/store/db.go b/store/db.go
index 2d9d596c..4a00ae7b 100644
--- a/store/db.go
+++ b/store/db.go
@@ -2,7 +2,9 @@ package store
 
 import (
 	"database/sql"
+	"io/ioutil"
 	"os"
+	"path/filepath"
 
 	_ "github.com/mattn/go-sqlite3"
 )
@@ -18,7 +20,6 @@ func InitDBConn() {
 
 	if _, err := os.Stat(dbFilePath); err != nil {
 		dbFilePath = "./resources/memos.db"
-		resetDataInDefaultDatabase()
 		println("use the default database")
 	} else {
 		println("use the custom database")
@@ -32,6 +33,10 @@ func InitDBConn() {
 		DB = db
 		println("connect to sqlite succeed")
 	}
+
+	if dbFilePath == "./resources/memos.db" {
+		resetDataInDefaultDatabase()
+	}
 }
 
 func FormatDBError(err error) error {
@@ -46,5 +51,16 @@ func FormatDBError(err error) error {
 }
 
 func resetDataInDefaultDatabase() {
-	// do nth
+	initialSQLFilePath := filepath.Join("resources", "initial_db.sql")
+	c, err := ioutil.ReadFile(initialSQLFilePath)
+
+	if err != nil {
+		// do nth
+		return
+	}
+
+	sql := string(c)
+	DB.Exec(sql)
+
+	println("Initial data succeed")
 }