nginx-ui/server/api/cert.go

package api

import (
	"github.com/0xJacky/Nginx-UI/server/model"
	"github.com/0xJacky/Nginx-UI/server/pkg/cert"
	"github.com/0xJacky/Nginx-UI/server/pkg/nginx"
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"log"
	"net/http"
	"os"
)

func IssueCert(c *gin.Context) {
	domain := c.Param("domain")

	var upGrader = websocket.Upgrader{
		CheckOrigin: func(r *http.Request) bool {
			return true
		},
	}

	// upgrade http to websocket
	ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		log.Println(err)
		return
	}

	defer func(ws *websocket.Conn) {
		err := ws.Close()
		if err != nil {
			log.Println("defer websocket close err", err)
		}
	}(ws)

	// read
	mt, message, err := ws.ReadMessage()
	if err != nil {
		log.Println(err)
		return
	}

	if mt == websocket.TextMessage && string(message) == "go" {

		err = cert.IssueCert(domain)

		if err != nil {

			log.Println(err)

			err = ws.WriteJSON(gin.H{
				"status":  "error",
				"message": err.Error(),
			})

			if err != nil {
				log.Println(err)
				return
			}

			return
		}

		sslCertificatePath := nginx.GetNginxConfPath("ssl/" + domain + "/fullchain.cer")
		_, err = os.Stat(sslCertificatePath)

		if err != nil {
			log.Println(err)
			return
		}

		log.Println("[found]", "fullchain.cer")

		err = ws.WriteJSON(gin.H{
			"status":  "success",
			"message": "[found] fullchain.cer",
		})

		if err != nil {
			log.Println(err)
			return
		}

		sslCertificateKeyPath := nginx.GetNginxConfPath("ssl/" + domain + "/" + domain + ".key")
		_, err = os.Stat(sslCertificateKeyPath)

		if err != nil {
			log.Println(err)
			return
		}

		log.Println("[found]", "cert key")
		err = ws.WriteJSON(gin.H{
			"status":  "success",
			"message": "[found] Certificate Key",
		})

		if err != nil {
			log.Println(err)
			return
		}

		certModel, err := model.FirstCert(domain)

		if err != nil {
			log.Println(err)
			return
		}

		err = certModel.Updates(&model.Cert{
			SSLCertificatePath: sslCertificatePath,
		})

		if err != nil {
			log.Println(err)
			return
		}

		err = ws.WriteJSON(gin.H{
			"status":              "success",
			"message":             "Issued certificate successfully",
			"ssl_certificate":     sslCertificatePath,
			"ssl_certificate_key": sslCertificateKeyPath,
		})

		if err != nil {
			log.Println(err)
			return
		}
	}
}