From 0b9c1a74d0491de91a396a52bfc8619cb4f15c4f Mon Sep 17 00:00:00 2001
From: "David J. Allen" <davidallendj@gmail.com>
Date: Mon, 11 Mar 2024 14:32:10 -0600
Subject: [PATCH] Added target query param to login endpoint

---
 internal/server/server.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/internal/server/server.go b/internal/server/server.go
index 0d31fcd..462cb65 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -49,6 +49,10 @@ func (s *Server) Login(buttons string, provider *oidc.IdentityProvider, client *
 		http.Redirect(w, r, "/login", http.StatusSeeOther)
 	})
 	r.HandleFunc("/login", func(w http.ResponseWriter, r *http.Request) {
+		// add target if query exists
+		if r != nil {
+			target = r.URL.Query().Get("target")
+		}
 		// show login page with notice to redirect
 		template, err := gonja.FromFile("pages/index.html")
 		if err != nil {
@@ -129,6 +133,7 @@ func (s *Server) Login(buttons string, provider *oidc.IdentityProvider, client *
 		if err = template.Execute(w, data); err != nil { // Prints: Hello Bob!
 			panic(err)
 		}
+		// try and send access code to target if set
 		if target != "" {
 			httpx.MakeHttpRequest(target, http.MethodPost, []byte(accessToken), httpx.Headers{})
 		}