nodepyx 1.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (184) hide show
  1. package/LICENSE +22 -0
  2. package/README.md +399 -0
  3. package/binding.gyp +73 -0
  4. package/dist/core/PyCallable.d.ts +65 -0
  5. package/dist/core/PyCallable.d.ts.map +1 -0
  6. package/dist/core/PyCallable.js +109 -0
  7. package/dist/core/PyCallable.js.map +1 -0
  8. package/dist/core/PyContext.d.ts +76 -0
  9. package/dist/core/PyContext.d.ts.map +1 -0
  10. package/dist/core/PyContext.js +228 -0
  11. package/dist/core/PyContext.js.map +1 -0
  12. package/dist/core/PyIterator.d.ts +84 -0
  13. package/dist/core/PyIterator.d.ts.map +1 -0
  14. package/dist/core/PyIterator.js +243 -0
  15. package/dist/core/PyIterator.js.map +1 -0
  16. package/dist/core/PyModule.d.ts +55 -0
  17. package/dist/core/PyModule.d.ts.map +1 -0
  18. package/dist/core/PyModule.js +172 -0
  19. package/dist/core/PyModule.js.map +1 -0
  20. package/dist/core/PyProxy.d.ts +65 -0
  21. package/dist/core/PyProxy.d.ts.map +1 -0
  22. package/dist/core/PyProxy.js +483 -0
  23. package/dist/core/PyProxy.js.map +1 -0
  24. package/dist/core/PyRuntime.d.ts +105 -0
  25. package/dist/core/PyRuntime.d.ts.map +1 -0
  26. package/dist/core/PyRuntime.js +438 -0
  27. package/dist/core/PyRuntime.js.map +1 -0
  28. package/dist/env/CondaManager.d.ts +118 -0
  29. package/dist/env/CondaManager.d.ts.map +1 -0
  30. package/dist/env/CondaManager.js +401 -0
  31. package/dist/env/CondaManager.js.map +1 -0
  32. package/dist/env/PackageInstaller.d.ts +233 -0
  33. package/dist/env/PackageInstaller.d.ts.map +1 -0
  34. package/dist/env/PackageInstaller.js +609 -0
  35. package/dist/env/PackageInstaller.js.map +1 -0
  36. package/dist/env/PythonDetector.d.ts +103 -0
  37. package/dist/env/PythonDetector.d.ts.map +1 -0
  38. package/dist/env/PythonDetector.js +381 -0
  39. package/dist/env/PythonDetector.js.map +1 -0
  40. package/dist/env/VenvManager.d.ts +117 -0
  41. package/dist/env/VenvManager.d.ts.map +1 -0
  42. package/dist/env/VenvManager.js +331 -0
  43. package/dist/env/VenvManager.js.map +1 -0
  44. package/dist/index.d.ts +169 -0
  45. package/dist/index.d.ts.map +1 -0
  46. package/dist/index.js +393 -0
  47. package/dist/index.js.map +1 -0
  48. package/dist/plugins/Plugin.interface.d.ts +41 -0
  49. package/dist/plugins/Plugin.interface.d.ts.map +1 -0
  50. package/dist/plugins/Plugin.interface.js +12 -0
  51. package/dist/plugins/Plugin.interface.js.map +1 -0
  52. package/dist/plugins/PluginManager.d.ts +26 -0
  53. package/dist/plugins/PluginManager.d.ts.map +1 -0
  54. package/dist/plugins/PluginManager.js +174 -0
  55. package/dist/plugins/PluginManager.js.map +1 -0
  56. package/dist/plugins/builtin/NumpyPlugin.d.ts +17 -0
  57. package/dist/plugins/builtin/NumpyPlugin.d.ts.map +1 -0
  58. package/dist/plugins/builtin/NumpyPlugin.js +41 -0
  59. package/dist/plugins/builtin/NumpyPlugin.js.map +1 -0
  60. package/dist/plugins/builtin/PandasPlugin.d.ts +19 -0
  61. package/dist/plugins/builtin/PandasPlugin.d.ts.map +1 -0
  62. package/dist/plugins/builtin/PandasPlugin.js +57 -0
  63. package/dist/plugins/builtin/PandasPlugin.js.map +1 -0
  64. package/dist/plugins/builtin/TorchPlugin.d.ts +23 -0
  65. package/dist/plugins/builtin/TorchPlugin.d.ts.map +1 -0
  66. package/dist/plugins/builtin/TorchPlugin.js +50 -0
  67. package/dist/plugins/builtin/TorchPlugin.js.map +1 -0
  68. package/dist/plugins/index.d.ts +7 -0
  69. package/dist/plugins/index.d.ts.map +1 -0
  70. package/dist/plugins/index.js +12 -0
  71. package/dist/plugins/index.js.map +1 -0
  72. package/dist/serialization/DataFrameBridge.d.ts +141 -0
  73. package/dist/serialization/DataFrameBridge.d.ts.map +1 -0
  74. package/dist/serialization/DataFrameBridge.js +355 -0
  75. package/dist/serialization/DataFrameBridge.js.map +1 -0
  76. package/dist/serialization/MsgPackSerializer.d.ts +45 -0
  77. package/dist/serialization/MsgPackSerializer.d.ts.map +1 -0
  78. package/dist/serialization/MsgPackSerializer.js +242 -0
  79. package/dist/serialization/MsgPackSerializer.js.map +1 -0
  80. package/dist/serialization/NumpyBridge.d.ts +96 -0
  81. package/dist/serialization/NumpyBridge.d.ts.map +1 -0
  82. package/dist/serialization/NumpyBridge.js +323 -0
  83. package/dist/serialization/NumpyBridge.js.map +1 -0
  84. package/dist/serialization/Serializer.d.ts +78 -0
  85. package/dist/serialization/Serializer.d.ts.map +1 -0
  86. package/dist/serialization/Serializer.js +281 -0
  87. package/dist/serialization/Serializer.js.map +1 -0
  88. package/dist/types/PythonTypeMapper.d.ts +87 -0
  89. package/dist/types/PythonTypeMapper.d.ts.map +1 -0
  90. package/dist/types/PythonTypeMapper.js +449 -0
  91. package/dist/types/PythonTypeMapper.js.map +1 -0
  92. package/dist/types/StubCache.d.ts +109 -0
  93. package/dist/types/StubCache.d.ts.map +1 -0
  94. package/dist/types/StubCache.js +333 -0
  95. package/dist/types/StubCache.js.map +1 -0
  96. package/dist/types/TypeGenerator.d.ts +139 -0
  97. package/dist/types/TypeGenerator.d.ts.map +1 -0
  98. package/dist/types/TypeGenerator.js +372 -0
  99. package/dist/types/TypeGenerator.js.map +1 -0
  100. package/dist/types/addon.d.ts +114 -0
  101. package/dist/types/addon.d.ts.map +1 -0
  102. package/dist/types/addon.js +32 -0
  103. package/dist/types/addon.js.map +1 -0
  104. package/dist/types/config.d.ts +175 -0
  105. package/dist/types/config.d.ts.map +1 -0
  106. package/dist/types/config.js +35 -0
  107. package/dist/types/config.js.map +1 -0
  108. package/dist/types/index.d.ts +10 -0
  109. package/dist/types/index.d.ts.map +1 -0
  110. package/dist/types/index.js +12 -0
  111. package/dist/types/index.js.map +1 -0
  112. package/dist/types/python.d.ts +235 -0
  113. package/dist/types/python.d.ts.map +1 -0
  114. package/dist/types/python.js +7 -0
  115. package/dist/types/python.js.map +1 -0
  116. package/dist/utils/ErrorTranslator.d.ts +83 -0
  117. package/dist/utils/ErrorTranslator.d.ts.map +1 -0
  118. package/dist/utils/ErrorTranslator.js +210 -0
  119. package/dist/utils/ErrorTranslator.js.map +1 -0
  120. package/dist/utils/Logger.d.ts +27 -0
  121. package/dist/utils/Logger.d.ts.map +1 -0
  122. package/dist/utils/Logger.js +115 -0
  123. package/dist/utils/Logger.js.map +1 -0
  124. package/dist/utils/MemoryMonitor.d.ts +44 -0
  125. package/dist/utils/MemoryMonitor.d.ts.map +1 -0
  126. package/dist/utils/MemoryMonitor.js +143 -0
  127. package/dist/utils/MemoryMonitor.js.map +1 -0
  128. package/package.json +177 -0
  129. package/python/error_handler.py +433 -0
  130. package/python/nodepyx_runtime.py +575 -0
  131. package/python/serializer.py +379 -0
  132. package/python/type_inspector.py +288 -0
  133. package/scripts/build-native.js +68 -0
  134. package/scripts/download-prebuilds.js +99 -0
  135. package/scripts/generate-stubs.js +405 -0
  136. package/scripts/install.js +260 -0
  137. package/src/core/PyCallable.ts +137 -0
  138. package/src/core/PyContext.ts +296 -0
  139. package/src/core/PyIterator.ts +294 -0
  140. package/src/core/PyModule.ts +194 -0
  141. package/src/core/PyProxy.ts +605 -0
  142. package/src/core/PyRuntime.ts +504 -0
  143. package/src/env/CondaManager.ts +451 -0
  144. package/src/env/PackageInstaller.ts +738 -0
  145. package/src/env/PythonDetector.ts +414 -0
  146. package/src/env/VenvManager.ts +396 -0
  147. package/src/index.ts +425 -0
  148. package/src/native/gil_guard.cpp +26 -0
  149. package/src/native/gil_guard.h +175 -0
  150. package/src/native/nodepyx_addon.cpp +886 -0
  151. package/src/native/python_bridge.cpp +790 -0
  152. package/src/native/python_bridge.h +257 -0
  153. package/src/native/thread_pool.cpp +336 -0
  154. package/src/native/thread_pool.h +175 -0
  155. package/src/native/type_converter.cpp +901 -0
  156. package/src/native/type_converter.h +272 -0
  157. package/src/nextjs/PyProvider.tsx +123 -0
  158. package/src/nextjs/index.ts +21 -0
  159. package/src/nextjs/usePython.ts +106 -0
  160. package/src/nextjs/withnodepyx.ts +88 -0
  161. package/src/plugins/Plugin.interface.ts +51 -0
  162. package/src/plugins/PluginManager.ts +155 -0
  163. package/src/plugins/builtin/NumpyPlugin.ts +36 -0
  164. package/src/plugins/builtin/PandasPlugin.ts +49 -0
  165. package/src/plugins/builtin/TorchPlugin.ts +56 -0
  166. package/src/plugins/index.ts +7 -0
  167. package/src/serialization/DataFrameBridge.ts +398 -0
  168. package/src/serialization/MsgPackSerializer.ts +220 -0
  169. package/src/serialization/NumpyBridge.ts +332 -0
  170. package/src/serialization/Serializer.ts +320 -0
  171. package/src/types/PythonTypeMapper.ts +495 -0
  172. package/src/types/StubCache.ts +340 -0
  173. package/src/types/TypeGenerator.ts +491 -0
  174. package/src/types/addon.ts +170 -0
  175. package/src/types/config.ts +226 -0
  176. package/src/types/index.ts +55 -0
  177. package/src/types/python.ts +309 -0
  178. package/src/types/stubs/numpy.d.ts +441 -0
  179. package/src/types/stubs/pandas.d.ts +575 -0
  180. package/src/types/stubs/sklearn.d.ts +728 -0
  181. package/src/types/stubs/torch.d.ts +694 -0
  182. package/src/utils/ErrorTranslator.ts +220 -0
  183. package/src/utils/Logger.ts +119 -0
  184. package/src/utils/MemoryMonitor.ts +175 -0
