feat: add user mode type (lite / pro) to ui store

mrgnlabs · Dec 1, 2023 · b4104a4 · b4104a4
1 parent 4377985
commit b4104a4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/apps/marginfi-v2-ui/src/store/uiStore.ts b/apps/marginfi-v2-ui/src/store/uiStore.ts
@@ -1,6 +1,6 @@
 import { create, StateCreator } from "zustand";
 
-import { LendingModes, PoolTypes, SortType, sortDirection, SortAssetOption } from "~/types";
+import { LendingModes, PoolTypes, SortType, sortDirection, SortAssetOption, UserMode } from "~/types";
 
 const SORT_OPTIONS_MAP: { [key in SortType]: SortAssetOption } = {
   APY_DESC: {
@@ -42,6 +42,7 @@ interface UiState {
   lendingMode: LendingModes;
   poolFilter: PoolTypes;
   sortOption: SortAssetOption;
+  userMode: UserMode;
 
   // Actions
   setIsMenuDrawerOpen: (isOpen: boolean) => void;
@@ -53,6 +54,7 @@ interface UiState {
   setLendingMode: (lendingMode: LendingModes) => void;
   setPoolFilter: (poolType: PoolTypes) => void;
   setSortOption: (sortOption: SortAssetOption) => void;
+  setUserMode: (userMode: UserMode) => void;
 }
 
 function createUiStore() {
@@ -70,6 +72,7 @@ const stateCreator: StateCreator<UiState, [], []> = (set, get) => ({
   lendingMode: LendingModes.LEND,
   poolFilter: PoolTypes.ALL,
   sortOption: SORT_OPTIONS_MAP[SortType.TVL_DESC],
+  userMode: UserMode.LITE,
 
   // Actions
   setIsMenuDrawerOpen: (isOpen: boolean) => set({ isMenuDrawerOpen: isOpen }),
@@ -82,6 +85,7 @@ const stateCreator: StateCreator<UiState, [], []> = (set, get) => ({
   setLendingMode: (lendingMode: LendingModes) => set({ lendingMode: lendingMode }),
   setPoolFilter: (poolType: PoolTypes) => set({ poolFilter: poolType }),
   setSortOption: (sortOption: SortAssetOption) => set({ sortOption: sortOption }),
+  setUserMode: (userMode: UserMode) => set({ userMode: userMode }),
 });
 
 export { createUiStore, SORT_OPTIONS_MAP };

diff --git a/apps/marginfi-v2-ui/src/types.ts b/apps/marginfi-v2-ui/src/types.ts
@@ -110,3 +110,8 @@ export enum SortType {
   TVL_ASC = "TVL_ASC",
   TVL_DESC = "TVL_DESC",
 }
+
+export enum UserMode {
+  LITE = "lite",
+  PRO = "pro",
+}