@jingyi0605/codingns 0.8.1 → 0.8.5

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 (249) hide show
  1. package/bin/codingns.mjs +180 -0
  2. package/dist/public/assets/{AdaptiveButlerPage-b7s6F_sd.js → AdaptiveButlerPage-CJw8Ae62.js} +3 -3
  3. package/dist/public/assets/App-BOHBGFOd.js +30 -0
  4. package/dist/public/assets/{BootstrapPage-jFfl7KNK.js → BootstrapPage-BxHQT4nA.js} +1 -1
  5. package/dist/public/assets/{ConversationPage-sj_rn4q0.js → ConversationPage-DWFsF6BB.js} +6 -6
  6. package/dist/public/assets/{DesktopDetachPreviewPage-C5c88SOX.js → DesktopDetachPreviewPage-DOgEjYEf.js} +1 -1
  7. package/dist/public/assets/DesktopModal-D_A8sgQU.js +1 -0
  8. package/dist/public/assets/DesktopWindowPage-DK7L7osV.js +2 -0
  9. package/dist/public/assets/FileContextPanel-BdCoubcJ.js +1 -0
  10. package/dist/public/assets/GitSidebar-BeZ0hj7A.js +6 -0
  11. package/dist/public/assets/MobileCreateSessionSheet-DfLMVu8q.js +1 -0
  12. package/dist/public/assets/MobileSheet-5kZ-w-gU.js +1 -0
  13. package/dist/public/assets/{MobileTopHeaderFrame-_I0t7cC_.js → MobileTopHeaderFrame-lcp2GscV.js} +1 -1
  14. package/dist/public/assets/MobileWorkspaceSwitcherHeader-C6JMiOq_.js +1 -0
  15. package/dist/public/assets/PluginAccessOverview-DGcKAMQl.js +1 -0
  16. package/dist/public/assets/PluginContainerPage-BlY-xJDh.js +1 -0
  17. package/dist/public/assets/PluginDetailPage-CAJ7LFpD.js +1 -0
  18. package/dist/public/assets/PluginsListPage-BxZG1NyT.js +1 -0
  19. package/dist/public/assets/{RelayConnectEntryPage-CMtDWyGS.js → RelayConnectEntryPage-CfNO_TIl.js} +1 -1
  20. package/dist/public/assets/ServerSettingsModal-by36Z_5k.js +1 -0
  21. package/dist/public/assets/SessionIndexPage-DkBp9Mqz.js +1 -0
  22. package/dist/public/assets/SettingsPage-C-ASmJAG.js +2 -0
  23. package/dist/public/assets/TerminalManagerPanel-NVZRxxmH.js +1 -0
  24. package/dist/public/assets/{TerminalPage-CBIKHI33.js → TerminalPage-C4LNoPBp.js} +3 -3
  25. package/dist/public/assets/TerminalRuntimeFallbackModal-Bzum5nZ0.js +1 -0
  26. package/dist/public/assets/ToolFilesPage-47zbdgTW.js +1 -0
  27. package/dist/public/assets/ToolGitPage-Fuk_b_jg.js +1 -0
  28. package/dist/public/assets/ToolProcessesPage-sWSMWD-9.js +1 -0
  29. package/dist/public/assets/ToolsHomePage-R1mZlbZi.js +1 -0
  30. package/dist/public/assets/WorkbenchLandingPage-CqmiFH2u.js +1 -0
  31. package/dist/public/assets/WorkbenchLayout-OFi6CWgH.js +244 -0
  32. package/dist/public/assets/WorkbenchModal-C7qoQElW.js +1 -0
  33. package/dist/public/assets/WorkbenchShellRoute-BAQe_E0O.js +1 -0
  34. package/dist/public/assets/WorkspaceDebugDetailPage-DhKa6e9y.js +1 -0
  35. package/dist/public/assets/WorkspaceDetailPage-BPsrFffw.js +1 -0
  36. package/dist/public/assets/WorkspaceHomePage-KAtqZOAb.js +1 -0
  37. package/dist/public/assets/{client-runtime-manager-CdJP7nUp.js → client-runtime-manager-wmCJZKYd.js} +1 -1
  38. package/dist/public/assets/file-tree-icon-Mg1DiBRX.js +590 -0
  39. package/dist/public/assets/{index-VVxfF0s5.css → index-C4t-vvqk.css} +1 -1
  40. package/dist/public/assets/index-CL97fwWB.js +42 -0
  41. package/dist/public/assets/{login-direct-candidate-resolver-C-mo698r.js → login-direct-candidate-resolver-BOAgTuUf.js} +1 -1
  42. package/dist/public/assets/plugin-permission-copy-Cq99cnzV.js +1 -0
  43. package/dist/public/assets/plugins-api-BQTV5DOp.js +1 -0
  44. package/dist/public/assets/{preferences-service-CnaTvZq2.js → preferences-service-DJxbEEeg.js} +1 -1
  45. package/dist/public/assets/realtime-client-CLafKzzJ.js +1 -0
  46. package/dist/public/assets/{relay-entry-D7OgZZN-.js → relay-entry-D-LfvdiX.js} +1 -1
  47. package/dist/public/assets/{terminal-runtime-meta-BCtAYB0E.js → terminal-runtime-meta-BJmy8dyK.js} +1 -1
  48. package/dist/public/assets/useRegisteredDebugTemplates-DQAWVdCo.js +1 -0
  49. package/dist/public/assets/workbench-navigation-MEzCSmsK.js +1 -0
  50. package/dist/public/index.html +2 -2
  51. package/dist/server/config/env.d.ts +1 -0
  52. package/dist/server/config/env.js +4 -0
  53. package/dist/server/config/env.js.map +1 -1
  54. package/dist/server/helpers/wechat-claw-runtime/modules/runtime-state-store.js +1 -1
  55. package/dist/server/helpers/wechat-claw-runtime/modules/runtime-state-store.js.map +1 -1
  56. package/dist/server/modules/channels/wechat-claw-client.d.ts +51 -0
  57. package/dist/server/modules/channels/wechat-claw-client.js +245 -0
  58. package/dist/server/modules/channels/wechat-claw-client.js.map +1 -0
  59. package/dist/server/modules/file/file-access-guard.d.ts +2 -0
  60. package/dist/server/modules/file/file-access-guard.js +24 -2
  61. package/dist/server/modules/file/file-access-guard.js.map +1 -1
  62. package/dist/server/modules/file/file-controller.d.ts +153 -1
  63. package/dist/server/modules/file/file-controller.js +215 -1
  64. package/dist/server/modules/file/file-controller.js.map +1 -1
  65. package/dist/server/modules/file/file-preview-link-service.d.ts +2 -0
  66. package/dist/server/modules/file/file-preview-link-service.js +5 -1
  67. package/dist/server/modules/file/file-preview-link-service.js.map +1 -1
  68. package/dist/server/modules/file/runtime/codingns-workspace-bridge.js +911 -0
  69. package/dist/server/modules/file/workspace-file-bridge-service.d.ts +126 -0
  70. package/dist/server/modules/file/workspace-file-bridge-service.js +471 -0
  71. package/dist/server/modules/file/workspace-file-bridge-service.js.map +1 -0
  72. package/dist/server/modules/file/workspace-file-bridge-watch-service.d.ts +42 -0
  73. package/dist/server/modules/file/workspace-file-bridge-watch-service.js +304 -0
  74. package/dist/server/modules/file/workspace-file-bridge-watch-service.js.map +1 -0
  75. package/dist/server/modules/model-switch/cc-switch-adapter.js +1 -1
  76. package/dist/server/modules/model-switch/cc-switch-adapter.js.map +1 -1
  77. package/dist/server/modules/plugins/plugin-controller.d.ts +131 -0
  78. package/dist/server/modules/plugins/plugin-controller.js +277 -0
  79. package/dist/server/modules/plugins/plugin-controller.js.map +1 -0
  80. package/dist/server/modules/plugins/plugin-file-gateway-service.d.ts +38 -0
  81. package/dist/server/modules/plugins/plugin-file-gateway-service.js +178 -0
  82. package/dist/server/modules/plugins/plugin-file-gateway-service.js.map +1 -0
  83. package/dist/server/modules/plugins/plugin-manifest.d.ts +8 -0
  84. package/dist/server/modules/plugins/plugin-manifest.js +420 -0
  85. package/dist/server/modules/plugins/plugin-manifest.js.map +1 -0
  86. package/dist/server/modules/plugins/plugin-permission-service.d.ts +49 -0
  87. package/dist/server/modules/plugins/plugin-permission-service.js +267 -0
  88. package/dist/server/modules/plugins/plugin-permission-service.js.map +1 -0
  89. package/dist/server/modules/plugins/plugin-process-runner.d.ts +21 -0
  90. package/dist/server/modules/plugins/plugin-process-runner.js +127 -0
  91. package/dist/server/modules/plugins/plugin-process-runner.js.map +1 -0
  92. package/dist/server/modules/plugins/plugin-registry-service.d.ts +41 -0
  93. package/dist/server/modules/plugins/plugin-registry-service.js +238 -0
  94. package/dist/server/modules/plugins/plugin-registry-service.js.map +1 -0
  95. package/dist/server/modules/plugins/plugin-runtime-service.d.ts +55 -0
  96. package/dist/server/modules/plugins/plugin-runtime-service.js +252 -0
  97. package/dist/server/modules/plugins/plugin-runtime-service.js.map +1 -0
  98. package/dist/server/modules/plugins/plugin-runtime-session-service.d.ts +21 -0
  99. package/dist/server/modules/plugins/plugin-runtime-session-service.js +83 -0
  100. package/dist/server/modules/plugins/plugin-runtime-session-service.js.map +1 -0
  101. package/dist/server/modules/plugins/plugin-scheduler-service.d.ts +31 -0
  102. package/dist/server/modules/plugins/plugin-scheduler-service.js +232 -0
  103. package/dist/server/modules/plugins/plugin-scheduler-service.js.map +1 -0
  104. package/dist/server/modules/plugins/plugin-static-service.d.ts +11 -0
  105. package/dist/server/modules/plugins/plugin-static-service.js +145 -0
  106. package/dist/server/modules/plugins/plugin-static-service.js.map +1 -0
  107. package/dist/server/modules/presentation/presentation-controller.d.ts +3 -0
  108. package/dist/server/modules/presentation/presentation-controller.js +11 -0
  109. package/dist/server/modules/presentation/presentation-controller.js.map +1 -1
  110. package/dist/server/modules/presentation/presentation-export-task-service.d.ts +6 -0
  111. package/dist/server/modules/presentation/presentation-export-task-service.js +52 -8
  112. package/dist/server/modules/presentation/presentation-export-task-service.js.map +1 -1
  113. package/dist/server/modules/presentation/presentation-pptx-export-service.js +200 -14
  114. package/dist/server/modules/presentation/presentation-pptx-export-service.js.map +1 -1
  115. package/dist/server/modules/presentation/presentation-renderer.d.ts +50 -0
  116. package/dist/server/modules/presentation/presentation-renderer.js +450 -37
  117. package/dist/server/modules/presentation/presentation-renderer.js.map +1 -1
  118. package/dist/server/modules/tasks/task-types.d.ts +2 -0
  119. package/dist/server/modules/tasks/task-types.js +2 -0
  120. package/dist/server/modules/tasks/task-types.js.map +1 -1
  121. package/dist/server/modules/terminal/runtime/terminal-log-writer-process.js +1 -1
  122. package/dist/server/modules/terminal/runtime/terminal-log-writer-process.js.map +1 -1
  123. package/dist/server/routes/files.js +40 -0
  124. package/dist/server/routes/files.js.map +1 -1
  125. package/dist/server/routes/plugins-public.d.ts +3 -0
  126. package/dist/server/routes/plugins-public.js +5 -0
  127. package/dist/server/routes/plugins-public.js.map +1 -0
  128. package/dist/server/routes/plugins.d.ts +3 -0
  129. package/dist/server/routes/plugins.js +19 -0
  130. package/dist/server/routes/plugins.js.map +1 -0
  131. package/dist/server/routes/presentation.js +1 -0
  132. package/dist/server/routes/presentation.js.map +1 -1
  133. package/dist/server/server/create-server.d.ts +20 -0
  134. package/dist/server/server/create-server.js +47 -1
  135. package/dist/server/server/create-server.js.map +1 -1
  136. package/dist/server/shared/http/error-handler.js +10 -6
  137. package/dist/server/shared/http/error-handler.js.map +1 -1
  138. package/dist/server/shared/runtime/better-sqlite3.d.ts +5 -0
  139. package/dist/server/shared/runtime/better-sqlite3.js +6 -0
  140. package/dist/server/shared/runtime/better-sqlite3.js.map +1 -0
  141. package/dist/server/storage/repositories/plugin-audit-event-repository.d.ts +8 -0
  142. package/dist/server/storage/repositories/plugin-audit-event-repository.js +49 -0
  143. package/dist/server/storage/repositories/plugin-audit-event-repository.js.map +1 -0
  144. package/dist/server/storage/repositories/plugin-definition-repository.d.ts +10 -0
  145. package/dist/server/storage/repositories/plugin-definition-repository.js +84 -0
  146. package/dist/server/storage/repositories/plugin-definition-repository.js.map +1 -0
  147. package/dist/server/storage/repositories/plugin-enablement-repository.d.ts +9 -0
  148. package/dist/server/storage/repositories/plugin-enablement-repository.js +74 -0
  149. package/dist/server/storage/repositories/plugin-enablement-repository.js.map +1 -0
  150. package/dist/server/storage/repositories/plugin-permission-grant-repository.d.ts +12 -0
  151. package/dist/server/storage/repositories/plugin-permission-grant-repository.js +142 -0
  152. package/dist/server/storage/repositories/plugin-permission-grant-repository.js.map +1 -0
  153. package/dist/server/storage/repositories/plugin-run-repository.d.ts +10 -0
  154. package/dist/server/storage/repositories/plugin-run-repository.js +111 -0
  155. package/dist/server/storage/repositories/plugin-run-repository.js.map +1 -0
  156. package/dist/server/storage/repositories/plugin-runtime-session-repository.d.ts +10 -0
  157. package/dist/server/storage/repositories/plugin-runtime-session-repository.js +86 -0
  158. package/dist/server/storage/repositories/plugin-runtime-session-repository.js.map +1 -0
  159. package/dist/server/storage/sqlite/client.d.ts +2 -2
  160. package/dist/server/storage/sqlite/client.js +250 -1
  161. package/dist/server/storage/sqlite/client.js.map +1 -1
  162. package/dist/server/storage/sqlite/schema.sql +140 -0
  163. package/dist/server/types/domain.d.ts +122 -0
  164. package/package.json +10 -3
  165. package/scripts/postinstall.mjs +248 -11
  166. package/vendor/better-sqlite3-win32-x64-node22/LICENSE +21 -0
  167. package/vendor/better-sqlite3-win32-x64-node22/README.md +12 -0
  168. package/vendor/better-sqlite3-win32-x64-node22/README.upstream.md +99 -0
  169. package/vendor/better-sqlite3-win32-x64-node22/binding.gyp +38 -0
  170. package/vendor/better-sqlite3-win32-x64-node22/build/Release/better_sqlite3.node +0 -0
  171. package/vendor/better-sqlite3-win32-x64-node22/deps/common.gypi +68 -0
  172. package/vendor/better-sqlite3-win32-x64-node22/deps/copy.js +31 -0
  173. package/vendor/better-sqlite3-win32-x64-node22/deps/defines.gypi +41 -0
  174. package/vendor/better-sqlite3-win32-x64-node22/deps/download.sh +122 -0
  175. package/vendor/better-sqlite3-win32-x64-node22/deps/patches/1208.patch +15 -0
  176. package/vendor/better-sqlite3-win32-x64-node22/deps/sqlite3/sqlite3.c +265994 -0
  177. package/vendor/better-sqlite3-win32-x64-node22/deps/sqlite3/sqlite3.h +13968 -0
  178. package/vendor/better-sqlite3-win32-x64-node22/deps/sqlite3/sqlite3ext.h +730 -0
  179. package/vendor/better-sqlite3-win32-x64-node22/deps/sqlite3.gyp +80 -0
  180. package/vendor/better-sqlite3-win32-x64-node22/deps/test_extension.c +21 -0
  181. package/vendor/better-sqlite3-win32-x64-node22/lib/database.js +90 -0
  182. package/vendor/better-sqlite3-win32-x64-node22/lib/index.js +3 -0
  183. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/aggregate.js +43 -0
  184. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/backup.js +67 -0
  185. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/function.js +31 -0
  186. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/inspect.js +7 -0
  187. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/pragma.js +12 -0
  188. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/serialize.js +16 -0
  189. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/table.js +189 -0
  190. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/transaction.js +78 -0
  191. package/vendor/better-sqlite3-win32-x64-node22/lib/methods/wrappers.js +54 -0
  192. package/vendor/better-sqlite3-win32-x64-node22/lib/sqlite-error.js +20 -0
  193. package/vendor/better-sqlite3-win32-x64-node22/lib/util.js +12 -0
  194. package/vendor/better-sqlite3-win32-x64-node22/package.json +38 -0
  195. package/vendor/better-sqlite3-win32-x64-node22/scripts/verify-runtime.cjs +42 -0
  196. package/vendor/better-sqlite3-win32-x64-node22/src/addon.cpp +47 -0
  197. package/vendor/better-sqlite3-win32-x64-node22/src/better_sqlite3.cpp +74 -0
  198. package/vendor/better-sqlite3-win32-x64-node22/src/objects/backup.cpp +120 -0
  199. package/vendor/better-sqlite3-win32-x64-node22/src/objects/backup.hpp +36 -0
  200. package/vendor/better-sqlite3-win32-x64-node22/src/objects/database.cpp +417 -0
  201. package/vendor/better-sqlite3-win32-x64-node22/src/objects/database.hpp +103 -0
  202. package/vendor/better-sqlite3-win32-x64-node22/src/objects/statement-iterator.cpp +113 -0
  203. package/vendor/better-sqlite3-win32-x64-node22/src/objects/statement-iterator.hpp +50 -0
  204. package/vendor/better-sqlite3-win32-x64-node22/src/objects/statement.cpp +383 -0
  205. package/vendor/better-sqlite3-win32-x64-node22/src/objects/statement.hpp +58 -0
  206. package/vendor/better-sqlite3-win32-x64-node22/src/util/bind-map.cpp +73 -0
  207. package/vendor/better-sqlite3-win32-x64-node22/src/util/binder.cpp +193 -0
  208. package/vendor/better-sqlite3-win32-x64-node22/src/util/constants.cpp +172 -0
  209. package/vendor/better-sqlite3-win32-x64-node22/src/util/custom-aggregate.cpp +121 -0
  210. package/vendor/better-sqlite3-win32-x64-node22/src/util/custom-function.cpp +59 -0
  211. package/vendor/better-sqlite3-win32-x64-node22/src/util/custom-table.cpp +409 -0
  212. package/vendor/better-sqlite3-win32-x64-node22/src/util/data-converter.cpp +17 -0
  213. package/vendor/better-sqlite3-win32-x64-node22/src/util/data.cpp +194 -0
  214. package/vendor/better-sqlite3-win32-x64-node22/src/util/helpers.cpp +109 -0
  215. package/vendor/better-sqlite3-win32-x64-node22/src/util/macros.cpp +83 -0
  216. package/vendor/better-sqlite3-win32-x64-node22/src/util/query-macros.cpp +71 -0
  217. package/vendor/better-sqlite3-win32-x64-node22/src/util/row-builder.cpp +49 -0
  218. package/vendor/node-pty-fork/README.md +1 -0
  219. package/vendor/node-pty-fork/scripts/runtime-install-context.cjs +24 -0
  220. package/vendor/node-pty-fork/scripts/verify-runtime.cjs +8 -0
  221. package/vendor/node-pty-fork/scripts/verify-runtime.test.cjs +30 -0
  222. package/vendor-src/better-sqlite3-win32-x64-node22/README.md +12 -0
  223. package/dist/public/assets/App-B0q8uDA-.js +0 -30
  224. package/dist/public/assets/DesktopWindowPage-BdKUZrKG.js +0 -2
  225. package/dist/public/assets/FileContextPanel-lM5mBcfn.js +0 -1
  226. package/dist/public/assets/GitSidebar-CmXAhyjo.js +0 -6
  227. package/dist/public/assets/MobileCreateSessionSheet-CceV0-AH.js +0 -1
  228. package/dist/public/assets/MobileWorkspaceSwitcherHeader-DhaM9fgx.js +0 -1
  229. package/dist/public/assets/ServerSettingsModal-Rkgn3jft.js +0 -1
  230. package/dist/public/assets/SessionIndexPage-BOumP3ij.js +0 -1
  231. package/dist/public/assets/SettingsPage-vx_c-00-.js +0 -2
  232. package/dist/public/assets/TerminalManagerPanel-D9JHCS6d.js +0 -1
  233. package/dist/public/assets/TerminalRuntimeFallbackModal-W5NjqFEz.js +0 -1
  234. package/dist/public/assets/ToolFilesPage-DWtpaN_7.js +0 -1
  235. package/dist/public/assets/ToolGitPage-MpGXsw4_.js +0 -1
  236. package/dist/public/assets/ToolProcessesPage-B1jHQKhZ.js +0 -1
  237. package/dist/public/assets/ToolsHomePage-CJrn8ZFl.js +0 -1
  238. package/dist/public/assets/WorkbenchLandingPage-CJQ1V5U-.js +0 -1
  239. package/dist/public/assets/WorkbenchLayout-D48i71PU.js +0 -244
  240. package/dist/public/assets/WorkbenchModal-BYNuMIoV.js +0 -1
  241. package/dist/public/assets/WorkbenchShellRoute-CGKjy4J-.js +0 -1
  242. package/dist/public/assets/WorkspaceDebugDetailPage-DCAYgays.js +0 -1
  243. package/dist/public/assets/WorkspaceDetailPage-D5v-nUIi.js +0 -1
  244. package/dist/public/assets/WorkspaceHomePage-I5G1p5SR.js +0 -1
  245. package/dist/public/assets/file-tree-icon-BFwaPxKs.js +0 -590
  246. package/dist/public/assets/index-CjewzOUG.js +0 -42
  247. package/dist/public/assets/model-switch-api-Br0ivfe5.js +0 -1
  248. package/dist/public/assets/realtime-client-BlQRqVmm.js +0 -1
  249. package/dist/public/assets/useRegisteredDebugTemplates-CXmxu69X.js +0 -1
