Add function to get number of pages

importer.go

@@ -108,6 +108,16 @@ func (this *Importer) SetSourceStream(rs *io.ReadSeeker) {
 	}
 }
 
+func (this *Importer) GetNumPages() int {
+	result, err := this.GetReader().getNumPages()
+
+	if err != nil {
+		panic(err)
+	}
+
+	return result
+}
+
 func (this *Importer) GetPageSizes() map[int]map[string]map[string]float64 {
 	result, err := this.GetReader().getAllPageBoxes(1.0)
 

reader.go

@@ -28,6 +28,7 @@ type PdfReader struct {
 	sourceFile     string
 	curPage        int
 	alreadyRead    bool
+	pageCount      int
 }
 
 func NewPdfReaderFromStream(rs io.ReadSeeker) (*PdfReader, error) {
@@ -1238,6 +1239,7 @@ func (this *PdfReader) readPages() error {
 	if err != nil {
 		return errors.Wrap(err, "Failed to get page count")
 	}
+	this.pageCount = pageCount.Int
 
 	// Allocate pages
 	this.pages = make([]*PdfValue, pageCount.Int)
@@ -1432,6 +1434,14 @@ func (this *PdfReader) rebuildContentStream(content *PdfValue) ([]byte, error) {
 	return stream, nil
 }
 
+func (this *PdfReader) getNumPages() (int, error) {
+	if this.pageCount == 0 {
+		return 0, errors.New("Page count is 0")
+	}
+
+	return this.pageCount, nil
+}
+
 func (this *PdfReader) getAllPageBoxes(k float64) (map[int]map[string]map[string]float64, error) {
 	var err error