如何获取在使用Selenium路由到另一个URL之前临时显示的重定向URL?

我有一个用来获取代码的URL。然后,我需要使用此代码并获取令牌。当我访问此URL时 mysite.com/oauth/authorize?response_type=code&client_id=ewrtergdgdfg324324242&redirect_uri=myNewUrl/oauth-check ,我被引导到一个登录页面。

在那里,当我输入凭证并按登录时,一个临时的URL myNewUrl/oauthcheck?code=dsfdsfs3242343242423 显示一个代码(我需要提取),然后重定向到我的应用程序的主页(mynewurl)。我需要从这个临时URL中提取“代码”。但这个网址只是出现和消失。我尝试了下面的代码来获取这个代码。当我尝试getcurrentURL时,我只得到新的主页URL。不是这个临时URL。

System.setProperty("webdriver.chrome.driver",
                "/Users/me/Documents/chromedriver");
        ChromeOptions options = new ChromeOptions();
        options.addArguments("headless");
        options.addArguments("window-size=1200x600");
        WebDriver driver = new ChromeDriver(options);
        driver.get("mysite.com/oauth/authorize?response_type=code&client_id=ewrtergdgdfg324324242&redirect_uri=https://myNewUrl/oauth-check");

        ((ChromeDriver) driver).findElementById("login_email").sendKeys("myuser@mysite.com");
        ((ChromeDriver) driver).findElementById("login_password").sendKeys("password");
        ((ChromeDriver) driver).findElementByXPath("//*[contains(@class, 'btn btn--primary btn--regular login-btn')]").click();

        String url = driver.getCurrentUrl();

如何从这个临时URL中提取“代码”?任何帮助都将不胜感激。