diff --git a/certificates.go b/certificates.go index bfa4d11ac..abbad7a4c 100644 --- a/certificates.go +++ b/certificates.go @@ -229,6 +229,12 @@ func certHandler(c *gin.Context) { }) } +func deleteCertHandler(c *gin.Context) { + os.Remove("ca.cert.pem") + os.Remove("ca.cert.cer") + os.Remove("ca.key.pem") +} + const noFirefoxTemplateHTML = ` diff --git a/conn.go b/conn.go index f8285633a..4989ebd6a 100644 --- a/conn.go +++ b/conn.go @@ -3,11 +3,12 @@ package main import ( + "net/http" + "strconv" + log "github.com/Sirupsen/logrus" "github.com/gin-gonic/gin" "github.com/googollee/go-socket.io" - "net/http" - "strconv" ) type connection struct { @@ -101,6 +102,7 @@ func wsHandler() *WsServer { so.On("command", func(message string) { h.broadcast <- []byte(message) }) + so.On("disconnection", func() { h.unregister <- c }) diff --git a/main.go b/main.go index e04ba5477..0d951af56 100755 --- a/main.go +++ b/main.go @@ -235,6 +235,7 @@ func main() { r.GET("/", homeHandler) r.GET("/certificate.crt", certHandler) + r.DELETE("/certificate.crt", deleteCertHandler) r.POST("/upload", uploadHandler) r.GET("/socket.io/", socketHandler) r.POST("/socket.io/", socketHandler)