Add initiallySelected prop to io.select.table

src/components/selectTable.ts

@@ -15,35 +15,45 @@ type PublicProps<Row> = Omit<T_IO_PROPS<'SELECT_TABLE'>, 'data' | 'columns'> & {
   data: Row[]
   columns?: (TableColumn<Row> | (string & keyof Row))[]
   rowMenuItems?: (row: Row) => MenuItem[]
+  initiallySelected?: (row: Row) => boolean
 }
 
 export default function selectTable(logger: Logger) {
   return function <Row extends z.input<typeof tableRow> = any>(
     props: PublicProps<Row>
   ) {
-    type DataList = typeof props['data']
+    type DataList = (typeof props)['data']
 
     const columns = columnsBuilder(props, column =>
       logger.warn(missingColumnMessage('io.select.table')(column))
     )
 
+    let selectedKeys: string[] = []
+
     // Rendering all rows on initialization is necessary for filtering and sorting
-    const data = props.data.map((row, index) =>
-      tableRowSerializer({
+    const data = props.data.map((row, index) => {
+      const rowData = tableRowSerializer({
         key: index.toString(),
         row,
         columns,
         menuBuilder: props.rowMenuItems,
         logger,
       })
-    )
+
+      if (props.initiallySelected?.(row)) {
+        selectedKeys.push(rowData.key)
+      }
+
+      return rowData
+    })
 
     return {
       props: {
         ...props,
         data: data.slice(0, TABLE_DATA_BUFFER_SIZE),
         totalRecords: data.length,
         columns,
+        selectedKeys,
       },
       getValue(response: T_IO_RETURNS<'SELECT_TABLE'>) {
         const indices = response.map(({ key }) => Number(key))

src/examples/basic/table.ts

@@ -302,7 +302,7 @@ export const async_table: IntervalActionHandler = async () => {
 export const select_table: IntervalActionHandler = async io => {
   faker.seed(0)
 
-  const data = generateRows(50_000)
+  const data = generateRows(500)
 
   const selected = await io.select.table('Display users', {
     data,
@@ -334,6 +334,7 @@ export const select_table: IntervalActionHandler = async io => {
         params: { email: row.email },
       },
     ],
+    initiallySelected: row => row.id % 2 === 0,
   })
 
   await io.display.table('Display users', {