Merge pull request #8 from superfly/hd

Allow hd hint in oauth flow
superfly · Oct 27, 2023 · b531999 · b531999
2 parents 5436dd8 + 684427d
commit b531999
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 2 deletions.
diff --git a/etc/ssokenizer.yml b/etc/ssokenizer.yml
@@ -126,4 +126,24 @@ identity_providers:
     client_secret: "$HEROKU_CLIENT_SECRET"
     return_url: "$HEROKU_STAGING_2_RETURN_URL"
     scopes:
-      - "$HEROKU_SCOPES"
+      - "$HEROKU_SCOPES"
+
+  google_auth:
+    secret_auth:
+      bearer_digest: "$AUTH_DIGEST"
+    profile: google
+    client_id: "$GOOGLE_CLIENT_ID"
+    client_secret: "$GOOGLE_CLIENT_SECRET"
+    return_url: "$GOOGLE_AUTH_RETURN_URL"
+    scopes:
+      - "$GOOGLE_SCOPES"
+
+  github_auth:
+    secret_auth:
+      bearer_digest: "$AUTH_DIGEST"
+    profile: github
+    client_id: "$GITHUB_CLIENT_ID"
+    client_secret: "$GITHUB_CLIENT_SECRET"
+    return_url: "$GITHUB_AUTH_RETURN_URL"
+    scopes:
+      - "$GITHUB_SCOPES"
diff --git a/oauth2/oauth2.go b/oauth2/oauth2.go
@@ -78,7 +78,14 @@ func (p *provider) handleStart(w http.ResponseWriter, r *http.Request) {
 	defer getLog(r).WithField("status", http.StatusFound).Info()
 
 	tr := ssokenizer.GetTransaction(r)
-	http.Redirect(w, r, p.config(r).AuthCodeURL(tr.Nonce, oauth2.AccessTypeOffline), http.StatusFound)
+
+	opts := []oauth2.AuthCodeOption{oauth2.AccessTypeOffline}
+
+	if hd := r.URL.Query().Get("hd"); hd != "" {
+		opts = append(opts, oauth2.SetAuthURLParam("hd", hd))
+	}
+
+	http.Redirect(w, r, p.config(r).AuthCodeURL(tr.Nonce, opts...), http.StatusFound)
 }
 
 func (p *provider) handleCallback(w http.ResponseWriter, r *http.Request) {