From ad21b2e1fd10bd9c1ae8e5beeb22e5496a24b0c9 Mon Sep 17 00:00:00 2001
From: Adria Navarro <adria@budibase.com>
Date: Tue, 24 Dec 2024 13:54:16 +0100
Subject: [PATCH] Type user store

---
 .../src/components/grid/stores/index.ts       |  3 +-
 .../grid/stores/{users.js => users.ts}        | 37 +++++++++++++++----
 packages/types/src/ui/stores/grid/index.ts    |  1 +
 packages/types/src/ui/stores/grid/user.ts     |  6 +++
 4 files changed, 38 insertions(+), 9 deletions(-)
 rename packages/frontend-core/src/components/grid/stores/{users.js => users.ts} (61%)
 create mode 100644 packages/types/src/ui/stores/grid/user.ts

diff --git a/packages/frontend-core/src/components/grid/stores/index.ts b/packages/frontend-core/src/components/grid/stores/index.ts
index 90e89cbf4c..8ecbc0e19c 100644
--- a/packages/frontend-core/src/components/grid/stores/index.ts
+++ b/packages/frontend-core/src/components/grid/stores/index.ts
@@ -61,7 +61,8 @@ export type Store = BaseStore &
   ViewV2.Store &
   NonPlus.Store &
   Datasource.Store &
-  Validation.Store & {
+  Validation.Store &
+  Users.Store & {
     // TODO while typing the rest of stores
     fetch: Writable<any>
     filter: Writable<any>
diff --git a/packages/frontend-core/src/components/grid/stores/users.js b/packages/frontend-core/src/components/grid/stores/users.ts
similarity index 61%
rename from packages/frontend-core/src/components/grid/stores/users.js
rename to packages/frontend-core/src/components/grid/stores/users.ts
index 64c1e27835..f8824f8329 100644
--- a/packages/frontend-core/src/components/grid/stores/users.js
+++ b/packages/frontend-core/src/components/grid/stores/users.ts
@@ -1,8 +1,29 @@
-import { writable, get, derived } from "svelte/store"
+import { writable, get, derived, Writable, Readable } from "svelte/store"
 import { helpers } from "@budibase/shared-core"
+import { Store as StoreContext } from "."
+import { UIUser } from "@budibase/types"
 
-export const createStores = () => {
-  const users = writable([])
+interface UsersStore {
+  users: Writable<UIUser[]>
+}
+
+interface DerivedUsersStore {
+  userCellMap: Readable<Record<string, UIUser>>
+}
+
+interface ActionUserStore {
+  users: UsersStore["users"] & {
+    actions: {
+      updateUser: (user: UIUser) => void
+      removeUser: (sessionId: string) => void
+    }
+  }
+}
+
+export type Store = UsersStore & DerivedUsersStore
+
+export const createStores = (): UsersStore => {
+  const users = writable<UIUser[]>([])
 
   const enrichedUsers = derived(users, $users => {
     return $users.map(user => ({
@@ -20,7 +41,7 @@ export const createStores = () => {
   }
 }
 
-export const deriveStores = context => {
+export const deriveStores = (context: StoreContext): DerivedUsersStore => {
   const { users, focusedCellId } = context
 
   // Generate a lookup map of cell ID to the user that has it selected, to make
@@ -28,7 +49,7 @@ export const deriveStores = context => {
   const userCellMap = derived(
     [users, focusedCellId],
     ([$users, $focusedCellId]) => {
-      let map = {}
+      let map: Record<string, UIUser> = {}
       $users.forEach(user => {
         const cellId = user.gridMetadata?.focusedCellId
         if (cellId && cellId !== $focusedCellId) {
@@ -44,10 +65,10 @@ export const deriveStores = context => {
   }
 }
 
-export const createActions = context => {
+export const createActions = (context: StoreContext): ActionUserStore => {
   const { users } = context
 
-  const updateUser = user => {
+  const updateUser = (user: UIUser) => {
     const $users = get(users)
     if (!$users.some(x => x.sessionId === user.sessionId)) {
       users.set([...$users, user])
@@ -60,7 +81,7 @@ export const createActions = context => {
     }
   }
 
-  const removeUser = sessionId => {
+  const removeUser = (sessionId: string) => {
     users.update(state => {
       return state.filter(x => x.sessionId !== sessionId)
     })
diff --git a/packages/types/src/ui/stores/grid/index.ts b/packages/types/src/ui/stores/grid/index.ts
index bcd7e3267d..b6a152ed73 100644
--- a/packages/types/src/ui/stores/grid/index.ts
+++ b/packages/types/src/ui/stores/grid/index.ts
@@ -2,3 +2,4 @@ export * from "./columns"
 export * from "./datasource"
 export * from "./table"
 export * from "./view"
+export * from "./user"
diff --git a/packages/types/src/ui/stores/grid/user.ts b/packages/types/src/ui/stores/grid/user.ts
new file mode 100644
index 0000000000..b6eb529805
--- /dev/null
+++ b/packages/types/src/ui/stores/grid/user.ts
@@ -0,0 +1,6 @@
+import { User } from "@budibase/types"
+
+export interface UIUser extends User {
+  sessionId: string
+  gridMetadata?: { focusedCellId?: string }
+}