appwrite-utils-cli 1.5.2 → 1.6.0

package/dist/main.js

@@ -14,6 +14,47 @@ import { listSpecifications } from "./functions/methods.js";
 import { MessageFormatter } from "./shared/messageFormatter.js";
 import { ConfirmationDialogs } from "./shared/confirmationDialogs.js";
 import path from "path";
+import fs from "fs";
+import { loadAppwriteProjectConfig, findAppwriteProjectConfig, projectConfigToAppwriteConfig } from "./utils/projectConfig.js";
+import { hasSessionAuth, getAvailableSessions, getAuthenticationStatus } from "./utils/sessionAuth.js";
+import { findYamlConfig, loadYamlConfigWithSession } from "./config/yamlConfig.js";
+/**
+ * Checks if the migration from collections to tables should be allowed
+ * Returns an object with:
+ * - allowed: boolean indicating if migration should proceed
+ * - reason: string explaining why migration was blocked (if not allowed)
+ */
+function checkMigrationConditions(configPath) {
+    const collectionsPath = path.join(configPath, "collections");
+    const tablesPath = path.join(configPath, "tables");
+    // Check if collections/ folder exists
+    if (!fs.existsSync(collectionsPath)) {
+        return {
+            allowed: false,
+            reason: "No collections/ folder found. Migration requires existing collections to migrate."
+        };
+    }
+    // Check if collections/ folder has YAML files
+    const collectionFiles = fs.readdirSync(collectionsPath).filter(file => file.endsWith(".yaml") || file.endsWith(".yml"));
+    if (collectionFiles.length === 0) {
+        return {
+            allowed: false,
+            reason: "No YAML files found in collections/ folder. Migration requires existing collection YAML files."
+        };
+    }
+    // Check if tables/ folder exists and has YAML files
+    if (fs.existsSync(tablesPath)) {
+        const tableFiles = fs.readdirSync(tablesPath).filter(file => file.endsWith(".yaml") || file.endsWith(".yml"));
+        if (tableFiles.length > 0) {
+            return {
+                allowed: false,
+                reason: `Tables folder already exists with ${tableFiles.length} YAML file(s). Migration appears to have already been completed.`
+            };
+        }
+    }
+    // All conditions met
+    return { allowed: true };
+}
 const argv = yargs(hideBin(process.argv))
     .option("config", {
     type: "string",
@@ -29,9 +70,9 @@ const argv = yargs(hideBin(process.argv))
     description: "Comma-separated list of database IDs to target (e.g., 'db1,db2,db3')",
 })
     .option("collectionIds", {
-    alias: ["collIds"],
+    alias: ["collIds", "tableIds", "tables"],
     type: "string",
-    description: "Comma-separated list of collection IDs to target (e.g., 'users,posts')",
+    description: "Comma-separated list of collection/table IDs to target (e.g., 'users,posts')",
 })
     .option("bucketIds", {
     type: "string",
@@ -43,7 +84,7 @@ const argv = yargs(hideBin(process.argv))
 })
     .option("wipeCollections", {
     type: "boolean",
-    description: "⚠️  DESTRUCTIVE: Wipe specific collections (requires --collectionIds)",
+    description: "⚠️  DESTRUCTIVE: Wipe specific collections/tables (requires --collectionIds or --tableIds)",
 })
     .option("transferUsers", {
     type: "boolean",
@@ -60,6 +101,31 @@ const argv = yargs(hideBin(process.argv))
     .option("backup", {
     type: "boolean",
     description: "Create a complete backup of your databases and collections",
+})
+    .option("backupFormat", {
+    type: "string",
+    choices: ["json", "zip"],
+    default: "json",
+    description: "Backup file format (json or zip)",
+})
+    .option("listBackups", {
+    type: "boolean",
+    description: "List all backups for databases",
+})
+    .option("comprehensiveBackup", {
+    alias: ["comprehensive", "backup-all"],
+    type: "boolean",
+    description: "🚀 Create comprehensive backup of ALL databases and ALL storage buckets",
+})
+    .option("trackingDatabaseId", {
+    alias: ["tracking-db"],
+    type: "string",
+    description: "Database ID to use for centralized backup tracking (interactive prompt if not specified)",
+})
+    .option("parallelDownloads", {
+    type: "number",
+    default: 10,
+    description: "Number of parallel file downloads for bucket backups (default: 10)",
 })
     .option("writeData", {
     type: "boolean",
@@ -174,25 +240,130 @@ const argv = yargs(hideBin(process.argv))
     type: "string",
     description: "Output directory for generated constants files (default: config-folder/constants)",
     default: "auto",
+})
+    .option("migrateCollectionsToTables", {
+    alias: ["migrate-collections"],
+    type: "boolean",
+    description: "Migrate collections to tables format for TablesDB API compatibility",
+})
+    .option("useSession", {
+    alias: ["session"],
+    type: "boolean",
+    description: "Use Appwrite CLI session authentication instead of API key",
+})
+    .option("sessionCookie", {
+    type: "string",
+    description: "Explicit session cookie to use for authentication",
 })
     .parse();
 async function main() {
     const startTime = Date.now();
     const operationStats = {};
+    // Early session detection for better user guidance
+    const availableSessions = getAvailableSessions();
+    let hasAnyValidSessions = availableSessions.length > 0;
     if (argv.it) {
         const cli = new InteractiveCLI(process.cwd());
         await cli.run();
     }
     else {
-        const directConfig = argv.endpoint || argv.projectId || argv.apiKey
-            ? {
-                appwriteEndpoint: argv.endpoint,
-                appwriteProject: argv.projectId,
-                appwriteKey: argv.apiKey,
-            }
-            : undefined;
-        const controller = new UtilsController(process.cwd(), directConfig);
-        await controller.init();
+        // Enhanced config creation with session and project file support
+        let directConfig = undefined;
+        // Show authentication status on startup if no config provided
+        if (!argv.config && !argv.endpoint && !argv.projectId && !argv.apiKey && !argv.useSession && !argv.sessionCookie) {
+            if (hasAnyValidSessions) {
+                MessageFormatter.info(`Found ${availableSessions.length} available session(s)`, { prefix: "Auth" });
+                availableSessions.forEach(session => {
+                    MessageFormatter.info(`  \u2022 ${session.projectId} (${session.email || 'unknown'}) at ${session.endpoint}`, { prefix: "Auth" });
+                });
+                MessageFormatter.info("Use --session to enable session authentication", { prefix: "Auth" });
+            }
+            else {
+                MessageFormatter.info("No active Appwrite sessions found", { prefix: "Auth" });
+                MessageFormatter.info("\u2022 Run 'appwrite login' to authenticate with session", { prefix: "Auth" });
+                MessageFormatter.info("\u2022 Or provide --apiKey for API key authentication", { prefix: "Auth" });
+            }
+        }
+        // Priority 1: Check for appwrite.json project configuration
+        const projectConfigPath = findAppwriteProjectConfig(process.cwd());
+        if (projectConfigPath) {
+            const projectConfig = loadAppwriteProjectConfig(projectConfigPath);
+            if (projectConfig) {
+                directConfig = projectConfigToAppwriteConfig(projectConfig);
+                MessageFormatter.info(`Loaded project configuration from ${projectConfigPath}`, { prefix: "CLI" });
+            }
+        }
+        // Priority 2: CLI arguments override project config
+        if (argv.endpoint || argv.projectId || argv.apiKey || argv.useSession || argv.sessionCookie) {
+            directConfig = {
+                ...directConfig,
+                appwriteEndpoint: argv.endpoint || directConfig?.appwriteEndpoint,
+                appwriteProject: argv.projectId || directConfig?.appwriteProject,
+                appwriteKey: argv.apiKey || directConfig?.appwriteKey,
+            };
+        }
+        // Priority 3: Session authentication support with improved detection
+        let sessionAuthAvailable = false;
+        if (directConfig?.appwriteEndpoint && directConfig?.appwriteProject) {
+            sessionAuthAvailable = hasSessionAuth(directConfig.appwriteEndpoint, directConfig.appwriteProject);
+        }
+        if (argv.useSession || argv.sessionCookie) {
+            if (argv.sessionCookie) {
+                // Explicit session cookie provided
+                MessageFormatter.info("Using explicit session cookie for authentication", { prefix: "Auth" });
+            }
+            else if (sessionAuthAvailable) {
+                MessageFormatter.info("Session authentication detected and will be used", { prefix: "Auth" });
+            }
+            else {
+                MessageFormatter.warning("Session authentication requested but no valid session found", { prefix: "Auth" });
+                const availableSessions = getAvailableSessions();
+                if (availableSessions.length > 0) {
+                    MessageFormatter.info(`Available sessions: ${availableSessions.map(s => `${s.projectId} (${s.email || 'unknown'})`).join(", ")}`, { prefix: "Auth" });
+                    MessageFormatter.info("Use --session flag to enable session authentication", { prefix: "Auth" });
+                }
+                else {
+                    MessageFormatter.warning("No Appwrite CLI sessions found. Please run 'appwrite login' first.", { prefix: "Auth" });
+                }
+                MessageFormatter.error("Session authentication requested but not available", undefined, { prefix: "Auth" });
+                return; // Exit early if session auth was requested but not available
+            }
+        }
+        else if (sessionAuthAvailable && !argv.apiKey) {
+            // Auto-detect session authentication when no API key is provided
+            MessageFormatter.info("Session authentication detected - no API key required", { prefix: "Auth" });
+            MessageFormatter.info("Use --session flag to explicitly enable session authentication", { prefix: "Auth" });
+        }
+        // Enhanced session authentication support:
+        // 1. If session auth is explicitly requested via flags, use it
+        // 2. If no API key is provided but sessions are available, offer to use session auth
+        // 3. Auto-detect session authentication when possible
+        let finalDirectConfig = directConfig;
+        if ((argv.useSession || argv.sessionCookie) &&
+            (!directConfig || !directConfig.appwriteEndpoint || !directConfig.appwriteProject)) {
+            // Don't pass incomplete directConfig - let UtilsController load YAML config normally
+            finalDirectConfig = null;
+        }
+        else if (finalDirectConfig && !finalDirectConfig.appwriteKey && !argv.useSession && !argv.sessionCookie) {
+            // Auto-detect session authentication when no API key provided
+            if (sessionAuthAvailable) {
+                MessageFormatter.info("No API key provided, but session authentication is available", { prefix: "Auth" });
+                MessageFormatter.info("Automatically using session authentication (add --session to suppress this message)", { prefix: "Auth" });
+                // Implicitly enable session authentication
+                argv.useSession = true;
+            }
+        }
+        // Create controller with session authentication support
+        const controller = new UtilsController(process.cwd(), finalDirectConfig);
+        // Pass session authentication options to the controller
+        const initOptions = {};
+        if (argv.useSession || argv.sessionCookie) {
+            initOptions.useSession = true;
+            if (argv.sessionCookie) {
+                initOptions.sessionCookie = argv.sessionCookie;
+            }
+        }
+        await controller.init(initOptions);
         if (argv.setup) {
             await setupDirsFiles(false, process.cwd());
             return;
@@ -229,11 +400,114 @@ async function main() {
             MessageFormatter.success(`Constants generated in ${outputDir}`, { prefix: "Constants" });
             return;
         }
+        if (argv.migrateCollectionsToTables) {
+            try {
+                if (!controller.config) {
+                    MessageFormatter.error("No Appwrite configuration found", undefined, { prefix: "Migration" });
+                    return;
+                }
+                // Get the config path from the controller or use .appwrite in current directory
+                let configPath = controller.getAppwriteFolderPath();
+                if (!configPath) {
+                    // Try .appwrite in current directory
+                    const defaultPath = path.join(process.cwd(), ".appwrite");
+                    if (fs.existsSync(defaultPath)) {
+                        configPath = defaultPath;
+                    }
+                    else {
+                        MessageFormatter.error("Could not determine configuration folder path", undefined, { prefix: "Migration" });
+                        MessageFormatter.info("Make sure you have a .appwrite/ folder in your current directory", { prefix: "Migration" });
+                        return;
+                    }
+                }
+                // Check if migration conditions are met
+                const migrationCheck = checkMigrationConditions(configPath);
+                if (!migrationCheck.allowed) {
+                    MessageFormatter.error(`Migration not allowed: ${migrationCheck.reason}`, undefined, { prefix: "Migration" });
+                    MessageFormatter.info("Migration requirements:", { prefix: "Migration" });
+                    MessageFormatter.info("  • Configuration must be loaded (use --config or have .appwrite/ folder)", { prefix: "Migration" });
+                    MessageFormatter.info("  • collections/ folder must exist with YAML files", { prefix: "Migration" });
+                    MessageFormatter.info("  • tables/ folder must not exist or be empty", { prefix: "Migration" });
+                    return;
+                }
+                const { migrateCollectionsToTables } = await import("./config/configMigration.js");
+                MessageFormatter.info("Starting collections to tables migration...", { prefix: "Migration" });
+                const result = migrateCollectionsToTables(controller.config, {
+                    strategy: "full_migration",
+                    validateResult: true,
+                    dryRun: false
+                });
+                if (result.success) {
+                    operationStats.migratedCollections = result.changes.length;
+                    MessageFormatter.success("Collections migration completed successfully", { prefix: "Migration" });
+                }
+                else {
+                    MessageFormatter.error(`Migration failed: ${result.errors.join(", ")}`, undefined, { prefix: "Migration" });
+                    process.exit(1);
+                }
+            }
+            catch (error) {
+                MessageFormatter.error("Migration failed", error instanceof Error ? error : new Error(String(error)), { prefix: "Migration" });
+                process.exit(1);
+            }
+            return;
+        }
         if (!controller.config) {
-            MessageFormatter.error("No Appwrite connection found", undefined, { prefix: "CLI" });
+            // Provide better guidance based on available authentication methods
+            const availableSessions = getAvailableSessions();
+            if (availableSessions.length > 0) {
+                MessageFormatter.error("No Appwrite configuration found", undefined, { prefix: "CLI" });
+                MessageFormatter.info("Available authentication options:", { prefix: "Auth" });
+                MessageFormatter.info("• Session authentication: Add --session flag", { prefix: "Auth" });
+                MessageFormatter.info("• API key authentication: Add --apiKey YOUR_API_KEY", { prefix: "Auth" });
+                MessageFormatter.info(`• Available sessions: ${availableSessions.map(s => `${s.projectId} (${s.email || 'unknown'})`).join(", ")}`, { prefix: "Auth" });
+            }
+            else {
+                MessageFormatter.error("No Appwrite configuration found", undefined, { prefix: "CLI" });
+                MessageFormatter.info("Authentication options:", { prefix: "Auth" });
+                MessageFormatter.info("• Login with Appwrite CLI: Run 'appwrite login' then use --session flag", { prefix: "Auth" });
+                MessageFormatter.info("• Use API key: Add --apiKey YOUR_API_KEY", { prefix: "Auth" });
+                MessageFormatter.info("• Create config file: Run with --setup to initialize project configuration", { prefix: "Auth" });
+            }
             return;
         }
         const parsedArgv = argv;
+        // List backups if requested
+        if (parsedArgv.listBackups) {
+            const { AdapterFactory } = await import("./adapters/AdapterFactory.js");
+            const { listBackups } = await import("./shared/backupTracking.js");
+            if (!controller.config) {
+                MessageFormatter.error("No Appwrite configuration found", undefined, { prefix: "Backups" });
+                return;
+            }
+            const { adapter } = await AdapterFactory.create({
+                appwriteEndpoint: controller.config.appwriteEndpoint,
+                appwriteProject: controller.config.appwriteProject,
+                appwriteKey: controller.config.appwriteKey
+            });
+            const databases = parsedArgv.dbIds
+                ? await controller.getDatabasesByIds(parsedArgv.dbIds.split(","))
+                : await fetchAllDatabases(controller.database);
+            if (!databases || databases.length === 0) {
+                MessageFormatter.info("No databases found", { prefix: "Backups" });
+                return;
+            }
+            for (const db of databases) {
+                const backups = await listBackups(adapter, db.$id);
+                MessageFormatter.info(`\nBackups for database: ${db.name} (${db.$id})`, { prefix: "Backups" });
+                if (backups.length === 0) {
+                    MessageFormatter.info("  No backups found", { prefix: "Backups" });
+                }
+                else {
+                    backups.forEach((backup, index) => {
+                        const date = new Date(backup.$createdAt).toLocaleString();
+                        const size = MessageFormatter.formatBytes(backup.sizeBytes);
+                        MessageFormatter.info(`  ${index + 1}. ${date} - ${backup.format.toUpperCase()} - ${size} - ${backup.collections} collections, ${backup.documents} documents`, { prefix: "Backups" });
+                    });
+                }
+            }
+            return;
+        }
         const options = {
             databases: parsedArgv.dbIds
                 ? await controller.getDatabasesByIds(parsedArgv.dbIds.split(","))
@@ -261,13 +535,14 @@ async function main() {
             }
             await controller.updateFunctionSpecifications(parsedArgv.functionId, parsedArgv.specification);
         }
-        // Add default databases if not specified
-        if (!options.databases || options.databases.length === 0) {
+        // Add default databases if not specified (only if we need them for operations)
+        const needsDatabases = options.doBackup || options.wipeDatabase ||
+            options.wipeDocumentStorage || options.wipeUsers ||
+            options.wipeCollections || options.importData ||
+            parsedArgv.sync || parsedArgv.transfer;
+        if (needsDatabases && (!options.databases || options.databases.length === 0)) {
             const allDatabases = await fetchAllDatabases(controller.database);
-            options.databases = allDatabases.filter((db) => {
-                const useMigrations = controller.config?.useMigrations ?? true;
-                return useMigrations || db.name.toLowerCase() !== "migrations";
-            });
+            options.databases = allDatabases;
         }
         // Add default collections if not specified
         if (!options.collections || options.collections.length === 0) {
@@ -278,10 +553,78 @@ async function main() {
                 options.collections = [];
             }
         }
+        // Comprehensive backup (all databases + all buckets)
+        if (parsedArgv.comprehensiveBackup) {
+            const { comprehensiveBackup } = await import("./backups/operations/comprehensiveBackup.js");
+            const { AdapterFactory } = await import("./adapters/AdapterFactory.js");
+            // Get tracking database ID (interactive prompt if not specified)
+            let trackingDatabaseId = parsedArgv.trackingDatabaseId;
+            if (!trackingDatabaseId) {
+                // Fetch all databases for selection
+                const allDatabases = await fetchAllDatabases(controller.database);
+                if (allDatabases.length === 0) {
+                    MessageFormatter.error("No databases found. Cannot create comprehensive backup without a tracking database.", undefined, { prefix: "Backup" });
+                    return;
+                }
+                if (allDatabases.length === 1) {
+                    trackingDatabaseId = allDatabases[0].$id;
+                    MessageFormatter.info(`Using only available database for tracking: ${allDatabases[0].name} (${trackingDatabaseId})`, { prefix: "Backup" });
+                }
+                else {
+                    // Interactive selection
+                    const inquirer = (await import("inquirer")).default;
+                    const answer = await inquirer.prompt([{
+                            type: 'list',
+                            name: 'trackingDb',
+                            message: 'Select database to store backup tracking metadata:',
+                            choices: allDatabases.map(db => ({
+                                name: `${db.name} (${db.$id})`,
+                                value: db.$id
+                            }))
+                        }]);
+                    trackingDatabaseId = answer.trackingDb;
+                }
+            }
+            // Ensure trackingDatabaseId is defined before proceeding
+            if (!trackingDatabaseId) {
+                throw new Error('Tracking database ID is required for comprehensive backup');
+            }
+            MessageFormatter.info(`Using tracking database: ${trackingDatabaseId}`, { prefix: "Backup" });
+            // Create adapter for backup tracking
+            const { adapter } = await AdapterFactory.create({
+                appwriteEndpoint: controller.config.appwriteEndpoint,
+                appwriteProject: controller.config.appwriteProject,
+                appwriteKey: controller.config.appwriteKey,
+                sessionCookie: controller.config.sessionCookie
+            });
+            const result = await comprehensiveBackup(controller.config, controller.database, controller.storage, adapter, {
+                trackingDatabaseId,
+                backupFormat: parsedArgv.backupFormat || 'zip',
+                parallelDownloads: parsedArgv.parallelDownloads || 10,
+                onProgress: (message) => {
+                    MessageFormatter.info(message, { prefix: "Backup" });
+                }
+            });
+            operationStats.comprehensiveBackup = 1;
+            operationStats.databasesBackedUp = result.databaseBackups.length;
+            operationStats.bucketsBackedUp = result.bucketBackups.length;
+            operationStats.totalBackupSize = result.totalSizeBytes;
+            if (result.status === 'completed') {
+                MessageFormatter.success(`Comprehensive backup completed successfully (ID: ${result.backupId})`, { prefix: "Backup" });
+            }
+            else if (result.status === 'partial') {
+                MessageFormatter.warning(`Comprehensive backup completed with errors (ID: ${result.backupId})`, { prefix: "Backup" });
+                result.errors.forEach(err => MessageFormatter.warning(err, { prefix: "Backup" }));
+            }
+            else {
+                MessageFormatter.error(`Comprehensive backup failed (ID: ${result.backupId})`, undefined, { prefix: "Backup" });
+                result.errors.forEach(err => MessageFormatter.error(err, undefined, { prefix: "Backup" }));
+            }
+        }
         if (options.doBackup && options.databases) {
-            MessageFormatter.info(`Creating backups for ${options.databases.length} database(s)`, { prefix: "Backup" });
+            MessageFormatter.info(`Creating backups for ${options.databases.length} database(s) in ${parsedArgv.backupFormat} format`, { prefix: "Backup" });
             for (const db of options.databases) {
-                await controller.backupDatabase(db);
+                await controller.backupDatabase(db, parsedArgv.backupFormat || 'json');
             }
             operationStats.backups = options.databases.length;
             MessageFormatter.success(`Backup completed for ${options.databases.length} database(s)`, { prefix: "Backup" });

package/dist/migrations/afterImportActions.js

@@ -1,26 +1,18 @@
-import { Databases, Storage, Query, ID, Client, Compression, } from "node-appwrite";
+import { Databases, Storage, Query, ID, Compression, } from "node-appwrite";
 import { InputFile } from "node-appwrite/file";
 import path from "path";
 import fs from "fs";
 import os from "os";
 import { logger } from "../shared/logging.js";
 import { tryAwaitWithRetry, } from "appwrite-utils";
+import { getClientFromConfig } from "../utils/getClientFromConfig.js";
+import { MessageFormatter } from "../shared/messageFormatter.js";
 export const getDatabaseFromConfig = (config) => {
-    if (!config.appwriteClient) {
-        config.appwriteClient = new Client()
-            .setEndpoint(config.appwriteEndpoint)
-            .setProject(config.appwriteProject)
-            .setKey(config.appwriteKey);
-    }
+    getClientFromConfig(config); // Sets config.appwriteClient if missing
     return new Databases(config.appwriteClient);
 };
 export const getStorageFromConfig = (config) => {
-    if (!config.appwriteClient) {
-        config.appwriteClient = new Client()
-            .setEndpoint(config.appwriteEndpoint)
-            .setProject(config.appwriteProject)
-            .setKey(config.appwriteKey);
-    }
+    getClientFromConfig(config); // Sets config.appwriteClient if missing
     return new Storage(config.appwriteClient);
 };
 export const afterImportActions = {
@@ -30,7 +22,7 @@ export const afterImportActions = {
             await tryAwaitWithRetry(async () => await db.updateDocument(dbId, collId, docId, data));
         }
         catch (error) {
-            console.error("Error updating document: ", error);
+            MessageFormatter.error("Error updating document", error instanceof Error ? error : new Error(String(error)), { prefix: "Import" });
         }
     },
     checkAndUpdateFieldInDocument: async (config, dbId, collId, docId, fieldName, oldFieldValue, newFieldValue) => {
@@ -44,7 +36,7 @@ export const afterImportActions = {
             }
         }
         catch (error) {
-            console.error("Error updating document: ", error);
+            MessageFormatter.error("Error updating document", error instanceof Error ? error : new Error(String(error)), { prefix: "Import" });
         }
     },
     setFieldFromOtherCollectionDocument: async (config, dbId, collIdOrName, docId, fieldName, otherCollIdOrName, otherDocId, otherFieldName) => {
@@ -76,10 +68,10 @@ export const afterImportActions = {
                     [fieldName]: valueToSet,
                 }));
             }
-            console.log(`Field ${fieldName} updated successfully in document ${docId}.`);
+            MessageFormatter.success(`Field ${fieldName} updated successfully in document ${docId}`, { prefix: "Import" });
         }
         catch (error) {
-            console.error("Error setting field from other collection document: ", error);
+            MessageFormatter.error("Error setting field from other collection document", error instanceof Error ? error : new Error(String(error)), { prefix: "Import" });
         }
     },
     /**
@@ -136,11 +128,11 @@ export const afterImportActions = {
                     ? { [fieldName]: documentIds }
                     : { [fieldName]: documentIds[0] };
                 await tryAwaitWithRetry(async () => await db.updateDocument(dbId, targetCollectionId, docId, updatePayload));
-                console.log(`Field ${fieldName} updated successfully in document ${docId} with ${documentIds.length} document IDs.`);
+                MessageFormatter.success(`Field ${fieldName} updated successfully in document ${docId} with ${documentIds.length} document IDs`, { prefix: "Import" });
             }
         }
         catch (error) {
-            console.error("Error setting field from other collection documents: ", error);
+            MessageFormatter.error("Error setting field from other collection documents", error instanceof Error ? error : new Error(String(error)), { prefix: "Import" });
         }
     },
     setTargetFieldFromOtherCollectionDocumentsByMatchingField: async (config, dbId, collIdOrName, docId, fieldName, otherCollIdOrName, matchingFieldName, matchingFieldValue, targetField) => {
@@ -190,11 +182,11 @@ export const afterImportActions = {
                     ? { [fieldName]: targetFieldValues }
                     : { [fieldName]: targetFieldValues[0] };
                 await tryAwaitWithRetry(async () => await db.updateDocument(dbId, targetCollectionId, docId, updatePayload));
-                console.log(`Field ${fieldName} updated successfully in document ${docId} with values from field ${targetField}.`);
+                MessageFormatter.success(`Field ${fieldName} updated successfully in document ${docId} with values from field ${targetField}`, { prefix: "Import" });
             }
         }
         catch (error) {
-            console.error("Error setting field from other collection documents: ", error);
+            MessageFormatter.error("Error setting field from other collection documents", error instanceof Error ? error : new Error(String(error)), { prefix: "Import" });
         }
     },
     createOrGetBucket: async (config, bucketName, bucketId, permissions, fileSecurity, enabled, maxFileSize, allowedExtensions, compression, encryption, antivirus) => {
@@ -217,7 +209,7 @@ export const afterImportActions = {
             }
         }
         catch (error) {
-            console.error("Error creating or getting bucket: ", error);
+            MessageFormatter.error("Error creating or getting bucket", error instanceof Error ? error : new Error(String(error)), { prefix: "Import" });
         }
     },
     createFileAndUpdateField: async (config, dbId, collId, docId, fieldName, bucketId, filePath, fileName) => {
@@ -231,12 +223,12 @@ export const afterImportActions = {
             //   `Processing field ${fieldName} in collection ${collId} for document ${docId} in database ${dbId} in bucket ${bucketId} with path ${filePath} and name ${fileName}...`
             // );
             if (filePath.length === 0 || fileName.length === 0) {
-                console.error(`File path or name is empty for field ${fieldName} in collection ${collId}, skipping...`);
+                MessageFormatter.error(`File path or name is empty for field ${fieldName} in collection ${collId}, skipping...`, undefined, { prefix: "Import" });
                 return;
             }
             let isArray = false;
             if (!attribute) {
-                console.log(`Field ${fieldName} not found in collection ${collId}, weird, skipping...`);
+                MessageFormatter.warning(`Field ${fieldName} not found in collection ${collId}, weird, skipping...`, { prefix: "Import" });
                 return;
             }
             else if (attribute.array === true) {
@@ -260,7 +252,7 @@ export const afterImportActions = {
                 // Download the file using fetch
                 const response = await tryAwaitWithRetry(async () => await fetch(filePath));
                 if (!response.ok)
-                    console.error(`Failed to fetch ${filePath}: ${response.statusText} for document ${docId} with field ${fieldName}`);
+                    MessageFormatter.error(`Failed to fetch ${filePath}: ${response.statusText} for document ${docId} with field ${fieldName}`, undefined, { prefix: "Import" });
                 // Use arrayBuffer if buffer is not available
                 const arrayBuffer = await response.arrayBuffer();
                 const buffer = Buffer.from(arrayBuffer);
@@ -269,7 +261,7 @@ export const afterImportActions = {
                 const inputFile = InputFile.fromPath(tempFilePath, fileName);
                 // Use the full file name (with extension) for creating the file
                 const file = await tryAwaitWithRetry(async () => await storage.createFile(bucketId, ID.unique(), inputFile));
-                console.log("Created file from URL: ", file.$id);
+                MessageFormatter.success(`Created file from URL: ${file.$id}`, { prefix: "Import" });
                 // After uploading, adjust the updateData based on whether the field is an array or not
                 if (isArray) {
                     updateData = [...updateData, file.$id]; // Append the new file ID
@@ -287,7 +279,7 @@ export const afterImportActions = {
                 const files = fs.readdirSync(filePath);
                 const fileFullName = files.find((file) => file.includes(fileName));
                 if (!fileFullName) {
-                    console.error(`File starting with '${fileName}' not found in '${filePath}'`);
+                    MessageFormatter.error(`File starting with '${fileName}' not found in '${filePath}'`, undefined, { prefix: "Import" });
                     return;
                 }
                 const pathToFile = path.join(filePath, fileFullName);
@@ -302,13 +294,13 @@ export const afterImportActions = {
                 tryAwaitWithRetry(async () => await db.updateDocument(dbId, collId, doc.$id, {
                     [fieldName]: updateData,
                 }));
-                console.log("Created file from path: ", file.$id);
+                MessageFormatter.success(`Created file from path: ${file.$id}`, { prefix: "Import" });
             }
         }
         catch (error) {
             logger.error(`Error creating file and updating field, params were:\ndbId: ${dbId}, collId: ${collId}, docId: ${docId}, fieldName: ${fieldName}, filePath: ${filePath}, fileName: ${fileName}\n\nError: ${error}`);
-            console.error("Error creating file and updating field: ", error);
-            console.log(`Params were: dbId: ${dbId}, collId: ${collId}, docId: ${docId}, fieldName: ${fieldName}, filePath: ${filePath}, fileName: ${fileName}`);
+            MessageFormatter.error("Error creating file and updating field", error instanceof Error ? error : new Error(String(error)), { prefix: "Import" });
+            MessageFormatter.info(`Params were: dbId: ${dbId}, collId: ${collId}, docId: ${docId}, fieldName: ${fieldName}, filePath: ${filePath}, fileName: ${fileName}`, { prefix: "Import" });
         }
     },
 };