@@ -0,0 +1,417 @@
1
+ const int Database::MAX_BUFFER_SIZE = (
2
+ node::Buffer::kMaxLength > INT_MAX
3
+ ? INT_MAX
4
+ : static_cast<int>(node::Buffer::kMaxLength)
5
+ );
6
+
7
+ const int Database::MAX_STRING_SIZE = (
8
+ v8::String::kMaxLength > INT_MAX
9
+ ? INT_MAX
10
+ : static_cast<int>(v8::String::kMaxLength)
11
+ );
12
+
13
+ Database::Database(
14
+ v8::Isolate* isolate,
15
+ Addon* addon,
16
+ sqlite3* db_handle,
17
+ v8::Local<v8::Value> logger
18
+ ) :
19
+ node::ObjectWrap(),
20
+ db_handle(db_handle),
21
+ open(true),
22
+ busy(false),
23
+ safe_ints(false),
24
+ unsafe_mode(false),
25
+ was_js_error(false),
26
+ has_logger(logger->IsFunction()),
27
+ iterators(0),
28
+ addon(addon),
29
+ logger(isolate, logger),
30
+ stmts(),
31
+ backups() {
32
+ assert(db_handle != NULL);
33
+ addon->dbs.insert(this);
34
+ }
35
+
36
+ Database::~Database() {
37
+ if (open) addon->dbs.erase(this);
38
+ CloseHandles();
39
+ }
40
+
41
+ // Whenever this is used, addon->dbs.erase() must be invoked beforehand.
42
+ void Database::CloseHandles() {
43
+ if (open) {
44
+ open = false;
45
+ for (Statement* stmt : stmts) stmt->CloseHandles();
46
+ for (Backup* backup : backups) backup->CloseHandles();
47
+ stmts.clear();
48
+ backups.clear();
49
+ int status = sqlite3_close(db_handle);
50
+ assert(status == SQLITE_OK); ((void)status);
51
+ }
52
+ }
53
+
54
+ void Database::ThrowDatabaseError() {
55
+ if (was_js_error) was_js_error = false;
56
+ else ThrowSqliteError(addon, db_handle);
57
+ }
58
+
59
+ void Database::ThrowSqliteError(Addon* addon, sqlite3* db_handle) {
60
+ assert(db_handle != NULL);
61
+ ThrowSqliteError(addon, sqlite3_errmsg(db_handle), sqlite3_extended_errcode(db_handle));
62
+ }
63
+
64
+ void Database::ThrowSqliteError(Addon* addon, const char* message, int code) {
65
+ assert(message != NULL);
66
+ assert((code & 0xff) != SQLITE_OK);
67
+ assert((code & 0xff) != SQLITE_ROW);
68
+ assert((code & 0xff) != SQLITE_DONE);
69
+ EasyIsolate;
70
+ v8::Local<v8::Value> args[2] = {
71
+ StringFromUtf8(isolate, message, -1),
72
+ addon->cs.Code(isolate, code)
73
+ };
74
+ isolate->ThrowException(addon->SqliteError.Get(isolate)
75
+ ->NewInstance(OnlyContext, 2, args)
76
+ .ToLocalChecked());
77
+ }
78
+
79
+ // Allows Statements to log their executed SQL.
80
+ bool Database::Log(v8::Isolate* isolate, sqlite3_stmt* handle) {
81
+ assert(was_js_error == false);
82
+ if (!has_logger) return false;
83
+ char* expanded = sqlite3_expanded_sql(handle);
84
+ v8::Local<v8::Value> arg = StringFromUtf8(isolate, expanded ? expanded : sqlite3_sql(handle), -1);
85
+ was_js_error = logger.Get(isolate).As<v8::Function>()
86
+ ->Call(OnlyContext, v8::Undefined(isolate), 1, &arg)
87
+ .IsEmpty();
88
+ if (expanded) sqlite3_free(expanded);
89
+ return was_js_error;
90
+ }
91
+
92
+ bool Database::Deserialize(
93
+ v8::Local<v8::Object> buffer,
94
+ Addon* addon,
95
+ sqlite3* db_handle,
96
+ bool readonly
97
+ ) {
98
+ size_t length = node::Buffer::Length(buffer);
99
+ unsigned char* data = (unsigned char*)sqlite3_malloc64(length);
100
+ unsigned int flags = SQLITE_DESERIALIZE_FREEONCLOSE | SQLITE_DESERIALIZE_RESIZEABLE;
101
+
102
+ if (readonly) {
103
+ flags |= SQLITE_DESERIALIZE_READONLY;
104
+ }
105
+ if (length) {
106
+ if (!data) {
107
+ ThrowError("Out of memory");
108
+ return false;
109
+ }
110
+ memcpy(data, node::Buffer::Data(buffer), length);
111
+ }
112
+
113
+ int status = sqlite3_deserialize(db_handle, "main", data, length, length, flags);
114
+ if (status != SQLITE_OK) {
115
+ ThrowSqliteError(addon, status == SQLITE_ERROR ? "unable to deserialize database" : sqlite3_errstr(status), status);
116
+ return false;
117
+ }
118
+
119
+ return true;
120
+ }
121
+
122
+ void Database::FreeSerialization(char* data, void* _) {
123
+ sqlite3_free(data);
124
+ }
125
+
126
+ INIT(Database::Init) {
127
+ v8::Local<v8::FunctionTemplate> t = NewConstructorTemplate(isolate, data, JS_new, "Database");
128
+ SetPrototypeMethod(isolate, data, t, "prepare", JS_prepare);
129
+ SetPrototypeMethod(isolate, data, t, "exec", JS_exec);
130
+ SetPrototypeMethod(isolate, data, t, "backup", JS_backup);
131
+ SetPrototypeMethod(isolate, data, t, "serialize", JS_serialize);
132
+ SetPrototypeMethod(isolate, data, t, "function", JS_function);
133
+ SetPrototypeMethod(isolate, data, t, "aggregate", JS_aggregate);
134
+ SetPrototypeMethod(isolate, data, t, "table", JS_table);
135
+ SetPrototypeMethod(isolate, data, t, "loadExtension", JS_loadExtension);
136
+ SetPrototypeMethod(isolate, data, t, "close", JS_close);
137
+ SetPrototypeMethod(isolate, data, t, "defaultSafeIntegers", JS_defaultSafeIntegers);
138
+ SetPrototypeMethod(isolate, data, t, "unsafeMode", JS_unsafeMode);
139
+ SetPrototypeGetter(isolate, data, t, "open", JS_open);
140
+ SetPrototypeGetter(isolate, data, t, "inTransaction", JS_inTransaction);
141
+ return t->GetFunction(OnlyContext).ToLocalChecked();
142
+ }
143
+
144
+ NODE_METHOD(Database::JS_new) {
145
+ assert(info.IsConstructCall());
146
+ REQUIRE_ARGUMENT_STRING(first, v8::Local<v8::String> filename);
147
+ REQUIRE_ARGUMENT_STRING(second, v8::Local<v8::String> filenameGiven);
148
+ REQUIRE_ARGUMENT_BOOLEAN(third, bool in_memory);
149
+ REQUIRE_ARGUMENT_BOOLEAN(fourth, bool readonly);
150
+ REQUIRE_ARGUMENT_BOOLEAN(fifth, bool must_exist);
151
+ REQUIRE_ARGUMENT_INT32(sixth, int timeout);
152
+ REQUIRE_ARGUMENT_ANY(seventh, v8::Local<v8::Value> logger);
153
+ REQUIRE_ARGUMENT_ANY(eighth, v8::Local<v8::Value> buffer);
154
+
155
+ UseAddon;
156
+ UseIsolate;
157
+ sqlite3* db_handle;
158
+ v8::String::Utf8Value utf8(isolate, filename);
159
+ int mask = readonly ? SQLITE_OPEN_READONLY
160
+ : must_exist ? SQLITE_OPEN_READWRITE
161
+ : (SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE);
162
+
163
+ if (sqlite3_open_v2(*utf8, &db_handle, mask, NULL) != SQLITE_OK) {
164
+ ThrowSqliteError(addon, db_handle);
165
+ int status = sqlite3_close(db_handle);
166
+ assert(status == SQLITE_OK); ((void)status);
167
+ return;
168
+ }
169
+
170
+ assert(sqlite3_db_mutex(db_handle) == NULL);
171
+ sqlite3_extended_result_codes(db_handle, 1);
172
+ sqlite3_busy_timeout(db_handle, timeout);
173
+ sqlite3_limit(db_handle, SQLITE_LIMIT_LENGTH, MAX_BUFFER_SIZE < MAX_STRING_SIZE ? MAX_BUFFER_SIZE : MAX_STRING_SIZE);
174
+ sqlite3_limit(db_handle, SQLITE_LIMIT_SQL_LENGTH, MAX_STRING_SIZE);
175
+ int status = sqlite3_db_config(db_handle, SQLITE_DBCONFIG_ENABLE_LOAD_EXTENSION, 1, NULL);
176
+ assert(status == SQLITE_OK); ((void)status);
177
+ status = sqlite3_db_config(db_handle, SQLITE_DBCONFIG_DEFENSIVE, 1, NULL);
178
+ assert(status == SQLITE_OK); ((void)status);
179
+
180
+ if (node::Buffer::HasInstance(buffer) && !Deserialize(buffer.As<v8::Object>(), addon, db_handle, readonly)) {
181
+ int status = sqlite3_close(db_handle);
182
+ assert(status == SQLITE_OK); ((void)status);
183
+ return;
184
+ }
185
+
186
+ UseContext;
187
+ Database* db = new Database(isolate, addon, db_handle, logger);
188
+ db->Wrap(info.This());
189
+ SetFrozen(isolate, ctx, info.This(), addon->cs.memory, v8::Boolean::New(isolate, in_memory));
190
+ SetFrozen(isolate, ctx, info.This(), addon->cs.readonly, v8::Boolean::New(isolate, readonly));
191
+ SetFrozen(isolate, ctx, info.This(), addon->cs.name, filenameGiven);
192
+
193
+ info.GetReturnValue().Set(info.This());
194
+ }
195
+
196
+ NODE_METHOD(Database::JS_prepare) {
197
+ REQUIRE_ARGUMENT_STRING(first, v8::Local<v8::String> source);
198
+ REQUIRE_ARGUMENT_OBJECT(second, v8::Local<v8::Object> database);
199
+ REQUIRE_ARGUMENT_BOOLEAN(third, bool pragmaMode);
200
+ (void)source;
201
+ (void)database;
202
+ (void)pragmaMode;
203
+ UseAddon;
204
+ UseIsolate;
205
+ v8::Local<v8::Function> c = addon->Statement.Get(isolate);
206
+ addon->privileged_info = &info;
207
+ v8::MaybeLocal<v8::Object> maybeStatement = c->NewInstance(OnlyContext, 0, NULL);
208
+ addon->privileged_info = NULL;
209
+ if (!maybeStatement.IsEmpty()) info.GetReturnValue().Set(maybeStatement.ToLocalChecked());
210
+ }
211
+
212
+ NODE_METHOD(Database::JS_exec) {
213
+ Database* db = Unwrap<Database>(info.This());
214
+ REQUIRE_ARGUMENT_STRING(first, v8::Local<v8::String> source);
215
+ REQUIRE_DATABASE_OPEN(db);
216
+ REQUIRE_DATABASE_NOT_BUSY(db);
217
+ REQUIRE_DATABASE_NO_ITERATORS_UNLESS_UNSAFE(db);
218
+ db->busy = true;
219
+
220
+ UseIsolate;
221
+ v8::String::Utf8Value utf8(isolate, source);
222
+ const char* sql = *utf8;
223
+ const char* tail;
224
+
225
+ int status;
226
+ const bool has_logger = db->has_logger;
227
+ sqlite3* const db_handle = db->db_handle;
228
+ sqlite3_stmt* handle;
229
+
230
+ for (;;) {
231
+ while (IS_SKIPPED(*sql)) ++sql;
232
+ status = sqlite3_prepare_v2(db_handle, sql, -1, &handle, &tail);
233
+ sql = tail;
234
+ if (!handle) break;
235
+ if (has_logger && db->Log(isolate, handle)) {
236
+ sqlite3_finalize(handle);
237
+ status = -1;
238
+ break;
239
+ }
240
+ do status = sqlite3_step(handle);
241
+ while (status == SQLITE_ROW);
242
+ status = sqlite3_finalize(handle);
243
+ if (status != SQLITE_OK) break;
244
+ }
245
+
246
+ db->busy = false;
247
+ if (status != SQLITE_OK) {
248
+ db->ThrowDatabaseError();
249
+ }
250
+ }
251
+
252
+ NODE_METHOD(Database::JS_backup) {
253
+ REQUIRE_ARGUMENT_OBJECT(first, v8::Local<v8::Object> database);
254
+ REQUIRE_ARGUMENT_STRING(second, v8::Local<v8::String> attachedName);
255
+ REQUIRE_ARGUMENT_STRING(third, v8::Local<v8::String> destFile);
256
+ REQUIRE_ARGUMENT_BOOLEAN(fourth, bool unlink);
257
+ (void)database;
258
+ (void)attachedName;
259
+ (void)destFile;
260
+ (void)unlink;
261
+ UseAddon;
262
+ UseIsolate;
263
+ v8::Local<v8::Function> c = addon->Backup.Get(isolate);
264
+ addon->privileged_info = &info;
265
+ v8::MaybeLocal<v8::Object> maybeBackup = c->NewInstance(OnlyContext, 0, NULL);
266
+ addon->privileged_info = NULL;
267
+ if (!maybeBackup.IsEmpty()) info.GetReturnValue().Set(maybeBackup.ToLocalChecked());
268
+ }
269
+
270
+ NODE_METHOD(Database::JS_serialize) {
271
+ Database* db = Unwrap<Database>(info.This());
272
+ REQUIRE_ARGUMENT_STRING(first, v8::Local<v8::String> attachedName);
273
+ REQUIRE_DATABASE_OPEN(db);
274
+ REQUIRE_DATABASE_NOT_BUSY(db);
275
+ REQUIRE_DATABASE_NO_ITERATORS(db);
276
+
277
+ UseIsolate;
278
+ v8::String::Utf8Value attached_name(isolate, attachedName);
279
+ sqlite3_int64 length = -1;
280
+ unsigned char* data = sqlite3_serialize(db->db_handle, *attached_name, &length, 0);
281
+
282
+ if (!data && length) {
283
+ ThrowError("Out of memory");
284
+ return;
285
+ }
286
+
287
+ info.GetReturnValue().Set(
288
+ SAFE_NEW_BUFFER(isolate, reinterpret_cast<char*>(data), length, FreeSerialization, NULL).ToLocalChecked()
289
+ );
290
+ }
291
+
292
+ NODE_METHOD(Database::JS_function) {
293
+ Database* db = Unwrap<Database>(info.This());
294
+ REQUIRE_ARGUMENT_FUNCTION(first, v8::Local<v8::Function> fn);
295
+ REQUIRE_ARGUMENT_STRING(second, v8::Local<v8::String> nameString);
296
+ REQUIRE_ARGUMENT_INT32(third, int argc);
297
+ REQUIRE_ARGUMENT_INT32(fourth, int safe_ints);
298
+ REQUIRE_ARGUMENT_BOOLEAN(fifth, bool deterministic);
299
+ REQUIRE_ARGUMENT_BOOLEAN(sixth, bool direct_only);
300
+ REQUIRE_DATABASE_OPEN(db);
301
+ REQUIRE_DATABASE_NOT_BUSY(db);
302
+ REQUIRE_DATABASE_NO_ITERATORS(db);
303
+
304
+ UseIsolate;
305
+ v8::String::Utf8Value name(isolate, nameString);
306
+ int mask = SQLITE_UTF8;
307
+ if (deterministic) mask |= SQLITE_DETERMINISTIC;
308
+ if (direct_only) mask |= SQLITE_DIRECTONLY;
309
+ safe_ints = safe_ints < 2 ? safe_ints : static_cast<int>(db->safe_ints);
310
+
311
+ if (sqlite3_create_function_v2(db->db_handle, *name, argc, mask, new CustomFunction(isolate, db, *name, fn, safe_ints), CustomFunction::xFunc, NULL, NULL, CustomFunction::xDestroy) != SQLITE_OK) {
312
+ db->ThrowDatabaseError();
313
+ }
314
+ }
315
+
316
+ NODE_METHOD(Database::JS_aggregate) {
317
+ Database* db = Unwrap<Database>(info.This());
318
+ REQUIRE_ARGUMENT_ANY(first, v8::Local<v8::Value> start);
319
+ REQUIRE_ARGUMENT_FUNCTION(second, v8::Local<v8::Function> step);
320
+ REQUIRE_ARGUMENT_ANY(third, v8::Local<v8::Value> inverse);
321
+ REQUIRE_ARGUMENT_ANY(fourth, v8::Local<v8::Value> result);
322
+ REQUIRE_ARGUMENT_STRING(fifth, v8::Local<v8::String> nameString);
323
+ REQUIRE_ARGUMENT_INT32(sixth, int argc);
324
+ REQUIRE_ARGUMENT_INT32(seventh, int safe_ints);
325
+ REQUIRE_ARGUMENT_BOOLEAN(eighth, bool deterministic);
326
+ REQUIRE_ARGUMENT_BOOLEAN(ninth, bool direct_only);
327
+ REQUIRE_DATABASE_OPEN(db);
328
+ REQUIRE_DATABASE_NOT_BUSY(db);
329
+ REQUIRE_DATABASE_NO_ITERATORS(db);
330
+
331
+ UseIsolate;
332
+ v8::String::Utf8Value name(isolate, nameString);
333
+ auto xInverse = inverse->IsFunction() ? CustomAggregate::xInverse : NULL;
334
+ auto xValue = xInverse ? CustomAggregate::xValue : NULL;
335
+ int mask = SQLITE_UTF8;
336
+ if (deterministic) mask |= SQLITE_DETERMINISTIC;
337
+ if (direct_only) mask |= SQLITE_DIRECTONLY;
338
+ safe_ints = safe_ints < 2 ? safe_ints : static_cast<int>(db->safe_ints);
339
+
340
+ if (sqlite3_create_window_function(db->db_handle, *name, argc, mask, new CustomAggregate(isolate, db, *name, start, step, inverse, result, safe_ints), CustomAggregate::xStep, CustomAggregate::xFinal, xValue, xInverse, CustomAggregate::xDestroy) != SQLITE_OK) {
341
+ db->ThrowDatabaseError();
342
+ }
343
+ }
344
+
345
+ NODE_METHOD(Database::JS_table) {
346
+ Database* db = Unwrap<Database>(info.This());
347
+ REQUIRE_ARGUMENT_FUNCTION(first, v8::Local<v8::Function> factory);
348
+ REQUIRE_ARGUMENT_STRING(second, v8::Local<v8::String> nameString);
349
+ REQUIRE_ARGUMENT_BOOLEAN(third, bool eponymous);
350
+ REQUIRE_DATABASE_OPEN(db);
351
+ REQUIRE_DATABASE_NOT_BUSY(db);
352
+ REQUIRE_DATABASE_NO_ITERATORS(db);
353
+
354
+ UseIsolate;
355
+ v8::String::Utf8Value name(isolate, nameString);
356
+ sqlite3_module* module = eponymous ? &CustomTable::EPONYMOUS_MODULE : &CustomTable::MODULE;
357
+
358
+ db->busy = true;
359
+ if (sqlite3_create_module_v2(db->db_handle, *name, module, new CustomTable(isolate, db, *name, factory), CustomTable::Destructor) != SQLITE_OK) {
360
+ db->ThrowDatabaseError();
361
+ }
362
+ db->busy = false;
363
+ }
364
+
365
+ NODE_METHOD(Database::JS_loadExtension) {
366
+ Database* db = Unwrap<Database>(info.This());
367
+ v8::Local<v8::String> entryPoint;
368
+ REQUIRE_ARGUMENT_STRING(first, v8::Local<v8::String> filename);
369
+ if (info.Length() > 1) { REQUIRE_ARGUMENT_STRING(second, entryPoint); }
370
+ REQUIRE_DATABASE_OPEN(db);
371
+ REQUIRE_DATABASE_NOT_BUSY(db);
372
+ REQUIRE_DATABASE_NO_ITERATORS(db);
373
+ UseIsolate;
374
+ char* error;
375
+ int status = sqlite3_load_extension(
376
+ db->db_handle,
377
+ *v8::String::Utf8Value(isolate, filename),
378
+ entryPoint.IsEmpty() ? NULL : *v8::String::Utf8Value(isolate, entryPoint),
379
+ &error
380
+ );
381
+ if (status != SQLITE_OK) {
382
+ ThrowSqliteError(db->addon, error, status);
383
+ }
384
+ sqlite3_free(error);
385
+ }
386
+
387
+ NODE_METHOD(Database::JS_close) {
388
+ Database* db = Unwrap<Database>(info.This());
389
+ if (db->open) {
390
+ REQUIRE_DATABASE_NOT_BUSY(db);
391
+ REQUIRE_DATABASE_NO_ITERATORS(db);
392
+ db->addon->dbs.erase(db);
393
+ db->CloseHandles();
394
+ }
395
+ }
396
+
397
+ NODE_METHOD(Database::JS_defaultSafeIntegers) {
398
+ Database* db = Unwrap<Database>(info.This());
399
+ if (info.Length() == 0) db->safe_ints = true;
400
+ else { REQUIRE_ARGUMENT_BOOLEAN(first, db->safe_ints); }
401
+ }
402
+
403
+ NODE_METHOD(Database::JS_unsafeMode) {
404
+ Database* db = Unwrap<Database>(info.This());
405
+ if (info.Length() == 0) db->unsafe_mode = true;
406
+ else { REQUIRE_ARGUMENT_BOOLEAN(first, db->unsafe_mode); }
407
+ sqlite3_db_config(db->db_handle, SQLITE_DBCONFIG_DEFENSIVE, static_cast<int>(!db->unsafe_mode), NULL);
408
+ }
409
+
410
+ NODE_GETTER(Database::JS_open) {
411
+ info.GetReturnValue().Set(Unwrap<Database>(PROPERTY_HOLDER(info))->open);
412
+ }
413
+
414
+ NODE_GETTER(Database::JS_inTransaction) {
415
+ Database* db = Unwrap<Database>(PROPERTY_HOLDER(info));
416
+ info.GetReturnValue().Set(db->open && !static_cast<bool>(sqlite3_get_autocommit(db->db_handle)));
417
+ }
@@ -0,0 +1,103 @@
1
+ class Database : public node::ObjectWrap {
2
+ public:
3
+
4
+ ~Database();
5
+
6
+ // Whenever this is used, addon->dbs.erase() must be invoked beforehand.
7
+ void CloseHandles();
8
+
9
+ // Used to support ordered containers.
10
+ class CompareDatabase { public:
11
+ inline bool operator() (Database const * const a, Database const * const b) const {
12
+ return a < b;
13
+ }
14
+ };
15
+ class CompareStatement { public:
16
+ inline bool operator() (Statement const * const a, Statement const * const b) const {
17
+ return Statement::Compare(a, b);
18
+ }
19
+ };
20
+ class CompareBackup { public:
21
+ inline bool operator() (Backup const * const a, Backup const * const b) const {
22
+ return Backup::Compare(a, b);
23
+ }
24
+ };
25
+
26
+ // Proper error handling logic for when an sqlite3 operation fails.
27
+ void ThrowDatabaseError();
28
+ static void ThrowSqliteError(Addon* addon, sqlite3* db_handle);
29
+ static void ThrowSqliteError(Addon* addon, const char* message, int code);
30
+
31
+ // Allows Statements to log their executed SQL.
32
+ bool Log(v8::Isolate* isolate, sqlite3_stmt* handle);
33
+
34
+ // Allow Statements to manage themselves when created and garbage collected.
35
+ inline void AddStatement(Statement* stmt) { stmts.insert(stmts.end(), stmt); }
36
+ inline void RemoveStatement(Statement* stmt) { stmts.erase(stmt); }
37
+
38
+ // Allow Backups to manage themselves when created and garbage collected.
39
+ inline void AddBackup(Backup* backup) { backups.insert(backups.end(), backup); }
40
+ inline void RemoveBackup(Backup* backup) { backups.erase(backup); }
41
+
42
+ // A view for Statements to see and modify Database state.
43
+ // The order of these fields must exactly match their actual order.
44
+ struct State {
45
+ const bool open;
46
+ bool busy;
47
+ const bool safe_ints;
48
+ const bool unsafe_mode;
49
+ bool was_js_error;
50
+ const bool has_logger;
51
+ unsigned short iterators;
52
+ Addon* const addon;
53
+ };
54
+
55
+ inline State* GetState() { return reinterpret_cast<State*>(&open); }
56
+ inline sqlite3* GetHandle() { return db_handle; }
57
+ inline Addon* GetAddon() { return addon; }
58
+
59
+ static INIT(Init);
60
+
61
+ private:
62
+
63
+ explicit Database(
64
+ v8::Isolate* isolate,
65
+ Addon* addon,
66
+ sqlite3* db_handle,
67
+ v8::Local<v8::Value> logger
68
+ );
69
+
70
+ static NODE_METHOD(JS_new);
71
+ static NODE_METHOD(JS_prepare);
72
+ static NODE_METHOD(JS_exec);
73
+ static NODE_METHOD(JS_backup);
74
+ static NODE_METHOD(JS_serialize);
75
+ static NODE_METHOD(JS_function);
76
+ static NODE_METHOD(JS_aggregate);
77
+ static NODE_METHOD(JS_table);
78
+ static NODE_METHOD(JS_loadExtension);
79
+ static NODE_METHOD(JS_close);
80
+ static NODE_METHOD(JS_defaultSafeIntegers);
81
+ static NODE_METHOD(JS_unsafeMode);
82
+ static NODE_GETTER(JS_open);
83
+ static NODE_GETTER(JS_inTransaction);
84
+
85
+ static bool Deserialize(v8::Local<v8::Object> buffer, Addon* addon, sqlite3* db_handle, bool readonly);
86
+ static void FreeSerialization(char* data, void* _);
87
+
88
+ static const int MAX_BUFFER_SIZE;
89
+ static const int MAX_STRING_SIZE;
90
+
91
+ sqlite3* const db_handle;
92
+ bool open;
93
+ bool busy;
94
+ bool safe_ints;
95
+ bool unsafe_mode;
96
+ bool was_js_error;
97
+ const bool has_logger;
98
+ unsigned short iterators;
99
+ Addon* const addon;
100
+ const v8::Global<v8::Value> logger;
101
+ std::set<Statement*, CompareStatement> stmts;
102
+ std::set<Backup*, CompareBackup> backups;
103
+ };
@@ -0,0 +1,113 @@
1
+ StatementIterator::StatementIterator(Statement* stmt, bool bound) :
2
+ node::ObjectWrap(),
3
+ stmt(stmt),
4
+ handle(stmt->handle),
5
+ db_state(stmt->db->GetState()),
6
+ bound(bound),
7
+ safe_ints(stmt->safe_ints),
8
+ mode(stmt->mode),
9
+ alive(true),
10
+ logged(!db_state->has_logger) {
11
+ assert(stmt != NULL);
12
+ assert(handle != NULL);
13
+ assert(stmt->bound == bound);
14
+ assert(stmt->alive == true);
15
+ assert(stmt->locked == false);
16
+ assert(db_state->iterators < USHRT_MAX);
17
+ stmt->locked = true;
18
+ db_state->iterators += 1;
19
+ }
20
+
21
+ // The ~Statement destructor currently covers any state this object creates.
22
+ // Additionally, we actually DON'T want to revert stmt->locked or db_state
23
+ // ->iterators in this destructor, to ensure deterministic database access.
24
+ StatementIterator::~StatementIterator() {}
25
+
26
+ void StatementIterator::Next(NODE_ARGUMENTS info) {
27
+ assert(alive == true);
28
+ db_state->busy = true;
29
+ if (!logged) {
30
+ logged = true;
31
+ if (stmt->db->Log(OnlyIsolate, handle)) {
32
+ db_state->busy = false;
33
+ Throw();
34
+ return;
35
+ }
36
+ }
37
+ int status = sqlite3_step(handle);
38
+ db_state->busy = false;
39
+ if (status == SQLITE_ROW) {
40
+ UseIsolate;
41
+ UseContext;
42
+ info.GetReturnValue().Set(
43
+ NewRecord(isolate, ctx, Data::GetRowJS(isolate, ctx, handle, safe_ints, mode), db_state->addon, false)
44
+ );
45
+ } else {
46
+ if (status == SQLITE_DONE) Return(info);
47
+ else Throw();
48
+ }
49
+ }
50
+
51
+ void StatementIterator::Return(NODE_ARGUMENTS info) {
52
+ Cleanup();
53
+ STATEMENT_RETURN_LOGIC(DoneRecord(OnlyIsolate, db_state->addon));
54
+ }
55
+
56
+ void StatementIterator::Throw() {
57
+ Cleanup();
58
+ Database* db = stmt->db;
59
+ STATEMENT_THROW_LOGIC();
60
+ }
61
+
62
+ void StatementIterator::Cleanup() {
63
+ assert(alive == true);
64
+ alive = false;
65
+ stmt->locked = false;
66
+ db_state->iterators -= 1;
67
+ sqlite3_reset(handle);
68
+ }
69
+
70
+ INIT(StatementIterator::Init) {
71
+ v8::Local<v8::FunctionTemplate> t = NewConstructorTemplate(isolate, data, JS_new, "StatementIterator");
72
+ SetPrototypeMethod(isolate, data, t, "next", JS_next);
73
+ SetPrototypeMethod(isolate, data, t, "return", JS_return);
74
+ SetPrototypeSymbolMethod(isolate, data, t, v8::Symbol::GetIterator(isolate), JS_symbolIterator);
75
+ return t->GetFunction(OnlyContext).ToLocalChecked();
76
+ }
77
+
78
+ NODE_METHOD(StatementIterator::JS_new) {
79
+ UseAddon;
80
+ if (!addon->privileged_info) return ThrowTypeError("Disabled constructor");
81
+ assert(info.IsConstructCall());
82
+
83
+ StatementIterator* iter;
84
+ {
85
+ NODE_ARGUMENTS info = *addon->privileged_info;
86
+ STATEMENT_START_LOGIC(REQUIRE_STATEMENT_RETURNS_DATA, DOES_ADD_ITERATOR);
87
+ iter = new StatementIterator(stmt, bound);
88
+ }
89
+ UseIsolate;
90
+ UseContext;
91
+ iter->Wrap(info.This());
92
+ SetFrozen(isolate, ctx, info.This(), addon->cs.statement, addon->privileged_info->This());
93
+
94
+ info.GetReturnValue().Set(info.This());
95
+ }
96
+
97
+ NODE_METHOD(StatementIterator::JS_next) {
98
+ StatementIterator* iter = Unwrap<StatementIterator>(info.This());
99
+ REQUIRE_DATABASE_NOT_BUSY(iter->db_state);
100
+ if (iter->alive) iter->Next(info);
101
+ else info.GetReturnValue().Set(DoneRecord(OnlyIsolate, iter->db_state->addon));
102
+ }
103
+
104
+ NODE_METHOD(StatementIterator::JS_return) {
105
+ StatementIterator* iter = Unwrap<StatementIterator>(info.This());
106
+ REQUIRE_DATABASE_NOT_BUSY(iter->db_state);
107
+ if (iter->alive) iter->Return(info);
108
+ else info.GetReturnValue().Set(DoneRecord(OnlyIsolate, iter->db_state->addon));
109
+ }
110
+
111
+ NODE_METHOD(StatementIterator::JS_symbolIterator) {
112
+ info.GetReturnValue().Set(info.This());
113
+ }
@@ -0,0 +1,50 @@
1
+ class StatementIterator : public node::ObjectWrap {
2
+ public:
3
+
4
+ // The ~Statement destructor currently covers any state this object creates.
5
+ // Additionally, we actually DON'T want to revert stmt->locked or db_state
6
+ // ->iterators in this destructor, to ensure deterministic database access.
7
+ ~StatementIterator();
8
+
9
+ static INIT(Init);
10
+
11
+ private:
12
+
13
+ explicit StatementIterator(Statement* stmt, bool bound);
14
+
15
+ void Next(NODE_ARGUMENTS info);
16
+ void Return(NODE_ARGUMENTS info);
17
+ void Throw();
18
+ void Cleanup();
19
+
20
+ static inline v8::Local<v8::Object> NewRecord(
21
+ v8::Isolate* isolate,
22
+ v8::Local<v8::Context> ctx,
23
+ v8::Local<v8::Value> value,
24
+ Addon* addon,
25
+ bool done
26
+ ) {
27
+ v8::Local<v8::Object> record = v8::Object::New(isolate);
28
+ record->Set(ctx, addon->cs.value.Get(isolate), value).FromJust();
29
+ record->Set(ctx, addon->cs.done.Get(isolate), v8::Boolean::New(isolate, done)).FromJust();
30
+ return record;
31
+ }
32
+
33
+ static inline v8::Local<v8::Object> DoneRecord(v8::Isolate* isolate, Addon* addon) {
34
+ return NewRecord(isolate, OnlyContext, v8::Undefined(isolate), addon, true);
35
+ }
36
+
37
+ static NODE_METHOD(JS_new);
38
+ static NODE_METHOD(JS_next);
39
+ static NODE_METHOD(JS_return);
40
+ static NODE_METHOD(JS_symbolIterator);
41
+
42
+ Statement* const stmt;
43
+ sqlite3_stmt* const handle;
44
+ Database::State* const db_state;
45
+ const bool bound;
46
+ const bool safe_ints;
47
+ const char mode;
48
+ bool alive;
49
+ bool logged;
50
+ };