@tokscale/core 1.0.17 → 1.0.18

package/index.d.ts

@@ -28,14 +28,13 @@ export interface DataSummary {
   models: Array<string>
 }
 
-/** Finalize graph with pricing */
-export declare function finalizeGraph(options: FinalizeGraphOptions): GraphResult
+/** Finalize graph */
+export declare function finalizeGraph(options: FinalizeGraphOptions): Promise<GraphResult>
 
 /** Options for finalizing graph */
 export interface FinalizeGraphOptions {
   homeDir?: string
   localMessages: ParsedMessages
-  pricing: Array<PricingEntry>
   includeCursor: boolean
   since?: string
   until?: string
@@ -46,49 +45,42 @@ export interface FinalizeGraphOptions {
 export interface FinalizeMonthlyOptions {
   homeDir?: string
   localMessages: ParsedMessages
-  pricing: Array<PricingEntry>
   includeCursor: boolean
   since?: string
   until?: string
   year?: string
 }
 
-/** Finalize monthly report with pricing */
-export declare function finalizeMonthlyReport(options: FinalizeMonthlyOptions): MonthlyReport
+/** Finalize monthly report */
+export declare function finalizeMonthlyReport(options: FinalizeMonthlyOptions): Promise<MonthlyReport>
 
 /** Finalize model report: apply pricing to local messages, add Cursor, aggregate */
-export declare function finalizeReport(options: FinalizeReportOptions): ModelReport
+export declare function finalizeReport(options: FinalizeReportOptions): Promise<ModelReport>
 
-/** Options for finalizing report with pricing */
+/**
+ * Finalize both report and graph in a single call with shared pricing
+ * This ensures consistent costs between report and graph data
+ */
+export declare function finalizeReportAndGraph(options: FinalizeReportOptions): Promise<ReportAndGraph>
+
+/** Options for finalizing report */
 export interface FinalizeReportOptions {
   homeDir?: string
   localMessages: ParsedMessages
-  pricing: Array<PricingEntry>
   includeCursor: boolean
   since?: string
   until?: string
   year?: string
 }
 
-/**
- * Generate graph data from all session sources
- *
- * This is the main entry point that orchestrates:
- * 1. Parallel file scanning
- * 2. Parallel session parsing
- * 3. Date filtering
- * 4. Parallel aggregation
- */
-export declare function generateGraph(options: GraphOptions): GraphResult
-
 /** Generate graph data with pricing calculation */
-export declare function generateGraphWithPricing(options: ReportOptions): GraphResult
+export declare function generateGraphWithPricing(options: ReportOptions): Promise<GraphResult>
 
 /** Get model usage report with pricing calculation */
-export declare function getModelReport(options: ReportOptions): ModelReport
+export declare function getModelReport(options: ReportOptions): Promise<ModelReport>
 
 /** Get monthly usage report with pricing calculation */
-export declare function getMonthlyReport(options: ReportOptions): MonthlyReport
+export declare function getMonthlyReport(options: ReportOptions): Promise<MonthlyReport>
 
 /** Metadata about the graph generation */
 export interface GraphMeta {
@@ -99,22 +91,6 @@ export interface GraphMeta {
   processingTimeMs: number
 }
 
-/** Configuration options for graph generation */
-export interface GraphOptions {
-  /** Home directory path (defaults to user's home) */
-  homeDir?: string
-  /** Sources to include: "opencode", "claude", "codex", "gemini" */
-  sources?: Array<string>
-  /** Start date filter (YYYY-MM-DD) */
-  since?: string
-  /** End date filter (YYYY-MM-DD) */
-  until?: string
-  /** Filter to specific year */
-  year?: string
-  /** Number of parallel threads (defaults to CPU count) */
-  threads?: number
-}
-
 /** Complete graph result */
 export interface GraphResult {
   meta: GraphMeta
@@ -135,13 +111,7 @@ export interface LocalParseOptions {
   year?: string
 }
 
-/** Pricing data for a single model (passed from TypeScript) */
-export interface ModelPricing {
-  inputCostPerToken: number
-  outputCostPerToken: number
-  cacheReadInputTokenCost?: number
-  cacheCreationInputTokenCost?: number
-}
+export declare function lookupPricing(modelId: string, provider?: string | undefined | null): Promise<PricingLookupResult>
 
 /** Model report result */
 export interface ModelReport {
@@ -188,6 +158,13 @@ export interface MonthlyUsage {
   cost: number
 }
 
+export interface NativePricing {
+  inputCostPerToken: number
+  outputCostPerToken: number
+  cacheReadInputTokenCost?: number
+  cacheCreationInputTokenCost?: number
+}
+
 export interface ParsedMessage {
   source: string
   modelId: string
@@ -210,6 +187,8 @@ export interface ParsedMessages {
   claudeCount: number
   codexCount: number
   geminiCount: number
+  ampCount: number
+  droidCount: number
   processingTimeMs: number
 }
 
@@ -219,41 +198,28 @@ export interface ParsedMessages {
  */
 export declare function parseLocalSources(options: LocalParseOptions): ParsedMessages
 
-/** Entry in the pricing map */
-export interface PricingEntry {
+export interface PricingLookupResult {
   modelId: string
-  pricing: ModelPricing
+  matchedKey: string
+  source: string
+  pricing: NativePricing
 }
 
-/** Options for reports with pricing */
+/** Combined result for report and graph (single pricing lookup) */
+export interface ReportAndGraph {
+  report: ModelReport
+  graph: GraphResult
+}
+
+/** Options for reports */
 export interface ReportOptions {
-  /** Home directory path (defaults to user's home) */
   homeDir?: string
-  /** Sources to include: "opencode", "claude", "codex", "gemini" */
   sources?: Array<string>
-  /** Pricing data for cost calculation */
-  pricing: Array<PricingEntry>
-  /** Start date filter (YYYY-MM-DD) */
   since?: string
-  /** End date filter (YYYY-MM-DD) */
   until?: string
-  /** Filter to specific year */
   year?: string
 }
 
-/** Scan for session files (for debugging/testing) */
-export declare function scanSessions(homeDir?: string | undefined | null, sources?: Array<string> | undefined | null): ScanStats
-
-/** Scan session files and return file counts per source */
-export interface ScanStats {
-  opencodeFiles: number
-  claudeFiles: number
-  codexFiles: number
-  geminiFiles: number
-  cursorFiles: number
-  totalFiles: number
-}
-
 /** Source contribution for a specific day */
 export interface SourceContribution {
   source: string

package/index.js

@@ -1,336 +1,587 @@
+// prettier-ignore
 /* eslint-disable */
+// @ts-nocheck
 /* auto-generated by NAPI-RS */
 
-const { existsSync, readFileSync } = require('fs')
-const { join } = require('path')
+const { readFileSync } = require('node:fs')
+let nativeBinding = null
+const loadErrors = []
 
-const { platform, arch } = process
+const isMusl = () => {
+  let musl = false
+  if (process.platform === 'linux') {
+    musl = isMuslFromFilesystem()
+    if (musl === null) {
+      musl = isMuslFromReport()
+    }
+    if (musl === null) {
+      musl = isMuslFromChildProcess()
+    }
+  }
+  return musl
+}
 
-let nativeBinding = null
-let localFileExisted = false
-let loadError = null
+const isFileMusl = (f) => f.includes('libc.musl-') || f.includes('ld-musl-')
 
-function isMusl() {
-  // For Node 10
-  if (!process.report || typeof process.report.getReport !== 'function') {
-    try {
-      const lddPath = require('child_process').execSync('which ldd').toString().trim()
-      return readFileSync(lddPath, 'utf8').includes('musl')
-    } catch (e) {
+const isMuslFromFilesystem = () => {
+  try {
+    return readFileSync('/usr/bin/ldd', 'utf-8').includes('musl')
+  } catch {
+    return null
+  }
+}
+
+const isMuslFromReport = () => {
+  let report = null
+  if (typeof process.report?.getReport === 'function') {
+    process.report.excludeNetwork = true
+    report = process.report.getReport()
+  }
+  if (!report) {
+    return null
+  }
+  if (report.header && report.header.glibcVersionRuntime) {
+    return false
+  }
+  if (Array.isArray(report.sharedObjects)) {
+    if (report.sharedObjects.some(isFileMusl)) {
       return true
     }
-  } else {
-    const { glibcVersionRuntime } = process.report.getReport().header
-    return !glibcVersionRuntime
   }
+  return false
 }
 
-switch (platform) {
-  case 'android':
-    switch (arch) {
-      case 'arm64':
-        localFileExisted = existsSync(join(__dirname, 'tokscale-core.android-arm64.node'))
+const isMuslFromChildProcess = () => {
+  try {
+    return require('child_process').execSync('ldd --version', { encoding: 'utf8' }).includes('musl')
+  } catch (e) {
+    // If we reach this case, we don't know if the system is musl or not, so is better to just fallback to false
+    return false
+  }
+}
+
+function requireNative() {
+  if (process.env.NAPI_RS_NATIVE_LIBRARY_PATH) {
+    try {
+      return require(process.env.NAPI_RS_NATIVE_LIBRARY_PATH);
+    } catch (err) {
+      loadErrors.push(err)
+    }
+  } else if (process.platform === 'android') {
+    if (process.arch === 'arm64') {
+      try {
+        return require('./tokscale-core.android-arm64.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-android-arm64')
+        const bindingPackageVersion = require('@tokscale/core-android-arm64/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else if (process.arch === 'arm') {
+      try {
+        return require('./tokscale-core.android-arm-eabi.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-android-arm-eabi')
+        const bindingPackageVersion = require('@tokscale/core-android-arm-eabi/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else {
+      loadErrors.push(new Error(`Unsupported architecture on Android ${process.arch}`))
+    }
+  } else if (process.platform === 'win32') {
+    if (process.arch === 'x64') {
+      if (process.config?.variables?.shlib_suffix === 'dll.a' || process.config?.variables?.node_target_type === 'shared_library') {
         try {
-          if (localFileExisted) {
-            nativeBinding = require('./tokscale-core.android-arm64.node')
-          } else {
-            nativeBinding = require('@tokscale/core-android-arm64')
-          }
+        return require('./tokscale-core.win32-x64-gnu.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-win32-x64-gnu')
+        const bindingPackageVersion = require('@tokscale/core-win32-x64-gnu/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      } else {
+        try {
+        return require('./tokscale-core.win32-x64-msvc.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-win32-x64-msvc')
+        const bindingPackageVersion = require('@tokscale/core-win32-x64-msvc/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      }
+    } else if (process.arch === 'ia32') {
+      try {
+        return require('./tokscale-core.win32-ia32-msvc.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-win32-ia32-msvc')
+        const bindingPackageVersion = require('@tokscale/core-win32-ia32-msvc/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else if (process.arch === 'arm64') {
+      try {
+        return require('./tokscale-core.win32-arm64-msvc.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-win32-arm64-msvc')
+        const bindingPackageVersion = require('@tokscale/core-win32-arm64-msvc/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else {
+      loadErrors.push(new Error(`Unsupported architecture on Windows: ${process.arch}`))
+    }
+  } else if (process.platform === 'darwin') {
+    try {
+      return require('./tokscale-core.darwin-universal.node')
+    } catch (e) {
+      loadErrors.push(e)
+    }
+    try {
+      const binding = require('@tokscale/core-darwin-universal')
+      const bindingPackageVersion = require('@tokscale/core-darwin-universal/package.json').version
+      if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+        throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+      }
+      return binding
+    } catch (e) {
+      loadErrors.push(e)
+    }
+    if (process.arch === 'x64') {
+      try {
+        return require('./tokscale-core.darwin-x64.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-darwin-x64')
+        const bindingPackageVersion = require('@tokscale/core-darwin-x64/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else if (process.arch === 'arm64') {
+      try {
+        return require('./tokscale-core.darwin-arm64.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-darwin-arm64')
+        const bindingPackageVersion = require('@tokscale/core-darwin-arm64/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else {
+      loadErrors.push(new Error(`Unsupported architecture on macOS: ${process.arch}`))
+    }
+  } else if (process.platform === 'freebsd') {
+    if (process.arch === 'x64') {
+      try {
+        return require('./tokscale-core.freebsd-x64.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-freebsd-x64')
+        const bindingPackageVersion = require('@tokscale/core-freebsd-x64/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else if (process.arch === 'arm64') {
+      try {
+        return require('./tokscale-core.freebsd-arm64.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-freebsd-arm64')
+        const bindingPackageVersion = require('@tokscale/core-freebsd-arm64/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else {
+      loadErrors.push(new Error(`Unsupported architecture on FreeBSD: ${process.arch}`))
+    }
+  } else if (process.platform === 'linux') {
+    if (process.arch === 'x64') {
+      if (isMusl()) {
+        try {
+          return require('./tokscale-core.linux-x64-musl.node')
         } catch (e) {
-          loadError = e
+          loadErrors.push(e)
         }
-        break
-      case 'arm':
-        localFileExisted = existsSync(join(__dirname, 'tokscale-core.android-arm-eabi.node'))
         try {
-          if (localFileExisted) {
-            nativeBinding = require('./tokscale-core.android-arm-eabi.node')
-          } else {
-            nativeBinding = require('@tokscale/core-android-arm-eabi')
+          const binding = require('@tokscale/core-linux-x64-musl')
+          const bindingPackageVersion = require('@tokscale/core-linux-x64-musl/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
+          return binding
         } catch (e) {
-          loadError = e
+          loadErrors.push(e)
         }
-        break
-      default:
-        throw new Error(`Unsupported architecture on Android ${arch}`)
-    }
-    break
-  case 'win32':
-    switch (arch) {
-      case 'x64':
-        localFileExisted = existsSync(
-          join(__dirname, 'tokscale-core.win32-x64-msvc.node')
-        )
+      } else {
         try {
-          if (localFileExisted) {
-            nativeBinding = require('./tokscale-core.win32-x64-msvc.node')
-          } else {
-            nativeBinding = require('@tokscale/core-win32-x64-msvc')
-          }
+          return require('./tokscale-core.linux-x64-gnu.node')
         } catch (e) {
-          loadError = e
+          loadErrors.push(e)
         }
-        break
-      case 'ia32':
-        localFileExisted = existsSync(
-          join(__dirname, 'tokscale-core.win32-ia32-msvc.node')
-        )
         try {
-          if (localFileExisted) {
-            nativeBinding = require('./tokscale-core.win32-ia32-msvc.node')
-          } else {
-            nativeBinding = require('@tokscale/core-win32-ia32-msvc')
+          const binding = require('@tokscale/core-linux-x64-gnu')
+          const bindingPackageVersion = require('@tokscale/core-linux-x64-gnu/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
+          return binding
         } catch (e) {
-          loadError = e
+          loadErrors.push(e)
         }
-        break
-      case 'arm64':
-        localFileExisted = existsSync(
-          join(__dirname, 'tokscale-core.win32-arm64-msvc.node')
-        )
+      }
+    } else if (process.arch === 'arm64') {
+      if (isMusl()) {
         try {
-          if (localFileExisted) {
-            nativeBinding = require('./tokscale-core.win32-arm64-msvc.node')
-          } else {
-            nativeBinding = require('@tokscale/core-win32-arm64-msvc')
+          return require('./tokscale-core.linux-arm64-musl.node')
+        } catch (e) {
+          loadErrors.push(e)
+        }
+        try {
+          const binding = require('@tokscale/core-linux-arm64-musl')
+          const bindingPackageVersion = require('@tokscale/core-linux-arm64-musl/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
+          return binding
         } catch (e) {
-          loadError = e
+          loadErrors.push(e)
         }
-        break
-      default:
-        throw new Error(`Unsupported architecture on Windows: ${arch}`)
-    }
-    break
-  case 'darwin':
-    localFileExisted = existsSync(join(__dirname, 'tokscale-core.darwin-universal.node'))
-    try {
-      if (localFileExisted) {
-        nativeBinding = require('./tokscale-core.darwin-universal.node')
       } else {
-        nativeBinding = require('@tokscale/core-darwin-universal')
-      }
-      break
-    } catch {}
-    switch (arch) {
-      case 'x64':
-        localFileExisted = existsSync(join(__dirname, 'tokscale-core.darwin-x64.node'))
         try {
-          if (localFileExisted) {
-            nativeBinding = require('./tokscale-core.darwin-x64.node')
-          } else {
-            nativeBinding = require('@tokscale/core-darwin-x64')
-          }
+          return require('./tokscale-core.linux-arm64-gnu.node')
         } catch (e) {
-          loadError = e
+          loadErrors.push(e)
         }
-        break
-      case 'arm64':
-        localFileExisted = existsSync(
-          join(__dirname, 'tokscale-core.darwin-arm64.node')
-        )
         try {
-          if (localFileExisted) {
-            nativeBinding = require('./tokscale-core.darwin-arm64.node')
-          } else {
-            nativeBinding = require('@tokscale/core-darwin-arm64')
+          const binding = require('@tokscale/core-linux-arm64-gnu')
+          const bindingPackageVersion = require('@tokscale/core-linux-arm64-gnu/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
+          return binding
         } catch (e) {
-          loadError = e
+          loadErrors.push(e)
         }
-        break
-      default:
-        throw new Error(`Unsupported architecture on macOS: ${arch}`)
-    }
-    break
-  case 'freebsd':
-    if (arch !== 'x64') {
-      throw new Error(`Unsupported architecture on FreeBSD: ${arch}`)
-    }
-    localFileExisted = existsSync(join(__dirname, 'tokscale-core.freebsd-x64.node'))
-    try {
-      if (localFileExisted) {
-        nativeBinding = require('./tokscale-core.freebsd-x64.node')
-      } else {
-        nativeBinding = require('@tokscale/core-freebsd-x64')
       }
-    } catch (e) {
-      loadError = e
-    }
-    break
-  case 'linux':
-    switch (arch) {
-      case 'x64':
-        if (isMusl()) {
-          localFileExisted = existsSync(
-            join(__dirname, 'tokscale-core.linux-x64-musl.node')
-          )
-          try {
-            if (localFileExisted) {
-              nativeBinding = require('./tokscale-core.linux-x64-musl.node')
-            } else {
-              nativeBinding = require('@tokscale/core-linux-x64-musl')
-            }
-          } catch (e) {
-            loadError = e
-          }
-        } else {
-          localFileExisted = existsSync(
-            join(__dirname, 'tokscale-core.linux-x64-gnu.node')
-          )
-          try {
-            if (localFileExisted) {
-              nativeBinding = require('./tokscale-core.linux-x64-gnu.node')
-            } else {
-              nativeBinding = require('@tokscale/core-linux-x64-gnu')
-            }
-          } catch (e) {
-            loadError = e
-          }
+    } else if (process.arch === 'arm') {
+      if (isMusl()) {
+        try {
+          return require('./tokscale-core.linux-arm-musleabihf.node')
+        } catch (e) {
+          loadErrors.push(e)
         }
-        break
-      case 'arm64':
-        if (isMusl()) {
-          localFileExisted = existsSync(
-            join(__dirname, 'tokscale-core.linux-arm64-musl.node')
-          )
-          try {
-            if (localFileExisted) {
-              nativeBinding = require('./tokscale-core.linux-arm64-musl.node')
-            } else {
-              nativeBinding = require('@tokscale/core-linux-arm64-musl')
-            }
-          } catch (e) {
-            loadError = e
-          }
-        } else {
-          localFileExisted = existsSync(
-            join(__dirname, 'tokscale-core.linux-arm64-gnu.node')
-          )
-          try {
-            if (localFileExisted) {
-              nativeBinding = require('./tokscale-core.linux-arm64-gnu.node')
-            } else {
-              nativeBinding = require('@tokscale/core-linux-arm64-gnu')
-            }
-          } catch (e) {
-            loadError = e
+        try {
+          const binding = require('@tokscale/core-linux-arm-musleabihf')
+          const bindingPackageVersion = require('@tokscale/core-linux-arm-musleabihf/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
+          return binding
+        } catch (e) {
+          loadErrors.push(e)
+        }
+      } else {
+        try {
+          return require('./tokscale-core.linux-arm-gnueabihf.node')
+        } catch (e) {
+          loadErrors.push(e)
         }
-        break
-      case 'arm':
-        if (isMusl()) {
-          localFileExisted = existsSync(
-            join(__dirname, 'tokscale-core.linux-arm-musleabihf.node')
-          )
-          try {
-            if (localFileExisted) {
-              nativeBinding = require('./tokscale-core.linux-arm-musleabihf.node')
-            } else {
-              nativeBinding = require('@tokscale/core-linux-arm-musleabihf')
-            }
-          } catch (e) {
-            loadError = e
+        try {
+          const binding = require('@tokscale/core-linux-arm-gnueabihf')
+          const bindingPackageVersion = require('@tokscale/core-linux-arm-gnueabihf/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
-        } else {
-          localFileExisted = existsSync(
-            join(__dirname, 'tokscale-core.linux-arm-gnueabihf.node')
-          )
-          try {
-            if (localFileExisted) {
-              nativeBinding = require('./tokscale-core.linux-arm-gnueabihf.node')
-            } else {
-              nativeBinding = require('@tokscale/core-linux-arm-gnueabihf')
-            }
-          } catch (e) {
-            loadError = e
+          return binding
+        } catch (e) {
+          loadErrors.push(e)
+        }
+      }
+    } else if (process.arch === 'loong64') {
+      if (isMusl()) {
+        try {
+          return require('./tokscale-core.linux-loong64-musl.node')
+        } catch (e) {
+          loadErrors.push(e)
+        }
+        try {
+          const binding = require('@tokscale/core-linux-loong64-musl')
+          const bindingPackageVersion = require('@tokscale/core-linux-loong64-musl/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
+          return binding
+        } catch (e) {
+          loadErrors.push(e)
         }
-        break
-      case 'riscv64':
-        if (isMusl()) {
-          localFileExisted = existsSync(
-            join(__dirname, 'tokscale-core.linux-riscv64-musl.node')
-          )
-          try {
-            if (localFileExisted) {
-              nativeBinding = require('./tokscale-core.linux-riscv64-musl.node')
-            } else {
-              nativeBinding = require('@tokscale/core-linux-riscv64-musl')
-            }
-          } catch (e) {
-            loadError = e
+      } else {
+        try {
+          return require('./tokscale-core.linux-loong64-gnu.node')
+        } catch (e) {
+          loadErrors.push(e)
+        }
+        try {
+          const binding = require('@tokscale/core-linux-loong64-gnu')
+          const bindingPackageVersion = require('@tokscale/core-linux-loong64-gnu/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
-        } else {
-          localFileExisted = existsSync(
-            join(__dirname, 'tokscale-core.linux-riscv64-gnu.node')
-          )
-          try {
-            if (localFileExisted) {
-              nativeBinding = require('./tokscale-core.linux-riscv64-gnu.node')
-            } else {
-              nativeBinding = require('@tokscale/core-linux-riscv64-gnu')
-            }
-          } catch (e) {
-            loadError = e
+          return binding
+        } catch (e) {
+          loadErrors.push(e)
+        }
+      }
+    } else if (process.arch === 'riscv64') {
+      if (isMusl()) {
+        try {
+          return require('./tokscale-core.linux-riscv64-musl.node')
+        } catch (e) {
+          loadErrors.push(e)
+        }
+        try {
+          const binding = require('@tokscale/core-linux-riscv64-musl')
+          const bindingPackageVersion = require('@tokscale/core-linux-riscv64-musl/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
+          return binding
+        } catch (e) {
+          loadErrors.push(e)
+        }
+      } else {
+        try {
+          return require('./tokscale-core.linux-riscv64-gnu.node')
+        } catch (e) {
+          loadErrors.push(e)
         }
-        break
-      case 's390x':
-        localFileExisted = existsSync(
-          join(__dirname, 'tokscale-core.linux-s390x-gnu.node')
-        )
         try {
-          if (localFileExisted) {
-            nativeBinding = require('./tokscale-core.linux-s390x-gnu.node')
-          } else {
-            nativeBinding = require('@tokscale/core-linux-s390x-gnu')
+          const binding = require('@tokscale/core-linux-riscv64-gnu')
+          const bindingPackageVersion = require('@tokscale/core-linux-riscv64-gnu/package.json').version
+          if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+            throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
           }
+          return binding
         } catch (e) {
-          loadError = e
+          loadErrors.push(e)
+        }
+      }
+    } else if (process.arch === 'ppc64') {
+      try {
+        return require('./tokscale-core.linux-ppc64-gnu.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-linux-ppc64-gnu')
+        const bindingPackageVersion = require('@tokscale/core-linux-ppc64-gnu/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
         }
-        break
-      default:
-        throw new Error(`Unsupported architecture on Linux: ${arch}`)
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else if (process.arch === 's390x') {
+      try {
+        return require('./tokscale-core.linux-s390x-gnu.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-linux-s390x-gnu')
+        const bindingPackageVersion = require('@tokscale/core-linux-s390x-gnu/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else {
+      loadErrors.push(new Error(`Unsupported architecture on Linux: ${process.arch}`))
     }
-    break
-  default:
-    throw new Error(`Unsupported OS: ${platform}, architecture: ${arch}`)
+  } else if (process.platform === 'openharmony') {
+    if (process.arch === 'arm64') {
+      try {
+        return require('./tokscale-core.openharmony-arm64.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-openharmony-arm64')
+        const bindingPackageVersion = require('@tokscale/core-openharmony-arm64/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else if (process.arch === 'x64') {
+      try {
+        return require('./tokscale-core.openharmony-x64.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-openharmony-x64')
+        const bindingPackageVersion = require('@tokscale/core-openharmony-x64/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else if (process.arch === 'arm') {
+      try {
+        return require('./tokscale-core.openharmony-arm.node')
+      } catch (e) {
+        loadErrors.push(e)
+      }
+      try {
+        const binding = require('@tokscale/core-openharmony-arm')
+        const bindingPackageVersion = require('@tokscale/core-openharmony-arm/package.json').version
+        if (bindingPackageVersion !== '1.0.17' && process.env.NAPI_RS_ENFORCE_VERSION_CHECK && process.env.NAPI_RS_ENFORCE_VERSION_CHECK !== '0') {
+          throw new Error(`Native binding package version mismatch, expected 1.0.17 but got ${bindingPackageVersion}. You can reinstall dependencies to fix this issue.`)
+        }
+        return binding
+      } catch (e) {
+        loadErrors.push(e)
+      }
+    } else {
+      loadErrors.push(new Error(`Unsupported architecture on OpenHarmony: ${process.arch}`))
+    }
+  } else {
+    loadErrors.push(new Error(`Unsupported OS: ${process.platform}, architecture: ${process.arch}`))
+  }
+}
+
+nativeBinding = requireNative()
+
+if (!nativeBinding || process.env.NAPI_RS_FORCE_WASI) {
+  let wasiBinding = null
+  let wasiBindingError = null
+  try {
+    wasiBinding = require('./tokscale-core.wasi.cjs')
+    nativeBinding = wasiBinding
+  } catch (err) {
+    if (process.env.NAPI_RS_FORCE_WASI) {
+      wasiBindingError = err
+    }
+  }
+  if (!nativeBinding) {
+    try {
+      wasiBinding = require('@tokscale/core-wasm32-wasi')
+      nativeBinding = wasiBinding
+    } catch (err) {
+      if (process.env.NAPI_RS_FORCE_WASI) {
+        wasiBindingError.cause = err
+        loadErrors.push(err)
+      }
+    }
+  }
+  if (process.env.NAPI_RS_FORCE_WASI === 'error' && !wasiBinding) {
+    const error = new Error('WASI binding not found and NAPI_RS_FORCE_WASI is set to error')
+    error.cause = wasiBindingError
+    throw error
+  }
 }
 
 if (!nativeBinding) {
-  if (loadError) {
-    throw loadError
+  if (loadErrors.length > 0) {
+    throw new Error(
+      `Cannot find native binding. ` +
+        `npm has a bug related to optional dependencies (https://github.com/npm/cli/issues/4828). ` +
+        'Please try `npm i` again after removing both package-lock.json and node_modules directory.',
+      {
+        cause: loadErrors.reduce((err, cur) => {
+          cur.cause = err
+          return cur
+        }),
+      },
+    )
   }
   throw new Error(`Failed to load native binding`)
 }
 
-const {
-  version,
-  healthCheck,
-  generateGraph,
-  scanSessions,
-  getModelReport,
-  getMonthlyReport,
-  generateGraphWithPricing,
-  // Two-phase processing functions (parallel optimization)
-  parseLocalSources,
-  finalizeReport,
-  finalizeMonthlyReport,
-  finalizeGraph,
-} = nativeBinding
-
-module.exports.version = version
-module.exports.healthCheck = healthCheck
-module.exports.generateGraph = generateGraph
-module.exports.scanSessions = scanSessions
-module.exports.getModelReport = getModelReport
-module.exports.getMonthlyReport = getMonthlyReport
-module.exports.generateGraphWithPricing = generateGraphWithPricing
-// Two-phase processing functions (parallel optimization)
-module.exports.parseLocalSources = parseLocalSources
-module.exports.finalizeReport = finalizeReport
-module.exports.finalizeMonthlyReport = finalizeMonthlyReport
-module.exports.finalizeGraph = finalizeGraph
+module.exports = nativeBinding
+module.exports.finalizeGraph = nativeBinding.finalizeGraph
+module.exports.finalizeMonthlyReport = nativeBinding.finalizeMonthlyReport
+module.exports.finalizeReport = nativeBinding.finalizeReport
+module.exports.finalizeReportAndGraph = nativeBinding.finalizeReportAndGraph
+module.exports.generateGraph = nativeBinding.generateGraph
+module.exports.generateGraphWithPricing = nativeBinding.generateGraphWithPricing
+module.exports.getModelReport = nativeBinding.getModelReport
+module.exports.getMonthlyReport = nativeBinding.getMonthlyReport
+module.exports.healthCheck = nativeBinding.healthCheck
+module.exports.lookupPricing = nativeBinding.lookupPricing
+module.exports.parseLocalSources = nativeBinding.parseLocalSources
+module.exports.scanSessions = nativeBinding.scanSessions
+module.exports.version = nativeBinding.version

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tokscale/core",
-  "version": "1.0.17",
+  "version": "1.0.18",
   "description": "Native Rust core for tokscale CLI - high-performance session parsing",
   "main": "index.js",
   "types": "index.d.ts",
@@ -66,4 +66,4 @@
     "performance",
     "native"
   ]
-}
+}