mirror of
https://github.com/gravitl/netmaker.git
synced 2025-09-11 15:44:52 +08:00
a8a0dd066c
* add pending hosts apis, migration logic for network auto join field * fix pending hosts logic on join * delete pending hosts on host delete * ignore pedning device request if host in the network already * add peer update on host approval
46 lines
1.7 KiB
Go
46 lines
1.7 KiB
Go
package schema
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/gravitl/netmaker/db"
|
|
"gorm.io/datatypes"
|
|
)
|
|
|
|
type PendingHost struct {
|
|
ID string `gorm:"id" json:"id"`
|
|
HostID string `gorm:"host_id" json:"host_id"`
|
|
Hostname string `gorm:"host_name" json:"host_name"`
|
|
Network string `gorm:"network" json:"network"`
|
|
PublicKey string `gorm:"public_key" json:"public_key"`
|
|
EnrollmentKey datatypes.JSON `gorm:"enrollment_key_id" json:"enrollment_key_id"`
|
|
OS string `gorm:"os" json:"os"`
|
|
Version string `gorm:"version" json:"version"`
|
|
Location string `gorm:"location" json:"location"` // Format: "lat,lon"
|
|
RequestedAt time.Time `gorm:"requested_at" json:"requested_at"`
|
|
}
|
|
|
|
func (p *PendingHost) Get(ctx context.Context) error {
|
|
return db.FromContext(ctx).Model(&PendingHost{}).First(&p).Where("id = ?", p.ID).Error
|
|
}
|
|
|
|
func (p *PendingHost) Create(ctx context.Context) error {
|
|
return db.FromContext(ctx).Model(&PendingHost{}).Create(&p).Error
|
|
}
|
|
|
|
func (p *PendingHost) List(ctx context.Context) (pendingHosts []PendingHost, err error) {
|
|
err = db.FromContext(ctx).Model(&PendingHost{}).Find(&pendingHosts).Error
|
|
return
|
|
}
|
|
|
|
func (p *PendingHost) Delete(ctx context.Context) error {
|
|
return db.FromContext(ctx).Model(&PendingHost{}).Where("id = ?", p.ID).Delete(&p).Error
|
|
}
|
|
func (p *PendingHost) CheckIfPendingHostExists(ctx context.Context) error {
|
|
return db.FromContext(ctx).Model(&PendingHost{}).Where("host_id = ? AND network = ?", p.HostID, p.Network).First(&p).Error
|
|
}
|
|
|
|
func (p *PendingHost) DeleteAllPendingHosts(ctx context.Context) error {
|
|
return db.FromContext(ctx).Model(&PendingHost{}).Where("host_id = ?", p.HostID).Delete(&p).Error
|
|
}
|