import {ApolloClient} from '@apollo/client';
import {getSession} from 'next-auth/react';
import {ProfileDocument, SettingDocument} from '../generated/graphql';
import {initializeApollo, APOLLO_STATE_PROP_NAME} from './apolloClient';

type ServerSideExtension = (
  context: any,
  apolloClient: ApolloClient<any>
) => Promise<Object | void>;

const getServerSideProps =
  (extension?: ServerSideExtension) => async (context: any) => {
    const session = await getSession(context);
    const {STRAPI_URL = 'http://localhost:1337'} = process.env;
    const apolloClient = await initializeApollo(
      `${STRAPI_URL}/graphql`,
      session
    );
    const locale = session?.user?.lang || 'fr';

    try {
      await apolloClient.query({
        query: SettingDocument,
        variables: {locale},
      });

      if (session)
        await apolloClient.query({
          query: ProfileDocument,
        });

      let extensionProps = {};
      if (extension)
        extensionProps = (await extension(context, apolloClient)) || {};

      return {
        props: {
          session,
          [APOLLO_STATE_PROP_NAME]: apolloClient.cache.extract(),
          ...extensionProps,
        },
      };
    } catch (error) {
      console.error(error);
      return {
        props: {session},
      };
    }
  };

export default {
  getServerSideProps,
};