@marimo-team/islands 0.25.2-dev97 → 0.25.2
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.
- package/dist/{ConnectedDataExplorerComponent-BZmqc_F0.js → ConnectedDataExplorerComponent-CNSxdKyJ.js} +15 -15
- package/dist/{ErrorBoundary-B-t5rRT6.js → ErrorBoundary-HusVNio-.js} +12 -12
- package/dist/{ImageComparisonComponent-Cpfz30OZ.js → ImageComparisonComponent-uU8FSKZE.js} +3 -3
- package/dist/{Plot-CTyjbQDT.js → Plot-DdYOe1UK.js} +2 -2
- package/dist/{abnfDiagram-VCTEODGH-C_Lmaa2j.js → abnfDiagram-VCTEODGH-DQDjNXt3.js} +3 -3
- package/dist/{any-language-editor-BFzqhOBR.js → any-language-editor-B9REGlD9.js} +7 -7
- package/dist/{architectureDiagram-5GKGNRK7-BKMataHB.js → architectureDiagram-5GKGNRK7-Nmls7Wxn.js} +5 -5
- package/dist/assets/__vite-browser-external-BXYkiFsk.js +1 -0
- package/dist/assets/worker-HPY9fbhz.js +114 -0
- package/dist/{blockDiagram-I7D4REHJ-BeHOjyDk.js → blockDiagram-I7D4REHJ-Cg25UrxT.js} +9 -9
- package/dist/{button-BtsM2UqM.js → button-CvjkwdSk.js} +5 -5
- package/dist/{c4Diagram-7LVT6UL2-Bao10t_O.js → c4Diagram-7LVT6UL2-BxFgrnNV.js} +7 -7
- package/dist/{capabilities-A5jrzjCJ.js → capabilities-CQ5IjhzL.js} +1 -1
- package/dist/{cells-DZi0ioak.js → cells-C8iCYvLU.js} +6644 -5604
- package/dist/{channel-BxCkvUgE.js → channel-CPObtJXj.js} +1 -1
- package/dist/{chat-ui-CwOWzoSO.js → chat-ui-BDpymeoe.js} +1589 -1396
- package/dist/{check-CNP79M3A.js → check-EDwSwP89.js} +1 -1
- package/dist/{chunk-4HAMMTFA-CH7Gpamd.js → chunk-4HAMMTFA-D4eFPyFh.js} +5 -5
- package/dist/{chunk-75Z2AOVW-AI7CvgKn.js → chunk-75Z2AOVW-D7gxIs2Q.js} +1 -1
- package/dist/{chunk-DU6HZSFF-BSvuTkX4.js → chunk-DU6HZSFF-BwiyaNpg.js} +1 -1
- package/dist/{chunk-F27PBJKO-CLshwT9T.js → chunk-F27PBJKO-ufseIn5y.js} +1 -1
- package/dist/{chunk-GMAD6QVW-wnHcn-lH.js → chunk-GMAD6QVW-BhaJx0X3.js} +3 -3
- package/dist/{chunk-GVQU2GXP-7O2p8hut.js → chunk-GVQU2GXP-Cwbt-lF0.js} +2 -2
- package/dist/{chunk-IMKFNOWR-Bu0Vh_Yi.js → chunk-IMKFNOWR-CoIFEo-1.js} +5 -5
- package/dist/{chunk-L3NEJ4N5-B9_Alrxb.js → chunk-L3NEJ4N5-n-zgIVSy.js} +5 -5
- package/dist/{chunk-OSK3NFVY-PeQPH2fD.js → chunk-OSK3NFVY-DXa-Cxbn.js} +6 -6
- package/dist/{chunk-P2QGCYS3-D9twiFuU.js → chunk-P2QGCYS3-B_Zd3-X0.js} +1 -1
- package/dist/{chunk-POPQ4Y6H-BUiNs18s.js → chunk-POPQ4Y6H-hYOXbYH_.js} +1 -1
- package/dist/{chunk-PWAF6VOD-D8S9NPCy.js → chunk-PWAF6VOD-CQw5qATZ.js} +1 -1
- package/dist/{chunk-SHT3W25Y-B2dyNZ31.js → chunk-SHT3W25Y-lHiGx3ZC.js} +7 -7
- package/dist/{chunk-SVP7TREG-D7UPvB10.js → chunk-SVP7TREG-BvsHIrY_.js} +2 -2
- package/dist/{chunk-TICWLB2K-BNaGoadd.js → chunk-TICWLB2K-BC8YIm_n.js} +5 -5
- package/dist/{chunk-YOKDWASO-DrWbKnxA.js → chunk-YOKDWASO-DPZk8vhS.js} +6 -6
- package/dist/classDiagram-ZZMXUADV-Bt98lkkv.js +30 -0
- package/dist/classDiagram-v2-VYDZK3BY-D9bfbcC3.js +30 -0
- package/dist/{code-visibility-B_vyV9G1.js → code-visibility-4aW5326K.js} +8 -8
- package/dist/{common-DrnmK6Aw.js → common-Cjo7bjRK.js} +6574 -7877
- package/dist/{compiler-runtime-Cj8ZWiNS.js → compiler-runtime-CldRdSsv.js} +2 -2
- package/dist/{copy-icon-BY7jgcaV.js → copy-icon-Ft7t9bqz.js} +6 -6
- package/dist/{cynefinDiagram-5FMLGOSQ-CCDn4Z9e.js → cynefinDiagram-5FMLGOSQ-CgWra7wt.js} +3 -3
- package/dist/{dagre-GXQ25YYZ-BJiUO_SV.js → dagre-GXQ25YYZ-CFZ8-35c.js} +10 -10
- package/dist/{data-grid-overlay-editor-CauDs7BG.js → data-grid-overlay-editor-CvdQXfSk.js} +7 -7
- package/dist/{data-grid-types-BucVmDDp.js → data-grid-types-BH_r7g7l.js} +1 -1
- package/dist/{diagram-S7CK7UJ4-BZfxFNRR.js → diagram-S7CK7UJ4-lCNQrW7d.js} +4 -4
- package/dist/{diagram-UQ7AKVKN-1xcsiIMg.js → diagram-UQ7AKVKN-CQPQ6pP_.js} +3 -3
- package/dist/{diagram-VSXAHHWV-DPAWJ9Nw.js → diagram-VSXAHHWV-BJWcVVJY.js} +2 -2
- package/dist/{diagram-VX7I27RA-DmRsnHYY.js → diagram-VX7I27RA-Cf3b0llz.js} +5 -5
- package/dist/{diagram-Z3DM3KII--b49ryKx.js → diagram-Z3DM3KII-Bt4QLrpn.js} +3 -3
- package/dist/{dialog-CmExhUDI.js → dialog-CEBFZNyc.js} +6 -6
- package/dist/{dist-CcyKTBTF.js → dist-DpBpA_n1.js} +2 -2
- package/dist/{dropdown-menu-Cz4TdOJh.js → dropdown-menu-CP_qCYHC.js} +139 -139
- package/dist/{ebnfDiagram-PWID7BFC-rSwVyh4y.js → ebnfDiagram-PWID7BFC-D3NAyPFY.js} +3 -3
- package/dist/{erDiagram-RLTQ6QDP-Bfd3iAoC.js → erDiagram-RLTQ6QDP-Tlu09_Nm.js} +12 -12
- package/dist/{error-banner-D30nA9nE.js → error-banner-Da600cwz.js} +6 -6
- package/dist/{error-entries-BovkFYNy.js → error-entries-CzoYkIQZ.js} +2 -2
- package/dist/{esm-DAVrGVKP.js → esm-AMFssoNp.js} +4 -4
- package/dist/{feedback-button-D7yOYYPr.js → feedback-button-ZuLnINhC.js} +13 -13
- package/dist/flowDiagram-HODETNUW-CAYbpJmV.js +16 -0
- package/dist/{formats-DhhbdyAH.js → formats-BGh2thRc.js} +6 -6
- package/dist/{ganttDiagram-EL5Y4UJY-B4m_YsKa.js → ganttDiagram-EL5Y4UJY-BQT16JSc.js} +2 -2
- package/dist/{gitGraphDiagram-WWUBYQGX-QRBbvWXh.js → gitGraphDiagram-WWUBYQGX-D5nSIzdE.js} +2 -2
- package/dist/{glide-data-editor-BVIswvcw.js → glide-data-editor-qM9FSJA9.js} +60 -60
- package/dist/{glide-utils-BVJ4fkxX.js → glide-utils-DJyqh0EM.js} +4 -4
- package/dist/{globals-oj-cWKZj.js → globals-Ci7B9t6C.js} +3 -3
- package/dist/{highlighted-body-KPVGNVTW-Ad3MKZ18.js → highlighted-body-KPVGNVTW-zwjq21tf.js} +4 -4
- package/dist/{html-to-image-Cn1_16-x.js → html-to-image-CZY1pIZH.js} +17 -17
- package/dist/{infoDiagram-27XIBGKW-DVQFYvhg.js → infoDiagram-27XIBGKW-XrDxhr0x.js} +2 -2
- package/dist/{ishikawaDiagram-5VMMS53U-viPND-nm.js → ishikawaDiagram-5VMMS53U-CD1muzov.js} +3 -3
- package/dist/{journeyDiagram-3NMN7TZE-BeeJdJo3.js → journeyDiagram-3NMN7TZE-BaLqn-dZ.js} +2 -2
- package/dist/{jsx-runtime-DgDPJV9S.js → jsx-runtime-3BD1kT0s.js} +1 -1
- package/dist/{kanban-definition-UXKFOSKX-Dcl_ECO9.js → kanban-definition-UXKFOSKX-Dg9W142_.js} +9 -9
- package/dist/{katex-BNCmKUol.js → katex-BIdZGsyO.js} +141 -116
- package/dist/{label-DQO1jFjf.js → label-BDKF3Afc.js} +10 -10
- package/dist/{loader-Cm-ATtou.js → loader-BQOfjTIJ.js} +1 -1
- package/dist/main.js +499 -467
- package/dist/{mermaid-BfhcnSj-.js → mermaid-CiAXjLpA.js} +56 -56
- package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +2 -0
- package/dist/{mhchem-BEc7A1or.js → mhchem-CdWPJK6g.js} +1 -1
- package/dist/{mindmap-definition-YA3MSWOX-BdxxhF-5.js → mindmap-definition-YA3MSWOX-DgCidgYJ.js} +11 -11
- package/dist/{number-overlay-editor-CS7fiUj7.js → number-overlay-editor-DunDxAzh.js} +5 -5
- package/dist/{pegDiagram-XKGWAZYB-Du5p2s0w.js → pegDiagram-XKGWAZYB-coOYYKB6.js} +3 -3
- package/dist/{pieDiagram-E7YTZNPT-DJg-SOIq.js → pieDiagram-E7YTZNPT-BnXTc3bj.js} +3 -3
- package/dist/{process-output-CUG57KRp.js → process-output-e4jj9Ijt.js} +4 -4
- package/dist/{quadrantDiagram-AXDQQJYC-keLfFx9f.js → quadrantDiagram-AXDQQJYC-Ccw0xMjN.js} +1 -1
- package/dist/{railroadDiagram-O6MQD6OU-C0ojm-6_.js → railroadDiagram-O6MQD6OU-B28-u83n.js} +3 -3
- package/dist/{react-dom-BUgJ-ab_.js → react-dom-DUmH0GXv.js} +29 -21
- package/dist/{react-ofJbxM4x.js → react-sHtZdjeP.js} +103 -94
- package/dist/{react-vega-ewj7q_U5.js → react-vega-BED0DjiB.js} +3 -3
- package/dist/{requirementDiagram-BXWQKSXE-CNwNBmI3.js → requirementDiagram-BXWQKSXE-CdYCTMu0.js} +11 -11
- package/dist/{reveal-component-taGJ10XF.js → reveal-component-A7HedxmE.js} +906 -878
- package/dist/{sankeyDiagram-P5KCCOFB-Bh4EsJox.js → sankeyDiagram-P5KCCOFB-ELvOcnZH.js} +1 -1
- package/dist/{select-Bs-nmvm6.js → select-BButFi-d.js} +11 -11
- package/dist/{sequenceDiagram-WJ2MYXX4-DQaW5qBG.js → sequenceDiagram-WJ2MYXX4-CM3aRAe_.js} +4 -4
- package/dist/{spec-DVsoMvYT.js → spec-BwarBpUJ.js} +4 -4
- package/dist/{stateDiagram-D77RDMKH-BDa515QC.js → stateDiagram-D77RDMKH-DOZUMVj8.js} +13 -13
- package/dist/stateDiagram-v2-MP3YSRHH-CISYxma6.js +30 -0
- package/dist/style.css +1 -1
- package/dist/{swimlanes-42K2YHIH-M4IdGi8J.js → swimlanes-42K2YHIH-BkSppxp8.js} +10 -10
- package/dist/swimlanesDiagram-VR7AAH4N-c8G5pGxC.js +30 -0
- package/dist/{swiper-component-CpN-hewZ.js → swiper-component-DKntyn8p.js} +5 -5
- package/dist/{timeline-definition-24CTP7MA-DH-kvakg.js → timeline-definition-24CTP7MA-CaMQayqK.js} +3 -3
- package/dist/{toDate-vEi5kTfY.js → toDate-BNJf01u1.js} +7 -7
- package/dist/{toast-DfEpJedt.js → toast-B9UMEi8u.js} +9 -9
- package/dist/{tooltip-CuNWg2ay.js → tooltip-BO1REbqs.js} +7 -7
- package/dist/{use-restore-focus-kmgwrL8q.js → use-restore-focus-BEM7sFWl.js} +2042 -1827
- package/dist/{use-vega-container-remeasure-4DwJbPWu.js → use-vega-container-remeasure-Bz4PZMtt.js} +2 -2
- package/dist/{useAsyncData-rkAlR_vh.js → useAsyncData-Deh5kszT.js} +3 -3
- package/dist/{useDateFormatter-C-tz65rD.js → useDateFormatter-D8XruNGG.js} +4 -4
- package/dist/{useDeepCompareMemoize-BlHFBj5_.js → useDeepCompareMemoize-DXwzRINI.js} +2 -2
- package/dist/{useIframeCapabilities-C1Bt04Iz.js → useIframeCapabilities-DBETvvPl.js} +2 -2
- package/dist/{useLifecycle-B68iY077.js → useLifecycle-j7D9uD1o.js} +6 -6
- package/dist/{useTheme-BwSgeAFb.js → useTheme-CRnjkku0.js} +7 -7
- package/dist/{vega-component-C-bPttRN.js → vega-component-DG29TAB7.js} +14 -14
- package/dist/{vega-embed-container-sSR5hAsy.js → vega-embed-container-BNHE18G9.js} +5 -5
- package/dist/{vennDiagram-4TSXK5OY-d5g7XeiI.js → vennDiagram-4TSXK5OY-CF8pfaK3.js} +3 -3
- package/dist/{wardleyDiagram-VM6X3IG4-BfX3bSzR.js → wardleyDiagram-VM6X3IG4-dGPu39d-.js} +3 -3
- package/dist/{xychartDiagram-S5SC5T6Z-C1QsLZg7.js → xychartDiagram-S5SC5T6Z-CP6-9q56.js} +5 -5
- package/package.json +23 -22
- package/src/components/data-table/__tests__/column_formatting.test.ts +30 -0
- package/src/components/data-table/__tests__/columns.test.tsx +70 -0
- package/src/components/data-table/column-formatting/feature.ts +20 -1
- package/src/components/data-table/columns.tsx +25 -18
- package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +4 -3
- package/src/components/editor/output/console/ConsoleOutput.tsx +5 -3
- package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +18 -0
- package/src/components/storage/__tests__/storage-file-viewer.test.tsx +2 -1
- package/src/core/cells/__tests__/document-roundtrip.test.ts +16 -0
- package/src/core/codemirror/copilot/__tests__/completion-integration.test.ts +447 -0
- package/src/core/codemirror/copilot/__tests__/inline-completion.test.ts +395 -0
- package/src/core/codemirror/copilot/__tests__/language-server.test.ts +511 -2
- package/src/core/codemirror/copilot/__tests__/rpc-cancellation.test.ts +201 -0
- package/src/core/codemirror/copilot/extension.ts +22 -11
- package/src/core/codemirror/copilot/language-server.ts +109 -60
- package/src/core/codemirror/language/__tests__/sql.test.ts +4 -3
- package/src/core/codemirror/language/embedded/__tests__/embedded-python.test.ts +138 -26
- package/src/core/codemirror/language/embedded/embedded-python.ts +47 -13
- package/src/core/codemirror/language/embedded/latex.ts +20 -13
- package/src/core/datasets/__tests__/data-source-discovery.test.ts +18 -11
- package/src/core/islands/bridge.ts +2 -2
- package/src/core/kernel/messages.ts +4 -7
- package/src/core/network/__tests__/api.test.ts +114 -1
- package/src/core/network/api.ts +3 -2
- package/src/core/wasm/bridge.ts +2 -2
- package/src/core/wasm/worker/types.ts +2 -2
- package/src/css/katex-css.test.ts +166 -0
- package/src/css/katex.min.css +33 -28
- package/src/hooks/__tests__/useEventListener.test.tsx +45 -1
- package/src/hooks/useEventListener.ts +6 -1
- package/src/plugins/impl/NumberPlugin.tsx +20 -3
- package/src/plugins/impl/__tests__/NumberPlugin.test.tsx +159 -1
- package/dist/assets/__vite-browser-external-BqAtzd2P.js +0 -1
- package/dist/assets/worker-Bf8dt6S-.js +0 -109
- package/dist/classDiagram-ZZMXUADV-Bbo2n384.js +0 -30
- package/dist/classDiagram-v2-VYDZK3BY-xR8cGP0r.js +0 -30
- package/dist/flowDiagram-HODETNUW-D7IBKS12.js +0 -16
- package/dist/mermaid-HWGCJPDP-DYCzSILp.js +0 -2
- package/dist/stateDiagram-v2-MP3YSRHH-ChNddepZ.js +0 -30
- package/dist/swimlanesDiagram-VR7AAH4N-FUSwmzEp.js +0 -30
|
@@ -2,13 +2,17 @@
|
|
|
2
2
|
|
|
3
3
|
import type { CompletionContext } from "@codemirror/autocomplete";
|
|
4
4
|
import { python } from "@codemirror/lang-python";
|
|
5
|
+
import { StreamLanguage } from "@codemirror/language";
|
|
6
|
+
import { stexMath } from "@codemirror/legacy-modes/mode/stex";
|
|
5
7
|
import { EditorState } from "@codemirror/state";
|
|
6
|
-
import
|
|
7
|
-
import {
|
|
8
|
+
import { classHighlighter, highlightTree } from "@lezer/highlight";
|
|
9
|
+
import { parser as markdownParser } from "@lezer/markdown";
|
|
10
|
+
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
8
11
|
import { cellId, variableName } from "@/__tests__/branded";
|
|
9
12
|
import { store } from "@/core/state/jotai";
|
|
10
13
|
import { variablesAtom } from "@/core/variables/state";
|
|
11
14
|
import { parsePython, variableCompletionSource } from "../embedded-python";
|
|
15
|
+
import { parseLatex } from "../latex";
|
|
12
16
|
|
|
13
17
|
const IS_ACTIVE = () => true;
|
|
14
18
|
|
|
@@ -23,25 +27,133 @@ describe("parsePython", () => {
|
|
|
23
27
|
]);
|
|
24
28
|
});
|
|
25
29
|
|
|
26
|
-
|
|
30
|
+
const cases = [
|
|
31
|
+
"{math.exp(3)}",
|
|
32
|
+
'${math.exp(3) + len("$}")} $',
|
|
33
|
+
"${math.exp(3)} + {math.sin(3)}$",
|
|
34
|
+
"${math.exp(3)}$",
|
|
35
|
+
"$$ {math.exp(3)} $$",
|
|
36
|
+
String.raw`\({math.exp(3)}\)`,
|
|
37
|
+
String.raw`\[{math.exp(3)}\]`,
|
|
38
|
+
String.raw`\\({math.exp(3)}\\)`,
|
|
39
|
+
String.raw`\\[{math.exp(3)}\\]`,
|
|
40
|
+
String.raw`$\frac{{{math.exp(3)}}}{{2}}$`,
|
|
41
|
+
String.raw`$\{math.exp(3)}$`,
|
|
42
|
+
'${ {"}": math.exp(3)} }$',
|
|
43
|
+
"${math.exp(3):{width}.{precision}f}$",
|
|
44
|
+
];
|
|
45
|
+
|
|
46
|
+
function highlightedProperties(doc: string, prefix: string) {
|
|
47
|
+
const parser = markdownParser.configure([
|
|
48
|
+
parseLatex(StreamLanguage.define(stexMath).parser),
|
|
49
|
+
parsePython(python().language.parser, () => prefix.includes("f")),
|
|
50
|
+
]);
|
|
51
|
+
const properties: string[] = [];
|
|
52
|
+
highlightTree(parser.parse(doc), classHighlighter, (from, to, classes) => {
|
|
53
|
+
if (classes.split(" ").includes("tok-propertyName")) {
|
|
54
|
+
properties.push(doc.slice(from, to));
|
|
55
|
+
}
|
|
56
|
+
});
|
|
57
|
+
return properties;
|
|
58
|
+
}
|
|
59
|
+
|
|
60
|
+
it.each(["f", "rf", "fr"])(
|
|
61
|
+
"highlights interpolation with %s strings",
|
|
62
|
+
(prefix) => {
|
|
63
|
+
for (const doc of cases) {
|
|
64
|
+
expect(highlightedProperties(doc, prefix)).toEqual(
|
|
65
|
+
doc.includes("math.sin") ? ["exp", "sin"] : ["exp"],
|
|
66
|
+
);
|
|
67
|
+
}
|
|
68
|
+
},
|
|
69
|
+
);
|
|
70
|
+
|
|
71
|
+
it.each(["", "r"])("does not interpolate %s strings", (prefix) => {
|
|
72
|
+
for (const doc of cases) {
|
|
73
|
+
expect(highlightedProperties(doc, prefix)).toEqual([]);
|
|
74
|
+
}
|
|
75
|
+
});
|
|
76
|
+
|
|
77
|
+
it("parses replacement boundaries once for a paragraph with many interpolations", () => {
|
|
27
78
|
const pythonParser = python().language.parser;
|
|
28
|
-
const
|
|
79
|
+
const parse = vi.spyOn(pythonParser, "parse");
|
|
80
|
+
try {
|
|
81
|
+
const doc = "{math.exp(3)} ".repeat(1000);
|
|
82
|
+
const parser = markdownParser.configure([
|
|
83
|
+
parseLatex(StreamLanguage.define(stexMath).parser),
|
|
84
|
+
parsePython(pythonParser, IS_ACTIVE),
|
|
85
|
+
]);
|
|
86
|
+
const properties: string[] = [];
|
|
87
|
+
highlightTree(
|
|
88
|
+
parser.parse(doc),
|
|
89
|
+
classHighlighter,
|
|
90
|
+
(from, to, classes) => {
|
|
91
|
+
if (classes.split(" ").includes("tok-propertyName")) {
|
|
92
|
+
properties.push(doc.slice(from, to));
|
|
93
|
+
}
|
|
94
|
+
},
|
|
95
|
+
);
|
|
96
|
+
expect(properties).toEqual(Array.from({ length: 1000 }, () => "exp"));
|
|
97
|
+
expect(parse).toHaveBeenCalledTimes(1);
|
|
98
|
+
} finally {
|
|
99
|
+
parse.mockRestore();
|
|
100
|
+
}
|
|
101
|
+
});
|
|
29
102
|
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
103
|
+
it("preserves raw backslashes between replacements", () => {
|
|
104
|
+
expect(
|
|
105
|
+
highlightedProperties(String.raw`{math.exp(3)} \N{math.sin(3)}`, "rf"),
|
|
106
|
+
).toEqual(["exp", "sin"]);
|
|
107
|
+
});
|
|
108
|
+
|
|
109
|
+
it("finds replacements after triple quotes in markdown text", () => {
|
|
110
|
+
expect(
|
|
111
|
+
highlightedProperties('{math.exp(3)} """ {math.sin(3)}', "f"),
|
|
112
|
+
).toEqual(["exp", "sin"]);
|
|
113
|
+
});
|
|
40
114
|
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
115
|
+
it("keeps replacement positions local to each paragraph", () => {
|
|
116
|
+
expect(
|
|
117
|
+
highlightedProperties("${math.exp(3)}$\n\n${math.sin(3)}$", "rf"),
|
|
118
|
+
).toEqual(["exp", "sin"]);
|
|
44
119
|
});
|
|
120
|
+
|
|
121
|
+
it("preserves LaTeX highlighting around Python replacements", () => {
|
|
122
|
+
const doc = String.raw`$\frac{{{math.exp(3)}}}{{2}} + \alpha$`;
|
|
123
|
+
const parser = markdownParser.configure([
|
|
124
|
+
parseLatex(StreamLanguage.define(stexMath).parser),
|
|
125
|
+
parsePython(python().language.parser, IS_ACTIVE),
|
|
126
|
+
]);
|
|
127
|
+
const highlighted: string[] = [];
|
|
128
|
+
highlightTree(parser.parse(doc), classHighlighter, (from, to) => {
|
|
129
|
+
highlighted.push(doc.slice(from, to));
|
|
130
|
+
});
|
|
131
|
+
expect(highlighted).toEqual(
|
|
132
|
+
expect.arrayContaining(["\\frac", "exp", "\\alpha"]),
|
|
133
|
+
);
|
|
134
|
+
});
|
|
135
|
+
|
|
136
|
+
it.each(["${math.exp(3)$", "${math.exp(3)", "${{math.exp(3)}$"])(
|
|
137
|
+
"does not treat incomplete or escaped braces as replacements: %s",
|
|
138
|
+
(doc) => {
|
|
139
|
+
expect(highlightedProperties(doc, "f")).toEqual([]);
|
|
140
|
+
},
|
|
141
|
+
);
|
|
142
|
+
|
|
143
|
+
it.each(["f", "rf", "fr"])(
|
|
144
|
+
"keeps doubled braces literal in %s strings",
|
|
145
|
+
(prefix) => {
|
|
146
|
+
for (const doc of [
|
|
147
|
+
"{{math.exp(3)}}",
|
|
148
|
+
"${{math.exp(3)}}$",
|
|
149
|
+
"$ {{{{math.exp(3)}}}} $",
|
|
150
|
+
String.raw`$\{{math.exp(3)}}$`,
|
|
151
|
+
String.raw`$\frac{{1}}{{2}}$`,
|
|
152
|
+
]) {
|
|
153
|
+
expect(highlightedProperties(doc, prefix)).toEqual([]);
|
|
154
|
+
}
|
|
155
|
+
},
|
|
156
|
+
);
|
|
45
157
|
});
|
|
46
158
|
|
|
47
159
|
describe("variableCompletionSource", () => {
|
|
@@ -72,10 +184,10 @@ describe("variableCompletionSource", () => {
|
|
|
72
184
|
const context: Partial<CompletionContext> = {
|
|
73
185
|
pos: 8,
|
|
74
186
|
explicit: false,
|
|
75
|
-
matchBefore: (
|
|
187
|
+
matchBefore: (_expr: RegExp) => ({ from: 7, to: 8, text: "v" }),
|
|
76
188
|
state,
|
|
77
189
|
aborted: false,
|
|
78
|
-
tokenBefore: (
|
|
190
|
+
tokenBefore: (_types: readonly string[]) => null,
|
|
79
191
|
};
|
|
80
192
|
|
|
81
193
|
const result = variableCompletionSource(context as CompletionContext);
|
|
@@ -92,10 +204,10 @@ describe("variableCompletionSource", () => {
|
|
|
92
204
|
const context: Partial<CompletionContext> = {
|
|
93
205
|
pos: 7,
|
|
94
206
|
explicit: false,
|
|
95
|
-
matchBefore: (
|
|
207
|
+
matchBefore: (_expr: RegExp) => ({ from: 6, to: 7, text: "v" }),
|
|
96
208
|
state,
|
|
97
209
|
aborted: false,
|
|
98
|
-
tokenBefore: (
|
|
210
|
+
tokenBefore: (_types: readonly string[]) => null,
|
|
99
211
|
};
|
|
100
212
|
|
|
101
213
|
const result = variableCompletionSource(context as CompletionContext);
|
|
@@ -110,10 +222,10 @@ describe("variableCompletionSource", () => {
|
|
|
110
222
|
const context: Partial<CompletionContext> = {
|
|
111
223
|
pos: 13,
|
|
112
224
|
explicit: false,
|
|
113
|
-
matchBefore: (
|
|
225
|
+
matchBefore: (_expr: RegExp) => ({ from: 12, to: 13, text: "v" }),
|
|
114
226
|
state,
|
|
115
227
|
aborted: false,
|
|
116
|
-
tokenBefore: (
|
|
228
|
+
tokenBefore: (_types: readonly string[]) => null,
|
|
117
229
|
};
|
|
118
230
|
|
|
119
231
|
const result = variableCompletionSource(context as CompletionContext);
|
|
@@ -131,7 +243,7 @@ describe("variableCompletionSource", () => {
|
|
|
131
243
|
matchBefore: () => null,
|
|
132
244
|
state,
|
|
133
245
|
aborted: false,
|
|
134
|
-
tokenBefore: (
|
|
246
|
+
tokenBefore: (_types: readonly string[]) => null,
|
|
135
247
|
};
|
|
136
248
|
|
|
137
249
|
const result = variableCompletionSource(context as CompletionContext);
|
|
@@ -146,10 +258,10 @@ describe("variableCompletionSource", () => {
|
|
|
146
258
|
const context: Partial<CompletionContext> = {
|
|
147
259
|
pos: 9,
|
|
148
260
|
explicit: false,
|
|
149
|
-
matchBefore: (
|
|
261
|
+
matchBefore: (_expr: RegExp) => ({ from: 8, to: 9, text: "v" }),
|
|
150
262
|
state,
|
|
151
263
|
aborted: false,
|
|
152
|
-
tokenBefore: (
|
|
264
|
+
tokenBefore: (_types: readonly string[]) => null,
|
|
153
265
|
};
|
|
154
266
|
|
|
155
267
|
const result = variableCompletionSource(context as CompletionContext);
|
|
@@ -16,11 +16,8 @@ import { getVariableCompletions } from "../../completion/variable-completions";
|
|
|
16
16
|
const PYTHON = "Python";
|
|
17
17
|
|
|
18
18
|
const OPEN_BRACE = 123; // {
|
|
19
|
-
const CLOSE_BRACE = 125; // }
|
|
20
19
|
const DELIMITER_LENGTH = 1;
|
|
21
20
|
|
|
22
|
-
const MARK = { mark: `${PYTHON}Mark`, resolve: PYTHON };
|
|
23
|
-
|
|
24
21
|
/**
|
|
25
22
|
* Define a Python code block parser for Markdown.
|
|
26
23
|
*
|
|
@@ -31,6 +28,11 @@ export function parsePython(
|
|
|
31
28
|
pythonParser: Parser,
|
|
32
29
|
isActivated: () => boolean,
|
|
33
30
|
): MarkdownConfig {
|
|
31
|
+
const replacements = new WeakMap<
|
|
32
|
+
InlineContext,
|
|
33
|
+
{ ends: Map<number, number>; parsedTo: number }
|
|
34
|
+
>();
|
|
35
|
+
|
|
34
36
|
const defineNodes = [
|
|
35
37
|
{ name: PYTHON, style: tags.emphasis },
|
|
36
38
|
{ name: `${PYTHON}Mark`, style: tags.processingInstruction },
|
|
@@ -41,26 +43,58 @@ export function parsePython(
|
|
|
41
43
|
parseInline: [
|
|
42
44
|
{
|
|
43
45
|
name: PYTHON,
|
|
46
|
+
before: "Escape",
|
|
44
47
|
parse(cx: InlineContext, next: number, pos: number): number {
|
|
45
48
|
if (!isActivated()) {
|
|
46
49
|
return -1;
|
|
47
50
|
}
|
|
48
51
|
|
|
49
|
-
|
|
52
|
+
// A backslash does not escape a brace in Python f-strings.
|
|
53
|
+
if (next === 92 && cx.char(pos + 1) === OPEN_BRACE) {
|
|
54
|
+
return pos + 1;
|
|
55
|
+
}
|
|
56
|
+
if (next !== OPEN_BRACE) {
|
|
50
57
|
return -1;
|
|
51
58
|
}
|
|
59
|
+
if (cx.char(pos + 1) === OPEN_BRACE) {
|
|
60
|
+
return pos + 2;
|
|
61
|
+
}
|
|
52
62
|
|
|
53
|
-
|
|
54
|
-
if (
|
|
63
|
+
let cached = replacements.get(cx);
|
|
64
|
+
if (!cached || pos >= cached.parsedTo) {
|
|
65
|
+
// Cache all replacements so each brace does not reparse the paragraph.
|
|
66
|
+
// Keep Markdown backslashes from consuming braces as Python escapes.
|
|
67
|
+
const prefix = 'rf"""';
|
|
68
|
+
const tree = pythonParser.parse(
|
|
69
|
+
`${prefix}${cx.slice(pos, cx.end)}\n"""`,
|
|
70
|
+
);
|
|
71
|
+
const formatString = tree.topNode
|
|
72
|
+
.getChild("ExpressionStatement")
|
|
73
|
+
?.getChild("FormatString");
|
|
74
|
+
const offset = pos - prefix.length;
|
|
75
|
+
const ends = new Map<number, number>();
|
|
76
|
+
for (const replacement of formatString?.getChildren(
|
|
77
|
+
"FormatReplacement",
|
|
78
|
+
) ?? []) {
|
|
79
|
+
if (replacement.lastChild?.name === "}") {
|
|
80
|
+
ends.set(offset + replacement.from, offset + replacement.to);
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
cached = {
|
|
84
|
+
ends,
|
|
85
|
+
parsedTo: formatString ? offset + formatString.to : cx.end,
|
|
86
|
+
};
|
|
87
|
+
replacements.set(cx, cached);
|
|
88
|
+
}
|
|
89
|
+
const end = cached.ends.get(pos);
|
|
90
|
+
if (end === undefined) {
|
|
55
91
|
return -1;
|
|
56
92
|
}
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
next === OPEN_BRACE,
|
|
63
|
-
next === CLOSE_BRACE,
|
|
93
|
+
return cx.addElement(
|
|
94
|
+
cx.elt(PYTHON, pos, end, [
|
|
95
|
+
cx.elt(`${PYTHON}Mark`, pos, pos + 1),
|
|
96
|
+
cx.elt(`${PYTHON}Mark`, end - 1, end),
|
|
97
|
+
]),
|
|
64
98
|
);
|
|
65
99
|
},
|
|
66
100
|
},
|
|
@@ -29,7 +29,7 @@
|
|
|
29
29
|
// OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
30
30
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
31
31
|
|
|
32
|
-
import { type Parser, parseMixed } from "@lezer/common";
|
|
32
|
+
import { IterMode, type Parser, parseMixed } from "@lezer/common";
|
|
33
33
|
import { tags } from "@lezer/highlight";
|
|
34
34
|
import type {
|
|
35
35
|
DelimiterType,
|
|
@@ -231,24 +231,31 @@ export function parseLatex(latexParser: Parser): MarkdownConfig {
|
|
|
231
231
|
// Test if the node type is one of the math expression
|
|
232
232
|
const delimiterLength = DELIMITER_LENGTH[node.type.name];
|
|
233
233
|
if (delimiterLength) {
|
|
234
|
-
const from = node.from + delimiterLength;
|
|
235
|
-
const to = node.to - delimiterLength;
|
|
234
|
+
const from = node.node.firstChild?.to ?? node.from + delimiterLength;
|
|
235
|
+
const to = node.node.lastChild?.from ?? node.to - delimiterLength;
|
|
236
236
|
|
|
237
237
|
// Ensure valid ranges
|
|
238
238
|
if (from >= to || from < 0 || to < 0) {
|
|
239
239
|
return null;
|
|
240
240
|
}
|
|
241
241
|
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
242
|
+
// Leave Python replacements to their own parser and highlighter.
|
|
243
|
+
const overlay: { from: number; to: number }[] = [];
|
|
244
|
+
let start = from;
|
|
245
|
+
node.node.cursor(IterMode.IgnoreMounts).iterate((child) => {
|
|
246
|
+
if (child.name !== "Python") {
|
|
247
|
+
return;
|
|
248
|
+
}
|
|
249
|
+
if (start < child.from) {
|
|
250
|
+
overlay.push({ from: start, to: child.from });
|
|
251
|
+
}
|
|
252
|
+
start = child.to;
|
|
253
|
+
return false;
|
|
254
|
+
});
|
|
255
|
+
if (start < to) {
|
|
256
|
+
overlay.push({ from: start, to });
|
|
257
|
+
}
|
|
258
|
+
return overlay.length ? { parser: latexParser, overlay } : null;
|
|
252
259
|
}
|
|
253
260
|
|
|
254
261
|
return null;
|
|
@@ -1,5 +1,7 @@
|
|
|
1
1
|
/* Copyright 2026 Marimo. All rights reserved. */
|
|
2
2
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
3
|
+
import { requestId } from "@/__tests__/branded";
|
|
4
|
+
import type { DataSourceDiscoveryResult } from "@/core/kernel/messages";
|
|
3
5
|
import { connectionAtom } from "@/core/network/connection";
|
|
4
6
|
import { store } from "@/core/state/jotai";
|
|
5
7
|
import { ConnectionState } from "@/core/websocket/types";
|
|
@@ -41,7 +43,7 @@ describe("fetchDataSourceDiscovery", () => {
|
|
|
41
43
|
it("shares a single in-flight request across concurrent callers", async () => {
|
|
42
44
|
const request = vi
|
|
43
45
|
.spyOn(DiscoverDataSources, "request")
|
|
44
|
-
.mockResolvedValue({ request_id: "request-id", sources: [] });
|
|
46
|
+
.mockResolvedValue({ request_id: requestId("request-id"), sources: [] });
|
|
45
47
|
|
|
46
48
|
await Promise.all([
|
|
47
49
|
fetchDataSourceDiscovery(),
|
|
@@ -55,7 +57,7 @@ describe("fetchDataSourceDiscovery", () => {
|
|
|
55
57
|
it("re-runs discovery after the cache is invalidated", async () => {
|
|
56
58
|
const request = vi
|
|
57
59
|
.spyOn(DiscoverDataSources, "request")
|
|
58
|
-
.mockResolvedValue({ request_id: "request-id", sources: [] });
|
|
60
|
+
.mockResolvedValue({ request_id: requestId("request-id"), sources: [] });
|
|
59
61
|
|
|
60
62
|
await fetchDataSourceDiscovery();
|
|
61
63
|
invalidateDataSourceDiscovery();
|
|
@@ -68,7 +70,10 @@ describe("fetchDataSourceDiscovery", () => {
|
|
|
68
70
|
const request = vi
|
|
69
71
|
.spyOn(DiscoverDataSources, "request")
|
|
70
72
|
.mockRejectedValueOnce(new Error("boom"))
|
|
71
|
-
.mockResolvedValueOnce({
|
|
73
|
+
.mockResolvedValueOnce({
|
|
74
|
+
request_id: requestId("request-id"),
|
|
75
|
+
sources: [],
|
|
76
|
+
});
|
|
72
77
|
|
|
73
78
|
await expect(fetchDataSourceDiscovery()).rejects.toThrow("boom");
|
|
74
79
|
await expect(fetchDataSourceDiscovery()).resolves.toEqual([]);
|
|
@@ -77,16 +82,18 @@ describe("fetchDataSourceDiscovery", () => {
|
|
|
77
82
|
|
|
78
83
|
it("does not clear a newer cache entry when an invalidated request fails", async () => {
|
|
79
84
|
let rejectFirstRequest: ((error: Error) => void) | undefined;
|
|
80
|
-
const firstRequest = new Promise<
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
});
|
|
85
|
+
const firstRequest = new Promise<DataSourceDiscoveryResult>(
|
|
86
|
+
(_resolve, reject) => {
|
|
87
|
+
rejectFirstRequest = reject;
|
|
88
|
+
},
|
|
89
|
+
);
|
|
86
90
|
const request = vi
|
|
87
91
|
.spyOn(DiscoverDataSources, "request")
|
|
88
92
|
.mockReturnValueOnce(firstRequest)
|
|
89
|
-
.mockResolvedValueOnce({
|
|
93
|
+
.mockResolvedValueOnce({
|
|
94
|
+
request_id: requestId("request-id"),
|
|
95
|
+
sources: [],
|
|
96
|
+
});
|
|
90
97
|
|
|
91
98
|
const staleResult = fetchDataSourceDiscovery().catch((error) => error);
|
|
92
99
|
await vi.waitFor(() => expect(request).toHaveBeenCalledTimes(1));
|
|
@@ -116,7 +123,7 @@ describe("fetchDataSourceDiscovery", () => {
|
|
|
116
123
|
}),
|
|
117
124
|
];
|
|
118
125
|
vi.spyOn(DiscoverDataSources, "request").mockResolvedValue({
|
|
119
|
-
request_id: "request-id",
|
|
126
|
+
request_id: requestId("request-id"),
|
|
120
127
|
sources,
|
|
121
128
|
});
|
|
122
129
|
|
|
@@ -7,7 +7,7 @@ import { throwNotImplemented } from "@/utils/functions";
|
|
|
7
7
|
import { Logger } from "@/utils/Logger";
|
|
8
8
|
import { generateUUID } from "@/utils/uuid";
|
|
9
9
|
import { initialNotebookState, notebookAtom } from "../cells/cells";
|
|
10
|
-
import type {
|
|
10
|
+
import type { ClientCommand } from "../kernel/messages";
|
|
11
11
|
import type { EditRequests, RunRequests } from "../network/types";
|
|
12
12
|
import { store as defaultStore } from "../state/jotai";
|
|
13
13
|
import { getMarimoExportContext } from "../static/export-context";
|
|
@@ -392,7 +392,7 @@ export class IslandsPyodideBridge implements RunRequests, EditRequests {
|
|
|
392
392
|
// The kernel uses msgspec to parse control requests, which requires a 'type'
|
|
393
393
|
// field for discriminated union deserialization.
|
|
394
394
|
private async putControlRequest(
|
|
395
|
-
operation:
|
|
395
|
+
operation: ClientCommand,
|
|
396
396
|
session?: AppSession,
|
|
397
397
|
): Promise<void> {
|
|
398
398
|
session ??= await this.getActiveSession();
|
|
@@ -2,7 +2,7 @@
|
|
|
2
2
|
import type { components } from "@marimo-team/marimo-api";
|
|
3
3
|
|
|
4
4
|
export type schemas = components["schemas"];
|
|
5
|
-
export type DataType = schemas["
|
|
5
|
+
export type DataType = schemas["DataType"];
|
|
6
6
|
export const DATA_TYPES = [
|
|
7
7
|
"string",
|
|
8
8
|
"boolean",
|
|
@@ -24,10 +24,7 @@ export type DatabaseSchema = schemas["Schema"];
|
|
|
24
24
|
export type DataSourceConnection = schemas["DataSourceConnection"];
|
|
25
25
|
export type OutputChannel = schemas["CellChannel"];
|
|
26
26
|
export type CellOutput = schemas["CellOutput"];
|
|
27
|
-
export type MarimoError =
|
|
28
|
-
CellOutput["data"],
|
|
29
|
-
{ type: string }[]
|
|
30
|
-
>[number];
|
|
27
|
+
export type MarimoError = schemas["MarimoError"];
|
|
31
28
|
export type OutputMessage = schemas["CellOutput"];
|
|
32
29
|
export type CompletionOption =
|
|
33
30
|
schemas["CompletionResultNotification"]["options"][0];
|
|
@@ -54,8 +51,8 @@ export type Capabilities =
|
|
|
54
51
|
NotificationMessageData<"kernel-ready">["capabilities"];
|
|
55
52
|
export type CacheInfoFetched = NotificationMessageData<"cache-info">;
|
|
56
53
|
|
|
57
|
-
export type NotificationMessage = schemas["
|
|
58
|
-
export type
|
|
54
|
+
export type NotificationMessage = schemas["NotificationMessage"];
|
|
55
|
+
export type ClientCommand = schemas["ClientCommand"];
|
|
59
56
|
|
|
60
57
|
export type NotificationMessageType = NotificationMessage["op"];
|
|
61
58
|
export interface NotificationPayload {
|
|
@@ -1,7 +1,8 @@
|
|
|
1
1
|
/* Copyright 2026 Marimo. All rights reserved. */
|
|
2
2
|
|
|
3
|
-
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
3
|
+
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
4
4
|
import { HTTPError } from "@/utils/errors";
|
|
5
|
+
import { Logger } from "@/utils/Logger";
|
|
5
6
|
import { API } from "../api";
|
|
6
7
|
|
|
7
8
|
// Mock fetch globally
|
|
@@ -25,6 +26,118 @@ describe("API", () => {
|
|
|
25
26
|
baseUrl = "http://localhost:8000";
|
|
26
27
|
});
|
|
27
28
|
|
|
29
|
+
afterEach(() => {
|
|
30
|
+
vi.restoreAllMocks();
|
|
31
|
+
});
|
|
32
|
+
|
|
33
|
+
it.each(["before fetch", "during fetch"])(
|
|
34
|
+
"API.post propagates cancellation %s without logging it",
|
|
35
|
+
async (timing) => {
|
|
36
|
+
const log = vi.spyOn(Logger, "error").mockReturnValue(undefined);
|
|
37
|
+
const controller = new AbortController();
|
|
38
|
+
mockFetch.mockImplementation(
|
|
39
|
+
(_url: string, { signal }: RequestInit) =>
|
|
40
|
+
new Promise<Response>((_resolve, reject) => {
|
|
41
|
+
if (signal?.aborted) {
|
|
42
|
+
reject(signal.reason);
|
|
43
|
+
} else {
|
|
44
|
+
signal?.addEventListener("abort", () => reject(signal.reason), {
|
|
45
|
+
once: true,
|
|
46
|
+
});
|
|
47
|
+
}
|
|
48
|
+
}),
|
|
49
|
+
);
|
|
50
|
+
if (timing === "before fetch") {
|
|
51
|
+
controller.abort();
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
const response = API.post("/foo", {}, { signal: controller.signal });
|
|
55
|
+
controller.abort();
|
|
56
|
+
|
|
57
|
+
await expect(response).rejects.toBe(controller.signal.reason);
|
|
58
|
+
expect(mockFetch).toHaveBeenCalledWith(
|
|
59
|
+
"http://localhost:8000/api/foo",
|
|
60
|
+
expect.objectContaining({ signal: controller.signal }),
|
|
61
|
+
);
|
|
62
|
+
expect(log).not.toHaveBeenCalled();
|
|
63
|
+
},
|
|
64
|
+
);
|
|
65
|
+
|
|
66
|
+
it.each(["application/json", "text/plain"])(
|
|
67
|
+
"API.post propagates an abort while reading %s without logging it",
|
|
68
|
+
async (contentType) => {
|
|
69
|
+
const log = vi.spyOn(Logger, "error").mockReturnValue(undefined);
|
|
70
|
+
const error = new DOMException("Body read aborted", "AbortError");
|
|
71
|
+
const readBody = vi.fn().mockRejectedValue(error);
|
|
72
|
+
mockFetch.mockResolvedValue({
|
|
73
|
+
ok: true,
|
|
74
|
+
headers: new Headers({ "Content-Type": contentType }),
|
|
75
|
+
json: readBody,
|
|
76
|
+
text: readBody,
|
|
77
|
+
});
|
|
78
|
+
|
|
79
|
+
await expect(API.post("/foo", {})).rejects.toBe(error);
|
|
80
|
+
expect(readBody).toHaveBeenCalledExactlyOnceWith();
|
|
81
|
+
expect(log).not.toHaveBeenCalled();
|
|
82
|
+
},
|
|
83
|
+
);
|
|
84
|
+
|
|
85
|
+
it.each([new TypeError("Failed to fetch"), null, "Request failed"])(
|
|
86
|
+
"API.post logs and preserves non-abort fetch failures: %s",
|
|
87
|
+
async (error) => {
|
|
88
|
+
const log = vi.spyOn(Logger, "error").mockReturnValue(undefined);
|
|
89
|
+
const controller = new AbortController();
|
|
90
|
+
controller.abort();
|
|
91
|
+
mockFetch.mockRejectedValue(error);
|
|
92
|
+
|
|
93
|
+
await expect(
|
|
94
|
+
API.post("/foo", {}, { signal: controller.signal }),
|
|
95
|
+
).rejects.toBe(error);
|
|
96
|
+
expect(log).toHaveBeenCalledExactlyOnceWith(
|
|
97
|
+
"Error requesting http://localhost:8000/api/foo",
|
|
98
|
+
error,
|
|
99
|
+
);
|
|
100
|
+
},
|
|
101
|
+
);
|
|
102
|
+
|
|
103
|
+
it("API.post logs HTTP failures and preserves their status and body", async () => {
|
|
104
|
+
const log = vi.spyOn(Logger, "error").mockReturnValue(undefined);
|
|
105
|
+
const body = { detail: "Service unavailable" };
|
|
106
|
+
mockFetch.mockResolvedValue(
|
|
107
|
+
new Response(JSON.stringify(body), {
|
|
108
|
+
status: 503,
|
|
109
|
+
statusText: "Service Unavailable",
|
|
110
|
+
headers: { "Content-Type": "application/json" },
|
|
111
|
+
}),
|
|
112
|
+
);
|
|
113
|
+
|
|
114
|
+
await expect(API.post("/foo", {})).rejects.toMatchObject({
|
|
115
|
+
name: "HTTPError",
|
|
116
|
+
status: 503,
|
|
117
|
+
message: "Service Unavailable",
|
|
118
|
+
cause: body,
|
|
119
|
+
});
|
|
120
|
+
expect(log).toHaveBeenCalledExactlyOnceWith(
|
|
121
|
+
"Error requesting http://localhost:8000/api/foo",
|
|
122
|
+
expect.any(HTTPError),
|
|
123
|
+
);
|
|
124
|
+
});
|
|
125
|
+
|
|
126
|
+
it("API.post logs malformed JSON responses", async () => {
|
|
127
|
+
const log = vi.spyOn(Logger, "error").mockReturnValue(undefined);
|
|
128
|
+
mockFetch.mockResolvedValue(
|
|
129
|
+
new Response("invalid json", {
|
|
130
|
+
headers: { "Content-Type": "application/json" },
|
|
131
|
+
}),
|
|
132
|
+
);
|
|
133
|
+
|
|
134
|
+
await expect(API.post("/foo", {})).rejects.toBeInstanceOf(SyntaxError);
|
|
135
|
+
expect(log).toHaveBeenCalledExactlyOnceWith(
|
|
136
|
+
"Error requesting http://localhost:8000/api/foo",
|
|
137
|
+
expect.any(SyntaxError),
|
|
138
|
+
);
|
|
139
|
+
});
|
|
140
|
+
|
|
28
141
|
it("API.post calls fetch with POST and correct URL", async () => {
|
|
29
142
|
mockFetch.mockResolvedValue({
|
|
30
143
|
ok: true,
|
package/src/core/network/api.ts
CHANGED
|
@@ -101,8 +101,9 @@ export const API = {
|
|
|
101
101
|
return response.text() as unknown as RESP;
|
|
102
102
|
})
|
|
103
103
|
.catch((error) => {
|
|
104
|
-
|
|
105
|
-
|
|
104
|
+
if (error?.name !== "AbortError") {
|
|
105
|
+
Logger.error(`Error requesting ${fullUrl}`, error);
|
|
106
|
+
}
|
|
106
107
|
throw error;
|
|
107
108
|
});
|
|
108
109
|
},
|
package/src/core/wasm/bridge.ts
CHANGED
|
@@ -11,7 +11,7 @@ import { reloadSafe } from "@/utils/reload-safe";
|
|
|
11
11
|
import { generateUUID } from "@/utils/uuid";
|
|
12
12
|
import { createModuleWorker } from "@/utils/worker";
|
|
13
13
|
import { notebookIsRunningAtom } from "../cells/cells";
|
|
14
|
-
import type {
|
|
14
|
+
import type { ClientCommand } from "../kernel/messages";
|
|
15
15
|
import { getInitialAppMode } from "../mode";
|
|
16
16
|
import { API } from "../network/api";
|
|
17
17
|
import { withDevAssetUrl } from "../network/export-asset-url";
|
|
@@ -711,7 +711,7 @@ export class PyodideBridge implements RunRequests, EditRequests {
|
|
|
711
711
|
public listStorageEntries = throwNotImplemented;
|
|
712
712
|
public downloadStorage = throwNotImplemented;
|
|
713
713
|
|
|
714
|
-
private async putControlRequest(operation:
|
|
714
|
+
private async putControlRequest(operation: ClientCommand) {
|
|
715
715
|
await this.rpc.proxy.request.bridge({
|
|
716
716
|
functionName: "put_control_request",
|
|
717
717
|
payload: operation,
|
|
@@ -3,7 +3,7 @@ import type { PyodideInterface } from "pyodide";
|
|
|
3
3
|
import type { WasmRuntimeConfig } from "../runtime-config";
|
|
4
4
|
import type { UserConfig } from "@/core/config/config-schema";
|
|
5
5
|
import type {
|
|
6
|
-
|
|
6
|
+
ClientCommand,
|
|
7
7
|
NotificationPayload,
|
|
8
8
|
} from "@/core/kernel/messages";
|
|
9
9
|
import type { JsonString } from "@/utils/json/base64";
|
|
@@ -74,7 +74,7 @@ export interface WasmController {
|
|
|
74
74
|
}
|
|
75
75
|
|
|
76
76
|
export interface RawBridge {
|
|
77
|
-
put_control_request(operation:
|
|
77
|
+
put_control_request(operation: ClientCommand): Promise<string>;
|
|
78
78
|
put_input(input: string): Promise<string>;
|
|
79
79
|
code_complete(request: CodeCompletionRequest): Promise<string>;
|
|
80
80
|
read_code(): Promise<{ contents: string }>;
|