nginx-ui/server/model/cert.go

package model

import (
	"github.com/0xJacky/Nginx-UI/server/pkg/nginx"
	"os"
	"path/filepath"
)

const (
	AutoCertEnabled  = 1
	AutoCertDisabled = -1
)

type Cert struct {
	Model
	Name                  string `json:"name"`
	Domain                string `json:"domain"`
	SSLCertificatePath    string `json:"ssl_certificate_path"`
	SSLCertificateKeyPath string `json:"ssl_certificate_key_path"`
	AutoCert              int    `json:"auto_cert"`
}

func FirstCert(domain string) (c Cert, err error) {
	err = db.First(&c, &Cert{
		Domain: domain,
	}).Error

	return
}

func FirstOrCreateCert(domain string) (c Cert, err error) {
	err = db.FirstOrCreate(&c, &Cert{Domain: domain}).Error
	return
}

func GetAutoCertList() (c []Cert) {
	var t []Cert
	db.Where("auto_cert", AutoCertEnabled).Find(&t)

	// check if this domain is enabled
	enabledConfig, err := os.ReadDir(filepath.Join(nginx.GetNginxConfPath("sites-enabled")))

	if err != nil {
		return
	}

	enabledConfigMap := make(map[string]bool)
	for i := range enabledConfig {
		enabledConfigMap[enabledConfig[i].Name()] = true
	}

	for _, v := range t {
		if enabledConfigMap[v.Domain] == true {
			c = append(c, v)
		}
	}

	return
}

func GetCertList(name, domain string) (c []Cert) {
	tx := db
	if name != "" {
		tx = tx.Where("name LIKE ? or domain LIKE ?", "%"+name+"%", "%"+name+"%")
	}
	if domain != "" {
		tx = tx.Where("domain LIKE ?", "%"+domain+"%")
	}
	tx.Find(&c)
	return
}

func FirstCertByID(id int) (c Cert, err error) {
	err = db.First(&c, id).Error

	return
}

func (c *Cert) Updates(n *Cert) error {
	return db.Model(c).Updates(n).Error
}

func (c *Cert) Remove() error {
	return db.Where("domain", c.Domain).Delete(c).Error
}