feat(notebooks): add actions, thunks, reducer, selector for checking if org has notebooks (#6410)

* feat(notebooks): add flows actions, reducer, selector, and thunk

* feat(notebooks): add api call for Notebooks to GetOrganizations

Author: hoorayimhelping

src/flows/actions/flowsActions.ts

@@ -0,0 +1,10 @@
+import {Notebook} from 'src/client/notebooksRoutes'
+
+export const SET_NOTEBOOKS = 'SET_NOTEBOOKS'
+
+export type Actions = ReturnType<typeof setNotebooks>
+
+export const setNotebooks = (notebooks: Notebook[]) => ({
+  type: SET_NOTEBOOKS,
+  notebooks,
+})

src/flows/actions/flowsThunks.ts

@@ -0,0 +1,22 @@
+import {Dispatch} from 'redux'
+
+import {Notebook} from 'src/client/notebooksRoutes'
+
+import {getNotebooks as fetchNotebooks} from 'src/client/notebooksRoutes'
+import {setNotebooks} from 'src/flows/actions/flowsActions'
+
+export const getNotebooks =
+  (orgID: string) => async (dispatch: Dispatch<any>) => {
+    try {
+      const response = await fetchNotebooks({query: {orgID}})
+
+      if (response.status !== 200) {
+        throw new Error()
+      }
+
+      dispatch(setNotebooks(response.data.flows))
+    } catch {
+      const emptyNotebooks = [] as Notebook[]
+      dispatch(setNotebooks(emptyNotebooks))
+    }
+  }

src/flows/reducers/flowsReducer.ts

@@ -0,0 +1,24 @@
+import {Notebook} from 'src/client/notebooksRoutes'
+import {Actions, SET_NOTEBOOKS} from 'src/flows/actions/flowsActions'
+
+export interface NotebooksState {
+  notebooks: Notebook[]
+}
+
+const INITIAL_STATE = {
+  notebooks: [],
+}
+
+export const flowsReducer = (
+  state = INITIAL_STATE,
+  action: Actions
+): NotebooksState => {
+  switch (action.type) {
+    case SET_NOTEBOOKS: {
+      return {...state, notebooks: action.notebooks}
+    }
+    default: {
+      return state
+    }
+  }
+}

src/flows/selectors/flowsSelectors.ts

@@ -0,0 +1,6 @@
+import {Notebook} from 'src/client/notebooksRoutes'
+import {AppState} from 'src/types'
+
+export const selectNotebooks = (state: AppState): Notebook[] => {
+  return state.resources.notebooks.notebooks
+}

src/shared/containers/GetOrganizations.tsx

@@ -39,6 +39,7 @@ import {RemoteDataState} from 'src/types'
 
 // Thunks
 import {getQuartzIdentityThunk} from 'src/identity/actions/thunks'
+import {getNotebooks} from 'src/flows/actions/flowsThunks'
 
 const GetOrganizations: FunctionComponent = () => {
   const {status: orgLoadingStatus} = useSelector(getAllOrgs)
@@ -49,7 +50,6 @@ const GetOrganizations: FunctionComponent = () => {
 
   const dispatch = useDispatch()
 
-  // This doesn't require another API call.
   useEffect(() => {
     if (orgLoadingStatus === RemoteDataState.NotStarted) {
       dispatch(getOrganizations())
@@ -92,6 +92,12 @@ const GetOrganizations: FunctionComponent = () => {
     }
   }, [account.type, account.isUpgradeable])
 
+  useEffect(() => {
+    if (CLOUD && org?.id) {
+      dispatch(getNotebooks(org.id))
+    }
+  }, [org.id])
+
   return (
     <PageSpinner loading={orgLoadingStatus}>
       <Suspense fallback={<PageSpinner />}>

src/store/configureStore.ts

@@ -61,6 +61,7 @@ import alertBuilderReducer from 'src/alerting/reducers/alertBuilder'
 import perfReducer from 'src/perf/reducers'
 import quartzOrganizationReducer from 'src/identity/quartzOrganizations/reducers'
 import orgCreationAllowancesReducer from 'src/identity/allowances/reducers'
+import {flowsReducer} from 'src/flows/reducers/flowsReducer'
 
 // Types
 import {AppState, LocalStorage} from 'src/types'
@@ -110,6 +111,7 @@ export const rootReducer = (history: History) => (state, action) => {
       endpoints: endpointsReducer,
       labels: labelsReducer,
       members: membersReducer,
+      notebooks: flowsReducer,
       orgs: orgsReducer,
       rules: rulesReducer,
       scrapers: scrapersReducer,

src/types/resources.ts

@@ -21,6 +21,8 @@ import {
   Subscription,
 } from 'src/types'
 
+import {NotebooksState} from 'src/flows/reducers/flowsReducer'
+
 export enum ResourceType {
   Authorizations = 'tokens',
   Buckets = 'buckets',
@@ -31,6 +33,7 @@ export enum ResourceType {
   Labels = 'labels',
   Orgs = 'orgs',
   Members = 'members',
+  Notebooks = 'notebooks',
   NotificationRules = 'rules',
   NotificationEndpoints = 'endpoints',
   Plugins = 'plugins',
@@ -90,6 +93,7 @@ export interface ResourceState {
   [ResourceType.Dashboards]: DashboardsState
   [ResourceType.Labels]: NormalizedState<Label>
   [ResourceType.Members]: NormalizedState<Member>
+  [ResourceType.Notebooks]: NotebooksState
   [ResourceType.Orgs]: OrgsState
   [ResourceType.Scrapers]: NormalizedState<Scraper>
   [ResourceType.Secrets]: NormalizedState<Secret>