Support multiple cells

srcbookdev · Jul 17, 2024 · ac84ceb · ac84ceb
1 parent ba4991e
commit ac84ceb
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/packages/web/src/routes/session.tsx b/packages/web/src/routes/session.tsx
@@ -167,17 +167,19 @@ function Session(props: { session: SessionType; channel: SessionChannel; config:
   }
 
   async function insertGeneratedCells(idx: number, cells: Array<CodeCellType | MarkdownCellType>) {
-    for (const cell of cells) {
+    for (let i = 0; i < cells.length; i++) {
+      const cell = cells[i];
+      const insertIdx = idx + i;
       let newCell;
       switch (cell.type) {
         case 'code':
-          newCell = createCodeCell(idx, session.metadata.language, cell);
+          newCell = createCodeCell(insertIdx, session.metadata.language, cell);
           break;
         case 'markdown':
-          newCell = createMarkdownCell(idx, cell);
+          newCell = createMarkdownCell(insertIdx, cell);
           break;
       }
-      channel.push('cell:create', { sessionId: session.id, index: idx, cell: newCell });
+      channel.push('cell:create', { sessionId: session.id, index: insertIdx, cell: newCell });
     }
   }