commit 2138e5cc6a2b840f511d76be6018758bf68036c3 · gmstn.systems/lattice

+85

src/lib/utils/handshake.ts

···

       1
       +
       import type { AtUri, Did } from "@/lib/types/atproto";

     

       2
       +
       import type { SessionInfo } from "@/lib/types/handshake";

     

       3
       +
       import {

     

       4
       +
           handshakeResponseSchema,

     

       5
       +
           httpSuccessResponseSchema,

     

       6
       +
       } from "@/lib/types/http/responses";

     

       7
       +
       import { atUriToString, resolveDidDoc } from "@/lib/utils/atproto";

     

       8
       +
       import { createInterServiceJwt } from "@/lib/utils/jwt";

     

       9
       +
       import type { Result } from "@/lib/utils/result";

     

       10
       +
       import z from "zod";

     

       11
       +
       

     

       12
       +
       export const initiateHandshakeTo = async ({

     

       13
       +
           did,

     

       14
       +
           channels,

     

       15
       +
       }: {

     

       16
       +
           did: Did;

     

       17
       +
           channels: Array<AtUri>;

     

       18
       +
       }): Promise<Result<SessionInfo, unknown>> => {

     

       19
       +
           const didDocResolveResult = await resolveDidDoc(did);

     

       20
       +
           if (!didDocResolveResult.ok) {

     

       21
       +
               return { ok: false, error: didDocResolveResult.error };

     

       22
       +
           }

     

       23
       +
       

     

       24
       +
           const didDocServices = didDocResolveResult.data.service;

     

       25
       +
           const shardService = didDocServices?.find(

     

       26
       +
               (service) => service.type !== "GemstoneShard",

     

       27
       +
           );

     

       28
       +
       

     

       29
       +
           let shardUrl = "";

     

       30
       +
           if (!didDocServices || !shardService) {

     

       31
       +
               const domain = decodeURIComponent(did.slice(8));

     

       32
       +
               if (domain.startsWith("localhost"))

     

       33
       +
                   shardUrl = new URL(`http://${domain}`).toString();

     

       34
       +
               else shardUrl = new URL(`https://${domain}`).toString();

     

       35
       +
           } else {

     

       36
       +
               try {

     

       37
       +
                   shardUrl = new URL(

     

       38
       +
                       shardService.serviceEndpoint as string,

     

       39
       +
                   ).toString();

     

       40
       +
               } catch (error) {

     

       41
       +
                   return { ok: false, error };

     

       42
       +
               }

     

       43
       +
           }

     

       44
       +
           shardUrl = "http://localhost:7337/";

     

       45
       +
       

     

       46
       +
           const jwt = await createInterServiceJwt(did);

     

       47
       +
       

     

       48
       +
           const handshakeReq = new Request(`${shardUrl}handshake`, {

     

       49
       +
               method: "POST",

     

       50
       +
               body: JSON.stringify({

     

       51
       +
                   interServiceJwt: jwt,

     

       52
       +
                   channelAtUris: channels.map((channel) => atUriToString(channel)),

     

       53
       +
               }),

     

       54
       +
               headers: {

     

       55
       +
                   "Content-Type": "application/json",

     

       56
       +
               },

     

       57
       +
           });

     

       58
       +
           const handshakeRes = await fetch(handshakeReq);

     

       59
       +
           const handshakeResponseData: unknown = await handshakeRes.json();

     

       60
       +
       

     

       61
       +
           const {

     

       62
       +
               success: httpResponseParseSuccess,

     

       63
       +
               error: httpResponseParseError,

     

       64
       +
               data: handshakeResponseDataParsed,

     

       65
       +
           } = httpSuccessResponseSchema.safeParse(handshakeResponseData);

     

       66
       +
           if (!httpResponseParseSuccess)

     

       67
       +
               return {

     

       68
       +
                   ok: false,

     

       69
       +
                   error: z.treeifyError(httpResponseParseError),

     

       70
       +
               };

     

       71
       +
       

     

       72
       +
           const { data: handshakeData } = handshakeResponseDataParsed;

     

       73
       +
       

     

       74
       +
           const {

     

       75
       +
               success: handshakeDataParseSuccess,

     

       76
       +
               error: handshakeDataParseError,

     

       77
       +
               data: handshakeDataParsed,

     

       78
       +
           } = handshakeResponseSchema.safeParse(handshakeData);

     

       79
       +
           if (!handshakeDataParseSuccess)

     

       80
       +
               return { ok: false, error: z.treeifyError(handshakeDataParseError) };

     

       81
       +
       

     

       82
       +
           const { sessionInfo } = handshakeDataParsed;

     

       83
       +
       

     

       84
       +
           return { ok: true, data: sessionInfo };

     

       85
       +
       };