Add account switch endpoint for account switcher

Author: futurGH

bin/main.ml

@@ -40,6 +40,7 @@ let handlers =
   ; ( get
     , "/account/signup/check-handle"
     , Api.Account_.Signup.check_handle_handler )
+  ; (post, "/account/switch", Api.Account_.Login.switch_account_handler)
   ; (get, "/account/logout", Api.Account_.Logout.handler)
   ; (* admin ui *)
     (get, "/admin", Api.Admin_.Index.handler)

frontend/src/components/AccountSwitcher.mlx

@@ -24,6 +24,32 @@ let fallback handle avatar =
 
 let[@react.component] make ~current_user ~logged_in_users ~add_account_url
     ?(name = "did") ?(inline = false) ?onChange () =
+  let handleAccountSwitch =
+    [%browser_only
+      fun newDid ->
+        if newDid <> current_user.did then
+          let formData = Fetch.FormData.make () in
+          Fetch.FormData.set "did" newDid formData ;
+          let _ =
+            Fetch.fetchWithInit "/account/switch"
+              (Fetch.RequestInit.make ~method_:Fetch.Post
+                 ~body:(Fetch.BodyInit.makeWithFormData formData)
+                 () )
+            |> Js.Promise.then_ (fun response ->
+                   if Fetch.Response.ok response then (
+                     ignore ([%mel.raw {| window.location.reload() |}] : unit) ;
+                     Js.Promise.resolve () )
+                   else Js.Promise.resolve () )
+            |> Js.Promise.catch (fun _ -> Js.Promise.resolve ())
+          in
+          () ;
+          ( match onChange with
+          | Some f ->
+              f newDid
+          | None ->
+              () )
+        else ()]
+  in
   let button_class =
     if inline then
       "group inline-flex flex-row items-center px-1.5 py-1 -mx-0.75 -my-1 \
@@ -41,7 +67,7 @@ let[@react.component] make ~current_user ~logged_in_users ~add_account_url
   <ClientOnly fallback=(fallback current_user.handle current_user.avatar_data_uri)>
   [%browser_only
     (fun () ->
-      <Aria.Select name className="inline" defaultValue=current_user.did placeholder="select account" ?onChange>
+      <Aria.Select name className="inline" defaultValue=current_user.did placeholder="select account" onChange=handleAccountSwitch>
         <Aria.Button className=button_class>
           <Aria.SelectValue className=value_class />
         </Aria.Button>

pegasus/lib/api/account_/login.ml

@@ -11,6 +11,39 @@ let get_handler =
         (module Frontend.LoginPage)
         ~props:{redirect_url; csrf_token; error= None} )
 
+type switch_account_response =
+  {success: bool; error: string option [@default None]}
+[@@deriving yojson {strict= false}]
+
+let switch_account_handler =
+  Xrpc.handler (fun ctx ->
+      match%lwt Dream.form ctx.req with
+      | `Ok fields -> (
+          let did = List.assoc_opt "did" fields in
+          match did with
+          | Some did ->
+              let%lwt logged_in_dids = Session.Raw.get_logged_in_dids ctx.req in
+              if List.mem did logged_in_dids then
+                let%lwt () = Session.Raw.set_current_did ctx.req did in
+                Dream.json @@ Yojson.Safe.to_string
+                @@ switch_account_response_to_yojson {success= true; error= None}
+              else
+                Dream.json ~status:`Bad_Request
+                @@ Yojson.Safe.to_string
+                @@ switch_account_response_to_yojson
+                     { success= false
+                     ; error= Some "not logged in as this account" }
+          | None ->
+              Dream.json ~status:`Bad_Request
+              @@ Yojson.Safe.to_string
+              @@ switch_account_response_to_yojson
+                   {success= false; error= Some "missing did parameter"} )
+      | _ ->
+          Dream.json ~status:`Bad_Request
+          @@ Yojson.Safe.to_string
+          @@ switch_account_response_to_yojson
+               {success= false; error= Some "invalid form submission"} )
+
 let post_handler =
   Xrpc.handler (fun ctx ->
       let csrf_token = Dream.csrf_token ctx.req in