From d0f0ebc6c35c4d028728b7068a6466401f8284a9 Mon Sep 17 00:00:00 2001
From: Mateusz Paprocki <mattpap@gmail.com>
Date: Fri, 15 Apr 2022 14:28:51 +0200
Subject: [PATCH 1/2] Very rudimentary support for ESM and import maps

---
 pyscriptjs/src/components/pyscript.ts | 44 +++++++++++++++++++++++++--
 1 file changed, 41 insertions(+), 3 deletions(-)

diff --git a/pyscriptjs/src/components/pyscript.ts b/pyscriptjs/src/components/pyscript.ts
index 80cfcbe..30aef4a 100644
--- a/pyscriptjs/src/components/pyscript.ts
+++ b/pyscriptjs/src/components/pyscript.ts
@@ -45,6 +45,11 @@ function htmlDecode(input) {
   return doc.documentElement.textContent;
 }
 
+// TODO: use type declaractions
+type PyodideInterface = {
+  registerJsModule(name: string, module: object): void
+}
+
 class Script {
   source: string;
   state: string;
@@ -241,17 +246,50 @@ export class PyScript extends HTMLElement {
       // pkg.do_something();
     }
 
-    async evaluate() {
+    protected async _register_esm(pyodide: PyodideInterface): Promise<void> {
+      const imports: {[key: string]: unknown} = {}
 
+      for (const node of document.querySelectorAll("script[type='importmap']")) {
+        const importmap = (() => {
+          try {
+            return JSON.parse(node.textContent)
+          } catch {
+            return null
+          }
+        })()
+
+        if (importmap?.imports == null)
+          continue
+
+        for (const [name, url] of Object.entries(importmap.imports)) {
+          if (typeof name != "string" || typeof url != "string")
+            continue
+
+          try {
+            // XXX: pyodide doesn't like Module(), failing with
+            // "can't read 'name' of undefined" at import time
+            imports[name] = {...await import(url)}
+          } catch {
+            console.error(`failed to fetch '${url}' for '${name}'`)
+          }
+        }
+      }
+
+      pyodide.registerJsModule("esm", imports)
+    }
+
+    async evaluate(): Promise<void> {
         console.log('evaluate');
+
         if (this.source){
           this.loadFromFile(this.source)
         }else{
-          let pyodide = await pyodideReadyPromise;
+          const pyodide = await pyodideReadyPromise;
+          await this._register_esm(pyodide)
           // debugger
           try {
             // @ts-ignore
-            let source = htmlDecode(this.editor.state.doc.toString());
+            const source = htmlDecode(this.editor.state.doc.toString());
             let output;
             if (source.includes("asyncio")){
               output = await pyodide.runPythonAsync(source);

From a80dcf13d35523f8001f38da507f45827f808617 Mon Sep 17 00:00:00 2001
From: Mateusz Paprocki <mattpap@gmail.com>
Date: Fri, 15 Apr 2022 14:30:32 +0200
Subject: [PATCH 2/2] Use ESM and import maps in d3 example

---
 pyscriptjs/examples/d3.html | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/pyscriptjs/examples/d3.html b/pyscriptjs/examples/d3.html
index c5bdc68..fcae36f 100644
--- a/pyscriptjs/examples/d3.html
+++ b/pyscriptjs/examples/d3.html
@@ -3,8 +3,6 @@
     <title>d3: JavaScript & PyScript visualizations side-by-side</title>
     <meta charset="utf-8">
 
-    <script src="https://d3js.org/d3.v7.min.js"></script>
-
     <link rel="stylesheet" href="../build/pyscript.css" />
     <script defer src="../build/pyscript.js"></script>
 
@@ -46,7 +44,17 @@
         </div>
       </div>
 
-<script>
+<script type="importmap">
+{
+  "imports": {
+    "d3": "https://cdn.skypack.dev/d3@7"
+  }
+}
+</script>
+
+<script type="module">
+import * as d3 from "d3"
+
 const fruits = [
   {name: "🍊", count: 21},
   {name: "🍇", count: 13},
@@ -55,7 +63,7 @@ const fruits = [
   {name: "🍐", count: 3},
   {name: "🍋", count: 2},
   {name: "🍎", count: 1},
-  {name: "🍉", count: 1}
+  {name: "🍉", count: 1},
 ]
 
 const fn = (d) => d.count
@@ -101,7 +109,7 @@ for (const d of data) {
 
 <py-script>
 from pyodide import create_proxy, to_js
-from js import d3
+from esm import d3
 
 fruits = [
   dict(name="🍊", count=21),