🦋

pkg/api/base.go

@@ -0,0 +1,62 @@
+package api
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/aykhans/bsky-feedgen/pkg/api/handler"
+	"github.com/aykhans/bsky-feedgen/pkg/config"
+	"github.com/aykhans/bsky-feedgen/pkg/feed"
+	"github.com/aykhans/bsky-feedgen/pkg/logger"
+)
+
+func Run(
+	ctx context.Context,
+	apiConfig *config.APIConfig,
+	feeds []feed.Feed,
+) error {
+	baseHandler, err := handler.NewBaseHandler(apiConfig.FeedgenHostname, apiConfig.ServiceDID)
+	if err != nil {
+		return err
+	}
+	feedHandler := handler.NewFeedHandler(feeds, apiConfig.FeedgenPublisherDID)
+
+	mux := http.NewServeMux()
+
+	mux.HandleFunc("GET /.well-known/did.json", baseHandler.GetWellKnownDIDDoc)
+	mux.HandleFunc("GET /xrpc/app.bsky.feed.describeFeedGenerator", feedHandler.DescribeFeeds)
+	mux.HandleFunc(
+		"GET /xrpc/app.bsky.feed.getFeedSkeleton",
+		feedHandler.GetFeedSkeleton,
+	)
+
+	httpServer := &http.Server{
+		Addr:    fmt.Sprintf(":%d", apiConfig.APIPort),
+		Handler: mux,
+	}
+
+	listenerErrChan := make(chan error)
+
+	logger.Log.Info(fmt.Sprintf("Starting server on port %d", apiConfig.APIPort))
+	go func() {
+		listenerErrChan <- httpServer.ListenAndServe()
+	}()
+
+	select {
+	case err := <-listenerErrChan:
+		if err != nil && err != http.ErrServerClosed {
+			return fmt.Errorf("error while serving http: %v", err)
+		}
+	case <-ctx.Done():
+		shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), time.Second*10)
+		defer shutdownCancel()
+		if err := httpServer.Shutdown(shutdownCtx); err != nil {
+			return fmt.Errorf("error while shutting down http server: %v", err)
+		}
+	}
+	logger.Log.Info(fmt.Sprintf("Server on port %d stopped", apiConfig.APIPort))
+
+	return nil
+}