@superblocksteam/vite-plugin-file-sync 1.99.99-next.2

package/dist/socket-manager.d.ts

@@ -0,0 +1,39 @@
+import EventEmitter from "node:events";
+import type { AiService } from "./ai-service/index.js";
+import type { FileSyncManager } from "./file-system-manager.js";
+import type { LockService } from "./lock-service/index.js";
+import type { SyncService } from "./sync-service/index.js";
+import type { Tracer } from "@opentelemetry/api";
+import type { EditorClientMethods, LibraryClientMethods } from "@superblocksteam/library-shared/types";
+import type { ISocketClient } from "@superblocksteam/shared";
+import type http from "node:http";
+type SocketManagerEvents = {
+    "connect:editor": [ISocketClient<EditorClientMethods>];
+    "connect:library": [ISocketClient<LibraryClientMethods>];
+};
+export declare class SocketManager extends EventEmitter<SocketManagerEvents> {
+    private peerIds;
+    private editorSockets;
+    private librarySockets;
+    private connectedUserPeerIdsMap;
+    constructor();
+    private socketServer;
+    startSocketServer({ server, fileSyncManager, syncService, lockService, aiService, path, tracer, }: {
+        server: http.Server;
+        fileSyncManager: FileSyncManager;
+        syncService?: SyncService;
+        lockService?: LockService;
+        aiService?: AiService;
+        path: string;
+        tracer?: Tracer;
+    }): void;
+    /**
+     * If we have no more socket connections, release the lock
+     * @param lockService
+     */
+    checkSocketsRemainingAndShutdownIfNeeded(lockService?: LockService): Promise<void>;
+    callEditorClients(fn: (socket: ISocketClient<EditorClientMethods>) => Promise<void>): void;
+    callLibraryClients(fn: (socket: ISocketClient<LibraryClientMethods>) => Promise<void>): void;
+}
+export {};
+//# sourceMappingURL=socket-manager.d.ts.map

package/dist/socket-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-manager.d.ts","sourceRoot":"","sources":["../src/socket-manager.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,aAAa,CAAC;AAKvC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAChE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,KAAK,EACV,mBAAmB,EACnB,oBAAoB,EAErB,MAAM,uCAAuC,CAAC;AAC/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAGlC,KAAK,mBAAmB,GAAG;IACzB,gBAAgB,EAAE,CAAC,aAAa,CAAC,mBAAmB,CAAC,CAAC,CAAC;IACvD,iBAAiB,EAAE,CAAC,aAAa,CAAC,oBAAoB,CAAC,CAAC,CAAC;CAC1D,CAAC;AAEF,qBAAa,aAAc,SAAQ,YAAY,CAAC,mBAAmB,CAAC;IAClE,OAAO,CAAC,OAAO,CAA8C;IAE7D,OAAO,CAAC,aAAa,CAAiD;IACtE,OAAO,CAAC,cAAc,CAAkD;IACxE,OAAO,CAAC,uBAAuB,CAAuC;;IAMtE,OAAO,CAAC,YAAY,CAA8B;IAClD,iBAAiB,CAAC,EAChB,MAAM,EACN,eAAe,EACf,WAAW,EACX,WAAW,EACX,SAAS,EACT,IAAI,EACJ,MAAM,GACP,EAAE;QACD,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC;QACpB,eAAe,EAAE,eAAe,CAAC;QACjC,WAAW,CAAC,EAAE,WAAW,CAAC;QAC1B,WAAW,CAAC,EAAE,WAAW,CAAC;QAC1B,SAAS,CAAC,EAAE,SAAS,CAAC;QACtB,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB;IAmKD;;;OAGG;IACG,wCAAwC,CAAC,WAAW,CAAC,EAAE,WAAW;IAoBxE,iBAAiB,CACf,EAAE,EAAE,CAAC,MAAM,EAAE,aAAa,CAAC,mBAAmB,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC;IAOnE,kBAAkB,CAChB,EAAE,EAAE,CAAC,MAAM,EAAE,aAAa,CAAC,oBAAoB,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC;CAMrE"}

package/dist/socket-manager.js

