chore: clean up code

This commit is contained in:
0xJacky 2023-04-13 09:35:41 +08:00
parent 3cdc675e4f
commit 0f259e4331
No known key found for this signature in database
GPG key ID: B6E4A6E4A561BAF0
9 changed files with 92 additions and 121 deletions

View file

@ -193,7 +193,12 @@ function add_server() {
} }
function remove_server(index: number) { function remove_server(index: number) {
props.ngx_config?.servers?.splice(index, 1) Modal.confirm({
title: $gettext('Do you want to remove this server?'),
mask: false,
centered: true,
onOk: () => props.ngx_config?.servers?.splice(index, 1)
})
} }
</script> </script>

View file

@ -1,75 +1,76 @@
package api package api
import ( import (
"github.com/0xJacky/Nginx-UI/server/model" "github.com/0xJacky/Nginx-UI/server/model"
"github.com/0xJacky/Nginx-UI/server/query" "github.com/0xJacky/Nginx-UI/server/query"
"github.com/0xJacky/Nginx-UI/server/settings" "github.com/0xJacky/Nginx-UI/server/settings"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/google/uuid" "github.com/google/uuid"
"golang.org/x/crypto/bcrypt" "golang.org/x/crypto/bcrypt"
"net/http" "net/http"
) )
func installLockStatus() bool { func installLockStatus() bool {
return "" != settings.ServerSettings.JwtSecret return "" != settings.ServerSettings.JwtSecret
} }
func InstallLockCheck(c *gin.Context) { func InstallLockCheck(c *gin.Context) {
c.JSON(http.StatusOK, gin.H{ c.JSON(http.StatusOK, gin.H{
"lock": installLockStatus(), "lock": installLockStatus(),
}) })
} }
type InstallJson struct { type InstallJson struct {
Email string `json:"email" binding:"required,email"` Email string `json:"email" binding:"required,email"`
Username string `json:"username" binding:"required,max=255"` Username string `json:"username" binding:"required,max=255"`
Password string `json:"password" binding:"required,max=255"` Password string `json:"password" binding:"required,max=255"`
Database string `json:"database"` Database string `json:"database"`
} }
func InstallNginxUI(c *gin.Context) { func InstallNginxUI(c *gin.Context) {
// Visit this api after installed is forbidden // Visit this api after installed is forbidden
if installLockStatus() { if installLockStatus() {
c.JSON(http.StatusForbidden, gin.H{ c.JSON(http.StatusForbidden, gin.H{
"error": "installed", "error": "installed",
}) })
return return
} }
var json InstallJson var json InstallJson
ok := BindAndValid(c, &json) ok := BindAndValid(c, &json)
if !ok { if !ok {
return return
} }
settings.ServerSettings.JwtSecret = uuid.New().String() settings.ServerSettings.JwtSecret = uuid.New().String()
settings.ServerSettings.Email = json.Email settings.ServerSettings.Email = json.Email
if "" != json.Database { if "" != json.Database {
settings.ServerSettings.Database = json.Database settings.ServerSettings.Database = json.Database
} }
settings.ReflectFrom() settings.ReflectFrom()
err := settings.Save() err := settings.Save()
if err != nil { if err != nil {
ErrHandler(c, err) ErrHandler(c, err)
return return
} }
// Init model // Init model
db := model.Init() db := model.Init()
query.Init(db) query.Init(db)
curd := model.NewCurd(&model.Auth{}) pwd, _ := bcrypt.GenerateFromPassword([]byte(json.Password), bcrypt.DefaultCost)
pwd, _ := bcrypt.GenerateFromPassword([]byte(json.Password), bcrypt.DefaultCost)
err = curd.Add(&model.Auth{
Name: json.Username,
Password: string(pwd),
})
if err != nil { u := query.Auth
ErrHandler(c, err) err = u.Create(&model.Auth{
return Name: json.Username,
} Password: string(pwd),
c.JSON(http.StatusOK, gin.H{ })
"message": "ok",
}) if err != nil {
ErrHandler(c, err)
return
}
c.JSON(http.StatusOK, gin.H{
"message": "ok",
})
} }

View file