@@ -0,0 +1,449 @@
1
+ "use strict";
2
+ /**
3
+ * nodepyx — PythonTypeMapper
4
+ * Maps Python type annotation strings → TypeScript type strings.
5
+ *
6
+ * Handles:
7
+ * - Python built-in types (str, int, float, bool, …)
8
+ * - typing module generics (List[X], Dict[K,V], Optional[X], Union[A,B], …)
9
+ * - numpy / pandas / torch special types
10
+ * - Forward references (strings in quotes)
11
+ * - PEP 604 union syntax (X | Y)
12
+ */
13
+ Object.defineProperty(exports, "__esModule", { value: true });
14
+ exports.PythonTypeMapper = void 0;
15
+ exports.mapType = mapType;
16
+ exports.mapReturnType = mapReturnType;
17
+ exports.mapDefaultValue = mapDefaultValue;
18
+ exports.splitGenericParams = splitGenericParams;
19
+ // ─── Core mapping table ────────────────────────────────────────────────────
20
+ const BUILTIN_MAP = {
21
+ // ── Primitives ──────────────────────────────────────────────────────────
22
+ 'str': { tsType: 'string' },
23
+ 'int': { tsType: 'number' },
24
+ 'float': { tsType: 'number' },
25
+ 'complex': { tsType: '{ real: number; imag: number }' },
26
+ 'bool': { tsType: 'boolean' },
27
+ 'bytes': { tsType: 'Uint8Array' },
28
+ 'bytearray': { tsType: 'Uint8Array' },
29
+ 'memoryview': { tsType: 'Uint8Array' },
30
+ // ── None / null-like ────────────────────────────────────────────────────
31
+ 'None': { tsType: 'null' },
32
+ 'NoneType': { tsType: 'null' },
33
+ 'type(None)': { tsType: 'null' },
34
+ // ── Collections (unparameterized) ───────────────────────────────────────
35
+ 'list': { tsType: 'unknown[]' },
36
+ 'tuple': { tsType: 'readonly unknown[]' },
37
+ 'dict': { tsType: 'Record<string, unknown>' },
38
+ 'set': { tsType: 'Set<unknown>' },
39
+ 'frozenset': { tsType: 'ReadonlySet<unknown>' },
40
+ // ── typing module ───────────────────────────────────────────────────────
41
+ 'Any': { tsType: 'unknown' },
42
+ 'NoReturn': { tsType: 'never' },
43
+ 'Never': { tsType: 'never' },
44
+ 'ClassVar': { tsType: 'unknown' },
45
+ 'Final': { tsType: 'unknown' },
46
+ 'Literal': { tsType: 'unknown' },
47
+ // ── IO / streams ────────────────────────────────────────────────────────
48
+ 'IO': { tsType: 'unknown' },
49
+ 'TextIO': { tsType: 'string' },
50
+ 'BinaryIO': { tsType: 'Uint8Array' },
51
+ // ── Callable ────────────────────────────────────────────────────────────
52
+ 'Callable': { tsType: '(...args: unknown[]) => unknown' },
53
+ // ── Numeric tower ───────────────────────────────────────────────────────
54
+ 'Number': { tsType: 'number' },
55
+ 'Integer': { tsType: 'number' },
56
+ 'Float': { tsType: 'number' },
57
+ 'Decimal': { tsType: 'number' },
58
+ 'Fraction': { tsType: 'number' },
59
+ // ── datetime ────────────────────────────────────────────────────────────
60
+ 'datetime': { tsType: 'Date' },
61
+ 'date': { tsType: 'Date' },
62
+ 'time': { tsType: 'Date' },
63
+ 'timedelta': { tsType: 'number' },
64
+ 'datetime.datetime': { tsType: 'Date' },
65
+ 'datetime.date': { tsType: 'Date' },
66
+ 'datetime.time': { tsType: 'Date' },
67
+ 'datetime.timedelta': { tsType: 'number' },
68
+ // ── pathlib ─────────────────────────────────────────────────────────────
69
+ 'Path': { tsType: 'string' },
70
+ 'PurePath': { tsType: 'string' },
71
+ 'PosixPath': { tsType: 'string' },
72
+ 'WindowsPath': { tsType: 'string' },
73
+ 'pathlib.Path': { tsType: 'string' },
74
+ // ── uuid ─────────────────────────────────────────────────────────────────
75
+ 'UUID': { tsType: 'string' },
76
+ 'uuid.UUID': { tsType: 'string' },
77
+ // ── NumPy ────────────────────────────────────────────────────────────────
78
+ 'ndarray': { tsType: 'Float64Array | Float32Array | Int32Array | Uint8Array' },
79
+ 'np.ndarray': { tsType: 'Float64Array | Float32Array | Int32Array | Uint8Array' },
80
+ 'numpy.ndarray': { tsType: 'Float64Array | Float32Array | Int32Array | Uint8Array' },
81
+ 'np.float32': { tsType: 'Float32Array' },
82
+ 'np.float64': { tsType: 'Float64Array' },
83
+ 'np.int32': { tsType: 'Int32Array' },
84
+ 'np.int64': { tsType: 'BigInt64Array' },
85
+ 'np.uint8': { tsType: 'Uint8Array' },
86
+ 'np.uint16': { tsType: 'Uint16Array' },
87
+ 'np.uint32': { tsType: 'Uint32Array' },
88
+ 'np.bool_': { tsType: 'Uint8Array' },
89
+ 'numpy.float64': { tsType: 'number' },
90
+ 'numpy.float32': { tsType: 'number' },
91
+ 'numpy.int64': { tsType: 'number' },
92
+ 'numpy.int32': { tsType: 'number' },
93
+ // ── Pandas ───────────────────────────────────────────────────────────────
94
+ 'DataFrame': { tsType: 'DataFrameResult' },
95
+ 'pd.DataFrame': { tsType: 'DataFrameResult' },
96
+ 'pandas.DataFrame': { tsType: 'DataFrameResult' },
97
+ 'Series': { tsType: 'SeriesResult' },
98
+ 'pd.Series': { tsType: 'SeriesResult' },
99
+ 'pandas.Series': { tsType: 'SeriesResult' },
100
+ 'Index': { tsType: 'unknown[]' },
101
+ 'pd.Index': { tsType: 'unknown[]' },
102
+ 'MultiIndex': { tsType: 'unknown[][]' },
103
+ // ── PyTorch ───────────────────────────────────────────────────────────────
104
+ 'Tensor': { tsType: 'Float32Array | Float64Array' },
105
+ 'torch.Tensor': { tsType: 'Float32Array | Float64Array' },
106
+ 'torch.nn.Module': { tsType: 'PyProxy' },
107
+ 'torch.optim.Optimizer': { tsType: 'PyProxy' },
108
+ // ── sklearn ───────────────────────────────────────────────────────────────
109
+ 'BaseEstimator': { tsType: 'PyProxy' },
110
+ 'ClassifierMixin': { tsType: 'PyProxy' },
111
+ 'RegressorMixin': { tsType: 'PyProxy' },
112
+ 'TransformerMixin': { tsType: 'PyProxy' },
113
+ // ── fallback ─────────────────────────────────────────────────────────────
114
+ 'object': { tsType: 'unknown' },
115
+ 'type': { tsType: 'PyProxy' },
116
+ 'module': { tsType: 'PyProxy' },
117
+ };
118
+ // ─── Generic type constructors ────────────────────────────────────────────
119
+ /** Patterns for parameterized generics like List[str], Dict[str, int], … */
120
+ const GENERIC_PATTERNS = [
121
+ // Iterator[T]
122
+ {
123
+ re: /^Iterator\[(.+)\]$/i,
124
+ map: ([inner], ctx) => `AsyncIterator<${mapType(inner, ctx)}>`,
125
+ },
126
+ // Set[T]
127
+ {
128
+ re: /^Set\[(.+)\]$/i,
129
+ map: ([inner], ctx) => `Set<${mapType(inner, ctx)}>`,
130
+ },
131
+ // FrozenSet[T]
132
+ {
133
+ re: /^FrozenSet\[(.+)\]$/i,
134
+ map: ([inner], ctx) => `ReadonlySet<${mapType(inner, ctx)}>`,
135
+ },
136
+ // Generator[YieldType, ...] — handled before the generic catch-all
137
+ {
138
+ re: /^Generator\[(.+)\]$/i,
139
+ map: ([inner], ctx) => {
140
+ const parts = splitGenericParams(inner);
141
+ return `AsyncGenerator<${mapType(parts[0], ctx)}>`;
142
+ },
143
+ },
144
+ // List[T] / Sequence[T] / Iterable[T] / MutableSequence[T] / Collection[T] / etc.
145
+ {
146
+ re: /^(?:List|Sequence|Iterable|MutableSequence|Collection|AbstractSet|MutableSet|Deque|Queue)\[(.+)\]$/i,
147
+ map: ([inner], ctx) => `Array<${mapType(inner, ctx)}>`,
148
+ },
149
+ // Tuple[A, B, C]
150
+ {
151
+ re: /^(?:Tuple)\[(.+)\]$/i,
152
+ map: ([inner], ctx) => {
153
+ if (inner === '...') {
154
+ return `unknown[]`;
155
+ }
156
+ const parts = splitGenericParams(inner);
157
+ if (parts.length === 2 && parts[1] === '...') {
158
+ return `Array<${mapType(parts[0], ctx)}>`;
159
+ }
160
+ return `[${parts.map(p => mapType(p, ctx)).join(', ')}]`;
161
+ },
162
+ },
163
+ // Dict[K, V] / Mapping[K, V] / MutableMapping[K, V]
164
+ {
165
+ re: /^(?:Dict|Mapping|MutableMapping|OrderedDict|DefaultDict|ChainMap)\[(.+)\]$/i,
166
+ map: ([inner], ctx) => {
167
+ const parts = splitGenericParams(inner);
168
+ if (parts.length === 2) {
169
+ return `Record<${mapType(parts[0], ctx)}, ${mapType(parts[1], ctx)}>`;
170
+ }
171
+ return `Record<string, unknown>`;
172
+ },
173
+ },
174
+ // Optional[T] → T | null
175
+ {
176
+ re: /^Optional\[(.+)\]$/i,
177
+ map: ([inner], ctx) => `${mapType(inner, ctx)} | null`,
178
+ },
179
+ // Union[A, B, C]
180
+ {
181
+ re: /^Union\[(.+)\]$/i,
182
+ map: ([inner], ctx) => {
183
+ const parts = splitGenericParams(inner);
184
+ return parts.map(p => mapType(p, ctx)).join(' | ');
185
+ },
186
+ },
187
+ // PEP 604 — A | B | C (raw union)
188
+ // Note: params[0] is the first capture group; we reconstruct the full string
189
+ // by combining all capture groups back with ' | ' to split properly.
190
+ {
191
+ re: /^(.+)\s*\|\s*(.+)$/,
192
+ map: (params, ctx) => {
193
+ // params[0..n] are capture groups; rejoin and re-split at top-level pipes
194
+ const full = params.join(' | ');
195
+ const parts = splitPipeUnion(full);
196
+ return parts.map(p => mapType(p.trim(), ctx)).join(' | ');
197
+ },
198
+ },
199
+ // Callable[[A, B], R]
200
+ {
201
+ re: /^Callable\[\[(.*)?\],\s*(.+)\]$/i,
202
+ map: ([params, ret], ctx) => {
203
+ if (!params) {
204
+ return `() => ${mapType(ret, ctx)}`;
205
+ }
206
+ const paramTypes = splitGenericParams(params).map((p, i) => `arg${i}: ${mapType(p, ctx)}`);
207
+ return `(${paramTypes.join(', ')}) => ${mapType(ret, ctx)}`;
208
+ },
209
+ },
210
+ // Type[T] / ClassVar[T] / Final[T]
211
+ {
212
+ re: /^(?:Type|ClassVar|Final)\[(.+)\]$/i,
213
+ map: ([inner], ctx) => mapType(inner, ctx),
214
+ },
215
+ // Awaitable[T] / Coroutine[…, T]
216
+ {
217
+ re: /^(?:Awaitable|Coroutine)\[.*?,?\s*(.+)\]$/i,
218
+ map: ([ret], ctx) => `Promise<${mapType(ret, ctx)}>`,
219
+ },
220
+ // AsyncIterator[T] / AsyncIterable[T] / AsyncGenerator[T]
221
+ {
222
+ re: /^(?:AsyncIterator|AsyncIterable|AsyncGenerator)\[(.+)\]$/i,
223
+ map: ([inner], ctx) => `AsyncIterable<${mapType(inner, ctx)}>`,
224
+ },
225
+ ];
226
+ // ─── Main mapType function ────────────────────────────────────────────────
227
+ /**
228
+ * Map a Python type annotation string to a TypeScript type string.
229
+ *
230
+ * @example
231
+ * ```typescript
232
+ * mapType('List[int]') // → 'Array<number>'
233
+ * mapType('Dict[str, float]') // → 'Record<string, number>'
234
+ * mapType('Optional[DataFrame]') // → 'DataFrameResult | null'
235
+ * mapType('Union[str, int, None]') // → 'string | number | null'
236
+ * mapType('Tuple[int, str, bool]') // → '[number, string, boolean]'
237
+ * ```
238
+ */
239
+ function mapType(pythonType, ctx = {}) {
240
+ if (!pythonType) {
241
+ return 'unknown';
242
+ }
243
+ const trimmed = pythonType.trim();
244
+ // ── Direct lookup ───────────────────────────────────────────────────────
245
+ const direct = BUILTIN_MAP[trimmed];
246
+ if (direct) {
247
+ return direct.tsType;
248
+ }
249
+ // ── Strip module qualifiers for a second lookup ─────────────────────────
250
+ const unqualified = trimmed.includes('.')
251
+ ? trimmed.split('.').pop()
252
+ : trimmed;
253
+ const unqualLookup = BUILTIN_MAP[unqualified];
254
+ if (unqualLookup) {
255
+ return unqualLookup.tsType;
256
+ }
257
+ // ── Generic patterns ────────────────────────────────────────────────────
258
+ for (const pattern of GENERIC_PATTERNS) {
259
+ const match = trimmed.match(pattern.re);
260
+ if (match) {
261
+ const capturedGroups = match.slice(1);
262
+ // Pass the full raw string for PEP-604 union
263
+ return pattern.map(capturedGroups, ctx);
264
+ }
265
+ }
266
+ // ── Known class from current module ─────────────────────────────────────
267
+ if (ctx.knownClasses?.has(trimmed) || ctx.knownClasses?.has(unqualified)) {
268
+ return 'PyProxy';
269
+ }
270
+ // ── String forward reference (PEP 484) ──────────────────────────────────
271
+ if ((trimmed.startsWith('"') && trimmed.endsWith('"')) ||
272
+ (trimmed.startsWith("'") && trimmed.endsWith("'"))) {
273
+ return mapType(trimmed.slice(1, -1), ctx);
274
+ }
275
+ // ── Fallback ─────────────────────────────────────────────────────────────
276
+ // Unknown types always map to 'unknown' (not PyProxy) for better type safety
277
+ return 'unknown';
278
+ }
279
+ // ─── mapReturnType (adds Promise<> wrapping) ────────────────────────────────
280
+ /**
281
+ * Map a Python return type annotation.
282
+ * Automatically wraps the result in `Promise<>` since all Python calls
283
+ * are asynchronous from the JavaScript side.
284
+ *
285
+ * Generator / Iterator returns are mapped to `AsyncGenerator`.
286
+ * None → Promise<void>
287
+ */
288
+ function mapReturnType(pythonReturnType, ctx = {}) {
289
+ if (!pythonReturnType || pythonReturnType === 'None' || pythonReturnType === 'NoneType') {
290
+ return 'Promise<void>';
291
+ }
292
+ if (pythonReturnType === 'NoReturn' || pythonReturnType === 'Never') {
293
+ return 'never';
294
+ }
295
+ // Generator → AsyncGenerator
296
+ if (/^Generator\[/.test(pythonReturnType)) {
297
+ const inner = pythonReturnType.match(/^Generator\[(.+)\]/)?.[1] ?? 'unknown';
298
+ const parts = splitGenericParams(inner);
299
+ return `AsyncGenerator<${mapType(parts[0], ctx)}>`;
300
+ }
301
+ // Coroutine / Awaitable → strip wrapper, it becomes the resolved type
302
+ const awaitMatch = pythonReturnType.match(/^(?:Awaitable|Coroutine)\[.*?,?\s*(.+)\]$/i);
303
+ if (awaitMatch) {
304
+ return `Promise<${mapType(awaitMatch[1], ctx)}>`;
305
+ }
306
+ const inner = mapType(pythonReturnType, ctx);
307
+ return `Promise<${inner}>`;
308
+ }
309
+ // ─── Parameter default value helper ──────────────────────────────────────────
310
+ /**
311
+ * Convert a Python default value representation to a TypeScript literal.
312
+ */
313
+ function mapDefaultValue(pythonDefault) {
314
+ if (pythonDefault === 'None') {
315
+ return 'null';
316
+ }
317
+ if (pythonDefault === 'True') {
318
+ return 'true';
319
+ }
320
+ if (pythonDefault === 'False') {
321
+ return 'false';
322
+ }
323
+ if (/^-?\d+(\.\d+)?$/.test(pythonDefault)) {
324
+ return pythonDefault;
325
+ }
326
+ if (pythonDefault.startsWith("'") || pythonDefault.startsWith('"')) {
327
+ return pythonDefault.replace(/^'|'$/g, '"');
328
+ }
329
+ return `undefined /* ${pythonDefault} */`;
330
+ }
331
+ // ─── Splitting helpers ─────────────────────────────────────────────────────
332
+ /**
333
+ * Split generic parameters at the top level (respects nested brackets).
334
+ * e.g. "str, Dict[str, int], bool" → ["str", "Dict[str, int]", "bool"]
335
+ */
336
+ function splitGenericParams(params) {
337
+ const result = [];
338
+ let depth = 0;
339
+ let current = '';
340
+ for (const ch of params) {
341
+ if (ch === '[') {
342
+ depth++;
343
+ current += ch;
344
+ }
345
+ else if (ch === ']') {
346
+ depth--;
347
+ current += ch;
348
+ }
349
+ else if (ch === ',' && depth === 0) {
350
+ result.push(current.trim());
351
+ current = '';
352
+ }
353
+ else {
354
+ current += ch;
355
+ }
356
+ }
357
+ if (current.trim()) {
358
+ result.push(current.trim());
359
+ }
360
+ return result;
361
+ }
362
+ /**
363
+ * Split a PEP 604 union "A | B | C" at top-level pipes.
364
+ */
365
+ function splitPipeUnion(type) {
366
+ const result = [];
367
+ let depth = 0;
368
+ let current = '';
369
+ for (const ch of type) {
370
+ if (ch === '[') {
371
+ depth++;
372
+ current += ch;
373
+ }
374
+ else if (ch === ']') {
375
+ depth--;
376
+ current += ch;
377
+ }
378
+ else if (ch === '|' && depth === 0) {
379
+ result.push(current.trim());
380
+ current = '';
381
+ }
382
+ else {
383
+ current += ch;
384
+ }
385
+ }
386
+ if (current.trim()) {
387
+ result.push(current.trim());
388
+ }
389
+ return result;
390
+ }
391
+ // ─── PythonTypeMapper class (stateful, module-aware) ──────────────────────
392
+ /**
393
+ * Stateful type mapper that accumulates required imports
394
+ * and knows which class names belong to the current module.
395
+ *
396
+ * @example
397
+ * ```typescript
398
+ * const mapper = new PythonTypeMapper({ moduleName: 'sklearn.linear_model' });
399
+ * mapper.registerClass('LinearRegression');
400
+ *
401
+ * mapper.map('LinearRegression') // → 'LinearRegression' (local class ref)
402
+ * mapper.map('DataFrame') // → 'DataFrameResult'
403
+ * ```
404
+ */
405
+ class PythonTypeMapper {
406
+ _ctx;
407
+ _imports = new Map();
408
+ constructor(ctx = {}) {
409
+ this._ctx = {
410
+ ...ctx,
411
+ knownClasses: ctx.knownClasses ?? new Set(),
412
+ };
413
+ }
414
+ registerClass(name) {
415
+ this._ctx.knownClasses.add(name);
416
+ }
417
+ map(pyType) {
418
+ const result = mapType(pyType, this._ctx);
419
+ // Auto-track imports for well-known result types
420
+ if (result === 'DataFrameResult') {
421
+ this.addImport('DataFrameResult', 'nodepyx');
422
+ }
423
+ else if (result === 'SeriesResult') {
424
+ this.addImport('SeriesResult', 'nodepyx');
425
+ }
426
+ else if (result === 'NumPyArrayResult') {
427
+ this.addImport('NumPyArrayResult', 'nodepyx');
428
+ }
429
+ return result;
430
+ }
431
+ mapReturn(pyType) {
432
+ return mapReturnType(pyType, this._ctx);
433
+ }
434
+ addImport(symbol, from) {
435
+ if (!this._imports.has(from)) {
436
+ this._imports.set(from, new Set());
437
+ }
438
+ this._imports.get(from).add(symbol);
439
+ }
440
+ generateImportBlock() {
441
+ const lines = [];
442
+ for (const [from, symbols] of this._imports) {
443
+ lines.push(`import type { ${[...symbols].join(', ')} } from '${from}';`);
444
+ }
445
+ return lines.join('\n');
446
+ }
447
+ }
448
+ exports.PythonTypeMapper = PythonTypeMapper;
449
+ //# sourceMappingURL=PythonTypeMapper.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"PythonTypeMapper.js","sourceRoot":"","sources":["../../src/types/PythonTypeMapper.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;GAUG;;;AAwQH,0BA6CC;AAYD,sCA2BC;AAOD,0CASC;AAQD,gDAyBC;AAnXD,4LAA4L;AAE5L,MAAM,WAAW,GAA2C;IAC1D,mMAAmM;IACnM,KAAK,EAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;IAClC,KAAK,EAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;IAClC,OAAO,EAAO,EAAE,MAAM,EAAE,QAAQ,EAAE;IAClC,SAAS,EAAK,EAAE,MAAM,EAAE,gCAAgC,EAAE;IAC1D,MAAM,EAAQ,EAAE,MAAM,EAAE,SAAS,EAAE;IACnC,OAAO,EAAO,EAAE,MAAM,EAAE,YAAY,EAAE;IACtC,WAAW,EAAG,EAAE,MAAM,EAAE,YAAY,EAAE;IACtC,YAAY,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE;IAEtC,uLAAuL;IACvL,MAAM,EAAQ,EAAE,MAAM,EAAE,MAAM,EAAE;IAChC,UAAU,EAAI,EAAE,MAAM,EAAE,MAAM,EAAE;IAChC,YAAY,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;IAEhC,6JAA6J;IAC7J,MAAM,EAAQ,EAAE,MAAM,EAAE,WAAW,EAAE;IACrC,OAAO,EAAO,EAAE,MAAM,EAAE,oBAAoB,EAAE;IAC9C,MAAM,EAAQ,EAAE,MAAM,EAAE,yBAAyB,EAAE;IACnD,KAAK,EAAS,EAAE,MAAM,EAAE,cAAc,EAAE;IACxC,WAAW,EAAG,EAAE,MAAM,EAAE,sBAAsB,EAAE;IAEhD,6LAA6L;IAC7L,KAAK,EAAa,EAAE,MAAM,EAAE,SAAS,EAAE;IACvC,UAAU,EAAQ,EAAE,MAAM,EAAE,OAAO,EAAE;IACrC,OAAO,EAAW,EAAE,MAAM,EAAE,OAAO,EAAE;IACrC,UAAU,EAAQ,EAAE,MAAM,EAAE,SAAS,EAAE;IACvC,OAAO,EAAW,EAAE,MAAM,EAAE,SAAS,EAAE;IACvC,SAAS,EAAS,EAAE,MAAM,EAAE,SAAS,EAAE;IAEvC,+LAA+L;IAC/L,IAAI,EAAc,EAAE,MAAM,EAAE,SAAS,EAAE;IACvC,QAAQ,EAAU,EAAE,MAAM,EAAE,QAAQ,EAAE;IACtC,UAAU,EAAQ,EAAE,MAAM,EAAE,YAAY,EAAE;IAE1C,uMAAuM;IACvM,UAAU,EAAQ,EAAE,MAAM,EAAE,iCAAiC,EAAE;IAE/D,6LAA6L;IAC7L,QAAQ,EAAU,EAAE,MAAM,EAAE,QAAQ,EAAE;IACtC,SAAS,EAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;IACtC,OAAO,EAAW,EAAE,MAAM,EAAE,QAAQ,EAAE;IACtC,SAAS,EAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;IACtC,UAAU,EAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE;IAEtC,uMAAuM;IACvM,UAAU,EAAY,EAAE,MAAM,EAAE,MAAM,EAAE;IACxC,MAAM,EAAgB,EAAE,MAAM,EAAE,MAAM,EAAE;IACxC,MAAM,EAAgB,EAAE,MAAM,EAAE,MAAM,EAAE;IACxC,WAAW,EAAW,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,mBAAmB,EAAG,EAAE,MAAM,EAAE,MAAM,EAAE;IACxC,eAAe,EAAO,EAAE,MAAM,EAAE,MAAM,EAAE;IACxC,eAAe,EAAO,EAAE,MAAM,EAAE,MAAM,EAAE;IACxC,oBAAoB,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE;IAE1C,yMAAyM;IACzM,MAAM,EAAgB,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,UAAU,EAAY,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,WAAW,EAAW,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,aAAa,EAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,cAAc,EAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE;IAE1C,kNAAkN;IAClN,MAAM,EAAgB,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,WAAW,EAAW,EAAE,MAAM,EAAE,QAAQ,EAAE;IAE1C,gNAAgN;IAChN,SAAS,EAAa,EAAE,MAAM,EAAE,uDAAuD,EAAE;IACzF,YAAY,EAAU,EAAE,MAAM,EAAE,uDAAuD,EAAE;IACzF,eAAe,EAAO,EAAE,MAAM,EAAE,uDAAuD,EAAE;IACzF,YAAY,EAAU,EAAE,MAAM,EAAE,cAAc,EAAE;IAChD,YAAY,EAAU,EAAE,MAAM,EAAE,cAAc,EAAE;IAChD,UAAU,EAAY,EAAE,MAAM,EAAE,YAAY,EAAE;IAC9C,UAAU,EAAY,EAAE,MAAM,EAAE,eAAe,EAAE;IACjD,UAAU,EAAY,EAAE,MAAM,EAAE,YAAY,EAAE;IAC9C,WAAW,EAAW,EAAE,MAAM,EAAE,aAAa,EAAE;IAC/C,WAAW,EAAW,EAAE,MAAM,EAAE,aAAa,EAAE;IAC/C,UAAU,EAAY,EAAE,MAAM,EAAE,YAAY,EAAE;IAC9C,eAAe,EAAO,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,eAAe,EAAO,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,aAAa,EAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;IAC1C,aAAa,EAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;IAE1C,8MAA8M;IAC9M,WAAW,EAAW,EAAE,MAAM,EAAE,iBAAiB,EAAE;IACnD,cAAc,EAAQ,EAAE,MAAM,EAAE,iBAAiB,EAAE;IACnD,kBAAkB,EAAI,EAAE,MAAM,EAAE,iBAAiB,EAAE;IACnD,QAAQ,EAAc,EAAE,MAAM,EAAE,cAAc,EAAE;IAChD,WAAW,EAAW,EAAE,MAAM,EAAE,cAAc,EAAE;IAChD,eAAe,EAAO,EAAE,MAAM,EAAE,cAAc,EAAE;IAChD,OAAO,EAAe,EAAE,MAAM,EAAE,WAAW,EAAE;IAC7C,UAAU,EAAY,EAAE,MAAM,EAAE,WAAW,EAAE;IAC7C,YAAY,EAAU,EAAE,MAAM,EAAE,aAAa,EAAE;IAE/C,+MAA+M;IAC/M,QAAQ,EAAc,EAAE,MAAM,EAAE,6BAA6B,EAAE;IAC/D,cAAc,EAAQ,EAAE,MAAM,EAAE,6BAA6B,EAAE;IAC/D,iBAAiB,EAAK,EAAE,MAAM,EAAE,SAAS,EAAE;IAC3C,uBAAuB,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;IAE9C,+MAA+M;IAC/M,eAAe,EAAQ,EAAE,MAAM,EAAE,SAAS,EAAE;IAC5C,iBAAiB,EAAM,EAAE,MAAM,EAAE,SAAS,EAAE;IAC5C,gBAAgB,EAAO,EAAE,MAAM,EAAE,SAAS,EAAE;IAC5C,kBAAkB,EAAK,EAAE,MAAM,EAAE,SAAS,EAAE;IAE5C,0MAA0M;IAC1M,QAAQ,EAAM,EAAE,MAAM,EAAE,SAAS,EAAE;IACnC,MAAM,EAAQ,EAAE,MAAM,EAAE,SAAS,EAAE;IACnC,QAAQ,EAAM,EAAE,MAAM,EAAE,SAAS,EAAE;CACpC,CAAC;AAEF,2KAA2K;AAE3K,8EAA8E;AAC9E,MAAM,gBAAgB,GAGjB;IACH,cAAc;IACd;QACE,EAAE,EAAE,qBAAqB;QACzB,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,iBAAiB,OAAO,CAAC,KAAM,EAAE,GAAG,CAAC,GAAG;KAChE;IACD,SAAS;IACT;QACE,EAAE,EAAE,gBAAgB;QACpB,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,OAAO,OAAO,CAAC,KAAM,EAAE,GAAG,CAAC,GAAG;KACtD;IACD,eAAe;IACf;QACE,EAAE,EAAE,sBAAsB;QAC1B,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,eAAe,OAAO,CAAC,KAAM,EAAE,GAAG,CAAC,GAAG;KAC9D;IACD,qEAAqE;IACrE;QACE,EAAE,EAAE,sBAAsB;QAC1B,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE;YACpB,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAM,CAAC,CAAC;YACzC,OAAO,kBAAkB,OAAO,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,GAAG,CAAC,GAAG,CAAC;QACtD,CAAC;KACF;IACD,kFAAkF;IAClF;QACE,EAAE,EAAE,qGAAqG;QACzG,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,SAAS,OAAO,CAAC,KAAM,EAAE,GAAG,CAAC,GAAG;KACxD;IACD,iBAAiB;IACjB;QACE,EAAE,EAAE,sBAAsB;QAC1B,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE;YACpB,IAAI,KAAK,KAAK,KAAK,EAAE,CAAC;gBAAA,OAAO,WAAW,CAAC;YAAA,CAAC;YAC1C,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAM,CAAC,CAAC;YACzC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,KAAK,EAAE,CAAC;gBAC7C,OAAO,SAAS,OAAO,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,GAAG,CAAC,GAAG,CAAC;YAC7C,CAAC;YACD,OAAO,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QAC3D,CAAC;KACF;IACD,oDAAoD;IACpD;QACE,EAAE,EAAE,6EAA6E;QACjF,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE;YACpB,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAM,CAAC,CAAC;YACzC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO,UAAU,OAAO,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,GAAG,CAAC,KAAK,OAAO,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,GAAG,CAAC,GAAG,CAAC;YAC1E,CAAC;YACD,OAAO,yBAAyB,CAAC;QACnC,CAAC;KACF;IACD,2BAA2B;IAC3B;QACE,EAAE,EAAE,qBAAqB;QACzB,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,OAAO,CAAC,KAAM,EAAE,GAAG,CAAC,SAAS;KACxD;IACD,iBAAiB;IACjB;QACE,EAAE,EAAE,kBAAkB;QACtB,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE;YACpB,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAM,CAAC,CAAC;YACzC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrD,CAAC;KACF;IACD,oCAAoC;IACpC,6EAA6E;IAC7E,qEAAqE;IACrE;QACE,EAAE,EAAE,oBAAoB;QACxB,GAAG,EAAE,CAAC,MAAgB,EAAE,GAAmB,EAAE,EAAE;YAC7C,0EAA0E;YAC1E,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,KAAK,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;YACnC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC;KACF;IACD,sBAAsB;IACtB;QACE,EAAE,EAAE,kCAAkC;QACtC,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,GAAG,EAAE,EAAE;YAC1B,IAAI,CAAC,MAAM,EAAE,CAAC;gBAAA,OAAO,SAAS,OAAO,CAAC,GAAI,EAAE,GAAG,CAAC,EAAE,CAAC;YAAA,CAAC;YACpD,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,OAAO,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YAC3F,OAAO,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,OAAO,CAAC,GAAI,EAAE,GAAG,CAAC,EAAE,CAAC;QAC/D,CAAC;KACF;IACD,mCAAmC;IACnC;QACE,EAAE,EAAE,oCAAoC;QACxC,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,KAAM,EAAE,GAAG,CAAC;KAC5C;IACD,mCAAmC;IACnC;QACE,EAAE,EAAE,4CAA4C;QAChD,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,WAAW,OAAO,CAAC,GAAI,EAAE,GAAG,CAAC,GAAG;KACtD;IACD,0DAA0D;IAC1D;QACE,EAAE,EAAE,2DAA2D;QAC/D,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,iBAAiB,OAAO,CAAC,KAAM,EAAE,GAAG,CAAC,GAAG;KAChE;CACF,CAAC;AAEF,mLAAmL;AAEnL;;;;;;;;;;;GAWG;AACH,SAAgB,OAAO,CACrB,UAAkB,EAClB,MAAsB,EAAE;IAExB,IAAI,CAAC,UAAU,EAAE,CAAC;QAAA,OAAO,SAAS,CAAC;IAAA,CAAC;IAEpC,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,EAAE,CAAC;IAElC,6LAA6L;IAC7L,MAAM,MAAM,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACpC,IAAI,MAAM,EAAE,CAAC;QAAA,OAAO,MAAM,CAAC,MAAM,CAAC;IAAA,CAAC;IAEnC,iIAAiI;IACjI,MAAM,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC;QACvC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAG;QAC3B,CAAC,CAAC,OAAO,CAAC;IACZ,MAAM,YAAY,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;IAC9C,IAAI,YAAY,EAAE,CAAC;QAAA,OAAO,YAAY,CAAC,MAAM,CAAC;IAAA,CAAC;IAE/C,uLAAuL;IACvL,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,cAAc,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACtC,6CAA6C;YAC7C,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,yJAAyJ;IACzJ,IAAI,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;QACzE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,mJAAmJ;IACnJ,IACE,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QAClD,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAClD,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC5C,CAAC;IAED,0MAA0M;IAC1M,6EAA6E;IAC7E,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,qJAAqJ;AAErJ;;;;;;;GAOG;AACH,SAAgB,aAAa,CAC3B,gBAAwB,EACxB,MAAsB,EAAE;IAExB,IAAI,CAAC,gBAAgB,IAAI,gBAAgB,KAAK,MAAM,IAAI,gBAAgB,KAAK,UAAU,EAAE,CAAC;QACxF,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,gBAAgB,KAAK,UAAU,IAAI,gBAAgB,KAAK,OAAO,EAAE,CAAC;QACpE,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,+BAA+B;IAC/B,IAAI,cAAc,CAAC,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC1C,MAAM,KAAK,GAAG,gBAAgB,CAAC,KAAK,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;QAC7E,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACxC,OAAO,kBAAkB,OAAO,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,GAAG,CAAC,GAAG,CAAC;IACtD,CAAC;IAED,wEAAwE;IACxE,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;IACxF,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,WAAW,OAAO,CAAC,UAAU,CAAC,CAAC,CAAE,EAAE,GAAG,CAAC,GAAG,CAAC;IACpD,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;IAC7C,OAAO,WAAW,KAAK,GAAG,CAAC;AAC7B,CAAC;AAED,0KAA0K;AAE1K;;GAEG;AACH,SAAgB,eAAe,CAAC,aAAqB;IACnD,IAAI,aAAa,KAAK,MAAM,EAAE,CAAC;QAAA,OAAO,MAAM,CAAC;IAAA,CAAC;IAC9C,IAAI,aAAa,KAAK,MAAM,EAAG,CAAC;QAAA,OAAO,MAAM,CAAC;IAAA,CAAC;IAC/C,IAAI,aAAa,KAAK,OAAO,EAAE,CAAC;QAAA,OAAO,OAAO,CAAC;IAAA,CAAC;IAChD,IAAI,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;QAAA,OAAO,aAAa,CAAC;IAAA,CAAC;IAClE,IAAI,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,aAAa,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACnE,OAAO,aAAa,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,gBAAgB,aAAa,KAAK,CAAC;AAC5C,CAAC;AAED,8LAA8L;AAE9L;;;GAGG;AACH,SAAgB,kBAAkB,CAAC,MAAc;IAC/C,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,OAAO,GAAG,EAAE,CAAC;IAEjB,KAAK,MAAM,EAAE,IAAI,MAAM,EAAE,CAAC;QACxB,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;YACf,KAAK,EAAE,CAAC;YACR,OAAO,IAAI,EAAE,CAAC;QAChB,CAAC;aAAM,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;YACtB,KAAK,EAAE,CAAC;YACR,OAAO,IAAI,EAAE,CAAC;QAChB,CAAC;aAAM,IAAI,EAAE,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAC5B,OAAO,GAAG,EAAE,CAAC;QACf,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QACnB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC9B,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAY;IAClC,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,OAAO,GAAG,EAAE,CAAC;IAEjB,KAAK,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC;QACtB,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;YAAC,KAAK,EAAE,CAAC;YAAC,OAAO,IAAI,EAAE,CAAC;QAAC,CAAC;aACtC,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;YAAC,KAAK,EAAE,CAAC;YAAC,OAAO,IAAI,EAAE,CAAC;QAAC,CAAC;aAC3C,IAAI,EAAE,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAC5B,OAAO,GAAG,EAAE,CAAC;QACf,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAAA,CAAC;IAClD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,+HAA+H;AAE/H;;;;;;;;;;;;GAYG;AACH,MAAa,gBAAgB;IACV,IAAI,CAAiB;IACrB,QAAQ,GAAG,IAAI,GAAG,EAAuB,CAAC;IAE3D,YAAY,MAAsB,EAAE;QAClC,IAAI,CAAC,IAAI,GAAG;YACV,GAAG,GAAG;YACN,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,IAAI,GAAG,EAAE;SAC5C,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,IAAY;QACxB,IAAI,CAAC,IAAI,CAAC,YAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,GAAG,CAAC,MAAc;QAChB,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,iDAAiD;QACjD,IAAI,MAAM,KAAK,iBAAiB,EAAE,CAAC;YACjC,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC;QAC/C,CAAC;aAAM,IAAI,MAAM,KAAK,cAAc,EAAE,CAAC;YACrC,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QAC5C,CAAC;aAAM,IAAI,MAAM,KAAK,kBAAkB,EAAE,CAAC;YACzC,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,OAAO,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED,SAAS,CAAC,MAAc,EAAE,IAAY;QACpC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QACrC,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACvC,CAAC;IAED,mBAAmB;QACjB,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC5C,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AA9CD,4CA8CC"}
@@ -0,0 +1,109 @@
1
+ /**
2
+ * nodepyx — StubCache
3
+ * Manages the on-disk cache of generated TypeScript stub files (.d.ts).
4
+ *
5
+ * Cache layout on disk (default: <project>/.nodepyx/stubs/):
6
+ * <stubsDir>/
7
+ * pandas.d.ts ← generated stub
8
+ * numpy.d.ts
9
+ * …
10
+ * _meta.json ← cache metadata / timestamps
11
+ *
12
+ * A cached stub is considered stale when:
13
+ * 1. The Python library version changes (checked via pip show).
14
+ * 2. nodepyx itself is updated (checked via package version hash).
15
+ * 3. maxAgeMs has elapsed (default: 7 days).
16
+ *
17
+ * The cache falls back gracefully when the disk is unavailable (read-only
18
+ * filesystem, CI environments, etc.).
19
+ */
20
+ export interface StubCacheOptions {
21
+ /** Directory where stub files are stored. Default: './.nodepyx/stubs' */
22
+ stubsDir?: string;
23
+ /** Time-to-live for cached stubs in ms. Default: 7 days */
24
+ maxAgeMs?: number;
25
+ /** Disable the cache entirely (always regenerate). Default: false */
26
+ disabled?: boolean;
27
+ /** nodepyx package version string (used in staleness checks) */
28
+ nodepyxVersion?: string;
29
+ }
30
+ /**
31
+ * StubCache — read / write TypeScript stubs to the on-disk cache.
32
+ *
33
+ * @example
34
+ * ```typescript
35
+ * const cache = new StubCache({ stubsDir: './.nodepyx/stubs' });
36
+ *
37
+ * // Check if a fresh stub exists
38
+ * const hit = cache.get('pandas');
39
+ * if (hit) {
40
+ * console.log(hit); // .d.ts content
41
+ * } else {
42
+ * const dts = await generateStub('pandas');
43
+ * cache.set('pandas', dts, { libVersion: '2.1.0' });
44
+ * }
45
+ * ```
46
+ */
47
+ export declare class StubCache {
48
+ private readonly _stubsDir;
49
+ private readonly _maxAgeMs;
50
+ private readonly _disabled;
51
+ private readonly _nodepyxVersion;
52
+ private _meta;
53
+ private _initialized;
54
+ constructor(options?: StubCacheOptions);
55
+ /**
56
+ * Get cached stub content for a module.
57
+ * Returns null on miss, staleness, or if the cache is disabled.
58
+ */
59
+ get(moduleName: string, libVersion?: string): string | null;
60
+ /**
61
+ * Store a stub in the cache.
62
+ */
63
+ set(moduleName: string, content: string, options?: {
64
+ libVersion?: string;
65
+ }): void;
66
+ /**
67
+ * Remove a single stub from the cache.
68
+ */
69
+ invalidate(moduleName: string): void;
70
+ /**
71
+ * Clear all cached stubs.
72
+ */
73
+ clear(): void;
74
+ /**
75
+ * List all cached module names.
76
+ */
77
+ list(): string[];
78
+ /**
79
+ * Check whether a fresh stub is cached.
80
+ */
81
+ has(moduleName: string, libVersion?: string): boolean;
82
+ /**
83
+ * Return the absolute path to the stub file for a module.
84
+ */
85
+ stubPath(moduleName: string): string;
86
+ /**
87
+ * Return the stubs directory path.
88
+ */
89
+ get stubsDir(): string;
90
+ /**
91
+ * Copy a pre-built stub from the package's bundled stubs directory
92
+ * to the project's stub cache. Pre-built stubs ship inside the nodepyx
93
+ * package for common libraries (pandas, numpy, sklearn, torch).
94
+ */
95
+ copyPrebuilt(moduleName: string): boolean;
96
+ /**
97
+ * Pre-populate the cache with all bundled stubs.
98
+ */
99
+ initPrebuilts(): void;
100
+ private _ensureInit;
101
+ private _ensureDir;
102
+ private _saveMeta;
103
+ private _stubPath;
104
+ private _isStale;
105
+ private _readnodepyxVersion;
106
+ /** djb2 hash — fast, not cryptographic */
107
+ private _hash;
108
+ }
109
+ //# sourceMappingURL=StubCache.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"StubCache.d.ts","sourceRoot":"","sources":["../../src/types/StubCache.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AA2BH,MAAM,WAAW,gBAAgB;IAC/B,yEAAyE;IACzE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,2DAA2D;IAC3D,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,qEAAqE;IACrE,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,gEAAgE;IAChE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAKD;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,SAAS;IACpB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAU;IACpC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,YAAY,CAAS;gBAEjB,OAAO,GAAE,gBAAqB;IAS1C;;;OAGG;IACH,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAwB3D;;OAEG;IACH,GAAG,CACD,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAA;KAAO,GACpC,IAAI;IAwBP;;OAEG;IACH,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAkBpC;;OAEG;IACH,KAAK,IAAI,IAAI;IAmBb;;OAEG;IACH,IAAI,IAAI,MAAM,EAAE;IAUhB;;OAEG;IACH,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO;IAIrD;;OAEG;IACH,QAAQ,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM;IAIpC;;OAEG;IACH,IAAI,QAAQ,IAAI,MAAM,CAErB;IAID;;;;OAIG;IACH,YAAY,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;IAgBzC;;OAEG;IACH,aAAa,IAAI,IAAI;IAWrB,OAAO,CAAC,WAAW;IAkBnB,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,SAAS;IASjB,OAAO,CAAC,SAAS;IAMjB,OAAO,CAAC,QAAQ;IAchB,OAAO,CAAC,mBAAmB;IAa3B,4CAA4C;IAC5C,OAAO,CAAC,KAAK;CAOd"}