From 1315282c7ff46f1ac048022f7eeb0b4d09eedb57 Mon Sep 17 00:00:00 2001
From: Frédéric Guillot <fred@miniflux.net>
Date: Sun, 11 Nov 2018 16:21:57 -0800
Subject: Add the possiblity to listen on Unix socket

---
 service/httpd/httpd.go | 33 +++++++++++++++++++++++++++++----
 1 file changed, 29 insertions(+), 4 deletions(-)

(limited to 'service')

diff --git a/service/httpd/httpd.go b/service/httpd/httpd.go
index 4ab7cfd..a365353 100644
--- a/service/httpd/httpd.go
+++ b/service/httpd/httpd.go
@@ -6,7 +6,10 @@ package httpd // import "miniflux.app/service/httpd"
 
 import (
 	"crypto/tls"
+	"net"
 	"net/http"
+	"os"
+	"strings"
 	"time"
 
 	"miniflux.app/api"
@@ -29,27 +32,49 @@ func Serve(cfg *config.Config, store *storage.Storage, pool *worker.Pool, feedHa
 	keyFile := cfg.KeyFile()
 	certDomain := cfg.CertDomain()
 	certCache := cfg.CertCache()
+	listenAddr := cfg.ListenAddr()
 	server := &http.Server{
 		ReadTimeout:  30 * time.Second,
 		WriteTimeout: 30 * time.Second,
 		IdleTimeout:  60 * time.Second,
-		Addr:         cfg.ListenAddr(),
 		Handler:      setupHandler(cfg, store, feedHandler, pool),
 	}
 
-	if certDomain != "" && certCache != "" {
+	switch {
+	case strings.HasPrefix(listenAddr, "/"):
+		startUnixSocketServer(server, listenAddr)
+	case certDomain != "" && certCache != "":
 		cfg.IsHTTPS = true
 		startAutoCertTLSServer(server, certDomain, certCache)
-	} else if certFile != "" && keyFile != "" {
+	case certFile != "" && keyFile != "":
 		cfg.IsHTTPS = true
+		server.Addr = listenAddr
 		startTLSServer(server, certFile, keyFile)
-	} else {
+	default:
+		server.Addr = listenAddr
 		startHTTPServer(server)
 	}
 
 	return server
 }
 
+func startUnixSocketServer(server *http.Server, socketFile string) {
+	os.Remove(socketFile)
+
+	go func(sock string) {
+		listener, err := net.Listen("unix", sock)
+		if err != nil {
+			logger.Fatal(`Server failed to start: %v`, err)
+		}
+		defer listener.Close()
+
+		logger.Info(`Listening on Unix socket %q`, sock)
+		if err := server.Serve(listener); err != http.ErrServerClosed {
+			logger.Fatal(`Server failed to start: %v`, err)
+		}
+	}(socketFile)
+}
+
 func startAutoCertTLSServer(server *http.Server, certDomain, certCache string) {
 	server.Addr = ":https"
 	certManager := autocert.Manager{
-- 
cgit v1.2.3