X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=CollectionReader.h;h=72f652a50c5704266ceea2ed4c01c6fe3af57b37;hb=e9bd547c31bc06fac87b49cafa624c03c2d50c91;hp=2936e2573982ba02e594aaf11426d7b881bbf87d;hpb=e1f64147e3eedccbbc59658ddb6cf7992dbd50c1;p=ecparse.git

diff --git a/CollectionReader.h b/CollectionReader.h
index 2936e25..72f652a 100644
--- a/CollectionReader.h
+++ b/CollectionReader.h
@@ -2,10 +2,11 @@
  * © 2008 by David Roden <droden@gmail.com>
  */
 
-#ifndef __ECPARSE_COLLECTIONREADER_H__
-#define __ECPARSE_COLLECTIONREADER_H__
+#pragma once
 
 #include "ReaderInput.h"
+#include "ED2KLink.h"
+#include "GrowingBuffer.h"
 
 class CollectionReader {
 
@@ -13,9 +14,20 @@ public:
 	CollectionReader(ReaderInput* readerInput);
 	~CollectionReader();
 
+	ED2KLink* getNextLink();
+
+private:
+	void identifyCollectionType();
+	bool isLineBreakPresent();
+	bool ensureBufferCapacity(size_t byteCount);
+	void readMoreBytes();
+
 private:
 	ReaderInput* readerInput;
+	GrowingBuffer growingBuffer;
+	int version;
+	bool firstLink;
+	bool isTextCollection;
 
 };
 
-#endif