unisolating error surrounds when stx error happens in isolation

Author: azizghuloum

examples/arrow-function-2.ts.md

@@ -18,6 +18,8 @@ const foo = (x) => {
 ### Output Program
 
 ```typescript
-const /*>>>*/ x /*<<<*/ = 13;
+const foo$1 = (x$2) => {
+  const /*>>>*/ x /*<<<*/ = 13;
+};
 ```
 

examples/console-2.ts.md

@@ -15,6 +15,7 @@ consoel.log("hello world");
 ### Output Program
 
 ```typescript
-consoel;
+/*>>>*/ consoel /*<<<*/
+  .log("hello world");
 ```
 

src/library-manager.ts

@@ -21,7 +21,7 @@ import stringify from "json-stringify-pretty-compact";
 import { init_global_context } from "./global-module";
 import { parse_dts } from "./parse-dts";
 
-const cookie = "rewrite-ts-028";
+const cookie = "rewrite-ts-029";
 
 type module_state =
   | { type: "initial" }

src/stx-error.ts

@@ -32,7 +32,15 @@ export const in_isolation: <G extends { loc: Loc }>(
   loc: Loc,
   f: (loc: Loc) => Promise<G>,
 ) => Promise<Omit<G, "loc"> & { loc: Loc }> = async (loc, f) => {
-  return f(isolate(loc)).then(({ loc: res, ...g }) => ({ ...g, loc: unisolate(loc, res) }));
+  return f(isolate(loc))
+    .then(({ loc: res, ...g }) => ({ ...g, loc: unisolate(loc, res) }))
+    .catch((err) => {
+      if (err instanceof StxError) {
+        throw new StxError(err.name, unisolate(loc, err.loc), err.error, err.info);
+      } else {
+        throw err;
+      }
+    });
 };
 
 type LibraryManager = {

test-project/.rts/main.rts.json

@@ -1,6 +1,6 @@
 {
   "cid": "test-project/main.rts rewrite-ts-visualized 0.0.0",
-  "cookie": "rewrite-ts-028",
+  "cookie": "rewrite-ts-029",
   "imports": [
     {
       "pkg": {"name": "rewrite-ts-visualized", "version": "0.0.0"},

test-project/.rts/mod.rts.json

@@ -1,6 +1,6 @@
 {
   "cid": "test-project/mod.rts rewrite-ts-visualized 0.0.0",
-  "cookie": "rewrite-ts-028",
+  "cookie": "rewrite-ts-029",
   "imports": [],
   "exported_identifiers": {
     "x": [

test-project/.rts/sourcemap1.test.rts.json

@@ -1,6 +1,6 @@
 {
   "cid": "test-project/sourcemap1.test.rts rewrite-ts-visualized 0.0.0",
-  "cookie": "rewrite-ts-028",
+  "cookie": "rewrite-ts-029",
   "imports": [
     {
       "pkg": {"name": "vitest", "version": "2.1.8"},