From 55fdc8d718f0578de090784ef64351043158e468 Mon Sep 17 00:00:00 2001
From: Kelvin Hammond <hammond.kelvin@gmail.com>
Date: Sat, 9 Dec 2023 12:01:27 -0500
Subject: [PATCH] Added mediaCount to Archive

---
 src/archive.h    | 9 +++++++++
 src/index.d.ts   | 1 +
 test/zim.test.ts | 1 +
 3 files changed, 11 insertions(+)

diff --git a/src/archive.h b/src/archive.h
index 857825c..9d9c973 100644
--- a/src/archive.h
+++ b/src/archive.h
@@ -69,6 +69,14 @@ class Archive : public Napi::ObjectWrap<Archive> {
     }
   }
 
+  Napi::Value getMediaCount(const Napi::CallbackInfo &info) {
+    try {
+      return Napi::Value::From(info.Env(), archive_->getMediaCount());
+    } catch (const std::exception &err) {
+      throw Napi::Error::New(info.Env(), err.what());
+    }
+  }
+
   Napi::Value getUuid(const Napi::CallbackInfo &info) {
     try {
       // TODO(kelvinhammond): convert this to
@@ -446,6 +454,7 @@ class Archive : public Napi::ObjectWrap<Archive> {
             InstanceAccessor<&Archive::getAllEntryCount>("allEntryCount"),
             InstanceAccessor<&Archive::getEntryCount>("entryCount"),
             InstanceAccessor<&Archive::getArticleCount>("articleCount"),
+            InstanceAccessor<&Archive::getMediaCount>("mediaCount"),
             InstanceAccessor<&Archive::getUuid>("uuid"),
             InstanceMethod<&Archive::getMetadata>("getMetadata"),
             InstanceMethod<&Archive::getMetadataItem>("getMetadataItem"),
diff --git a/src/index.d.ts b/src/index.d.ts
index 93fde60..2287674 100644
--- a/src/index.d.ts
+++ b/src/index.d.ts
@@ -153,6 +153,7 @@ export class Archive {
   get allEntryCount(): number;
   get entryCount(): number;
   get articleCount(): number;
+  get mediaCount(): number;
   get uuid(): string;
   getMetadata(name: string): string;
   getMetadataItem(name: string): Item;
diff --git a/test/zim.test.ts b/test/zim.test.ts
index 8ca4168..61a4fff 100644
--- a/test/zim.test.ts
+++ b/test/zim.test.ts
@@ -334,6 +334,7 @@ describe("Archive", () => {
     expect(archive.allEntryCount).toBeGreaterThanOrEqual(items.length);
     expect(archive.entryCount).toBe(entries.length);
     expect(archive.articleCount).toBe(items.length);
+    expect(archive.mediaCount).toBe(0);
     expect(archive.uuid).toBeDefined();
 
     // test metadata