@ -2,6 +2,7 @@ package api
import ( import (
"github.com/0xJacky/Nginx-UI/server/model" "github.com/0xJacky/Nginx-UI/server/model"
"github.com/0xJacky/Nginx-UI/server/query"
"github.com/0xJacky/Nginx-UI/server/settings" "github.com/0xJacky/Nginx-UI/server/settings"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/spf13/cast" "github.com/spf13/cast"
@ -16,16 +17,17 @@ func GetUsers(c *gin.Context) {
} }
func GetUser(c *gin.Context) { func GetUser(c *gin.Context) {
curd := model.NewCurd(&model.Auth{}) id := cast.ToInt(c.Param("id"))
id := c.Param("id")
var user model.Auth u := query.Auth
err := curd.First(&user, id)
user, err := u.FirstByID(id)
if err != nil { if err != nil {
ErrHandler(c, err) ErrHandler(c, err)
return return
} }
c.JSON(http.StatusOK, user) c.JSON(http.StatusOK, user)
} }
@ -40,7 +42,8 @@ func AddUser(c *gin.Context) {
if !ok { if !ok {
return return
} }
curd := model.NewCurd(&model.Auth{})
u := query.Auth
pwd, err := bcrypt.GenerateFromPassword([]byte(json.Password), bcrypt.DefaultCost) pwd, err := bcrypt.GenerateFromPassword([]byte(json.Password), bcrypt.DefaultCost)
if err != nil { if err != nil {
@ -54,7 +57,7 @@ func AddUser(c *gin.Context) {
Password: json.Password, Password: json.Password,
} }
err = curd.Add(&user) err = u.Create(&user)
if err != nil { if err != nil {
ErrHandler(c, err) ErrHandler(c, err)
@ -80,17 +83,17 @@ func EditUser(c *gin.Context) {
if !ok { if !ok {
return return
} }
curd := model.NewCurd(&model.Auth{})
var user, edit model.Auth u := query.Auth
user, err := u.FirstByID(userId)
err := curd.First(&user, userId)
if err != nil { if err != nil {
ErrHandler(c, err) ErrHandler(c, err)
return return
} }
edit.Name = json.Name edit := &model.Auth{
Name: json.Name,
}
// encrypt password // encrypt password
if json.Password != "" { if json.Password != "" {
@ -103,7 +106,7 @@ func EditUser(c *gin.Context) {
edit.Password = string(pwd) edit.Password = string(pwd)
} }
err = curd.Edit(&user, &edit) _, err = u.Where(u.ID.Eq(userId)).Updates(&edit)
if err != nil { if err != nil {
ErrHandler(c, err) ErrHandler(c, err)
@ -114,7 +117,7 @@ func EditUser(c *gin.Context) {
} }
func DeleteUser(c *gin.Context) { func DeleteUser(c *gin.Context) {
id := c.Param("id") id := cast.ToInt(c.Param("id"))
if cast.ToInt(id) == 1 { if cast.ToInt(id) == 1 {
c.JSON(http.StatusNotAcceptable, gin.H{ c.JSON(http.StatusNotAcceptable, gin.H{
@ -123,8 +126,8 @@ func DeleteUser(c *gin.Context) {
return return
} }
curd := model.NewCurd(&model.Auth{}) u := query.Auth
err := curd.Delete(&model.Auth{}, "id", id) err := u.DeleteByID(id)
if err != nil { if err != nil {
ErrHandler(c, err) ErrHandler(c, err)
return return

View file

@ -1,38 +0,0 @@
package model
type Curd struct {
Model interface{}
}
func NewCurd(Model interface{}) *Curd {
return &Curd{Model: Model}
}
func (c *Curd) GetList(dest interface{}) (err error) {
err = db.Model(c.Model).Scan(dest).Error
return
}
func (c *Curd) First(dest interface{}, conds ...interface{}) (err error) {
err = db.Model(c.Model).First(dest, conds).Error
return
}
func (c *Curd) Add(value interface{}) (err error) {
err = db.Model(c.Model).Create(value).Error
if err != nil {
return err
}
err = db.Find(value).Error
return
}
func (c *Curd) Edit(orig interface{}, new interface{}) (err error) {
err = db.Model(orig).Updates(new).Error
return
}
func (c *Curd) Delete(value interface{}, conds ...interface{}) (err error) {
err = db.Model(c.Model).Delete(value, conds).Error
return
}

View file

@ -17,7 +17,7 @@ import (
var db *gorm.DB var db *gorm.DB
type Model struct { type Model struct {
ID uint `gorm:"primary_key" json:"id"` ID int `gorm:"primary_key" json:"id"`
CreatedAt time.Time `json:"created_at"` CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"` UpdatedAt time.Time `json:"updated_at"`
DeletedAt *time.Time `gorm:"index" json:"deleted_at"` DeletedAt *time.Time `gorm:"index" json:"deleted_at"`

View file

@ -28,7 +28,7 @@ func newAuth(db *gorm.DB, opts ...gen.DOOption) auth {
tableName := _auth.authDo.TableName() tableName := _auth.authDo.TableName()
_auth.ALL = field.NewAsterisk(tableName) _auth.ALL = field.NewAsterisk(tableName)
_auth.ID = field.NewUint(tableName, "id") _auth.ID = field.NewInt(tableName, "id")
_auth.CreatedAt = field.NewTime(tableName, "created_at") _auth.CreatedAt = field.NewTime(tableName, "created_at")
_auth.UpdatedAt = field.NewTime(tableName, "updated_at") _auth.UpdatedAt = field.NewTime(tableName, "updated_at")
_auth.DeletedAt = field.NewTime(tableName, "deleted_at") _auth.DeletedAt = field.NewTime(tableName, "deleted_at")
@ -44,7 +44,7 @@ type auth struct {
authDo authDo
ALL field.Asterisk ALL field.Asterisk
ID field.Uint ID field.Int
CreatedAt field.Time CreatedAt field.Time
UpdatedAt field.Time UpdatedAt field.Time
DeletedAt field.Time DeletedAt field.Time
@ -66,7 +66,7 @@ func (a auth) As(alias string) *auth {
func (a *auth) updateTableName(table string) *auth { func (a *auth) updateTableName(table string) *auth {
a.ALL = field.NewAsterisk(table) a.ALL = field.NewAsterisk(table)
a.ID = field.NewUint(table, "id") a.ID = field.NewInt(table, "id")
a.CreatedAt = field.NewTime(table, "created_at") a.CreatedAt = field.NewTime(table, "created_at")
a.UpdatedAt = field.NewTime(table, "updated_at") a.UpdatedAt = field.NewTime(table, "updated_at")
a.DeletedAt = field.NewTime(table, "deleted_at") a.DeletedAt = field.NewTime(table, "deleted_at")

View file

@ -28,7 +28,7 @@ func newCert(db *gorm.DB, opts ...gen.DOOption) cert {
tableName := _cert.certDo.TableName() tableName := _cert.certDo.TableName()
_cert.ALL = field.NewAsterisk(tableName) _cert.ALL = field.NewAsterisk(tableName)
_cert.ID = field.NewUint(tableName, "id") _cert.ID = field.NewInt(tableName, "id")
_cert.CreatedAt = field.NewTime(tableName, "created_at") _cert.CreatedAt = field.NewTime(tableName, "created_at")
_cert.UpdatedAt = field.NewTime(tableName, "updated_at") _cert.UpdatedAt = field.NewTime(tableName, "updated_at")
_cert.DeletedAt = field.NewTime(tableName, "deleted_at") _cert.DeletedAt = field.NewTime(tableName, "deleted_at")
@ -49,7 +49,7 @@ type cert struct {
certDo certDo
ALL field.Asterisk ALL field.Asterisk
ID field.Uint ID field.Int
CreatedAt field.Time CreatedAt field.Time
UpdatedAt field.Time UpdatedAt field.Time
DeletedAt field.Time DeletedAt field.Time
@ -76,7 +76,7 @@ func (c cert) As(alias string) *cert {
func (c *cert) updateTableName(table string) *cert { func (c *cert) updateTableName(table string) *cert {
c.ALL = field.NewAsterisk(table) c.ALL = field.NewAsterisk(table)
c.ID = field.NewUint(table, "id") c.ID = field.NewInt(table, "id")
c.CreatedAt = field.NewTime(table, "created_at") c.CreatedAt = field.NewTime(table, "created_at")
c.UpdatedAt = field.NewTime(table, "updated_at") c.UpdatedAt = field.NewTime(table, "updated_at")
c.DeletedAt = field.NewTime(table, "deleted_at") c.DeletedAt = field.NewTime(table, "deleted_at")

View file

@ -28,7 +28,7 @@ func newConfigBackup(db *gorm.DB, opts ...gen.DOOption) configBackup {
tableName := _configBackup.configBackupDo.TableName() tableName := _configBackup.configBackupDo.TableName()
_configBackup.ALL = field.NewAsterisk(tableName) _configBackup.ALL = field.NewAsterisk(tableName)
_configBackup.ID = field.NewUint(tableName, "id") _configBackup.ID = field.NewInt(tableName, "id")
_configBackup.CreatedAt = field.NewTime(tableName, "created_at") _configBackup.CreatedAt = field.NewTime(tableName, "created_at")
_configBackup.UpdatedAt = field.NewTime(tableName, "updated_at") _configBackup.UpdatedAt = field.NewTime(tableName, "updated_at")
_configBackup.DeletedAt = field.NewTime(tableName, "deleted_at") _configBackup.DeletedAt = field.NewTime(tableName, "deleted_at")
@ -45,7 +45,7 @@ type configBackup struct {
configBackupDo configBackupDo
ALL field.Asterisk ALL field.Asterisk
ID field.Uint ID field.Int
CreatedAt field.Time CreatedAt field.Time
UpdatedAt field.Time UpdatedAt field.Time
DeletedAt field.Time DeletedAt field.Time
@ -68,7 +68,7 @@ func (c configBackup) As(alias string) *configBackup {
func (c *configBackup) updateTableName(table string) *configBackup { func (c *configBackup) updateTableName(table string) *configBackup {
c.ALL = field.NewAsterisk(table) c.ALL = field.NewAsterisk(table)
c.ID = field.NewUint(table, "id") c.ID = field.NewInt(table, "id")
c.CreatedAt = field.NewTime(table, "created_at") c.CreatedAt = field.NewTime(table, "created_at")
c.UpdatedAt = field.NewTime(table, "updated_at") c.UpdatedAt = field.NewTime(table, "updated_at")
c.DeletedAt = field.NewTime(table, "deleted_at") c.DeletedAt = field.NewTime(table, "deleted_at")

View file

@ -28,7 +28,7 @@ func newSite(db *gorm.DB, opts ...gen.DOOption) site {
tableName := _site.siteDo.TableName() tableName := _site.siteDo.TableName()
_site.ALL = field.NewAsterisk(tableName) _site.ALL = field.NewAsterisk(tableName)
_site.ID = field.NewUint(tableName, "id") _site.ID = field.NewInt(tableName, "id")
_site.CreatedAt = field.NewTime(tableName, "created_at") _site.CreatedAt = field.NewTime(tableName, "created_at")
_site.UpdatedAt = field.NewTime(tableName, "updated_at") _site.UpdatedAt = field.NewTime(tableName, "updated_at")
_site.DeletedAt = field.NewTime(tableName, "deleted_at") _site.DeletedAt = field.NewTime(tableName, "deleted_at")
@ -44,7 +44,7 @@ type site struct {
siteDo siteDo
ALL field.Asterisk ALL field.Asterisk
ID field.Uint ID field.Int
CreatedAt field.Time CreatedAt field.Time
UpdatedAt field.Time UpdatedAt field.Time
DeletedAt field.Time DeletedAt field.Time
@ -66,7 +66,7 @@ func (s site) As(alias string) *site {
func (s *site) updateTableName(table string) *site { func (s *site) updateTableName(table string) *site {
s.ALL = field.NewAsterisk(table) s.ALL = field.NewAsterisk(table)
s.ID = field.NewUint(table, "id") s.ID = field.NewInt(table, "id")
s.CreatedAt = field.NewTime(table, "created_at") s.CreatedAt = field.NewTime(table, "created_at")
s.UpdatedAt = field.NewTime(table, "updated_at") s.UpdatedAt = field.NewTime(table, "updated_at")
s.DeletedAt = field.NewTime(table, "deleted_at") s.DeletedAt = field.NewTime(table, "deleted_at")