From c6bc22618075d9717d0f64e8bb1c7467bb28bbfd Mon Sep 17 00:00:00 2001
From: Kenny Daniel <platypii@gmail.com>
Date: Sat, 17 May 2025 17:50:32 -0700
Subject: [PATCH] parquetSchema more generic argument

---
 package.json            | 2 +-
 src/metadata.js         | 6 +++---
 test/schemaTree.test.js | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/package.json b/package.json
index b51342f..549820f 100644
--- a/package.json
+++ b/package.json
@@ -50,7 +50,7 @@
   "devDependencies": {
     "@types/node": "22.15.18",
     "@vitest/coverage-v8": "3.1.3",
-    "eslint": "9.26.0",
+    "eslint": "9.27.0",
     "eslint-plugin-jsdoc": "50.6.17",
     "hyparquet-compressors": "1.1.1",
     "typescript": "5.8.3",
diff --git a/src/metadata.js b/src/metadata.js
index 8600945..7c6d63d 100644
--- a/src/metadata.js
+++ b/src/metadata.js
@@ -187,11 +187,11 @@ export function parquetMetadata(arrayBuffer) {
 /**
  * Return a tree of schema elements from parquet metadata.
  *
- * @param {FileMetaData} metadata parquet metadata object
+ * @param {{schema: SchemaElement[]}} metadata parquet metadata object
  * @returns {SchemaTree} tree of schema elements
  */
-export function parquetSchema(metadata) {
-  return getSchemaPath(metadata.schema, [])[0]
+export function parquetSchema({ schema }) {
+  return getSchemaPath(schema, [])[0]
 }
 
 /**
diff --git a/test/schemaTree.test.js b/test/schemaTree.test.js
index 4189314..0f71951 100644
--- a/test/schemaTree.test.js
+++ b/test/schemaTree.test.js
@@ -6,8 +6,8 @@ describe('parquetSchema', () => {
   it('parse schema tree from rowgroups.parquet', async () => {
     const arrayBuffer = await asyncBufferFromFile('test/files/rowgroups.parquet')
     const metadata = await parquetMetadataAsync(arrayBuffer)
-    const result = parquetSchema(metadata)
-    expect(result).toEqual(rowgroupsSchema)
+    const schemaTree = parquetSchema(metadata)
+    expect(schemaTree).toEqual(rowgroupsSchema)
   })
 })