khairul169/vaulterm
1
0
Fork 0
mirror of https://github.com/khairul169/vaulterm.git synced 2026-09-15 17:03:30 +07:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add hosts management

Browse Source
This commit is contained in:
khairul169
2024-11-09 10:33:07 +00:00
parent 31c43836f4
commit d931235fb3
27 changed files with 742 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/app/hosts/repository.go
+15 -1
View File
@@ -14,7 +14,7 @@ func NewHostsRepository() *Hosts {
func (r *Hosts) GetAll() ([]*models.Host, error) {
var rows []*models.Host
ret := r.db.Order("created_at DESC").Find(&rows)
ret := r.db.Order("id DESC").Find(&rows)
return rows, ret.Error
}
@@ -49,6 +49,20 @@ func (r *Hosts) Get(id string) (*GetHostResult, error) {
return res, ret.Error
}
func (r *Hosts) Exists(id string) (bool, error) {
var count int64
ret := r.db.Model(&models.Host{}).Where("id = ?", id).Count(&count)
return count > 0, ret.Error
}
func (r *Hosts) Delete(id string) error {
return r.db.Delete(&models.Host{Model: models.Model{ID: id}}).Error
}
func (r *Hosts) Create(item *models.Host) error {
return r.db.Create(item).Error
}
func (r *Hosts) Update(item *models.Host) error {
return r.db.Save(item).Error
}
server/app/hosts/router.go
+54 -1
View File
@@ -1,6 +1,7 @@
package hosts
import (
"fmt"
"net/http"
"github.com/gofiber/fiber/v2"
@@ -13,6 +14,8 @@ func Router(app *fiber.App) {
router.Get("/", getAll)
router.Post("/", create)
router.Put("/:id", update)
router.Delete("/:id", delete)
}
func getAll(c *fiber.Ctx) error {
@@ -34,7 +37,6 @@ func create(c *fiber.Ctx) error {
}
repo := NewHostsRepository()
item := &models.Host{
Type: body.Type,
Label: body.Label,
@@ -51,3 +53,54 @@ func create(c *fiber.Ctx) error {
return c.Status(http.StatusCreated).JSON(item)
}
func update(c *fiber.Ctx) error {
var body CreateHostSchema
if err := c.BodyParser(&body); err != nil {
return utils.ResponseError(c, err, 500)
}
repo := NewHostsRepository()
id := c.Params("id")
exist, _ := repo.Exists(id)
if !exist {
return utils.ResponseError(c, fmt.Errorf("host %s not found", id), 404)
}
item := &models.Host{
Model: models.Model{ID: id},
Type: body.Type,
Label: body.Label,
Host: body.Host,
Port: body.Port,
Metadata: body.Metadata,
ParentID: body.ParentID,
KeyID: body.KeyID,
AltKeyID: body.AltKeyID,
}
if err := repo.Update(item); err != nil {
return utils.ResponseError(c, err, 500)
}
return c.JSON(item)
}
func delete(c *fiber.Ctx) error {
repo := NewHostsRepository()
id := c.Params("id")
exist, _ := repo.Exists(id)
if !exist {
return utils.ResponseError(c, fmt.Errorf("host %s not found", id), 404)
}
if err := repo.Delete(id); err != nil {
return utils.ResponseError(c, err, 500)
}
return c.JSON(fiber.Map{
"status": "ok",
"message": "Successfully deleted",
})
}
server/app/ws/term_incus.go
+1
View File
@@ -13,6 +13,7 @@ import (
)
type IncusWebsocketSession struct {
Type string `json:"type"` // "qemu" | "lxc"
Instance string `json:"instance"`
Shell string `json:"shell"`
}
server/models/base_model.go
+3 -3
View File
@@ -8,16 +8,16 @@ import (
"gorm.io/gorm"
)
type BaseModel struct {
type Model struct {
ID string `gorm:"primarykey;type:varchar(26)" json:"id"`
}
func (m *BaseModel) BeforeCreate(tx *gorm.DB) error {
func (m *Model) BeforeCreate(tx *gorm.DB) error {
m.ID = m.GenerateID()
return nil
}
func (m *BaseModel) GenerateID() string {
func (m *Model) GenerateID() string {
return strings.ToLower(ulid.Make().String())
}
server/models/host.go
+3 -3
View File
@@ -12,7 +12,7 @@ const (
)
type Host struct {
BaseModel
Model
Type string `json:"type" gorm:"not null;index:hosts_type_idx;type:varchar(16)"`
Label string `json:"label"`
@@ -23,9 +23,9 @@ type Host struct {
ParentID *string `json:"parentId" gorm:"index:hosts_parent_id_idx;type:varchar(26)"`
Parent *Host `json:"parent" gorm:"foreignKey:ParentID"`
KeyID *string `json:"keyId" gorm:"index:hosts_key_id_idx"`
Key Keychain `gorm:"foreignKey:KeyID"`
Key Keychain `json:"key" gorm:"foreignKey:KeyID"`
AltKeyID *string `json:"altKeyId" gorm:"index:hosts_altkey_id_idx"`
AltKey Keychain `gorm:"foreignKey:AltKeyID"`
AltKey Keychain `json:"altKey" gorm:"foreignKey:AltKeyID"`
Timestamps
SoftDeletes
server/models/keychain.go
+1 -1
View File
@@ -13,7 +13,7 @@ const (
)
type Keychain struct {
BaseModel
Model
Label string `json:"label"`
Type string `json:"type" gorm:"not null;index:keychains_type_idx;type:varchar(12)"`
server/models/user.go
+1 -1
View File
@@ -6,7 +6,7 @@ const (
)
type User struct {
BaseModel
Model
Name string `json:"name"`
Username string `json:"username" gorm:"unique"`
server/tests/hosts_test.go
+30 -1
View File
@@ -21,7 +21,7 @@ func TestHostsCreate(t *testing.T) {
test := NewTestWithAuth(t)
data := map[string]interface{}{
"type": "pve",
"type": "ssh",
"label": "test ssh",
"host": "10.0.0.102",
"port": 22,
@@ -72,3 +72,32 @@ func TestHostsCreate(t *testing.T) {
assert.Equal(t, http.StatusCreated, status)
assert.NotNil(t, res["id"])
}
func TestHostsUpdate(t *testing.T) {
test := NewTestWithAuth(t)
id := "01jc3v9w609f8e2wzw60amv195"
data := map[string]interface{}{
"type": "ssh",
"label": "test ssh update",
"host": "10.0.0.102",
"port": 22,
"keyId": "01jc3wkctzqrcz8qhwynr4p9pe",
}
res, status, err := test.Fetch("PUT", "/hosts/"+id, &FetchOptions{Body: data})
assert.NoError(t, err)
assert.Equal(t, http.StatusOK, status)
assert.NotNil(t, res["id"])
}
func TestHostsDelete(t *testing.T) {
test := NewTestWithAuth(t)
id := "01jc3v9w609f8e2wzw60amv195"
_, status, err := test.Fetch("DELETE", "/hosts/"+id, nil)
assert.NoError(t, err)
assert.Equal(t, http.StatusOK, status)
}