初步实现安装界面

internal/db/models/admin_dao.go

@@ -1,9 +1,12 @@
 package models
 
 import (
+	"github.com/TeaOSLab/EdgeAPI/internal/errors"
 	_ "github.com/go-sql-driver/mysql"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
+	stringutil "github.com/iwind/TeaGo/utils/string"
 )
 
 const (
@@ -24,7 +27,13 @@ func NewAdminDAO() *AdminDAO {
 	}).(*AdminDAO)
 }
 
-var SharedAdminDAO = NewAdminDAO()
+var SharedAdminDAO *AdminDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedAdminDAO = NewAdminDAO()
+	})
+}
 
 // 启用条目
 func (this *AdminDAO) EnableAdmin(id uint32) (rowsAffected int64, err error) {
@@ -71,7 +80,7 @@ func (this *AdminDAO) FindAdminFullname(adminId int) (string, error) {
 }
 
 // 检查用户名、密码
-func (this *AdminDAO) CheckAdminPassword(username string, encryptedPassword string) (int, error) {
+func (this *AdminDAO) CheckAdminPassword(username string, encryptedPassword string) (int64, error) {
 	if len(username) == 0 || len(encryptedPassword) == 0 {
 		return 0, nil
 	}
@@ -80,5 +89,46 @@ func (this *AdminDAO) CheckAdminPassword(username string, encryptedPassword stri
 		Attr("password", encryptedPassword).
 		Attr("state", AdminStateEnabled).
 		ResultPk().
-		FindIntCol(0)
+		FindInt64Col(0)
+}
+
+// 根据用户名查询管理员ID
+func (this *AdminDAO) FindAdminIdWithUsername(username string) (int64, error) {
+	one, err := this.Query().
+		Attr("username", username).
+		State(AdminStateEnabled).
+		ResultPk().
+		Find()
+	if err != nil {
+		return 0, err
+	}
+	if one == nil {
+		return 0, nil
+	}
+	return int64(one.(*Admin).Id), nil
+}
+
+// 更改管理员密码
+func (this *AdminDAO) UpdateAdminPassword(adminId int64, password string) error {
+	if adminId <= 0 {
+		return errors.New("invalid adminId")
+	}
+	op := NewAdminOperator()
+	op.Id = adminId
+	op.Password = stringutil.Md5(password)
+	_, err := this.Save(op)
+	return err
+}
+
+// 创建管理员
+func (this *AdminDAO) CreateAdmin(username string, password string, fullname string) (int64, error) {
+	op := NewAdminOperator()
+	op.Username = username
+	op.Password = stringutil.Md5(password)
+	op.Fullname = fullname
+	_, err := this.Save(op)
+	if err != nil {
+		return 0, err
+	}
+	return types.Int64(op.Id), nil
 }