import native from 'k6/x/neofs/native';
import registry from 'k6/x/neofs/registry';
import s3 from 'k6/x/neofs/s3';
import { sleep } from 'k6';
import exec from 'k6/execution';

/*
  ./k6 run -e CLIENTS=200 -e TIME_LIMIT=30 -e GRPC_ENDPOINTS=node4.data:8084 
    -e REGISTRY_FILE=registry.bolt scenarios/verify.js
*/

const obj_registry = registry.open(__ENV.REGISTRY_FILE);

// Time limit (in seconds) for the run
const time_limit = __ENV.TIME_LIMIT || "60";

// Connect to random gRPC endpoint
let grpc_client = undefined;
if (__ENV.GRPC_ENDPOINTS) {
    const grpcEndpoints = __ENV.GRPC_ENDPOINTS.split(',');
    const grpcEndpoint = grpcEndpoints[Math.floor(Math.random() * grpcEndpoints.length)];
    grpc_client = native.connect(grpcEndpoint, '');
}

// Connect to random S3 endpoint
let s3_client = undefined;
if (__ENV.S3_ENDPOINTS) {
    const s3_endpoints = __ENV.S3_ENDPOINTS.split(',');
    const s3_endpoint = s3_endpoints[Math.floor(Math.random() * s3_endpoints.length)];
    s3_client = s3.connect(`http://${s3_endpoint}`);
}

const scenarios = {
    verify: {
        executor: 'constant-vus',
        vus: __ENV.CLIENTS,
        duration: `${time_limit}s`,
        exec: 'obj_verify',
        gracefulStop: '5s',
    }
};

export const options = {
    scenarios: scenarios,
    setupTimeout: '5s',
};

export function obj_verify() {
    if (__ENV.SLEEP) {
        sleep(__ENV.SLEEP);
    }

    const obj = obj_registry.nextObjectToVerify();
    if (!obj) {
        // TODO: consider using a metric with abort condition to stop execution when
        // all VUs have no objects to verify. Alternative solution could be a
        // shared-iterations executor, but it might be not a good choice, as we need to
        // check same object several times (if specific request fails)

        // Allow time for other VUs to complete verification
        sleep(30.0);
        exec.test.abort("All objects have been verified");
    }
    console.log(`Verifying object ${obj.id}`);

    let result = undefined;
    if (obj.c_id && obj.o_id) {
        result = grpc_client.verifyHash(obj.c_id, obj.o_id, obj.payload_hash);
    } else if (obj.s3_bucket && obj.s3_key) {
        result = s3_client.verifyHash(obj.s3_bucket, obj.s3_key, obj.payload_hash);
    } else {
        console.log(`Object id=${obj.id} cannot be verified with supported protocols`);
        obj_registry.setObjectStatus(obj.id, "skipped");
    }

    if (result.success) {
        obj_registry.setObjectStatus(obj.id, "verified");
    } else {
        obj_registry.setObjectStatus(obj.id, "invalid");
        console.log(`Verify error on ${obj.c_id}/${obj.o_id}: {resp.error}`);
    }
}