Merge pull request #7 from mitodl/cg/add-redirect-url

add the redirectUrl argument and use it in loginUrl

index.js

@@ -4,21 +4,25 @@
  * @function initLoginButton
  * @param {String} containerId The id property of the container element to place the login button / status in
  * @param {String} baseUrl The base URL of the MIT Open instance
+ * @param {String} redirectUrl The URL to redirect to after login is complete
  * @param {String} buttonText The text to show on the Login button
  * @param {String} buttonClass The CSS class(es) to assign to the button
  * @param {String} loggedInTextClass The CSS class(es) to assign to the logged-in status text
  */
 export function initLoginButton(
   containerId,
   baseUrl,
+  redirectUrl = "",
   buttonText = "Login",
   buttonClass = "",
   loggedInTextClass = "",
 ) {
   const container = document.getElementById(containerId)
   const parsedBaseUrl = new URL(baseUrl)
   const currentUserUrl = `${parsedBaseUrl.origin}/api/v0/users/me/?format=json`
-  const loginUrl = `${parsedBaseUrl.origin}/login/ol-oidc/`
+  const redirectUrlParam =
+    redirectUrl !== "" ? `?next=${encodeURIComponent(redirectUrl)}` : ""
+  const loginUrl = `${parsedBaseUrl.origin}/login/ol-oidc/${redirectUrlParam}`
   fetch(currentUserUrl, {
     method: "GET",
     credentials: "include",