import { 
    Credentials, 
    Env, 
    LogLevel, 
    QueryRequest, 
    QueryResponse, 
    Skyflow, 
    SkyflowConfig, 
    SkyflowError, 
    VaultConfig 
} from 'skyflow-node';

/**
 * Skyflow Query Example
 * 
 * This example demonstrates how to:
 * 1. Configure credentials
 * 2. Set up vault configuration
 * 3. Execute a query on the vault
 * 4. Handle response and errors
 */
async function executeQuery() {
    try {
        // Step 1: Configure Credentials
        const credentials: Credentials = {
            // Using API Key authentication
            apiKey: 'your-skyflow-api-key',
        };

        // Step 2: Configure Vault
        const primaryVaultConfig: VaultConfig = {
            vaultId: 'your-vault-id',          // Unique vault identifier
            clusterId: 'your-cluster-id',      // From vault URL
            env: Env.PROD,                     // Deployment environment
            credentials: credentials           // Authentication method
        };

        // Step 3: Configure Skyflow Client
        const skyflowConfig: SkyflowConfig = {
            vaultConfigs: [primaryVaultConfig],
            logLevel: LogLevel.ERROR,               // Logging verbosity
        };

        // Initialize Skyflow Client
        const skyflowClient: Skyflow = new Skyflow(skyflowConfig);

        // Step 4: Prepare Query
        const query: string = 'select * from table_name limit 1'; // Example query
        const queryRequest: QueryRequest = new QueryRequest(query);

        // Step 5: Execute Query
        const response: QueryResponse = await skyflowClient
            .vault(primaryVaultConfig.vaultId)
            .query(queryRequest);

        // Handle Successful Response
        // fields, tokenizedData, and errors are always present in QueryResponse
        console.log('Query Result:', response);
        response.fields.forEach(record => {
            console.log('Fields:', record);
            console.log('Tokenized Data:', record.tokenizedData);
        });
        if (response.errors !== null) {
            console.error('Query Errors:', response.errors);
        }

    } catch (error) {
        // Comprehensive Error Handling
        if (error instanceof SkyflowError) {
            console.error('Skyflow Specific Error:', {
                httpCode: error.error?.httpCode,
                grpcCode: error.error?.grpcCode,
                httpStatus: error.error?.httpStatus,
                message: error.message,
                details: error.error?.details,
            });
        } else {
            console.error('Unexpected Error:', error);
        }
    }
}

// Invoke the query function
executeQuery();