mirror of
https://github.com/bakito/adguardhome-sync.git
synced 2025-01-10 01:02:11 +08:00
126 lines
2.9 KiB
Go
126 lines
2.9 KiB
Go
package sync
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/bakito/adguardhome-sync/pkg/log"
|
|
"github.com/bakito/adguardhome-sync/version"
|
|
)
|
|
|
|
func (w *worker) handleSync(rw http.ResponseWriter, req *http.Request) {
|
|
switch req.Method {
|
|
case http.MethodPost:
|
|
l.With("remote-addr", req.RemoteAddr).Info("Starting sync from API")
|
|
w.sync()
|
|
default:
|
|
http.Error(rw, "only POST allowed", http.StatusBadRequest)
|
|
}
|
|
}
|
|
|
|
func (w *worker) handleRoot(rw http.ResponseWriter, _ *http.Request) {
|
|
_, _ = rw.Write([]byte("adguardhome-sync"))
|
|
}
|
|
|
|
func (w *worker) handleLogs(rw http.ResponseWriter, _ *http.Request) {
|
|
_, _ = rw.Write([]byte(strings.Join(log.Logs(), "")))
|
|
}
|
|
|
|
func (w *worker) basicAuth(h http.HandlerFunc) http.HandlerFunc {
|
|
return func(rw http.ResponseWriter, r *http.Request) {
|
|
|
|
rw.Header().Set("WWW-Authenticate", `Basic realm="Restricted"`)
|
|
|
|
username, password, authOK := r.BasicAuth()
|
|
if !authOK {
|
|
http.Error(rw, "Not authorized", 401)
|
|
return
|
|
}
|
|
|
|
if username != w.cfg.API.Username || password != w.cfg.API.Password {
|
|
http.Error(rw, "Not authorized", 401)
|
|
return
|
|
}
|
|
|
|
h.ServeHTTP(rw, r)
|
|
}
|
|
}
|
|
|
|
func use(h http.HandlerFunc, middleware ...func(http.HandlerFunc) http.HandlerFunc) http.HandlerFunc {
|
|
for _, m := range middleware {
|
|
h = m(h)
|
|
}
|
|
|
|
return h
|
|
}
|
|
|
|
func (w *worker) listenAndServe() {
|
|
l.With("version", version.Version, "port", w.cfg.API.Port).Info("Starting API server")
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
mux := http.NewServeMux()
|
|
httpServer := &http.Server{
|
|
Addr: fmt.Sprintf(":%d", w.cfg.API.Port),
|
|
Handler: mux,
|
|
BaseContext: func(_ net.Listener) context.Context { return ctx },
|
|
}
|
|
|
|
var mw []func(http.HandlerFunc) http.HandlerFunc
|
|
if w.cfg.API.Username != "" && w.cfg.API.Password != "" {
|
|
mw = append(mw, w.basicAuth)
|
|
}
|
|
|
|
mux.HandleFunc("/api/v1/sync", use(w.handleSync, mw...))
|
|
mux.HandleFunc("/api/v1/logs", use(w.handleLogs, mw...))
|
|
mux.HandleFunc("/", use(w.handleRoot, mw...))
|
|
|
|
go func() {
|
|
if err := httpServer.ListenAndServe(); err != http.ErrServerClosed {
|
|
l.With("error", err).Fatalf("HTTP server ListenAndServe")
|
|
}
|
|
}()
|
|
|
|
signalChan := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(
|
|
signalChan,
|
|
syscall.SIGHUP, // kill -SIGHUP XXXX
|
|
syscall.SIGINT, // kill -SIGINT XXXX or Ctrl+c
|
|
syscall.SIGQUIT, // kill -SIGQUIT XXXX
|
|
)
|
|
|
|
<-signalChan
|
|
l.Info("os.Interrupt - shutting down...")
|
|
|
|
go func() {
|
|
<-signalChan
|
|
l.Fatal("os.Kill - terminating...")
|
|
}()
|
|
|
|
gracefullCtx, cancelShutdown := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancelShutdown()
|
|
|
|
if w.cron != nil {
|
|
l.Info("Stopping cron")
|
|
w.cron.Stop()
|
|
}
|
|
|
|
if err := httpServer.Shutdown(gracefullCtx); err != nil {
|
|
l.With("error", err).Error("Shutdown error")
|
|
defer os.Exit(1)
|
|
} else {
|
|
l.Info("API server stopped")
|
|
}
|
|
|
|
// manually cancel context if not using httpServer.RegisterOnShutdown(cancel)
|
|
cancel()
|
|
|
|
defer os.Exit(0)
|
|
}
|