nginx-proxy-manager/backend/internal/entity/certificateauthority/model.go

package certificateauthority

import (
	"fmt"
	"os"
	"path/filepath"
	"time"

	"npm/internal/database"
	"npm/internal/errors"
	"npm/internal/types"

	"github.com/rotisserie/eris"
)

const (
	tableName = "certificate_authority"
)

// Model is the user model
type Model struct {
	ID                  int          `json:"id" db:"id" filter:"id,integer"`
	CreatedOn           types.DBDate `json:"created_on" db:"created_on" filter:"created_on,integer"`
	ModifiedOn          types.DBDate `json:"modified_on" db:"modified_on" filter:"modified_on,integer"`
	Name                string       `json:"name" db:"name" filter:"name,string"`
	AcmeshServer        string       `json:"acmesh_server" db:"acmesh_server" filter:"acmesh_server,string"`
	CABundle            string       `json:"ca_bundle" db:"ca_bundle" filter:"ca_bundle,string"`
	MaxDomains          int          `json:"max_domains" db:"max_domains" filter:"max_domains,integer"`
	IsWildcardSupported bool         `json:"is_wildcard_supported" db:"is_wildcard_supported" filter:"is_wildcard_supported,boolean"`
	IsReadonly          bool         `json:"is_readonly" db:"is_readonly" filter:"is_readonly,boolean"`
	IsDeleted           bool         `json:"is_deleted,omitempty" db:"is_deleted"`
}

func (m *Model) getByQuery(query string, params []interface{}) error {
	return database.GetByQuery(m, query, params)
}

// LoadByID will load from an ID
func (m *Model) LoadByID(id int) error {
	query := fmt.Sprintf("SELECT * FROM `%s` WHERE id = ? AND is_deleted = ? LIMIT 1", tableName)
	params := []interface{}{id, 0}
	return m.getByQuery(query, params)
}

// Touch will update model's timestamp(s)
func (m *Model) Touch(created bool) {
	var d types.DBDate
	d.Time = time.Now()
	if created {
		m.CreatedOn = d
	}
	m.ModifiedOn = d
}

// Save will save this model to the DB
func (m *Model) Save() error {
	var err error

	if m.ID == 0 {
		m.ID, err = Create(m)
	} else {
		err = Update(m)
	}

	return err
}

// Delete will mark a certificate as deleted
func (m *Model) Delete() bool {
	m.Touch(false)
	m.IsDeleted = true
	if err := m.Save(); err != nil {
		return false
	}
	return true
}

// Check will ensure the ca bundle path exists if it's set
func (m *Model) Check() error {
	var err error

	if m.CABundle != "" {
		if _, fileerr := os.Stat(filepath.Clean(m.CABundle)); eris.Is(fileerr, os.ErrNotExist) {
			err = errors.ErrCABundleDoesNotExist
		}
	}

	return err
}