@forzalabs/remora 1.0.21 → 1.1.1

package/actions/init.js

@@ -1,27 +1,12 @@
-"use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.init = void 0;
-const chalk_1 = __importDefault(require("chalk"));
-const fs_extra_1 = __importDefault(require("fs-extra"));
-const ora_1 = __importDefault(require("ora"));
-const path_1 = __importDefault(require("path"));
-const init = () => __awaiter(void 0, void 0, void 0, function* () {
+import chalk from 'chalk';
+import fs from 'fs-extra';
+import ora from 'ora';
+import path from 'path';
+export const init = async () => {
     // TODO init single folder
-    console.log(chalk_1.default.blue.bold('📦 Initializing your application...'));
+    console.log(chalk.blue.bold('📦 Initializing your application...'));
     try {
-        const spinner = (0, ora_1.default)('Creating configuration files...').start();
+        const spinner = ora('Creating configuration files...').start();
         const directories = [
             'remora',
             'remora/consumers',
@@ -29,12 +14,12 @@ const init = () => __awaiter(void 0, void 0, void 0, function* () {
             'remora/schemas',
             'remora/sources'
         ];
-        const defaultSource = fs_extra_1.default.readFileSync(path_1.default.join(__dirname, `../documentation/default_resources/source.json`), 'utf-8');
-        const defaultConsumer = fs_extra_1.default.readFileSync(path_1.default.join(__dirname, `../documentation/default_resources/consumer.json`), 'utf-8');
-        const defaultProducer = fs_extra_1.default.readFileSync(path_1.default.join(__dirname, `../documentation/default_resources/producer.json`), 'utf-8');
+        const defaultSource = fs.readFileSync(path.join(__dirname, `../documentation/default_resources/source.json`), 'utf-8');
+        const defaultConsumer = fs.readFileSync(path.join(__dirname, `../documentation/default_resources/consumer.json`), 'utf-8');
+        const defaultProducer = fs.readFileSync(path.join(__dirname, `../documentation/default_resources/producer.json`), 'utf-8');
         // const defaultSchema = fs.readFileSync(path.join(__dirname, `../documentation/default_resources/schema.json`), 'utf-8')
-        const defaultRemoraProject = fs_extra_1.default.readFileSync(path_1.default.join(__dirname, `../documentation/default_resources/project.json`), 'utf-8');
-        const readme = fs_extra_1.default.readFileSync(path_1.default.join(__dirname, `../documentation/README.md`), 'utf-8');
+        const defaultRemoraProject = fs.readFileSync(path.join(__dirname, `../documentation/default_resources/project.json`), 'utf-8');
+        const readme = fs.readFileSync(path.join(__dirname, `../documentation/README.md`), 'utf-8');
         const files = [
             { path: 'remora/sources/.gitkeep', content: '' },
             { path: 'remora/sources/default-source.json', content: defaultSource },
@@ -48,30 +33,29 @@ const init = () => __awaiter(void 0, void 0, void 0, function* () {
         ];
         for (let i = 0; i < directories.length; i++) {
             const dir = directories[i];
-            yield fs_extra_1.default.ensureDir(dir);
+            await fs.ensureDir(dir);
         }
         for (let i = 0; i < files.length; i++) {
             const file = files[i];
-            yield fs_extra_1.default.ensureFile(file.path);
-            yield fs_extra_1.default.writeFile(file.path, file.content);
+            await fs.ensureFile(file.path);
+            await fs.writeFile(file.path, file.content);
         }
         spinner.succeed('Remora configuration files successfully created.');
-        console.log(chalk_1.default.green('✅ Initialization complete!'));
-        console.log(chalk_1.default.green('📁 A new folder named ') + chalk_1.default.cyan('remora/') + chalk_1.default.green(' has been created in your project.'));
-        console.log(chalk_1.default.green('✏️  Customize the generated configuration files inside this folder to setup Remora.'));
-        console.log(chalk_1.default.green('\n👣 Next steps: configure your environment variables.'));
-        console.log(chalk_1.default.green('\n🔧 Local development (add these to your .env file):'));
-        console.log(chalk_1.default.blue('• REMORA_WORKER_HOST') + chalk_1.default.gray(' → URL of your Remora Worker instance'));
-        console.log(chalk_1.default.blue('• REMORA_LICENCE_KEY') + chalk_1.default.gray(' → Your valid Remora license key'));
-        console.log(chalk_1.default.green('\n🔐 Production environment (store these in your Secret Manager):'));
-        console.log(chalk_1.default.yellow('• ROOT_TEMP_PASSWORD_FILE') + chalk_1.default.gray(' → Temporary root password for first-time setup'));
-        console.log(chalk_1.default.yellow('• INSTALLATION_ID') + chalk_1.default.gray(' → Unique ID for this installation instance'));
-        console.log(chalk_1.default.yellow('• ADMIN_JWT_SECRET_FILE') + chalk_1.default.gray(' → JWT secret for admin-level access'));
-        console.log(chalk_1.default.yellow('• JWT_SECRET_FILE') + chalk_1.default.gray(' → JWT secret for regular user authentication'));
+        console.log(chalk.green('✅ Initialization complete!'));
+        console.log(chalk.green('📁 A new folder named ') + chalk.cyan('remora/') + chalk.green(' has been created in your project.'));
+        console.log(chalk.green('✏️  Customize the generated configuration files inside this folder to setup Remora.'));
+        console.log(chalk.green('\n👣 Next steps: configure your environment variables.'));
+        console.log(chalk.green('\n🔧 Local development (add these to your .env file):'));
+        console.log(chalk.blue('• REMORA_WORKER_HOST') + chalk.gray(' → URL of your Remora Worker instance'));
+        console.log(chalk.blue('• REMORA_LICENCE_KEY') + chalk.gray(' → Your valid Remora license key'));
+        console.log(chalk.green('\n🔐 Production environment (store these in your Secret Manager):'));
+        console.log(chalk.yellow('• ROOT_TEMP_PASSWORD') + chalk.gray(' → Temporary root password for first-time setup'));
+        console.log(chalk.yellow('• INSTALLATION_ID') + chalk.gray(' → Unique ID for this installation instance'));
+        console.log(chalk.yellow('• ADMIN_JWT_SECRET_FILE') + chalk.gray(' → JWT secret for admin-level access'));
+        console.log(chalk.yellow('• JWT_SECRET_FILE') + chalk.gray(' → JWT secret for regular user authentication'));
     }
     catch (error) {
-        console.error(chalk_1.default.red('❌ Error during initialization:'), error instanceof Error ? error.message : String(error));
+        console.error(chalk.red('❌ Error during initialization:'), error instanceof Error ? error.message : String(error));
         process.exit(1);
     }
-});
-exports.init = init;
+};

package/actions/mock.js

@@ -1,43 +1,27 @@
-"use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.mock = void 0;
-const chalk_1 = __importDefault(require("chalk"));
-const ora_1 = __importDefault(require("ora"));
-const Environment_1 = __importDefault(require("../engines/Environment"));
-const DeveloperEngine_1 = __importDefault(require("../engines/ai/DeveloperEngine"));
-const compile_1 = require("./compile");
-const mock = (producerName, records) => __awaiter(void 0, void 0, void 0, function* () {
+import chalk from 'chalk';
+import ora from 'ora';
+import { Environment } from '@remora/app-common';
+import { DeveloperEngine } from '@remora/app-engines';
+import { compile } from './compile';
+export const mock = async (producerName, records) => {
     try {
-        (0, compile_1.compile)();
+        compile();
         console.log(); // needed for newline
-        const spinner = (0, ora_1.default)(chalk_1.default.blue(`Generating ${records} mock records for producer "${producerName}"...`)).start();
-        const producer = Environment_1.default.getProducer(producerName);
+        const spinner = ora(chalk.blue(`Generating ${records} mock records for producer "${producerName}"...`)).start();
+        const producer = Environment.getProducer(producerName);
         if (!producer) {
-            spinner.fail(chalk_1.default.red(`Producer "${producerName}" not found.`));
+            spinner.fail(chalk.red(`Producer "${producerName}" not found.`));
             process.exit(1);
         }
-        const result = yield DeveloperEngine_1.default.createMockData(producer, records);
+        const result = await DeveloperEngine.createMockData(producer, records);
         spinner.succeed(`Mock data generated successfully`);
-        console.log(chalk_1.default.green('\n✅ Mock data generation complete!'));
-        console.log(chalk_1.default.blue('📁 File path: ') + chalk_1.default.cyan(result.filePath));
-        console.log(chalk_1.default.blue('📊 Records generated: ') + chalk_1.default.cyan(result.recordCount.toString()));
+        console.log(chalk.green('\n✅ Mock data generation complete!'));
+        console.log(chalk.blue('📁 File path: ') + chalk.cyan(result.filePath));
+        console.log(chalk.blue('📊 Records generated: ') + chalk.cyan(result.recordCount.toString()));
         process.exit(0);
     }
     catch (err) {
-        console.error(chalk_1.default.red.bold('\n❌ Unexpected error during mock data generation:'), err instanceof Error ? err.message : String(err));
+        console.error(chalk.red.bold('\n❌ Unexpected error during mock data generation:'), err instanceof Error ? err.message : String(err));
         process.exit(1);
     }
-});
-exports.mock = mock;
+};

package/actions/run.js

@@ -1,62 +1,47 @@
-"use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.run = void 0;
-const chalk_1 = __importDefault(require("chalk"));
-const Environment_1 = __importDefault(require("../engines/Environment"));
-const compile_1 = require("./compile");
-const Helper_1 = __importDefault(require("../helper/Helper"));
-const LicenceManager_1 = __importDefault(require("../licencing/LicenceManager"));
-const ExecutorOrchestrator_1 = __importDefault(require("../executors/ExecutorOrchestrator"));
-const ProcessENVManager_1 = __importDefault(require("../engines/ProcessENVManager"));
-const run = (consumerName, options) => __awaiter(void 0, void 0, void 0, function* () {
+import chalk from 'chalk';
+import { compile } from './compile';
+import { Environment } from '@remora/app-common';
+import { Helper } from '@remora/app-helper';
+import { LicenceManager } from '@remora/app-auth';
+import { ExecutorOrchestrator } from '@remora/app-executors';
+import { ProcessENVManager } from '@remora/app-common';
+export const run = async (consumerName, options) => {
     try {
-        (0, compile_1.compile)();
+        compile();
         // needed for newline
         console.log();
         const consumersToExecute = [];
         if (consumerName && consumerName.length > 0) {
-            const cons = Environment_1.default.getConsumer(consumerName);
+            const cons = Environment.getConsumer(consumerName);
             if (!cons)
                 throw new Error(`Consumer with name "${consumerName}" was not found.`);
             consumersToExecute.push(cons);
         }
-        else if ((options === null || options === void 0 ? void 0 : options.project) && options.project.length > 0) {
-            const projectConsumers = Environment_1.default._env.consumers.filter(c => { var _a; return ((_a = c.project) === null || _a === void 0 ? void 0 : _a.toLowerCase()) === options.project.toLowerCase(); });
+        else if (options?.project && options.project.length > 0) {
+            const projectConsumers = Environment._env.consumers.filter(c => c.project?.toLowerCase() === options.project.toLowerCase());
             if (projectConsumers.length === 0) {
                 throw new Error(`No consumers found for project "${options.project}".`);
             }
             consumersToExecute.push(...projectConsumers);
         }
         else {
-            consumersToExecute.push(...Environment_1.default._env.consumers);
+            consumersToExecute.push(...Environment._env.consumers);
         }
         if (consumersToExecute.length === 1)
-            console.log(chalk_1.default.blue(`Running consumer ${consumersToExecute[0].name}...\n`));
+            console.log(chalk.blue(`Running consumer ${consumersToExecute[0].name}...\n`));
         else
-            console.log(chalk_1.default.blue(`Running consumers ${consumersToExecute.map(x => x.name).join(', ')}...\n`));
+            console.log(chalk.blue(`Running consumers ${consumersToExecute.map(x => x.name).join(', ')}...\n`));
         const results = [];
         for (let i = 0; i < consumersToExecute.length; i++) {
             const consumer = consumersToExecute[i];
             try {
-                const remoraLicenceKey = ProcessENVManager_1.default.getEnvVariable('REMORA_LICENCE_KEY');
-                const check = LicenceManager_1.default.validate(remoraLicenceKey);
+                const remoraLicenceKey = ProcessENVManager.getEnvVariable('REMORA_LICENCE_KEY');
+                const check = LicenceManager.validate(remoraLicenceKey);
                 if (!check.valid) {
                     console.error(`Invalid Remora licence key, the product is not active: remember to set "REMORA_LICENCE_KEY" environment variable.`);
                     process.exit(1);
                 }
-                const response = yield ExecutorOrchestrator_1.default.launch({
+                const response = await ExecutorOrchestrator.launch({
                     consumer,
                     details: {
                         invokedBy: 'CLI',
@@ -67,9 +52,9 @@ const run = (consumerName, options) => __awaiter(void 0, void 0, void 0, functio
                 results.push({ success: true, consumer, response });
             }
             catch (error) {
-                const myErr = Helper_1.default.asError(error);
+                const myErr = Helper.asError(error);
                 results.push({ success: false, consumer, error: myErr.message });
-                if (Helper_1.default.isDev())
+                if (Helper.isDev())
                     console.log(myErr.stack);
             }
         }
@@ -77,40 +62,37 @@ const run = (consumerName, options) => __awaiter(void 0, void 0, void 0, functio
             if (success) {
                 const { elapsedMS, outputCount, resultUri } = response;
                 const rowCount = outputCount;
-                const duration = Helper_1.default.formatDuration(elapsedMS);
-                const performanceInfo = chalk_1.default.gray(` (${rowCount} rows, ${duration})`);
+                const duration = Helper.formatDuration(elapsedMS);
+                const performanceInfo = chalk.gray(` (${rowCount} rows, ${duration})`);
                 if (resultUri)
-                    console.log(chalk_1.default.green(`• Consumer ${consumer.name} -> ${resultUri}`) + performanceInfo);
+                    console.log(chalk.green(`• Consumer ${consumer.name} -> ${resultUri}`) + performanceInfo);
                 else
-                    console.log(chalk_1.default.green(`• Consumer ${consumer.name} `) + performanceInfo);
+                    console.log(chalk.green(`• Consumer ${consumer.name} `) + performanceInfo);
             }
             else {
-                console.log(chalk_1.default.red(`• Consumer ${consumer.name} -> Failed: ${error}`));
+                console.log(chalk.red(`• Consumer ${consumer.name} -> Failed: ${error}`));
             }
         });
         // Calculate totals for successful runs
         const successfulResults = results.filter(x => x.success);
         const totalRows = successfulResults.reduce((sum, result) => {
-            var _a, _b;
-            return sum + ((_b = (_a = result.response) === null || _a === void 0 ? void 0 : _a.outputCount) !== null && _b !== void 0 ? _b : 0);
+            return sum + (result.response?.outputCount ?? 0);
         }, 0);
         const totalDuration = successfulResults.reduce((sum, result) => {
-            var _a;
-            return sum + (((_a = result.response) === null || _a === void 0 ? void 0 : _a.elapsedMS) || 0);
+            return sum + (result.response?.elapsedMS || 0);
         }, 0);
-        const totalsInfo = chalk_1.default.gray(` (${totalRows} rows, ${Helper_1.default.formatDuration(totalDuration)})`);
+        const totalsInfo = chalk.gray(` (${totalRows} rows, ${Helper.formatDuration(totalDuration)})`);
         if (results.some(x => !x.success))
-            console.log(chalk_1.default.blueBright('\nℹ️ Run completed with errors') + totalsInfo);
+            console.log(chalk.blueBright('\nℹ️ Run completed with errors') + totalsInfo);
         else
-            console.log(chalk_1.default.green('\n✅ Run complete!') + totalsInfo);
+            console.log(chalk.green('\n✅ Run complete!') + totalsInfo);
         process.exit(1);
     }
     catch (err) {
-        const myErr = Helper_1.default.asError(err);
-        console.error(chalk_1.default.red.bold('\n❌ Unexpected error during run:'), myErr.message);
-        if (Helper_1.default.isDev())
+        const myErr = Helper.asError(err);
+        console.error(chalk.red.bold('\n❌ Unexpected error during run:'), myErr.message);
+        if (Helper.isDev())
             console.log(myErr.stack);
         process.exit(1);
     }
-});
-exports.run = run;
+};

package/actions/sample.js

@@ -1,88 +1,72 @@
-"use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.sample = void 0;
-const chalk_1 = __importDefault(require("chalk"));
-const ora_1 = __importDefault(require("ora"));
-const Environment_1 = __importDefault(require("../engines/Environment"));
-const ProducerEngine_1 = __importDefault(require("../engines/producer/ProducerEngine"));
-const DatasetRecord_1 = __importDefault(require("../engines/dataset/DatasetRecord"));
-const compile_1 = require("./compile");
-const Helper_1 = __importDefault(require("../helper/Helper"));
-const sample = (resourceName_1, ...args_1) => __awaiter(void 0, [resourceName_1, ...args_1], void 0, function* (resourceName, sampleSize = 10) {
+import chalk from 'chalk';
+import ora from 'ora';
+import { Environment } from '@remora/app-common';
+import { ProducerEngine } from '@remora/app-engines';
+import { DatasetRecord } from '@remora/app-engines';
+import { compile } from './compile';
+import { Helper } from '@remora/app-helper';
+export const sample = async (resourceName, sampleSize = 10) => {
     try {
-        (0, compile_1.compile)();
+        compile();
         console.log(); // needed for newline
-        const spinner = (0, ora_1.default)(chalk_1.default.blue('Sampling dataset...')).start();
+        const spinner = ora(chalk.blue('Sampling dataset...')).start();
         // Try to find the resource as a producer first, then as a consumer
-        const producer = Environment_1.default.getProducer(resourceName);
-        const consumer = Environment_1.default.getConsumer(resourceName);
+        const producer = Environment.getProducer(resourceName);
+        const consumer = Environment.getConsumer(resourceName);
         if (!producer && !consumer) {
-            spinner.fail(chalk_1.default.red(`Resource "${resourceName}" not found. Please check if it exists as a producer or consumer.`));
+            spinner.fail(chalk.red(`Resource "${resourceName}" not found. Please check if it exists as a producer or consumer.`));
             process.exit(1);
         }
         let sampleData;
         let resourceType;
         if (producer) {
             resourceType = 'Producer';
-            spinner.text = chalk_1.default.blue(`Sampling from producer "${resourceName}"...`);
-            sampleData = yield ProducerEngine_1.default.readSampleData(producer, sampleSize, false);
+            spinner.text = chalk.blue(`Sampling from producer "${resourceName}"...`);
+            sampleData = await ProducerEngine.readSampleData(producer, sampleSize, false);
         }
         else {
             resourceType = 'Consumer';
-            spinner.text = chalk_1.default.blue(`Sampling from consumer "${resourceName}"...`);
-            sampleData = yield sampleFromConsumer(consumer, sampleSize);
+            spinner.text = chalk.blue(`Sampling from consumer "${resourceName}"...`);
+            sampleData = await sampleFromConsumer(consumer, sampleSize);
         }
-        spinner.succeed(chalk_1.default.green(`Sample data retrieved from ${resourceType.toLowerCase()} "${resourceName}"`));
+        spinner.succeed(chalk.green(`Sample data retrieved from ${resourceType.toLowerCase()} "${resourceName}"`));
         if (sampleData.length === 0) {
-            console.log(chalk_1.default.yellow('No data found in the dataset.'));
+            console.log(chalk.yellow('No data found in the dataset.'));
             return;
         }
         // Display the sample data
-        console.log(chalk_1.default.cyan(`\n📊 Sample Data (showing ${sampleData.length} rows):`));
-        console.log(chalk_1.default.gray('─'.repeat(80)));
+        console.log(chalk.cyan(`\n📊 Sample Data (showing ${sampleData.length} rows):`));
+        console.log(chalk.gray('─'.repeat(80)));
         displayDataAsTable(sampleData);
-        console.log(chalk_1.default.gray('─'.repeat(80)));
-        console.log(chalk_1.default.green(`✅ Successfully sampled ${sampleData.length} rows from ${resourceType.toLowerCase()} "${resourceName}"`));
+        console.log(chalk.gray('─'.repeat(80)));
+        console.log(chalk.green(`✅ Successfully sampled ${sampleData.length} rows from ${resourceType.toLowerCase()} "${resourceName}"`));
     }
     catch (err) {
-        const myErr = Helper_1.default.asError(err);
-        console.error(chalk_1.default.red.bold('\n❌ Error during sampling:'), myErr.message);
-        if (Helper_1.default.isDev())
+        const myErr = Helper.asError(err);
+        console.error(chalk.red.bold('\n❌ Error during sampling:'), myErr.message);
+        if (Helper.isDev())
             console.log(myErr.stack);
         process.exit(1);
     }
-});
-exports.sample = sample;
-const sampleFromConsumer = (consumer, sampleSize) => __awaiter(void 0, void 0, void 0, function* () {
+};
+const sampleFromConsumer = async (consumer, sampleSize) => {
     // For consumers, we need to get sample data from the first producer
     // and then apply the consumer's field mappings to show what the output would look like
     const firstProducerRef = consumer.producers[0];
     if (!firstProducerRef) {
         throw new Error(`Consumer "${consumer.name}" has no producers configured`);
     }
-    const producer = Environment_1.default.getProducer(firstProducerRef.name);
+    const producer = Environment.getProducer(firstProducerRef.name);
     if (!producer) {
-        const subConsumer = Environment_1.default.getConsumer(firstProducerRef.name);
+        const subConsumer = Environment.getConsumer(firstProducerRef.name);
         if (!subConsumer) {
             throw new Error(`Producer or consumer "${firstProducerRef.name}" not found for consumer "${consumer.name}"`);
         }
         // If it's a consumer that references another consumer, sample from that consumer
-        return yield sampleFromConsumer(subConsumer, sampleSize);
+        return await sampleFromConsumer(subConsumer, sampleSize);
     }
     // Get raw sample data from the producer
-    const rawSampleData = yield ProducerEngine_1.default.readSampleData(producer, sampleSize, false);
+    const rawSampleData = await ProducerEngine.readSampleData(producer, sampleSize, false);
     // For consumers with wildcard fields ("*"), return all data as-is
     const hasWildcard = consumer.fields.some(field => field.key === '*');
     if (hasWildcard) {
@@ -91,7 +75,7 @@ const sampleFromConsumer = (consumer, sampleSize) => __awaiter(void 0, void 0, v
     // For consumers with specific field mappings, show only the mapped fields
     // This gives users a preview of what the consumer output would look like
     const mappedData = rawSampleData.map(record => {
-        const mappedRecord = new DatasetRecord_1.default('', [], record._delimiter);
+        const mappedRecord = new DatasetRecord('', [], record._delimiter);
         consumer.fields.forEach(field => {
             if (field.key !== '*') {
                 const sourceValue = record.getValue(field.key);
@@ -102,7 +86,7 @@ const sampleFromConsumer = (consumer, sampleSize) => __awaiter(void 0, void 0, v
         return mappedRecord;
     });
     return mappedData;
-});
+};
 const displayDataAsTable = (data) => {
     if (data.length === 0)
         return;
@@ -132,7 +116,7 @@ const displayDataAsTable = (data) => {
         columnWidths[field] = Math.min(columnWidths[field], maxColumnWidth);
     });
     // Print header
-    const headerRow = fields.map(field => chalk_1.default.bold(field.padEnd(columnWidths[field]))).join(' │ ');
+    const headerRow = fields.map(field => chalk.bold(field.padEnd(columnWidths[field]))).join(' │ ');
     console.log('│ ' + headerRow + ' │');
     // Print separator
     const separator = fields.map(field => '─'.repeat(columnWidths[field])).join('─┼─');
@@ -149,28 +133,28 @@ const displayDataAsTable = (data) => {
         }).join(' │ ');
         // Alternate row colors for better readability
         if (index % 2 === 0) {
-            console.log('│ ' + chalk_1.default.white(dataRow) + ' │');
+            console.log('│ ' + chalk.white(dataRow) + ' │');
         }
         else {
-            console.log('│ ' + chalk_1.default.gray(dataRow) + ' │');
+            console.log('│ ' + chalk.gray(dataRow) + ' │');
         }
     });
 };
 const formatValue = (value) => {
     if (value === null || value === undefined) {
-        return chalk_1.default.dim('null');
+        return chalk.dim('null');
     }
     if (typeof value === 'string') {
         return value;
     }
     if (typeof value === 'number') {
-        return chalk_1.default.cyan(value.toString());
+        return chalk.cyan(value.toString());
     }
     if (typeof value === 'boolean') {
-        return chalk_1.default.yellow(value.toString());
+        return chalk.yellow(value.toString());
     }
     if (value instanceof Date) {
-        return chalk_1.default.magenta(value.toISOString());
+        return chalk.magenta(value.toISOString());
     }
-    return chalk_1.default.dim(JSON.stringify(value));
+    return chalk.dim(JSON.stringify(value));
 };