Merge pull request #28 from vivshankar/v0.44.x

Sync commits to v0.46.x
vivshankar · Jul 9, 2024 · 55da756 · 55da756
2 parents c2138c6 + 74cec9f
commit 55da756
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/handler/rfc8628/device_authorize_handler.go b/handler/rfc8628/device_authorize_handler.go
@@ -5,6 +5,7 @@ package rfc8628
 
 import (
 	"context"
+	"net/url"
 	"time"
 
 	"github.com/ory/fosite"
@@ -57,8 +58,16 @@ func (d *DeviceAuthorizeHandler) HandleDeviceAuthorizeEndpointRequest(ctx contex
 	resp.SetDeviceCode(deviceCode)
 	resp.SetUserCode(userCode)
 	resp.SetVerificationURI(d.Config.GetRFC8628UserVerificationURL(ctx))
-	resp.SetVerificationURIComplete(d.Config.GetRFC8628UserVerificationURL(ctx) + "?user_code=" + userCode)
+	resp.SetVerificationURIComplete(d.formCompleteURI(d.Config.GetRFC8628UserVerificationURL(ctx), userCode))
 	resp.SetExpiresIn(int64(time.Until(expireAt).Seconds()))
 	resp.SetInterval(int(d.Config.GetDeviceAuthTokenPollingInterval(ctx).Seconds()))
 	return nil
 }
+
+func (d *DeviceAuthorizeHandler) formCompleteURI(verificationURI, userCode string) string {
+	u, _ := url.Parse(verificationURI)
+	qp := u.Query()
+	qp["user_code"] = []string{userCode}
+	u.RawQuery = qp.Encode()
+	return u.String()
+}