rwsdk 0.1.15 → 0.1.16

package/dist/vite/transformServerFunctions.mjs

@@ -1,296 +0,0 @@
-import MagicString from "magic-string";
-import debug from "debug";
-import { hasDirective } from "./hasDirective.mjs";
-import { findExports } from "./findSpecifiers.mjs";
-import { parse as sgParse, Lang as SgLang, Lang } from "@ast-grep/napi";
-import path from "path";
-const log = debug("rwsdk:vite:transform-server-functions");
-export const findExportedFunctions = (code, normalizedId) => {
-    return findExportInfo(code, normalizedId).localFunctions;
-};
-export const findExportInfo = (code, normalizedId) => {
-    process.env.VERBOSE && log("Finding exported functions in source file");
-    const localFunctions = new Set();
-    const reExports = [];
-    const exportInfos = findExports(normalizedId || "file.ts", code, log);
-    for (const exportInfo of exportInfos) {
-        if (exportInfo.isReExport && exportInfo.moduleSpecifier) {
-            // For re-exports, we need to determine the original name by parsing the code
-            // For "export { default as multiply }", we want localName="multiply", originalName="default"
-            // For "export { sum }", we want localName="sum", originalName="sum"
-            let originalName = exportInfo.name;
-            // Check if this is a default re-export with alias
-            if (exportInfo.isDefault && exportInfo.alias) {
-                originalName = "default";
-            }
-            reExports.push({
-                localName: exportInfo.name,
-                originalName: originalName,
-                moduleSpecifier: exportInfo.moduleSpecifier,
-            });
-            process.env.VERBOSE &&
-                log("Found re-exported function: %s (original: %s) from %s", exportInfo.name, originalName, exportInfo.moduleSpecifier);
-        }
-        else {
-            localFunctions.add(exportInfo.name);
-            process.env.VERBOSE &&
-                log("Found exported function: %s", exportInfo.name);
-        }
-    }
-    log("Found %d local functions: %O", localFunctions.size, Array.from(localFunctions));
-    log("Found %d re-exports: %O", reExports.length, reExports.map((r) => `${r.localName} from ${r.moduleSpecifier}`));
-    return { localFunctions, reExports };
-};
-// Helper function to find default function names using ast-grep
-function findDefaultFunctionName(code, normalizedId) {
-    const ext = path.extname(normalizedId).toLowerCase();
-    const lang = ext === ".tsx" || ext === ".jsx" ? Lang.Tsx : SgLang.TypeScript;
-    try {
-        const root = sgParse(lang, code);
-        const matches = root
-            .root()
-            .findAll("export default function $NAME($$$) { $$$ }");
-        if (matches.length > 0) {
-            const nameCapture = matches[0].getMatch("NAME");
-            return nameCapture?.text() || null;
-        }
-    }
-    catch (err) {
-        process.env.VERBOSE && log("Error finding default function name: %O", err);
-    }
-    return null;
-}
-// Helper function to check if there's a default export (not re-export)
-function hasDefaultExport(code, normalizedId) {
-    const ext = path.extname(normalizedId).toLowerCase();
-    const lang = ext === ".tsx" || ext === ".jsx" ? Lang.Tsx : SgLang.TypeScript;
-    try {
-        const root = sgParse(lang, code);
-        // Check for any export default statements
-        const patterns = [
-            "export default function $$$",
-            "export default function($$$) { $$$ }",
-            "export default $$$",
-        ];
-        for (const pattern of patterns) {
-            const matches = root.root().findAll(pattern);
-            if (matches.length > 0) {
-                return true;
-            }
-        }
-    }
-    catch (err) {
-        process.env.VERBOSE && log("Error checking for default export: %O", err);
-    }
-    return false;
-}
-export const transformServerFunctions = (code, normalizedId, environment, serverFiles, addServerModule) => {
-    process.env.VERBOSE &&
-        log("Transform server functions called for normalizedId=%s, environment=%s", normalizedId, environment);
-    if (!hasDirective(code, "use server")) {
-        log("Skipping: no 'use server' directive in id=%s", normalizedId);
-        process.env.VERBOSE &&
-            log(":VERBOSE: Returning code unchanged for id=%s:\n%s", normalizedId, code);
-        return;
-    }
-    log("Processing 'use server' module: normalizedId=%s, environment=%s", normalizedId, environment);
-    addServerModule?.(environment, normalizedId);
-    if (environment === "ssr" || environment === "client") {
-        log(`Transforming for ${environment} environment: normalizedId=%s`, normalizedId);
-        const exportInfo = findExportInfo(code, normalizedId);
-        const allExports = new Set([
-            ...exportInfo.localFunctions,
-            ...exportInfo.reExports.map((r) => r.localName),
-        ]);
-        // Check for default function exports that should also be named exports
-        const defaultFunctionName = findDefaultFunctionName(code, normalizedId);
-        if (defaultFunctionName) {
-            allExports.add(defaultFunctionName);
-        }
-        // Generate completely new code for SSR
-        const s = new MagicString("");
-        if (environment === "ssr") {
-            s.append('import { createServerReference } from "rwsdk/__ssr";\n\n');
-        }
-        else {
-            s.append('import { createServerReference } from "rwsdk/client";\n\n');
-        }
-        for (const name of allExports) {
-            if (name !== "default" && name !== defaultFunctionName) {
-                s.append(`export let ${name} = createServerReference(${JSON.stringify(normalizedId)}, ${JSON.stringify(name)});\n`);
-                log(`Added ${environment} server reference for function: %s in normalizedId=%s`, name, normalizedId);
-            }
-        }
-        // Check for default export in the actual module (not re-exports)
-        if (hasDefaultExport(code, normalizedId)) {
-            s.append(`\nexport default createServerReference(${JSON.stringify(normalizedId)}, "default");\n`);
-            log(`Added ${environment} server reference for default export in normalizedId=%s`, normalizedId);
-        }
-        log(`${environment} transformation complete for normalizedId=%s`, normalizedId);
-        return {
-            code: s.toString(),
-            map: s.generateMap({
-                source: normalizedId,
-                includeContent: true,
-                hires: true,
-            }),
-        };
-    }
-    else if (environment === "worker") {
-        log("Transforming for worker environment: normalizedId=%s", normalizedId);
-        const exportInfo = findExportInfo(code, normalizedId);
-        const s = new MagicString(code);
-        // Remove "use server" directive first
-        const directiveRegex = /^(\s*)(['"]use server['"])\s*;?\s*$/gm;
-        let match;
-        while ((match = directiveRegex.exec(code)) !== null) {
-            const start = match.index;
-            const end = match.index + match[0].length;
-            s.remove(start, end);
-            process.env.VERBOSE &&
-                log("Removed 'use server' directive from normalizedId=%s", normalizedId);
-            break; // Only remove the first one
-        }
-        // Add imports at the very beginning
-        let importsToAdd = [];
-        // Add imports for re-exported functions so they exist in scope
-        for (const reExport of exportInfo.reExports) {
-            // Fix the import statement - the originalName is what we import, localName is the alias
-            const importStatement = reExport.originalName === "default"
-                ? `import { default as ${reExport.localName} } from "${reExport.moduleSpecifier}";`
-                : reExport.originalName === reExport.localName
-                    ? `import { ${reExport.originalName} } from "${reExport.moduleSpecifier}";`
-                    : `import { ${reExport.originalName} as ${reExport.localName} } from "${reExport.moduleSpecifier}";`;
-            importsToAdd.push(importStatement);
-            log("Added import for re-exported function: %s from %s in normalizedId=%s", reExport.localName, reExport.moduleSpecifier, normalizedId);
-        }
-        // Add registerServerReference import
-        importsToAdd.push('import { registerServerReference } from "rwsdk/worker";');
-        // Add imports - position depends on whether file starts with block comment
-        if (importsToAdd.length > 0) {
-            const trimmedCode = code.trim();
-            if (trimmedCode.startsWith("/*")) {
-                // Find the end of the block comment
-                const blockCommentEnd = code.indexOf("*/");
-                if (blockCommentEnd !== -1) {
-                    // Insert after the block comment
-                    const insertPos = blockCommentEnd + 2;
-                    // Find the next newline after the block comment
-                    const nextNewline = code.indexOf("\n", insertPos);
-                    const actualInsertPos = nextNewline !== -1 ? nextNewline + 1 : insertPos;
-                    s.appendLeft(actualInsertPos, importsToAdd.join("\n") + "\n");
-                }
-                else {
-                    s.prepend(importsToAdd.join("\n") + "\n");
-                }
-            }
-            else {
-                // No block comment at start, add at beginning
-                s.prepend(importsToAdd.join("\n") + "\n");
-            }
-        }
-        // Handle default export renaming if present
-        const hasDefExport = hasDefaultExport(code, normalizedId);
-        if (hasDefExport) {
-            // Find and rename default function export using ast-grep
-            const ext = path.extname(normalizedId).toLowerCase();
-            const lang = ext === ".tsx" || ext === ".jsx" ? Lang.Tsx : SgLang.TypeScript;
-            try {
-                const root = sgParse(lang, code);
-                // Handle named default function: export default function myFunc() {}
-                const namedMatches = root
-                    .root()
-                    .findAll("export default function $NAME($$$) { $$$ }");
-                if (namedMatches.length > 0) {
-                    const match = namedMatches[0];
-                    const range = match.range();
-                    const funcName = match.getMatch("NAME")?.text();
-                    if (funcName) {
-                        // Replace "export default function myFunc" with "function __defaultServerFunction__"
-                        const newText = match
-                            .text()
-                            .replace(`export default function ${funcName}`, "function __defaultServerFunction__");
-                        s.overwrite(range.start.index, range.end.index, newText);
-                        s.append("\nexport default __defaultServerFunction__;\n");
-                    }
-                }
-                else {
-                    // Handle anonymous default function: export default function() {}
-                    const anonMatches = root
-                        .root()
-                        .findAll("export default function($$$) { $$$ }");
-                    if (anonMatches.length > 0) {
-                        const match = anonMatches[0];
-                        const range = match.range();
-                        const newText = match
-                            .text()
-                            .replace("export default function", "function __defaultServerFunction__");
-                        s.overwrite(range.start.index, range.end.index, newText);
-                        s.append("\nexport default __defaultServerFunction__;\n");
-                    }
-                    else {
-                        const predefinedMatches = root
-                            .root()
-                            .findAll("export default $NAME");
-                        if (predefinedMatches.length > 0) {
-                            const match = predefinedMatches[0];
-                            const nameCapture = match.getMatch("NAME")?.text();
-                            if (nameCapture) {
-                                s.append(`const __defaultServerFunction__ = ${nameCapture};\n`);
-                            }
-                        }
-                    }
-                }
-            }
-            catch (err) {
-                process.env.VERBOSE &&
-                    log("Error processing default function: %O", err);
-            }
-        }
-        // Add registration calls at the end
-        let registrationCalls = [];
-        const registeredFunctions = new Set(); // Track to avoid duplicates
-        if (hasDefExport) {
-            registrationCalls.push(`registerServerReference(__defaultServerFunction__, ${JSON.stringify(normalizedId)}, "default")`);
-            registeredFunctions.add("default");
-            log("Registered worker server reference for default export in normalizedId=%s", normalizedId);
-        }
-        // Register local functions
-        const defaultFunctionName = findDefaultFunctionName(code, normalizedId);
-        for (const name of exportInfo.localFunctions) {
-            if (name === "__defaultServerFunction__" ||
-                name === "default" ||
-                name === defaultFunctionName)
-                continue;
-            // Skip if already registered
-            if (registeredFunctions.has(name))
-                continue;
-            registrationCalls.push(`registerServerReference(${name}, ${JSON.stringify(normalizedId)}, ${JSON.stringify(name)})`);
-            registeredFunctions.add(name);
-            log("Registered worker server reference for local function: %s in normalizedId=%s", name, normalizedId);
-        }
-        // Register re-exported functions
-        for (const reExport of exportInfo.reExports) {
-            // Skip if already registered
-            if (registeredFunctions.has(reExport.localName))
-                continue;
-            registrationCalls.push(`registerServerReference(${reExport.localName}, ${JSON.stringify(normalizedId)}, ${JSON.stringify(reExport.localName)})`);
-            registeredFunctions.add(reExport.localName);
-            log("Registered worker server reference for re-exported function: %s in normalizedId=%s", reExport.localName, normalizedId);
-        }
-        if (registrationCalls.length > 0) {
-            s.append(registrationCalls.join("\n") + "\n");
-        }
-        log("Worker transformation complete for normalizedId=%s", normalizedId);
-        return {
-            code: s.toString(),
-            map: s.generateMap({
-                source: normalizedId,
-                includeContent: true,
-                hires: true,
-            }),
-        };
-    }
-    process.env.VERBOSE &&
-        log("No transformation applied for environment=%s, normalizedId=%s", environment, normalizedId);
-};

package/dist/vite/transformServerFunctions.test.d.mts

@@ -1 +0,0 @@
-export {};

package/dist/vite/transformServerFunctions.test.mjs

@@ -1,124 +0,0 @@
-import { describe, expect, it } from "vitest";
-import { transformServerFunctions } from "./transformServerFunctions.mjs";
-describe("useServerPlugin", () => {
-    let COMMENT_CODE = `
-// Comment
-"use server";
-
-export function sum() {
-  return 1 + 1;
-}
-`;
-    let MULTI_LINE_COMMENT_CODE = `
-// Multi-line
-// Comment
-"use server";
-
-export function sum() {
-  return 1 + 1
-}
-`;
-    let COMMENT_BLOCK_CODE = `
-/* Giant
- * Comment
- * Block
- */
-"use server";
-
-export function sum() {
-  return 1 + 1
-}
-`;
-    let DEFAULT_EXPORT_CODE = `
-"use server";
-
-export default function sum() {
-  return 1 + 1;
-}
-`;
-    let DEFAULT_AND_NAMED_EXPORTS_CODE = `
-"use server";
-
-export function sum() {
-  return 1 + 1;
-}
-
-export default function sum() {
-  return 1 + 2;
-}
-`;
-    let PREDEFINED_DEFAULT_EXPORT_CODE = `
-"use server";
-
-const sum = () => {
-  return 1 + 1;
-}
-
-export default sum;
-`;
-    let EXPORT_DEFAULT_FUNCTION_CODE = `
-"use server";
-export default function sum() {
-  return 1 + 1;
-}
-`;
-    let NAMED_EXPORT_CODE = `
-"use server";
-
-export function sum() {
-  return 1 + 1
-}
-`;
-    let ARROW_FUNCTION_EXPORT_CODE = `
-"use server";
-
-export const sum = () => {
-  return 1 + 1
-}
-`;
-    let ASYNC_FUNCTION_EXPORT_CODE = `
-"use server";
-
-export async function sum() {
-  return 1 + 1
-}
-`;
-    let RE_EXPORT_CODE = `
-"use server";
-
-export { sum } from './math';
-export { default as multiply } from './multiply';
-export * from './utils';
-`;
-    const TEST_CASES = {
-        COMMENT_CODE,
-        MULTI_LINE_COMMENT_CODE,
-        COMMENT_BLOCK_CODE,
-        DEFAULT_EXPORT_CODE,
-        NAMED_EXPORT_CODE,
-        ARROW_FUNCTION_EXPORT_CODE,
-        ASYNC_FUNCTION_EXPORT_CODE,
-        DEFAULT_AND_NAMED_EXPORTS_CODE,
-        RE_EXPORT_CODE,
-        PREDEFINED_DEFAULT_EXPORT_CODE,
-        EXPORT_DEFAULT_FUNCTION_CODE,
-    };
-    describe("TRANSFORMS", () => {
-        for (const [key, CODE] of Object.entries(TEST_CASES)) {
-            describe(key, () => {
-                it(`CLIENT`, () => {
-                    const result = transformServerFunctions(CODE, "/test.tsx", "client");
-                    expect(result?.code).toMatchSnapshot();
-                });
-                it(`WORKER`, () => {
-                    const result = transformServerFunctions(CODE, "/test.tsx", "worker");
-                    expect(result?.code).toMatchSnapshot();
-                });
-                it(`SSR`, () => {
-                    const result = transformServerFunctions(CODE, "/test.tsx", "ssr");
-                    expect(result?.code).toMatchSnapshot();
-                });
-            });
-        }
-    });
-});

package/dist/vite/useClientLookupPlugin.d.mts

@@ -1,5 +0,0 @@
-import { Plugin } from "vite";
-export declare const useClientLookupPlugin: ({ projectRootDir, clientFiles, }: {
-    projectRootDir: string;
-    clientFiles: Set<string>;
-}) => Promise<Plugin>;

package/dist/vite/useClientLookupPlugin.mjs

@@ -1,15 +0,0 @@
-import { createDirectiveLookupPlugin } from "./createDirectiveLookupPlugin.mjs";
-export const useClientLookupPlugin = async ({ projectRootDir, clientFiles, }) => {
-    return createDirectiveLookupPlugin({
-        projectRootDir,
-        files: clientFiles,
-        config: {
-            kind: "client",
-            directive: "use client",
-            virtualModuleName: "virtual:use-client-lookup",
-            exportName: "useClientLookup",
-            pluginName: "use-client-lookup",
-            optimizeForEnvironments: ["ssr", "client"],
-        },
-    });
-};

package/dist/vite/useServerLookupPlugin.d.mts

@@ -1,5 +0,0 @@
-import { Plugin } from "vite";
-export declare const useServerLookupPlugin: ({ projectRootDir, serverFiles, }: {
-    projectRootDir: string;
-    serverFiles: Set<string>;
-}) => Promise<Plugin>;

package/dist/vite/useServerLookupPlugin.mjs

@@ -1,15 +0,0 @@
-import { createDirectiveLookupPlugin } from "./createDirectiveLookupPlugin.mjs";
-export const useServerLookupPlugin = async ({ projectRootDir, serverFiles, }) => {
-    return createDirectiveLookupPlugin({
-        projectRootDir,
-        files: serverFiles,
-        config: {
-            kind: "server",
-            directive: "use server",
-            virtualModuleName: "virtual:use-server-lookup",
-            exportName: "useServerLookup",
-            pluginName: "use-server-lookup",
-            optimizeForEnvironments: ["ssr", "worker"],
-        },
-    });
-};

package/dist/vite/useServerPlugin.d.mts

@@ -1 +0,0 @@
-export { transformServerFunctions } from "./transformServerFunctions.mjs";

package/dist/vite/useServerPlugin.mjs

@@ -1 +0,0 @@
-export { transformServerFunctions } from "./transformServerFunctions.mjs";

package/dist/vite/virtualPlugin.d.mts

@@ -1,2 +0,0 @@
-import { Plugin } from "vite";
-export declare const virtualPlugin: (name: string, load: Plugin["load"]) => Plugin;

package/dist/vite/virtualPlugin.mjs

@@ -1,18 +0,0 @@
-// port(justinvdm, 3 Dec 2024): From https://github.com/hi-ogawa/vite-environment-examples/blob/440212b4208fc66a14d69a1bcbc7c5254b7daa91/examples/react-server/src/features/utils/plugin.ts#L37
-export const virtualPlugin = (name, load) => {
-    name = "virtual:" + name;
-    return {
-        name: `rwsdk:virtual-${name}`,
-        resolveId(source, _importer, _options) {
-            if (source === name || source.startsWith(`${name}?`)) {
-                return `\0${source}`;
-            }
-            return;
-        },
-        load(id, options) {
-            if (id === `\0${name}` || id.startsWith(`\0${name}?`)) {
-                return load.apply(this, [id, options]);
-            }
-        },
-    };
-};

package/dist/vite/vitePreamblePlugin.d.mts

@@ -1 +0,0 @@
-export declare const vitePreamblePlugin: () => import("vite").Plugin<any>;

package/dist/vite/vitePreamblePlugin.mjs

@@ -1,11 +0,0 @@
-import MagicString from "magic-string";
-import { virtualPlugin } from "./virtualPlugin.mjs";
-export const vitePreamblePlugin = () => virtualPlugin("vite-preamble", async () => {
-    const s = new MagicString(`
-import RefreshRuntime from "/@react-refresh"; RefreshRuntime.injectIntoGlobalHook(window); window.$RefreshReg$ = () => {}; window.$RefreshSig$ = () => (type) => type; window.__vite_plugin_react_preamble_installed__ = true;
-    `);
-    return {
-        code: s.toString(),
-        map: s.generateMap(),
-    };
-});