@@ -0,0 +1,184 @@
+import EventEmitter from "node:events";
+import { createISocketClient, ISocket } from "@superblocksteam/shared";
+import { WebSocketServer } from "ws";
+import { getErrorMeta, getLogger } from "./util/logger.js";
+import { withMethodTracing } from "./util/tracing.js";
+export class SocketManager extends EventEmitter {
+    peerIds = new Map();
+    editorSockets = new Set();
+    librarySockets = new Set();
+    connectedUserPeerIdsMap = new Map();
+    constructor() {
+        super();
+    }
+    socketServer;
+    startSocketServer({ server, fileSyncManager, syncService, lockService, aiService, path, tracer, }) {
+        if (this.socketServer) {
+            return;
+        }
+        const wss = new WebSocketServer({ path, noServer: true });
+        this.socketServer = wss;
+        server.on("upgrade", (req, socket, head) => {
+            if (!req.url) {
+                return;
+            }
+            const url = new URL(req.url, "http://localhost");
+            // TODO: some sort of authorization? this socket shouldnt be open outside of superblocks
+            if (url.pathname === path) {
+                let isEditor = false;
+                if (req.headers["sec-websocket-protocol"] === "editor") {
+                    isEditor = true;
+                }
+                wss.handleUpgrade(req, socket, head, (ws) => {
+                    const peerId = url.searchParams.get("peerId");
+                    if (!peerId) {
+                        getLogger().error("No peerId in upgrade request");
+                        ws.close(1008, "No peerId found");
+                        return;
+                    }
+                    wss.emit("connection", ws, isEditor, url.searchParams.get("peerId"), url.searchParams.get("userId"));
+                });
+            }
+        });
+        wss.on("connection", (ws, isEditor, peerId, userId) => {
+            const socket = createISocketClient(new ISocket(ws, withMethodTracing(tracer, {
+                editor: {
+                    // Update widgets
+                    setProperties: [fileSyncManager.handleSetProperties],
+                    deleteProperties: [fileSyncManager.handleDeleteProperties],
+                    moveComponent: [fileSyncManager.handleReparent],
+                    createComponent: [fileSyncManager.handleCreateComponent],
+                    deleteComponents: [fileSyncManager.handleDeleteComponents],
+                    setProperty: [fileSyncManager.handleSetProperty],
+                    batchUpdate: [fileSyncManager.handleBatchUpdate],
+                    // Renames
+                    renameElement: [fileSyncManager.handleRenameElement],
+                    renameApi: [fileSyncManager.handleRenameApi],
+                    // Update entities (variables, timers, events)
+                    addEntity: [fileSyncManager.handleAddEntity],
+                    updateEntity: [fileSyncManager.handleUpdateEntity],
+                    deleteEntity: [fileSyncManager.handleDeleteEntity],
+                    // Update theme
+                    updateTheme: [fileSyncManager.handleUpdateTheme],
+                    // Update APIs
+                    apiUpdate: [fileSyncManager.handleUpdateApi],
+                    apiDelete: [fileSyncManager.handleDeleteApi],
+                    // Ensure files are synced
+                    ensureFilesSynced: [
+                        async () => {
+                            if (!syncService) {
+                                throw new Error("Syncing is disabled, cannot ensure files are synced");
+                            }
+                            return await syncService.uploadDirectoryNowIfNeeded();
+                        },
+                    ],
+                    // AI + draft handling
+                    aiGenerate: [
+                        async (request, _context, peer) => {
+                            await aiService?.handleAiGenerate(request, peer);
+                        },
+                    ],
+                    aiRejectDraft: [
+                        async (_request, _context, peer) => {
+                            await syncService?.discardLocalDraftChanges();
+                            await aiService?.handleUserRejectedDraft(peer);
+                        },
+                    ],
+                    aiAcceptDraft: [
+                        async (_request, _context, peer) => {
+                            await syncService?.commitLocalDraftChanges();
+                            await aiService?.handleUserAcceptedDraft(peer);
+                        },
+                    ],
+                    aiCancel: [
+                        async () => {
+                            await syncService?.discardLocalDraftChanges();
+                            await aiService?.handleUserCanceled();
+                        },
+                    ],
+                },
+            }), [], undefined, undefined));
+            // we are storing all of the sockets for a given peerId (editor+library pair of connections)
+            // which lets us handle synchronization of operations between the two
+            let existingSockets = this.peerIds.get(peerId);
+            if (!existingSockets) {
+                existingSockets = new Set();
+                this.peerIds.set(peerId, existingSockets);
+            }
+            if (isEditor) {
+                this.editorSockets.add(socket);
+                this.emit("connect:editor", socket);
+                if (userId) {
+                    lockService?.addConnectedPeerIdToUser(userId, peerId);
+                }
+            }
+            else {
+                this.librarySockets.add(socket);
+                this.emit("connect:library", socket);
+            }
+            existingSockets.add(socket);
+            lockService?.stopShutdownTimerIfNeeded();
+            ws.on("err", (e) => {
+                getLogger().error("Socket error", getErrorMeta(e));
+            });
+            ws.on("close", async (code, reason) => {
+                this.peerIds.get(peerId)?.delete(socket);
+                getLogger().info(`Socket closed peerId: ${peerId} code: ${code} reason: ${reason}`);
+                if (isEditor) {
+                    this.editorSockets.delete(socket);
+                    if (userId) {
+                        lockService?.removeConnectedPeerIdFromUser(userId, peerId);
+                    }
+                }
+                else {
+                    this.librarySockets.delete(socket);
+                }
+                await this.checkSocketsRemainingAndShutdownIfNeeded(lockService);
+            });
+        });
+    }
+    /**
+     * If we have no more socket connections, release the lock
+     * @param lockService
+     */
+    async checkSocketsRemainingAndShutdownIfNeeded(lockService) {
+        const logger = getLogger();
+        if (this.editorSockets.size === 0 && this.librarySockets.size === 0) {
+            if (process.env.SUPERBLOCKS_IS_CSB === "true") {
+                // Only shutdown everything immediately if it's in the cloud editing mode
+                logger.info("[vite-plugin-file-sync] No more sockets remaining, shutting down immediately in cloud editing mode");
+                await lockService?.shutdownAndExit();
+            }
+            else {
+                // If it's not in the cloud editing mode, we start a shutdown timer
+                // This will trigger a shutdown of the server after 5 minutes of inactivity
+                logger.info("[vite-plugin-file-sync] No more sockets remaining, starting shutdown timer in local editing mode");
+                await lockService?.startShutdownTimer();
+            }
+        }
+    }
+    callEditorClients(fn) {
+        for (const socket of this.editorSockets) {
+            fn(socket).catch(errorHandler);
+        }
+    }
+    callLibraryClients(fn) {
+        for (const socket of this.librarySockets) {
+            fn(socket).catch(errorHandler);
+        }
+    }
+}
+function errorHandler(err) {
+    // ignore if a method is not callable or doesnt exist, just log it so we can debug and fix
+    // otherwise we can crash the vite process due to unhandled promise rejection
+    if (err != null &&
+        typeof err === "object" &&
+        "code" in err &&
+        (err.code === 2 || err.code === 3)) {
+        getLogger().error(`[socket] ${err.message}`, getErrorMeta(err));
+    }
+    else {
+        throw err;
+    }
+}
+//# sourceMappingURL=socket-manager.js.map

