Add support for getting auth plugin providers via graphql and made th…

cmd/make-backend/main.go

@@ -10,7 +10,6 @@ import (
 	"make-backend/internal/gql"
 	"make-backend/internal/logging"
 	"make-backend/internal/plugins"
-	auth_plugin "make-backend/internal/plugins/auth"
 	"time"
 
 	acsmqtt "make-backend/internal/api/acs/acs-mqtt"
@@ -51,13 +50,6 @@ const mqttPort = 23002
 
 var glblPlugins = plugins.PluginStore{}
 
-func GetAuthPlugin() auth_plugin.AuthProvider {
-	for _, p := range glblPlugins.Auth {
-		return p
-	}
-	return nil
-}
-
 func main() {
 	sigs := make(chan os.Signal, 1)
 	done := make(chan bool, 1)
@@ -169,7 +161,11 @@ func StartReverseProxy(port string, httpPort, mqttPort int, pluginForwards []plu
 func startHttp(db *sql.DB, store *database.Store, logger *logging.Logger, port int, sessionManager *scs.SessionManager) *http.Server {
 
 	// GraphQL
-	graphqlConfig := gql.Config{Resolvers: &resolvers.Resolver{Store: store}}
+	graphqlConfig := gql.Config{Resolvers: &resolvers.Resolver{
+		Store:   store,
+		Logger:  logger,
+		Plugins: &glblPlugins,
+	}}
 	directives.SetupDirectives(&graphqlConfig, store)
 	srv := handler.New(gql.NewExecutableSchema(graphqlConfig))
 
@@ -191,31 +187,9 @@ func startHttp(db *sql.DB, store *database.Store, logger *logging.Logger, port i
 	mux.Handle("/playground", playground.Handler("GraphQL playground", "/query"))
 	mux.Handle("/query", protectedQueryHandler)
 
-	loginHandler := func(w http.ResponseWriter, r *http.Request) {
-		a := GetAuthPlugin()
-
-		req, err := a.GenerateLoginRequest(&auth_plugin.UserLoginStartRequest{
-			OriginalURL: "http://localhost:8080",
-		})
-		if err != nil {
-			slog.Error("failed to get login url", "err", err)
-			w.WriteHeader(http.StatusInternalServerError)
-			return
-		}
-		h := w.Header()
-		for _, header := range req.SetHeaders {
-			h.Add(header.Key, header.Value)
-		}
-		w.WriteHeader(int(req.Code))
-		_, err = w.Write(req.Body)
-		if err != nil {
-			slog.Error("failed to write login redirect", "err", err)
-		}
-	}
-
 	fileHandler := http.StripPrefix("/app/", http.FileServer(http.Dir("./client")))
 	mux.Handle("/app/", fileHandler)
-	mux.HandleFunc("/login", loginHandler)
+	mux.HandleFunc("/login", auth.LoginHandler(&glblPlugins))
 
 	mux.Handle("/", http.RedirectHandler("/app/", http.StatusFound))
 

internal/auth/sessions.go

@@ -2,6 +2,11 @@ package auth
 
 import (
 	"database/sql"
+	"log/slog"
+	"make-backend/internal/plugins"
+	auth_plugin "make-backend/internal/plugins/auth"
+
+	"net/http"
 
 	"github.com/alexedwards/scs/postgresstore"
 	"github.com/alexedwards/scs/v2"
@@ -13,3 +18,44 @@ func SetupSessions(db *sql.DB) *scs.SessionManager {
 	sessionManager.Store = postgresstore.New(db)
 	return sessionManager
 }
+
+func LoginHandler(plugins *plugins.PluginStore) func(w http.ResponseWriter, r *http.Request) {
+	return func(w http.ResponseWriter, r *http.Request) {
+		vs := r.URL.Query()
+		pluginId := vs.Get("plugin")
+		if pluginId == "" {
+			// no plugin specified. should redirect to login chooser page
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+		fromUrl := vs.Get("from")
+		if fromUrl == "" {
+			fromUrl = "http://localhost:8080"
+		}
+
+		a, ok := plugins.Auth[pluginId]
+		if !ok {
+			// no plugin found. should redirect to login chooser page
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+
+		req, err := a.Provider.GenerateLoginRequest(&auth_plugin.UserLoginStartRequest{
+			OriginalURL: fromUrl,
+		})
+		if err != nil {
+			slog.Error("failed to get login url", "err", err)
+			w.WriteHeader(http.StatusInternalServerError)
+			return
+		}
+		h := w.Header()
+		for _, header := range req.SetHeaders {
+			h.Add(header.Key, header.Value)
+		}
+		w.WriteHeader(int(req.Code))
+		_, err = w.Write(req.Body)
+		if err != nil {
+			slog.Error("failed to write login redirect", "err", err)
+		}
+	}
+}