panel/internal/biz/database_item.go

package biz

import (
	"time"

	"github.com/go-rat/utils/crypt"
	"gorm.io/gorm"

	"github.com/TheTNB/panel/internal/app"
)

type DatabaseItemStatus string

const (
	DatabaseItemStatusNormal  DatabaseItemStatus = "normal"
	DatabaseItemStatusInvalid DatabaseItemStatus = "invalid"
)

type DatabaseItem struct {
	ID         uint               `gorm:"primaryKey" json:"id"`
	DatabaseID uint               `gorm:"not null" json:"database_id"`
	Name       string             `gorm:"not null" json:"name"`
	Status     DatabaseItemStatus `gorm:"not null" json:"status"`
	Username   string             `gorm:"not null" json:"username"`
	Password   string             `gorm:"not null" json:"password"`
	Remark     string             `gorm:"not null" json:"remark"`
	CreatedAt  time.Time          `json:"created_at"`
	UpdatedAt  time.Time          `json:"updated_at"`

	Database *Database `json:"database"`
}

func (r *DatabaseItem) BeforeSave(tx *gorm.DB) error {
	crypter, err := crypt.NewXChacha20Poly1305([]byte(app.Key))
	if err != nil {
		return err
	}

	r.Password, err = crypter.Encrypt([]byte(r.Password))
	if err != nil {
		return err
	}

	return nil
}

func (r *DatabaseItem) AfterFind(tx *gorm.DB) error {
	crypter, err := crypt.NewXChacha20Poly1305([]byte(app.Key))
	if err != nil {
		return err
	}

	password, err := crypter.Decrypt(r.Password)
	if err == nil {
		r.Password = string(password)
	}

	return nil
}