Workbook package

packages/graph/behaviors/prefer-async.ts

@@ -0,0 +1,35 @@
+import { TimelinePipe, delay } from "@pnp/core";
+import { errorCheck, InjectHeaders } from "@pnp/queryable";
+import { GraphQueryable, _GraphQueryable, graphGet } from "../graphqueryable.js";
+import { RichLongRunningOperation } from "@microsoft/microsoft-graph-types";
+
+export function PreferAsync(pollIntervalMs: number = 25000, maxPolls: number = 4): TimelinePipe {
+    return (instance: _GraphQueryable) => {
+        instance.using(InjectHeaders({ "Prefer": "respond-async" }));
+        instance.on.parse(errorCheck);
+        instance.on.parse(async function (url: URL, response: Response, result: any) {
+            if (response.status === 202) {
+                const opLocation = response.headers.get("Location");
+                const opId = opLocation.split("/").at(-1);
+
+                const statusQuery = GraphQueryable(instance, `operations/${opId}`);
+                for (let i = 0; i < maxPolls; i++) {
+                    await delay(pollIntervalMs);
+
+                    const status = await statusQuery<RichLongRunningOperation>();
+                    if (status.status === 'succeeded') {
+                        let resultEndpoint = status.resourceLocation.split("/").at(-1);
+                        result = await graphGet(GraphQueryable(instance, resultEndpoint));
+                    } else if (status.status === 'failed') {
+                        throw status.error;
+                    }
+                }
+                throw new Error(`Timed out waiting for async operation after ${pollIntervalMs * maxPolls}ms`);
+            }
+
+            return [url, response, result];
+        });
+
+        return instance;
+    }
+}

packages/graph/workbooks/driveitem.ts

@@ -0,0 +1,32 @@
+import { addProp, body, InjectHeaders } from "@pnp/queryable";
+import { _DriveItem } from "../files/types.js";
+import { IWorkbook, IWorkbookWithSession, Workbook, WorkbookWithSession } from "./types.js";
+import { graphPost, GraphQueryable } from "../graphqueryable.js";
+import { WorkbookSessionInfo } from "@microsoft/microsoft-graph-types";
+
+declare module "../files/types.js" {
+    interface _DriveItem {
+        readonly workbook: IWorkbook;
+        getWorkbookSession(persistChanges: boolean): Promise<IWorkbookWithSession>;
+    }
+    interface DriveItem {
+        readonly workbook: IWorkbook;
+        getWorkbookSession(persistChanges: boolean): Promise<IWorkbookWithSession>;
+    }
+}
+
+addProp(_DriveItem, "workbook", Workbook);
+_DriveItem.prototype.getWorkbookSession = getWorkbookSession;
+
+export async function getWorkbookSession(this: _DriveItem, persistChanges: boolean): Promise<IWorkbookWithSession> {
+    const workbook = WorkbookWithSession(this);
+    const sessionResult = await graphPost<WorkbookSessionInfo>(
+        GraphQueryable(workbook, 'createSession'), body({
+            persistChanges
+        }));
+
+    if (!sessionResult.id) throw new Error("createSession did not respond with a session ID");
+
+    workbook.using(InjectHeaders({ "workbook-session-id": sessionResult.id }));
+    return workbook;
+}

packages/graph/workbooks/index.ts

@@ -0,0 +1,20 @@
+import "./driveitem.js";
+
+export { 
+    IWorkbook,
+    IWorksheet,
+    IWorksheets,
+    IRange,
+    ITable,
+    ITables,
+    ITableRow,
+    ITableRows,
+    ITableColumn,
+    ITableColumns,
+    IRangeFormat,
+    IRangeBorder,
+    IRangeBorders,
+    IRangeFill,
+    IRangeFont,
+    IRangeFormatProtection
+} from "./types.js";