added check for partial messages

AlCalzone · Dec 8, 2017 · 662005b · 662005b
1 parent 9e0d084
commit 662005b
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/build/Message.d.ts b/build/Message.d.ts
@@ -83,4 +83,8 @@ export declare class Message {
      * serializes this message into a buffer
      */
     serialize(): Buffer;
+    /**
+     * Checks if this message is part of a blockwise transfer
+     */
+    isPartialMessage(): boolean;
 }
diff --git a/build/Message.js b/build/Message.js
@@ -167,6 +167,19 @@ class Message {
         }
         return ret;
     }
+    /**
+     * Checks if this message is part of a blockwise transfer
+     */
+    isPartialMessage() {
+        // TODO: can we put the codes in an enum?
+        const block1option = this.options.find(o => o.code === 27 /* Block1 */);
+        const block2option = this.options.find(o => o.code === 23 /* Block2 */);
+        if (this.code.isRequest() && block1option != null)
+            return true;
+        if (this.code.isResponse() && block2option != null)
+            return true;
+        return false;
+    }
 }
 exports.Message = Message;
 /*

diff --git a/src/Message.ts b/src/Message.ts
@@ -1,4 +1,5 @@
 import { Option } from "./Option";
+import { fail } from "assert";
 
 export enum MessageType {
 	CON = 0, // Confirmable
@@ -202,6 +203,18 @@ export class Message {
 		return ret;
 	}
 
+	/**
+	 * Checks if this message is part of a blockwise transfer
+	 */
+	public isPartialMessage(): boolean {
+		// TODO: can we put the codes in an enum?
+		const block1option = this.options.find(o => o.code === 27 /* Block1 */);
+		const block2option = this.options.find(o => o.code === 23 /* Block2 */);
+		if (this.code.isRequest() && block1option != null) return true;
+		if (this.code.isResponse() && block2option != null) return true;
+		return false;
+	}
+
 }
 
 /*