From 4192ec93237d0cd7b23f89a063c21a1426f98c03 Mon Sep 17 00:00:00 2001
From: Richard Viney <richard.viney@heartlab.com>
Date: Fri, 23 Feb 2024 09:22:47 +1300
Subject: [PATCH] Attempt to pass request object to createStoreWritableStream

---
 index.d.ts     | 5 ++++-
 src/Network.js | 5 ++++-
 src/Server.js  | 3 ++-
 3 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/index.d.ts b/index.d.ts
index 85d734f..bcfa3fd 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -1427,7 +1427,10 @@ declare class Scp extends Network {
    * The default implementation creates a memory Writable stream that for, big instances,
    * could cause out of memory situations.
    */
-  createStoreWritableStream(acceptedPresentationContext: PresentationContext): Writable;
+  createStoreWritableStream(
+    acceptedPresentationContext: PresentationContext,
+    request: CStoreRequest
+  ): Writable;
 
   /**
    * Allows the caller to create a Dataset from the Writable stream used to
diff --git a/src/Network.js b/src/Network.js
index 77c3efc..8e6b326 100644
--- a/src/Network.js
+++ b/src/Network.js
@@ -534,7 +534,10 @@ class Network extends AsyncEventEmitter {
           if (this.dimse.getCommandFieldType() === CommandFieldType.CStoreRequest) {
             if (!this.dimseStoreStream) {
               this.dimseStream = undefined;
-              this.dimseStoreStream = this.createStoreWritableStream(presentationContext);
+              this.dimseStoreStream = this.createStoreWritableStream(
+                presentationContext,
+                this.dimse
+              );
             }
           } else {
             if (!this.dimseStream) {
diff --git a/src/Server.js b/src/Server.js
index 1fa572f..5f5da15 100644
--- a/src/Server.js
+++ b/src/Server.js
@@ -106,10 +106,11 @@ class Scp extends Network {
    * could cause out of memory situations.
    * @method
    * @param {PresentationContext} acceptedPresentationContext - The accepted presentation context.
+   * @param {CStoreRequest} request - C-STORE request.
    * @returns {Writable} The created store writable stream.
    */
   // eslint-disable-next-line no-unused-vars
-  createStoreWritableStream(acceptedPresentationContext) {
+  createStoreWritableStream(acceptedPresentationContext, request) {
     return super.createStoreWritableStream();
   }