package/dist/socket-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket-manager.js","sourceRoot":"","sources":["../src/socket-manager.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,mBAAmB,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AACvE,OAAO,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAoBtD,MAAM,OAAO,aAAc,SAAQ,YAAiC;IAC1D,OAAO,GAAG,IAAI,GAAG,EAAmC,CAAC;IAErD,aAAa,GAAG,IAAI,GAAG,EAAsC,CAAC;IAC9D,cAAc,GAAG,IAAI,GAAG,EAAuC,CAAC;IAChE,uBAAuB,GAA6B,IAAI,GAAG,EAAE,CAAC;IAEtE;QACE,KAAK,EAAE,CAAC;IACV,CAAC;IAEO,YAAY,CAA8B;IAClD,iBAAiB,CAAC,EAChB,MAAM,EACN,eAAe,EACf,WAAW,EACX,WAAW,EACX,SAAS,EACT,IAAI,EACJ,MAAM,GASP;QACC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,YAAY,GAAG,GAAG,CAAC;QAExB,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;YACzC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;gBACb,OAAO;YACT,CAAC;YACD,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;YAEjD,wFAAwF;YACxF,IAAI,GAAG,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;gBAC1B,IAAI,QAAQ,GAAG,KAAK,CAAC;gBACrB,IAAI,GAAG,CAAC,OAAO,CAAC,wBAAwB,CAAC,KAAK,QAAQ,EAAE,CAAC;oBACvD,QAAQ,GAAG,IAAI,CAAC;gBAClB,CAAC;gBAED,GAAG,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE;oBAC1C,MAAM,MAAM,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBAC9C,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,SAAS,EAAE,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;wBAClD,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;wBAClC,OAAO;oBACT,CAAC;oBAED,GAAG,CAAC,IAAI,CACN,YAAY,EACZ,EAAE,EACF,QAAQ,EACR,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,EAC9B,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAC/B,CAAC;gBACJ,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,EAAE,CACJ,YAAY,EACZ,CAAC,EAAa,EAAE,QAAiB,EAAE,MAAc,EAAE,MAAe,EAAE,EAAE;YACpE,MAAM,MAAM,GAAG,mBAAmB,CAChC,IAAI,OAAO,CACT,EAAE,EACF,iBAAiB,CAA0B,MAAM,EAAE;gBACjD,MAAM,EAAE;oBACN,iBAAiB;oBACjB,aAAa,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC;oBACpD,gBAAgB,EAAE,CAAC,eAAe,CAAC,sBAAsB,CAAC;oBAC1D,aAAa,EAAE,CAAC,eAAe,CAAC,cAAc,CAAC;oBAC/C,eAAe,EAAE,CAAC,eAAe,CAAC,qBAAqB,CAAC;oBACxD,gBAAgB,EAAE,CAAC,eAAe,CAAC,sBAAsB,CAAC;oBAC1D,WAAW,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;oBAChD,WAAW,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;oBAEhD,UAAU;oBACV,aAAa,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC;oBACpD,SAAS,EAAE,CAAC,eAAe,CAAC,eAAe,CAAC;oBAC5C,8CAA8C;oBAC9C,SAAS,EAAE,CAAC,eAAe,CAAC,eAAe,CAAC;oBAC5C,YAAY,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC;oBAClD,YAAY,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC;oBAElD,eAAe;oBACf,WAAW,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;oBAEhD,cAAc;oBACd,SAAS,EAAE,CAAC,eAAe,CAAC,eAAe,CAAC;oBAC5C,SAAS,EAAE,CAAC,eAAe,CAAC,eAAe,CAAC;oBAE5C,0BAA0B;oBAC1B,iBAAiB,EAAE;wBACjB,KAAK,IAAI,EAAE;4BACT,IAAI,CAAC,WAAW,EAAE,CAAC;gCACjB,MAAM,IAAI,KAAK,CACb,qDAAqD,CACtD,CAAC;4BACJ,CAAC;4BACD,OAAO,MAAM,WAAW,CAAC,0BAA0B,EAAE,CAAC;wBACxD,CAAC;qBACF;oBAED,sBAAsB;oBACtB,UAAU,EAAE;wBACV,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE;4BAChC,MAAM,SAAS,EAAE,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;wBACnD,CAAC;qBACF;oBACD,aAAa,EAAE;wBACb,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE;4BACjC,MAAM,WAAW,EAAE,wBAAwB,EAAE,CAAC;4BAC9C,MAAM,SAAS,EAAE,uBAAuB,CAAC,IAAI,CAAC,CAAC;wBACjD,CAAC;qBACF;oBACD,aAAa,EAAE;wBACb,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE;4BACjC,MAAM,WAAW,EAAE,uBAAuB,EAAE,CAAC;4BAC7C,MAAM,SAAS,EAAE,uBAAuB,CAAC,IAAI,CAAC,CAAC;wBACjD,CAAC;qBACF;oBACD,QAAQ,EAAE;wBACR,KAAK,IAAI,EAAE;4BACT,MAAM,WAAW,EAAE,wBAAwB,EAAE,CAAC;4BAC9C,MAAM,SAAS,EAAE,kBAAkB,EAAE,CAAC;wBACxC,CAAC;qBACF;iBACF;aACF,CAAC,EACF,EAAE,EACF,SAAS,EACT,SAAS,CACV,CACF,CAAC;YAEF,4FAA4F;YAC5F,qEAAqE;YACrE,IAAI,eAAe,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC/C,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,eAAe,GAAG,IAAI,GAAG,EAAE,CAAC;gBAC5B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;YAC5C,CAAC;YAED,IAAI,QAAQ,EAAE,CAAC;gBACb,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC/B,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;gBACpC,IAAI,MAAM,EAAE,CAAC;oBACX,WAAW,EAAE,wBAAwB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACxD,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAChC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;YACvC,CAAC;YAED,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC5B,WAAW,EAAE,yBAAyB,EAAE,CAAC;YAEzC,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE;gBACjB,SAAS,EAAE,CAAC,KAAK,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;gBACpC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;gBAEzC,SAAS,EAAE,CAAC,IAAI,CACd,yBAAyB,MAAM,UAAU,IAAI,YAAY,MAAM,EAAE,CAClE,CAAC;gBAEF,IAAI,QAAQ,EAAE,CAAC;oBACb,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBAClC,IAAI,MAAM,EAAE,CAAC;wBACX,WAAW,EAAE,6BAA6B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;oBAC7D,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACrC,CAAC;gBAED,MAAM,IAAI,CAAC,wCAAwC,CAAC,WAAW,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC;QACL,CAAC,CACF,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,wCAAwC,CAAC,WAAyB;QACtE,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,IAAI,IAAI,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACpE,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,KAAK,MAAM,EAAE,CAAC;gBAC9C,yEAAyE;gBACzE,MAAM,CAAC,IAAI,CACT,oGAAoG,CACrG,CAAC;gBACF,MAAM,WAAW,EAAE,eAAe,EAAE,CAAC;YACvC,CAAC;iBAAM,CAAC;gBACN,mEAAmE;gBACnE,2EAA2E;gBAC3E,MAAM,CAAC,IAAI,CACT,kGAAkG,CACnG,CAAC;gBACF,MAAM,WAAW,EAAE,kBAAkB,EAAE,CAAC;YAC1C,CAAC;QACH,CAAC;IACH,CAAC;IAED,iBAAiB,CACf,EAAiE;QAEjE,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACxC,EAAE,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED,kBAAkB,CAChB,EAAkE;QAElE,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACzC,EAAE,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;CACF;AAED,SAAS,YAAY,CAAC,GAAQ;IAC5B,0FAA0F;IAC1F,6EAA6E;IAC7E,IACE,GAAG,IAAI,IAAI;QACX,OAAO,GAAG,KAAK,QAAQ;QACvB,MAAM,IAAI,GAAG;QACb,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,CAAC,CAAC,EAClC,CAAC;QACD,SAAS,EAAE,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,OAAO,EAAE,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC;IAClE,CAAC;SAAM,CAAC;QACN,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/source-tracker.d.ts

