feat: add ws login (#43)

schema.graphql

@@ -52,7 +52,7 @@ input ListRecommendationsInput {
 }
 
 type Mutation {
-  joinWithGoogle(input: JoinWithGoogleInput!): User!
+  joinWithGoogle(input: JoinWithGoogleInput!): Void!
   logout: Void!
   readAllNotifications: Void!
   readNotification(input: ReadNotificationInput!): Void!

src/features/apollo/client.ts

@@ -9,6 +9,8 @@ import { onError } from '@apollo/client/link/error';
 import { GraphQLWsLink } from '@apollo/client/link/subscriptions';
 import { createClient } from 'graphql-ws';
 import { getMainDefinition } from '@apollo/client/utilities';
+import { setContext } from '@apollo/client/link/context';
+import { getCookie } from '../../utils/cookie';
 import { API_URL, WS_URL } from './constants';
 import errorHandler from './errorHandler';
 
@@ -17,6 +19,9 @@ export const linkError = onError(errorHandler);
 const wsLink = new GraphQLWsLink(
   createClient({
     url: `${WS_URL}/subscriptions`,
+    connectionParams: {
+      authorization: getCookie('Authentication'),
+    },
   }),
 );
 
@@ -25,6 +30,13 @@ const httpLink = new HttpLink({
   credentials: 'include',
 });
 
+const authLink = setContext((_, { headers }) => ({
+  headers: {
+    ...headers,
+    authorization: getCookie('Authentication'),
+  },
+}));
+
 const splitLink = split(
   ({ query }) => {
     const definition = getMainDefinition(query);
@@ -35,7 +47,7 @@ const splitLink = split(
     );
   },
   wsLink,
-  httpLink,
+  authLink.concat(httpLink),
 );
 
 const client = new ApolloClient({

src/features/join/features/google/queries.ts

@@ -3,8 +3,7 @@ import { gql } from '@apollo/client';
 export const MUTATION_JOIN_WITH_GOOGLE = gql`
   mutation JoinWithGoogle($input: JoinWithGoogleInput!) {
     joinWithGoogle(input: $input) {
-      name
-      email
+      __typename
     }
   }
 `;

src/generated/graphql.tsx

@@ -76,7 +76,7 @@ export type ListRecommendationsInput = {
 
 export type Mutation = {
   __typename?: 'Mutation';
-  joinWithGoogle: User;
+  joinWithGoogle: Void;
   logout: Void;
   readAllNotifications: Void;
   readNotification: Void;
@@ -398,7 +398,7 @@ export type JoinWithGoogleMutationVariables = Exact<{
 
 export type JoinWithGoogleMutation = {
   __typename?: 'Mutation';
-  joinWithGoogle: { __typename?: 'User'; name: string; email: string };
+  joinWithGoogle: { __typename: 'Void' };
 };
 
 export type ListNotificationsQueryVariables = Exact<{ [key: string]: never }>;
@@ -1195,8 +1195,7 @@ export type ListUpNextQueryResult = Apollo.QueryResult<
 export const JoinWithGoogleDocument = gql`
   mutation JoinWithGoogle($input: JoinWithGoogleInput!) {
     joinWithGoogle(input: $input) {
-      name
-      email
+      __typename
     }
   }
 `;

src/utils/cookie.ts

@@ -0,0 +1,13 @@
+/* eslint-disable no-useless-escape */
+
+function escape(string: string) {
+  return string.replace(/([.*+?\^$(){}|\[\]\/\\])/g, '\\$1');
+}
+
+export function getCookie(name: string) {
+  const match = document.cookie.match(
+    RegExp('(?:^|;\\s*)' + escape(name) + '=([^;]*)'),
+  );
+
+  return match ? match[1] : '';
+}