onlinecodesimulator/pyodide-worker.js

89 lines
2.3 KiB
JavaScript

importScripts("https://cdn.jsdelivr.net/pyodide/v0.23.4/full/pyodide.js");
let sensorData = {}; // ✅ Store sensor values
async function initializePyodide() {
self.pyodide = await loadPyodide({
indexURL: "https://cdn.jsdelivr.net/pyodide/v0.23.4/full/"
});
self.pyodide.globals.set("send_to_main", (event, data) => {
self.postMessage({ type: event, data: data });
});
// ✅ Expose sensor data to Python
self.pyodide.globals.set("get_sensor_data", (name) => {
return sensorData[name] ?? null;
});
// ✅ Run Python initialization
await self.pyodide.runPythonAsync(`
import sys
import pyodide
import asyncio
import time
class RobotModule:
def get_sensor(self, name):
return get_sensor_data(name)
def move(self, speed):
send_to_main("move", speed)
def fire(self):
send_to_main("fire", None)
def turn(self, deg):
send_to_main("turn", deg)
robot = RobotModule()
class ConsoleOutput:
def write(self, text):
if text.strip():
send_to_main("console", text)
def flush(self):
pass
sys.stdout = ConsoleOutput()
sys.stderr = ConsoleOutput()
sys.modules["robot"] = robot
# ✅ Override time.sleep() with an async version
_original_sleep = time.sleep # Backup original sleep
async def async_sleep(seconds):
await asyncio.sleep(seconds) # ✅ Allows JavaScript to process events while sleeping
time.sleep = async_sleep # ✅ Monkey-patch time.sleep()
`);
self.postMessage({ type: "ready" }); // ✅ Notify main thread that Pyodide is ready
}
initializePyodide();
self.onmessage = async (event) => {
if (!self.pyodide) {
self.postMessage({ type: "error", message: "Pyodide not initialized yet." });
return;
}
if (event.data.type === "sensor_update") {
// ✅ Update sensor data
Object.assign(sensorData, event.data.data);
} else if (event.data.type === "execute") {
try {
let result = await self.pyodide.runPythonAsync(event.data.code);
if (result !== undefined && result !== null && result !== "") {
self.postMessage({ type: "console", data: result });
}
} catch (error) {
self.postMessage({ type: "error", message: error.toString() });
}
}
};