2021-07-05 21:35:27 +07:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2024-09-07 23:55:39 +03:00
|
|
|
"errors"
|
|
|
|
"log/slog"
|
2021-07-05 21:35:27 +07:00
|
|
|
"net"
|
|
|
|
"net/http"
|
|
|
|
"os"
|
|
|
|
"strings"
|
|
|
|
"time"
|
2024-09-07 23:55:39 +03:00
|
|
|
|
|
|
|
"github.com/bouncepaw/mycorrhiza/internal/cfg"
|
2021-07-05 21:35:27 +07:00
|
|
|
)
|
|
|
|
|
2024-09-07 23:55:39 +03:00
|
|
|
func serveHTTP(handler http.Handler) (err error) {
|
2021-07-05 21:35:27 +07:00
|
|
|
server := &http.Server{
|
|
|
|
ReadTimeout: 300 * time.Second,
|
|
|
|
WriteTimeout: 300 * time.Second,
|
|
|
|
IdleTimeout: 300 * time.Second,
|
2021-07-16 00:46:35 +07:00
|
|
|
Handler: handler,
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|
|
|
|
|
|
|
|
if strings.HasPrefix(cfg.ListenAddr, "/") {
|
2024-09-07 23:55:39 +03:00
|
|
|
err = startUnixSocketServer(server, cfg.ListenAddr)
|
2021-07-05 21:35:27 +07:00
|
|
|
} else {
|
|
|
|
server.Addr = cfg.ListenAddr
|
2024-09-07 23:55:39 +03:00
|
|
|
err = startHTTPServer(server)
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|
2024-09-07 23:55:39 +03:00
|
|
|
return err
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|
|
|
|
|
2024-09-07 23:55:39 +03:00
|
|
|
func startUnixSocketServer(server *http.Server, socketPath string) error {
|
|
|
|
err := os.Remove(socketPath)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2021-07-05 21:35:27 +07:00
|
|
|
|
2024-09-07 23:55:39 +03:00
|
|
|
listener, err := net.Listen("unix", socketPath)
|
2021-07-05 21:35:27 +07:00
|
|
|
if err != nil {
|
2024-09-07 23:55:39 +03:00
|
|
|
slog.Error("Failed to start the server", "err", err)
|
|
|
|
return err
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|
2024-09-07 23:55:39 +03:00
|
|
|
defer func(listener net.Listener) {
|
|
|
|
_ = listener.Close()
|
|
|
|
}(listener)
|
2021-07-05 21:35:27 +07:00
|
|
|
|
2024-09-07 23:55:39 +03:00
|
|
|
if err := os.Chmod(socketPath, 0666); err != nil {
|
|
|
|
slog.Error("Failed to set socket permissions", "err", err)
|
|
|
|
return err
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|
|
|
|
|
2024-09-07 23:55:39 +03:00
|
|
|
slog.Info("Listening Unix socket", "addr", socketPath)
|
|
|
|
|
|
|
|
if err := server.Serve(listener); !errors.Is(err, http.ErrServerClosed) {
|
|
|
|
slog.Error("Failed to start the server", "err", err)
|
|
|
|
return err
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|
2024-09-07 23:55:39 +03:00
|
|
|
|
|
|
|
return nil
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|
|
|
|
|
2024-09-07 23:55:39 +03:00
|
|
|
func startHTTPServer(server *http.Server) error {
|
|
|
|
slog.Info("Listening over HTTP", "addr", server.Addr)
|
|
|
|
|
|
|
|
if err := server.ListenAndServe(); !errors.Is(err, http.ErrServerClosed) {
|
|
|
|
slog.Error("Failed to start the server", "err", err)
|
|
|
|
return err
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|
2024-09-07 23:55:39 +03:00
|
|
|
|
|
|
|
return nil
|
2021-07-05 21:35:27 +07:00
|
|
|
}
|