support multiple users

1.sh

@@ -0,0 +1,3 @@
+sudo docker-compose stop 
+CGO_ENABLED=0 GOOS=linux go build -a -installsuffix cgo -o AssassinGo
+sudo docker-compose up --build -d

config/config.go

@@ -0,0 +1,25 @@
+package config
+
+import (
+	"fmt"
+	"os"
+)
+
+var (
+	// RootDir of your app
+	RootDir = "./web"
+
+	// SecretKey computes sg_token
+	SecretKey string
+
+	// DB addr and passwd.
+	DB string
+)
+
+func init() {
+	DB = fmt.Sprintf("%v:%v@tcp(mariadb:3306)/%v?charset=utf8",
+		os.Getenv("DB_User"),
+		os.Getenv("DB_Passwd"),
+		os.Getenv("DB_Db"))
+	SecretKey = os.Getenv("SecretKey")
+}

docker-compose.yml

@@ -6,23 +6,29 @@ services:
     ports:
       - 8000:8000
     links:
-      - chromedp
       - mariadb
+      - chromedp
     volumes:
       - ./web/:/web/
       - ./dict/:/dict/
-
-  chromedp:
-    image: knqz/chrome-headless
-    ports:
-      - 9222:9222
-  
+    environment:
+      - DB_Db=ag
+      - DB_User=ag
+      - DB_Passwd=password
+      - SecretKey=biubiubiu
+
   mariadb:
     image: mariadb
     expose:
       - "3306"
     environment:
       - MYSQL_ROOT_PASSWORD=assassingo
+      - MYSQL_DATABASE=ag
       - MYSQL_USER=ag
-      - MYSQL_PASSWORD=agpasswd
-      - MYSQL_DATABASE=ag
+      - MYSQL_PASSWORD=password
+
+  chromedp:
+    image: knqz/chrome-headless
+    ports:
+      - 9222:9222
+

dockerfile

@@ -1,5 +1,7 @@
 FROM alpine
 
+RUN apk --no-cache add ca-certificates
+
 ADD AssassinGo /
 
 CMD [ "./AssassinGo" ]

web/controller.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"../assassin"
+	"../config"
 	"../logger"
 	"github.com/AmyangXYZ/sweetygo"
 	jwt "github.com/dgrijalva/jwt-go"
@@ -33,7 +34,7 @@ func signin(ctx *sweetygo.Context) {
 			claims := token.Claims.(jwt.MapClaims)
 			claims["username"] = username
 			claims["exp"] = time.Now().Add(time.Hour * 36).Unix()
-			t, _ := token.SignedString([]byte("secret-key"))
+			t, _ := token.SignedString([]byte(config.SecretKey))
 
 			ctx.JSON(200, 1, "success", map[string]string{"SG_Token": t})
 

web/model.go

@@ -5,20 +5,20 @@ import (
 	"fmt"
 	"time"
 
+	"../config"
 	//
 	_ "github.com/go-sql-driver/mysql"
 )
 
 var db *sql.DB
 
 func init() {
-	db, _ = sql.Open("mysql", "ag:agpasswd@tcp(mariadb:3306)/ag?charset=utf8")
+	db, _ = sql.Open("mysql", config.DB)
 
 	// mysql image starts need time.
 	for {
 		err := db.Ping()
 		if err == nil {
-			fmt.Println("db ok!")
 			break
 		}
 		fmt.Println(err)
@@ -36,6 +36,7 @@ func init() {
 			email VARCHAR(64) NULL DEFAULT NULL,
 			PRIMARY KEY (id)
 		);`)
+	db.Exec(`INSERT INTO users(username, password, email) values("admin","adminn","[email protected]");`)
 }
 
 func getPassword(username string) string {

web/router.go

@@ -1,6 +1,7 @@
 package web
 
 import (
+	"../config"
 	"github.com/AmyangXYZ/sweetygo"
 	"github.com/AmyangXYZ/sweetygo/middlewares"
 )
@@ -14,7 +15,7 @@ var (
 
 // SetMiddlewares sets middlewares.
 func SetMiddlewares(app *sweetygo.SweetyGo) {
-	app.USE(middlewares.JWT("Header", "secret-key", requireJWTMap))
+	app.USE(middlewares.JWT("Header", config.SecretKey, requireJWTMap))
 }
 
 // SetRouter sets router.

web/run.go

@@ -2,6 +2,7 @@ package web
 
 import (
 	"../assassin"
+	"../config"
 	"github.com/AmyangXYZ/sweetygo"
 )
 
@@ -15,7 +16,7 @@ func init() {
 
 // Run Web GUI.
 func Run() {
-	app := sweetygo.New("./web", nil)
+	app := sweetygo.New(config.RootDir, nil)
 	SetMiddlewares(app)
 	SetRouter(app)
 	app.RunServer(":8000")