@@ -0,0 +1,190 @@
+import { type PropertyInfo } from "@superblocksteam/library-shared";
+import { SbEntityType, type CreateRequest, type SbElement, type SbEntity, type SourceLocation, type Theme, type UpdateEntityRequest } from "@superblocksteam/library-shared/types";
+import type { SbScopeDefinition } from "./parsing/scope.js";
+import type { ParseResult } from "@babel/parser";
+import type { Node, NodePath } from "@babel/traverse";
+import type { File } from "@babel/types";
+type Position = {
+    source: SourceLocation;
+};
+export type SourceKind = "api" | "ts";
+interface SourceEditActions {
+    moveElement: (p: {
+        from: Position;
+        to: Position;
+    }) => Promise<void>;
+    addElement: (p: CreateRequest) => Promise<SbElement | undefined>;
+    deleteElement: (p: {
+        source: SourceLocation;
+        scopeName?: string;
+    }) => Promise<void>;
+    setProperties: (p: {
+        source: SourceLocation;
+        changes: Record<string, {
+            value: string | number | boolean | object;
+            type: PropertyInfo["type"];
+        }>;
+    }) => Promise<void>;
+    setProperty: (p: {
+        source: SourceLocation;
+        property: string;
+        value: string | number | boolean | object;
+        type: PropertyInfo["type"];
+    }) => Promise<void>;
+    addEntity: (p: {
+        scopeId: string;
+        entity: {
+            name: string;
+            type: SbEntityType;
+            attributes?: Record<string, unknown>;
+        };
+    }) => Promise<void>;
+    updateEntity: (p: UpdateEntityRequest) => Promise<void>;
+    deleteEntity: (p: {
+        entityId: SbElement;
+    }) => Promise<SbElement | undefined>;
+    updateTheme: (p: {
+        themeFilePath: string;
+        theme: Theme;
+    }) => Promise<void>;
+}
+export declare class SourceTracker implements SourceEditActions {
+    private renameManager;
+    private sbScopeManager;
+    private elementToLocation;
+    private elementToFilePath;
+    private scopeIdToFilePath;
+    private sbScopeNamesToFilePath;
+    private newlyAddedCount;
+    private fileToMeta;
+    private changedFiles;
+    constructor();
+    getCurrentFiles(): Record<string, {
+        code: string;
+        ast: ParseResult<File> | null;
+        imports: Record<string, boolean>;
+    }>;
+    /**
+     * Retrieves and clears the changes made to files, returning them as a record of file
+     * names and their corresponding code.
+     * @returns A Record<string, string> object containing the file names as keys and their corresponding code as values.
+     */
+    getAndFlushChanges: () => Promise<Array<{
+        fileName: string;
+        source: string;
+        kind: SourceKind;
+    }>>;
+    getElementToFilePath(): Record<SbElement, string>;
+    getElementToFilePath(el: SbElement): string;
+    getElementToLocation(): Record<SbElement, NodePath<Node>>;
+    getElementToLocation(el: SbElement): NodePath<Node>;
+    setElementToLocation(el: SbElement, path: NodePath<Node>): void;
+    setElementToFilePath(el: SbElement, filePath: string): void;
+    deleteElementToLocation(el: SbElement): void;
+    deleteElementToFilePath(el: SbElement): void;
+    getScopeDefinitionForPage(pageName: string): SbScopeDefinition | undefined;
+    private parseJsxFromEntrypoint;
+    initializeSourceTracker: ({ files }: {
+        files: Record<string, string>;
+    }) => void;
+    handleNonVisualChangeByDeletingIds: (fileName: string, fileContents: string) => Promise<void>;
+    removeFile: (fileName: string) => Promise<void>;
+    moveElement: ({ from, to, }: {
+        from: Position;
+        to: Position;
+    }) => Promise<void>;
+    addApi: ({ pageName, apiName, }: {
+        pageName: string;
+        apiName: string;
+    }) => Promise<string>;
+    deleteApi: ({ pageName, apiName, }: {
+        pageName: string;
+        apiName: string;
+    }) => Promise<string | undefined>;
+    addEntity: ({ scopeId, entity, }: {
+        scopeId: string;
+        entity: {
+            type: SbEntityType;
+            name: string;
+            attributes?: Partial<SbEntity>;
+        };
+    }) => Promise<void>;
+    updateEntity: (payload: UpdateEntityRequest) => Promise<void>;
+    deleteEntity: ({ entityId }: {
+        entityId: SbElement;
+    }) => Promise<`SB-${string}` | undefined>;
+    renameEntity: ({ entityId, oldName, newName, scopeName, }: {
+        entityId: SbElement;
+        oldName: string;
+        newName: string;
+        scopeName?: string;
+    }) => Promise<void>;
+    updateTheme: ({ themeFilePath, theme, }: {
+        themeFilePath: string;
+        theme: Theme;
+    }) => Promise<void>;
+    addElement: ({ parentElement, tagName, properties, children, id, scopeName, }: CreateRequest) => Promise<SbElement | undefined>;
+    deleteElement: ({ source, scopeName, }: {
+        source: SourceLocation;
+        scopeName?: string;
+    }) => Promise<void>;
+    private deleteSectionIfOnlyChild;
+    /**
+     * Set's a deeply nested property on a JSX attribute. For example, if you have an attribute that is
+     * an object like:
+     *   ```
+     *   columns={{
+     *     name: {
+     *       accessor: "name",
+     *     },
+     *   }}
+     *   ```
+     * And you want to update the `accessor` property, you can use this function to do so, passing a property path like
+     * `columns.name.accessor`, rather than the entire columns object.
+     */
+    private setNestedAttribute;
+    private setAttribute;
+    private deleteAttribute;
+    private transformSelfClosingTagToRegularTag;
+    setProperty: ({ source, property, value, type, }: {
+        source: SourceLocation;
+        property: string;
+        value: string | number | boolean | object;
+        type: PropertyInfo["type"];
+    }) => Promise<void>;
+    setProperties: ({ source, changes, }: {
+        source: SourceLocation;
+        changes: Record<string, {
+            value: any;
+            type: PropertyInfo["type"];
+        }>;
+    }) => Promise<void>;
+    deleteProperties: ({ source, properties, }: {
+        source: SourceLocation;
+        properties: string[];
+    }) => Promise<void>;
+    renameComponent: ({ widgetSourceId, newName, oldName, scopeName, }: {
+        widgetSourceId: SbElement;
+        newName: string;
+        oldName: string;
+        scopeName?: string;
+    }) => Promise<void>;
+    renameIdentifier: ({ oldName, newName, filePath, parentBinding, }: {
+        oldName: string;
+        newName: string;
+        filePath: string;
+        parentBinding?: string;
+    }) => void;
+    /**
+     * Change the name of the page element within a page file. Note: this does not actually
+     * rename the page file, but only the name of the page element, you have to perform the
+     * file system operation separately.
+     */
+    renamePage: ({ newName, filePath, }: {
+        newName: string;
+        filePath: string;
+    }) => Promise<void>;
+    getAddElementImports: ({ properties, children, }: Pick<CreateRequest, "properties" | "children">) => string[];
+}
+export {};
+//# sourceMappingURL=source-tracker.d.ts.map

