chore(atlas-service): move all sign in logic to atlas-service package (#4680)

* chore(atlas-service): move all sign in logic to atlas-service package

* chore(atlas-service): add more tests

Author: gribnoysup

package-lock.json

@@ -1 +1 @@
-module.exports = require('@mongodb-js/mocha-config-compass');
+module.exports = require('@mongodb-js/mocha-config-compass/react');

packages/atlas-service/.mocharc.js

@@ -67,9 +67,14 @@
     "typescript": "^5.0.4"
   },
   "dependencies": {
+    "@mongodb-js/compass-components": "^1.10.0",
     "@mongodb-js/devtools-connect": "^2.4.0",
     "@mongodb-js/oidc-plugin": "^0.3.0",
     "electron": "^23.3.10",
-    "node-fetch": "^2.6.7"
+    "node-fetch": "^2.6.7",
+    "react": "^17.0.2",
+    "react-redux": "^8.0.5",
+    "redux": "^4.2.1",
+    "redux-thunk": "^2.4.1"
   }
 }

packages/atlas-service/package.json

@@ -1,7 +1,5 @@
 import React from 'react';
 import { connect } from 'react-redux';
-import type { RootState } from '../../stores/query-bar-store';
-import { signIn, closeSignInModal } from '../../stores/atlas-signin-reducer';
 import {
   Badge,
   Button,
@@ -13,7 +11,9 @@ import {
   cx,
   spacing,
 } from '@mongodb-js/compass-components';
-import { AISignInImageBanner } from './ai-sign-in-banner-image';
+import { AISignInImageBanner } from './ai-signin-banner-image';
+import type { AtlasSignInState } from '../store/atlas-signin-reducer';
+import { closeSignInModal, signIn } from '../store/atlas-signin-reducer';
 
 type SignInModalProps = {
   isSignInModalVisible?: boolean;
@@ -120,10 +120,10 @@ const AISignInModal: React.FunctionComponent<SignInModalProps> = ({
 };
 
 export default connect(
-  (state: RootState) => {
+  (state: AtlasSignInState) => {
     return {
-      isSignInModalVisible: state.atlasSignIn.isModalOpen,
-      isSignInInProgress: state.atlasSignIn.state === 'in-progress',
+      isSignInModalVisible: state.isModalOpen,
+      isSignInInProgress: state.state === 'in-progress',
     };
   },
   { onSignInModalClose: closeSignInModal, onSignInClick: signIn }

…enerative-ai/ai-sign-in-banner-image.tsx …rc/components/ai-signin-banner-image.tsxpackages/compass-query-bar/src/components/generative-ai/ai-sign-in-banner-image.tsx renamed to packages/atlas-service/src/components/ai-signin-banner-image.tsx packages/compass-query-bar/src/components/generative-ai/ai-sign-in-banner-image.tsx renamed to packages/atlas-service/src/components/ai-signin-banner-image.tsx

@@ -0,0 +1,21 @@
+import React, { useEffect } from 'react';
+import { Provider, useDispatch } from 'react-redux';
+import AISignInModal from './ai-signin-modal';
+import { getStore } from '../store/atlas-signin-store';
+import { restoreSignInState } from '../store/atlas-signin-reducer';
+
+function Modal() {
+  const dispatch = useDispatch();
+  useEffect(() => {
+    dispatch(restoreSignInState() as any);
+  }, [dispatch]);
+  return <AISignInModal></AISignInModal>;
+}
+
+export const AtlasSignIn = () => {
+  return (
+    <Provider store={getStore()}>
+      <Modal></Modal>
+    </Provider>
+  );
+};

…nents/generative-ai/ai-sign-in-modal.tsx …rvice/src/components/ai-signin-modal.tsxpackages/compass-query-bar/src/components/generative-ai/ai-sign-in-modal.tsx renamed to packages/atlas-service/src/components/ai-signin-modal.tsx packages/compass-query-bar/src/components/generative-ai/ai-sign-in-modal.tsx renamed to packages/atlas-service/src/components/ai-signin-modal.tsx

@@ -1,5 +1,7 @@
 import type { AtlasService as AtlasServiceMain } from './main';
 import { ipcInvoke } from './util';
+import { signInWithModalPrompt } from './store/atlas-signin-reducer';
+import { dispatch } from './store/atlas-signin-store';
 
 export class AtlasService {
   private ipc = ipcInvoke<
@@ -21,7 +23,26 @@ export class AtlasService {
   introspect = this.ipc.introspect;
   isAuthenticated = this.ipc.isAuthenticated;
   getQueryFromUserPrompt = this.ipc.getQueryFromUserPrompt;
-  signIn = this.ipc.signIn;
+
+  async signIn(
+    options: {
+      signal?: AbortSignal;
+      promptType?: 'none' | 'ai-promo-modal';
+    } = {}
+  ) {
+    switch (options.promptType ?? 'none') {
+      case 'none':
+        await this.ipc.signIn({ signal: options.signal });
+        return;
+      case 'ai-promo-modal':
+        await dispatch(signInWithModalPrompt({ signal: options.signal }));
+        return;
+      default:
+        return Promise.reject(new Error('Not implemented'));
+    }
+  }
 }
 
+export { AtlasSignIn } from './components/atlas-signin';
+
 export type { UserInfo, IntrospectInfo, Token } from './util';