Remember requested URL after authentication flow

We would like to link to query outputs using URLs in the "table in new window" form (https://sqlpad/query-table/). The idea being if the client is authenticated they can follow the link and see the results.

If they are unauthenticated they are taken to the login view. They go through the login flow but are then taken to the interactive query editor, not the URL they followed.

Is it possible to make this work? I'm happy to make a PR if you have any hints on how to implement.