package/dist/source-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"source-tracker.d.ts","sourceRoot":"","sources":["../src/source-tracker.ts"],"names":[],"mappings":"AAGA,OAAO,EAKL,KAAK,YAAY,EAClB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EACL,YAAY,EACZ,KAAK,aAAa,EAClB,KAAK,SAAS,EACd,KAAK,QAAQ,EACb,KAAK,cAAc,EACnB,KAAK,KAAK,EACV,KAAK,mBAAmB,EACzB,MAAM,uCAAuC,CAAC;AAiC/C,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAC5D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EACV,IAAI,EAIL,MAAM,cAAc,CAAC;AAEtB,KAAK,QAAQ,GAAG;IACd,MAAM,EAAE,cAAc,CAAC;CACxB,CAAC;AAiBF,MAAM,MAAM,UAAU,GAAG,KAAK,GAAG,IAAI,CAAC;AAEtC,UAAU,iBAAiB;IACzB,WAAW,EAAE,CAAC,CAAC,EAAE;QAAE,IAAI,EAAE,QAAQ,CAAC;QAAC,EAAE,EAAE,QAAQ,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpE,UAAU,EAAE,CAAC,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC,CAAC;IACjE,aAAa,EAAE,CAAC,CAAC,EAAE;QACjB,MAAM,EAAE,cAAc,CAAC;QACvB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpB,aAAa,EAAE,CAAC,CAAC,EAAE;QACjB,MAAM,EAAE,cAAc,CAAC;QACvB,OAAO,EAAE,MAAM,CACb,MAAM,EACN;YACE,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,CAAC;YAC1C,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;SAC5B,CACF,CAAC;KACH,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpB,WAAW,EAAE,CAAC,CAAC,EAAE;QACf,MAAM,EAAE,cAAc,CAAC;QACvB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,CAAC;QAC1C,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;KAC5B,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpB,SAAS,EAAE,CAAC,CAAC,EAAE;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE;YACN,IAAI,EAAE,MAAM,CAAC;YACb,IAAI,EAAE,YAAY,CAAC;YACnB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;SACtC,CAAC;KACH,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpB,YAAY,EAAE,CAAC,CAAC,EAAE,mBAAmB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACxD,YAAY,EAAE,CAAC,CAAC,EAAE;QAAE,QAAQ,EAAE,SAAS,CAAA;KAAE,KAAK,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC,CAAC;IAC7E,WAAW,EAAE,CAAC,CAAC,EAAE;QAAE,aAAa,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,KAAK,CAAA;KAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC5E;AAQD,qBAAa,aAAc,YAAW,iBAAiB;IACrD,OAAO,CAAC,aAAa,CAAuB;IAC5C,OAAO,CAAC,cAAc,CAAkB;IACxC,OAAO,CAAC,iBAAiB,CAAyC;IAClE,OAAO,CAAC,iBAAiB,CAAiC;IAC1D,OAAO,CAAC,iBAAiB,CAA8B;IACvD,OAAO,CAAC,sBAAsB,CAOvB;IAEP,OAAO,CAAC,eAAe,CAAK;IAC5B,OAAO,CAAC,UAAU,CAOX;IAEP,OAAO,CAAC,YAAY,CAA0B;;IAM9C,eAAe;cAZL,MAAM;aACP,WAAW,CAAC,IAAI,CAAC,GAAG,IAAI;iBACpB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;;IAcpC;;;;OAIG;IACH,kBAAkB,QAAa,OAAO,CACpC,KAAK,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,UAAU,CAAA;KAAE,CAAC,CAC9D,CAcC;IAEF,oBAAoB,IAAI,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC;IACjD,oBAAoB,CAAC,EAAE,EAAE,SAAS,GAAG,MAAM;IAM3C,oBAAoB,IAAI,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;IACzD,oBAAoB,CAAC,EAAE,EAAE,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC;IAMnD,oBAAoB,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC;IAGxD,oBAAoB,CAAC,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM;IAIpD,uBAAuB,CAAC,EAAE,EAAE,SAAS;IAGrC,uBAAuB,CAAC,EAAE,EAAE,SAAS;IAIrC,yBAAyB,CAAC,QAAQ,EAAE,MAAM,GAAG,iBAAiB,GAAG,SAAS;IAS1E,OAAO,CAAC,sBAAsB;IA6B9B,uBAAuB,GAAI,WAAW;QAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,UAoCrE;IAEF,kCAAkC,GAChC,UAAU,MAAM,EAChB,cAAc,MAAM,mBAuDpB;IAEF,UAAU,GAAU,UAAU,MAAM,mBAwBlC;IAEF,WAAW,GAAU,eAGlB;QACD,IAAI,EAAE,QAAQ,CAAC;QACf,EAAE,EAAE,QAAQ,CAAC;KACd,KAAG,OAAO,CAAC,IAAI,CAAC,CAuGf;IAEF,MAAM,GAAU,wBAGb;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;KACjB,qBAsBC;IAEF,SAAS,GAAU,wBAGhB;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;KACjB,iCAmBC;IAEF,SAAS,GAAU,sBAGhB;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE;YACN,IAAI,EAAE,YAAY,CAAC;YACnB,IAAI,EAAE,MAAM,CAAC;YACb,UAAU,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;SAChC,CAAC;KACH,mBAKC;IAEF,YAAY,GAAU,SAAS,mBAAmB,mBAGhD;IAEF,YAAY,GAAU,cAAc;QAAE,QAAQ,EAAE,SAAS,CAAA;KAAE,yCAOzD;IAEF,YAAY,GAAU,4CAKnB;QACD,QAAQ,EAAE,SAAS,CAAC;QACpB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,mBAoCC;IAGF,WAAW,GAAU,2BAGlB;QACD,aAAa,EAAE,MAAM,CAAC;QACtB,KAAK,EAAE,KAAK,CAAC;KACd,mBAcC;IAEF,UAAU,GAAU,kEAOjB,aAAa,KAAG,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC,CAmM/C;IAEF,aAAa,GAAU,wBAGpB;QACD,MAAM,EAAE,cAAc,CAAC;QACvB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,mBAsEC;IAGF,OAAO,CAAC,wBAAwB,CA0B9B;IAEF;;;;;;;;;;;;OAYG;IACH,OAAO,CAAC,kBAAkB,CAgDxB;IAEF,OAAO,CAAC,YAAY,CA8ClB;IAEF,OAAO,CAAC,eAAe,CAWrB;IAEF,OAAO,CAAC,mCAAmC,CAUzC;IAEF,WAAW,GAAU,oCAKlB;QACD,MAAM,EAAE,cAAc,CAAC;QACvB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,CAAC;QAC1C,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;KAC5B,mBAKC;IAEF,aAAa,GAAU,sBAGpB;QACD,MAAM,EAAE,cAAc,CAAC;QACvB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE;YAAE,KAAK,EAAE,GAAG,CAAC;YAAC,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC,CAAA;SAAE,CAAC,CAAC;KACrE,KAAG,OAAO,CAAC,IAAI,CAAC,CAiHf;IAEF,gBAAgB,GAAU,yBAGvB;QACD,MAAM,EAAE,cAAc,CAAC;QACvB,UAAU,EAAE,MAAM,EAAE,CAAC;KACtB,mBAgCC;IAEF,eAAe,GAAU,kDAKtB;QACD,cAAc,EAAE,SAAS,CAAC;QAC1B,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,mBAqCC;IAEF,gBAAgB,GAAI,gDAKjB;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,aAAa,CAAC,EAAE,MAAM,CAAC;KACxB,UAaC;IAEF;;;;OAIG;IACH,UAAU,GAAU,wBAGjB;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;KAClB,mBAyBC;IAEF,oBAAoB,GAAI,2BAGrB,IAAI,CAAC,aAAa,EAAE,YAAY,GAAG,UAAU,CAAC,cAmD/C;CACH"}