rivet-design 0.5.7 → 0.5.9

package/dist/services/ScriptInjectionService.js

@@ -1,172 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ScriptInjectionService = exports.RIVET_SCRIPT_REMOVAL_PATTERN = void 0;
-const fs_1 = __importDefault(require("fs"));
-const path_1 = __importDefault(require("path"));
-const logger_1 = require("../utils/logger");
-const log = (0, logger_1.createLogger)('ScriptInjectionService');
-// Combination of next & vite inject paths
-const STANDARD_INJECT_PATHS = [
-    'app/layout.tsx',
-    'src/app/layout.tsx',
-    'pages/_app.tsx',
-    'src/pages/_app.tsx',
-    // Monorepo patterns
-    'apps/web/app/layout.tsx',
-    'apps/web/src/app/layout.tsx',
-    'apps/web/pages/_app.tsx',
-    'apps/web/src/pages/_app.tsx',
-    'apps/frontend/app/layout.tsx',
-    'apps/frontend/src/app/layout.tsx',
-    'apps/frontend/pages/_app.tsx',
-    'apps/frontend/src/pages/_app.tsx',
-    'packages/web/app/layout.tsx',
-    'packages/web/src/app/layout.tsx',
-    'packages/web/pages/_app.tsx',
-    'packages/web/src/pages/_app.tsx',
-    // Vite inject paths
-    'index.html',
-    'apps/web/index.html',
-    'apps/frontend/index.html',
-    'packages/web/index.html',
-];
-// Regex pattern used to remove the injected script tag
-exports.RIVET_SCRIPT_REMOVAL_PATTERN = /\s*<script[^>]*id="rivet-selection-script"[^>]*>[\s\S]*?<\/script>\s*/g;
-class ScriptInjectionService {
-    selectionScript;
-    constructor() {
-        // Load the selection script
-        try {
-            const scriptPath = path_1.default.join(__dirname, '../scripts/selection-script.js');
-            this.selectionScript = fs_1.default.readFileSync(scriptPath, 'utf8');
-        }
-        catch (error) {
-            log.error('Failed to load selection script:', error);
-            this.selectionScript = '// Selection script not found';
-        }
-    }
-    async injectScript(projectPath, customInjectPath, rivetPort = 4000) {
-        log.info('Injecting Rivet selection script...');
-        // Use custom inject path if provided, otherwise determine paths based on framework
-        let injectPaths;
-        if (customInjectPath) {
-            log.info(`Using custom inject path: ${customInjectPath}`);
-            injectPaths = [customInjectPath];
-        }
-        else {
-            // Default to standard paths for unknown frameworks
-            log.debug('Unknown framework, defaulting to standard paths');
-            injectPaths = STANDARD_INJECT_PATHS;
-        }
-        const layoutPaths = injectPaths.map((relativePath) => path_1.default.join(projectPath, relativePath));
-        let layoutPath = null;
-        for (const testPath of layoutPaths) {
-            log.debug(`🔍 Checking if file exists: ${testPath}`);
-            if (fs_1.default.existsSync(testPath)) {
-                layoutPath = testPath;
-                log.info(`✅ Found layout file at: ${layoutPath}`);
-                break;
-            }
-            else {
-                log.debug(`❌ File does not exist: ${testPath}`);
-            }
-        }
-        if (!layoutPath) {
-            log.warn('⚠️  No layout file found, skipping script injection');
-            log.debug('📂 Searched paths:', layoutPaths);
-            log.debug('🔍 Files in project root:');
-            try {
-                const rootFiles = fs_1.default
-                    .readdirSync(projectPath)
-                    .filter((f) => f.endsWith('.tsx') || f.endsWith('.ts') || f.endsWith('.html'));
-                log.debug('  ', rootFiles);
-            }
-            catch (e) {
-                log.debug('  Could not read project root', e);
-            }
-            return null;
-        }
-        try {
-            log.debug(`📖 Reading layout file: ${layoutPath}`);
-            const layoutContent = fs_1.default.readFileSync(layoutPath, 'utf8');
-            log.debug(`📄 Layout file length: ${layoutContent.length} characters`);
-            if (layoutContent.includes('rivet-selection-script')) {
-                log.info('✅ Selection script already injected');
-                return path_1.default.relative(projectPath, layoutPath);
-            }
-            log.debug('💉 Script not found, proceeding with injection...');
-            const scriptTag = `<script src="http://localhost:${rivetPort}/api/selection-script" id="rivet-selection-script"></script>`;
-            log.debug(`📝 Created script tag (${scriptTag.length} characters)`);
-            let modifiedContent;
-            let injectionMethod = '';
-            if (layoutContent.includes('</head>')) {
-                modifiedContent = layoutContent.replace('</head>', `  ${scriptTag}\n      </head>`);
-                injectionMethod = 'before </head>';
-            }
-            else if (layoutContent.includes('</body>')) {
-                modifiedContent = layoutContent.replace('</body>', `  ${scriptTag}\n      </body>`);
-                injectionMethod = 'before </body>';
-            }
-            else {
-                // For React components, inject at the end of the return statement
-                const returnMatch = layoutContent.match(/return\s*\(\s*(<[^>]+>)/);
-                if (returnMatch) {
-                    const firstTag = returnMatch[1];
-                    modifiedContent = layoutContent.replace(firstTag, firstTag + `\n        ${scriptTag}`);
-                    injectionMethod = `after first tag: ${firstTag}`;
-                }
-                else {
-                    log.warn('⚠️  Could not find injection point in layout file');
-                    log.debug('📄 Layout file preview (first 500 chars):');
-                    log.debug(layoutContent.substring(0, 500));
-                    return null;
-                }
-            }
-            log.debug(`🎯 Injection method: ${injectionMethod}`);
-            // Write back the modified content
-            log.debug(`💾 Writing modified content back to: ${layoutPath}`);
-            fs_1.default.writeFileSync(layoutPath, modifiedContent);
-            log.info(`✅ Selection script injected into ${layoutPath} (${injectionMethod})`);
-            return path_1.default.relative(projectPath, layoutPath);
-        }
-        catch (error) {
-            log.error('Failed to inject selection script:', error);
-            return null;
-        }
-    }
-    async removeScript(projectPath, customInjectPath) {
-        log.info('Removing Rivet selection script...');
-        // Use custom inject path if provided, otherwise determine paths based on framework
-        let injectPaths;
-        if (customInjectPath) {
-            log.info(`Using custom inject path for removal: ${customInjectPath}`);
-            injectPaths = [customInjectPath];
-        }
-        else {
-            injectPaths = STANDARD_INJECT_PATHS;
-        }
-        const layoutPaths = injectPaths.map((relativePath) => path_1.default.join(projectPath, relativePath));
-        for (const layoutPath of layoutPaths) {
-            if (!fs_1.default.existsSync(layoutPath))
-                continue;
-            try {
-                const layoutContent = fs_1.default.readFileSync(layoutPath, 'utf8');
-                // Remove the injected script
-                if (layoutContent.includes('rivet-selection-script')) {
-                    // Remove the entire script tag and surrounding whitespace
-                    const modifiedContent = layoutContent.replace(/\s*<script[^>]*id="rivet-selection-script"[^>]*>[\s\S]*?<\/script>\s*/g, '');
-                    fs_1.default.writeFileSync(layoutPath, modifiedContent);
-                    log.info(`Selection script removed from ${layoutPath}`);
-                }
-            }
-            catch (error) {
-                log.error(`Failed to remove selection script from ${layoutPath}:`, error);
-            }
-        }
-    }
-}
-exports.ScriptInjectionService = ScriptInjectionService;
-//# sourceMappingURL=ScriptInjectionService.js.map

package/dist/services/ScriptInjectionService.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ScriptInjectionService.js","sourceRoot":"","sources":["../../src/services/ScriptInjectionService.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAoB;AACpB,gDAAwB;AACxB,4CAA+C;AAE/C,MAAM,GAAG,GAAG,IAAA,qBAAY,EAAC,wBAAwB,CAAC,CAAC;AAEnD,0CAA0C;AAC1C,MAAM,qBAAqB,GAAG;IAC5B,gBAAgB;IAChB,oBAAoB;IACpB,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,yBAAyB;IACzB,6BAA6B;IAC7B,yBAAyB;IACzB,6BAA6B;IAC7B,8BAA8B;IAC9B,kCAAkC;IAClC,8BAA8B;IAC9B,kCAAkC;IAClC,6BAA6B;IAC7B,iCAAiC;IACjC,6BAA6B;IAC7B,iCAAiC;IACjC,oBAAoB;IACpB,YAAY;IACZ,qBAAqB;IACrB,0BAA0B;IAC1B,yBAAyB;CAC1B,CAAC;AAEF,uDAAuD;AAC1C,QAAA,4BAA4B,GACvC,wEAAwE,CAAC;AAE3E,MAAa,sBAAsB;IACzB,eAAe,CAAS;IAEhC;QACE,4BAA4B;QAC5B,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,gCAAgC,CAAC,CAAC;YAC1E,IAAI,CAAC,eAAe,GAAG,YAAE,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;YACrD,IAAI,CAAC,eAAe,GAAG,+BAA+B,CAAC;QACzD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,WAAmB,EACnB,gBAAyB,EACzB,YAAoB,IAAI;QAExB,GAAG,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;QAEhD,mFAAmF;QACnF,IAAI,WAAqB,CAAC;QAC1B,IAAI,gBAAgB,EAAE,CAAC;YACrB,GAAG,CAAC,IAAI,CAAC,6BAA6B,gBAAgB,EAAE,CAAC,CAAC;YAC1D,WAAW,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,mDAAmD;YACnD,GAAG,CAAC,KAAK,CAAC,iDAAiD,CAAC,CAAC;YAC7D,WAAW,GAAG,qBAAqB,CAAC;QACtC,CAAC;QAED,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CACnD,cAAI,CAAC,IAAI,CAAC,WAAW,EAAE,YAAY,CAAC,CACrC,CAAC;QAEF,IAAI,UAAU,GAAkB,IAAI,CAAC;QACrC,KAAK,MAAM,QAAQ,IAAI,WAAW,EAAE,CAAC;YACnC,GAAG,CAAC,KAAK,CAAC,+BAA+B,QAAQ,EAAE,CAAC,CAAC;YACrD,IAAI,YAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC5B,UAAU,GAAG,QAAQ,CAAC;gBACtB,GAAG,CAAC,IAAI,CAAC,2BAA2B,UAAU,EAAE,CAAC,CAAC;gBAClD,MAAM;YACR,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,KAAK,CAAC,0BAA0B,QAAQ,EAAE,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAED,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,GAAG,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAC;YAChE,GAAG,CAAC,KAAK,CAAC,oBAAoB,EAAE,WAAW,CAAC,CAAC;YAE7C,GAAG,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACvC,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,YAAE;qBACjB,WAAW,CAAC,WAAW,CAAC;qBACxB,MAAM,CACL,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CACjE,CAAC;gBACJ,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YAC7B,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,GAAG,CAAC,KAAK,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;YAChD,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,CAAC;YACH,GAAG,CAAC,KAAK,CAAC,2BAA2B,UAAU,EAAE,CAAC,CAAC;YACnD,MAAM,aAAa,GAAG,YAAE,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAC1D,GAAG,CAAC,KAAK,CAAC,0BAA0B,aAAa,CAAC,MAAM,aAAa,CAAC,CAAC;YAEvE,IAAI,aAAa,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;gBACrD,GAAG,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;gBAChD,OAAO,cAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;YAChD,CAAC;YAED,GAAG,CAAC,KAAK,CAAC,mDAAmD,CAAC,CAAC;YAE/D,MAAM,SAAS,GAAG,iCAAiC,SAAS,8DAA8D,CAAC;YAC3H,GAAG,CAAC,KAAK,CAAC,0BAA0B,SAAS,CAAC,MAAM,cAAc,CAAC,CAAC;YAEpE,IAAI,eAAuB,CAAC;YAC5B,IAAI,eAAe,GAAG,EAAE,CAAC;YAEzB,IAAI,aAAa,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACtC,eAAe,GAAG,aAAa,CAAC,OAAO,CACrC,SAAS,EACT,KAAK,SAAS,iBAAiB,CAChC,CAAC;gBACF,eAAe,GAAG,gBAAgB,CAAC;YACrC,CAAC;iBAAM,IAAI,aAAa,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC7C,eAAe,GAAG,aAAa,CAAC,OAAO,CACrC,SAAS,EACT,KAAK,SAAS,iBAAiB,CAChC,CAAC;gBACF,eAAe,GAAG,gBAAgB,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACN,kEAAkE;gBAClE,MAAM,WAAW,GAAG,aAAa,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;gBACnE,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;oBAChC,eAAe,GAAG,aAAa,CAAC,OAAO,CACrC,QAAQ,EACR,QAAQ,GAAG,aAAa,SAAS,EAAE,CACpC,CAAC;oBACF,eAAe,GAAG,oBAAoB,QAAQ,EAAE,CAAC;gBACnD,CAAC;qBAAM,CAAC;oBACN,GAAG,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAC;oBAC9D,GAAG,CAAC,KAAK,CAAC,2CAA2C,CAAC,CAAC;oBACvD,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;oBAC3C,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;YAED,GAAG,CAAC,KAAK,CAAC,wBAAwB,eAAe,EAAE,CAAC,CAAC;YAErD,kCAAkC;YAClC,GAAG,CAAC,KAAK,CAAC,wCAAwC,UAAU,EAAE,CAAC,CAAC;YAChE,YAAE,CAAC,aAAa,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;YAC9C,GAAG,CAAC,IAAI,CACN,oCAAoC,UAAU,KAAK,eAAe,GAAG,CACtE,CAAC;YAEF,OAAO,cAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YACvD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,WAAmB,EACnB,gBAAyB;QAEzB,GAAG,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;QAE/C,mFAAmF;QACnF,IAAI,WAAqB,CAAC;QAC1B,IAAI,gBAAgB,EAAE,CAAC;YACrB,GAAG,CAAC,IAAI,CAAC,yCAAyC,gBAAgB,EAAE,CAAC,CAAC;YACtE,WAAW,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,WAAW,GAAG,qBAAqB,CAAC;QACtC,CAAC;QAED,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CACnD,cAAI,CAAC,IAAI,CAAC,WAAW,EAAE,YAAY,CAAC,CACrC,CAAC;QAEF,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,UAAU,CAAC;gBAAE,SAAS;YAEzC,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,YAAE,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;gBAE1D,6BAA6B;gBAC7B,IAAI,aAAa,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;oBACrD,0DAA0D;oBAC1D,MAAM,eAAe,GAAG,aAAa,CAAC,OAAO,CAC3C,wEAAwE,EACxE,EAAE,CACH,CAAC;oBAEF,YAAE,CAAC,aAAa,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;oBAC9C,GAAG,CAAC,IAAI,CAAC,iCAAiC,UAAU,EAAE,CAAC,CAAC;gBAC1D,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CACP,0CAA0C,UAAU,GAAG,EACvD,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAhLD,wDAgLC"}

package/dist/types/proxy-types.d.ts

@@ -1,39 +0,0 @@
-type SchemaRegistry = 'CodeModificationSchema';
-export type ProxyRequest = {
-    provider: 'anthropic' | 'openai';
-    model: string;
-    messages: Array<{
-        role: 'user' | 'assistant' | 'system';
-        content: string | Array<{
-            type: string;
-            text?: string;
-            source?: {
-                type: string;
-                media_type: string;
-                data: string;
-            };
-        }>;
-    }>;
-    temperature?: number;
-    maxTokens?: number;
-    stream?: boolean;
-    schema?: SchemaRegistry;
-    prompt?: string;
-    image?: string;
-};
-export type ProxyResponse = {
-    success: boolean;
-    data?: {
-        text: string;
-        finishReason?: string;
-    };
-    error?: string;
-    usage?: {
-        promptTokens: number;
-        completionTokens: number;
-        totalTokens: number;
-        cost: number;
-    };
-};
-export {};
-//# sourceMappingURL=proxy-types.d.ts.map

package/dist/types/proxy-types.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"proxy-types.d.ts","sourceRoot":"","sources":["../../src/types/proxy-types.ts"],"names":[],"mappings":"AACA,KAAK,cAAc,GAAG,wBAAwB,CAAC;AAE/C,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,EAAE,WAAW,GAAG,QAAQ,CAAC;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,KAAK,CAAC;QACd,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;QACtC,OAAO,EACH,MAAM,GACN,KAAK,CAAC;YACJ,IAAI,EAAE,MAAM,CAAC;YACb,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,MAAM,CAAC,EAAE;gBAAE,IAAI,EAAE,MAAM,CAAC;gBAAC,UAAU,EAAE,MAAM,CAAC;gBAAC,IAAI,EAAE,MAAM,CAAA;aAAE,CAAC;SAC7D,CAAC,CAAC;KACR,CAAC,CAAC;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,cAAc,CAAC;IACxB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,CAAC,EAAE;QACL,IAAI,EAAE,MAAM,CAAC;QACb,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,CAAC;IACF,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;QACpB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;CACH,CAAC"}

package/dist/types/proxy-types.js

@@ -1,3 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-//# sourceMappingURL=proxy-types.js.map

package/dist/types/proxy-types.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"proxy-types.js","sourceRoot":"","sources":["../../src/types/proxy-types.ts"],"names":[],"mappings":""}