Return json from magic code authenticate endpoint if no callback_url is given (#9012)

Match the behaviour of the email+password authenticate endpoint, and
return a json response if the `callback_url` is not passed to the magic
code authenticate endpoint.

---------

Co-authored-by: Scott Trinh <[email protected]>

Author: jaclarke

edb/server/protocol/auth_ext/http.py

@@ -1710,15 +1710,21 @@ async def handle_magic_link_authenticate(
                 data = self._get_data_from_request(request)
 
                 _check_keyset(
-                    data, {"email", "code", "callback_url", "challenge"}
+                    data, {"email", "code", "challenge"}
                 )
 
                 email = data["email"]
                 code_str = data["code"]
-                callback_url = data["callback_url"]
                 challenge = data["challenge"]
 
-                if not self._is_url_allowed(callback_url):
+                maybe_callback_url = cast(
+                    Optional[str], data.get("callback_url")
+                )
+
+                if (
+                    maybe_callback_url and
+                    not self._is_url_allowed(maybe_callback_url)
+                ):
                     raise errors.InvalidData(
                         "Callback URL does not match any allowed URLs.",
                     )
@@ -1761,10 +1767,17 @@ async def handle_magic_link_authenticate(
                     datetime.datetime.now(datetime.timezone.utc),
                 )
 
-                return self._try_redirect(
-                    response,
-                    util.join_url_params(callback_url, {"code": auth_code}),
-                )
+                response_dict = {"code": auth_code}
+
+                if maybe_callback_url:
+                    return self._try_redirect(
+                        response,
+                        util.join_url_params(maybe_callback_url, response_dict),
+                    )
+                else:
+                    response.status = http.HTTPStatus.OK
+                    response.content_type = b"application/json"
+                    response.body = json.dumps(response_dict).encode()
 
             except Exception as ex:
                 redirect_on_failure = _maybe_get_search_param(