<?php

namespace DeliciousBrains\WP_Offload_Media\Gcp;

\chdir(\dirname(__FILE__));
require '../vendor/autoload.php';
// require_once '../Google/Cloud/Firestore/V1beta1/FirestoreClient.php';
// require_once '../Google/Cloud/Spanner/V1/SpannerClient.php';
$firestore_probes = (require './firestore_probes.php');
$spanner_probes = (require './spanner_probes.php');
require './stackdriver_util.php';
$_OAUTH_SCOPE = 'https://www.googleapis.com/auth/cloud-platform';
$_FIRESTORE_TARGET = 'firestore.googleapis.com:443';
$_SPANNER_TARGET = 'spanner.googleapis.com:443';
use DeliciousBrains\WP_Offload_Media\Gcp\Google\Auth\ApplicationDefaultCredentials;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Firestore\V1beta1\FirestoreGrpcClient;
use DeliciousBrains\WP_Offload_Media\Gcp\Google\Cloud\Spanner\V1\SpannerGrpcClient;
function getArgs()
{
    $options = \getopt('', ['api:', 'extension:']);
    return $options;
}
/*
function secureAuthorizedChannel($credentials, $request, $target, $kwargs){
	$metadata_plugin = $transport_grpc->AuthMetadataPlugin($credentials, $request);
	$ssl_credentials = Grpc\ChannelCredentials::createSsl();
	$composit_credentials = $grpc->composite_channel_credentials($ssl_credentials, $google_auth_credentials);
	return $grpc_gcp->secure_channel($target, $composit_credentials, $kwargs);
}

function getStubChannel($target){
	$res = $auth->default([$_OAUTH_SCOPE]);
	$cred = $res[0];
	return secureAuthorizedChannel($cred, Request(), $target);
}*/
function executeProbes($api)
{
    global $_OAUTH_SCOPE;
    global $_SPANNER_TARGET;
    global $_FIRESTORE_TARGET;
    global $spanner_probes;
    global $firestore_probes;
    $util = new StackdriverUtil($api);
    $auth = Google\Auth\ApplicationDefaultCredentials::getCredentials($_OAUTH_SCOPE);
    $opts = ['credentials' => \Grpc\ChannelCredentials::createSsl(), 'update_metadata' => $auth->getUpdateMetadataFunc()];
    if ($api == 'spanner') {
        $client = new SpannerGrpcClient($_SPANNER_TARGET, $opts);
        $probe_functions = $spanner_probes;
    } else {
        if ($api == 'firestore') {
            $client = new FirestoreGrpcClient($_FIRESTORE_TARGET, $opts);
            $probe_functions = $firestore_probes;
        } else {
            echo 'grpc not implemented for ' . $api;
            exit(1);
        }
    }
    $total = \sizeof($probe_functions);
    $success = 0;
    $metrics = [];
    # Execute all probes for given api
    foreach ($probe_functions as $probe_name => $probe_function) {
        try {
            $probe_function($client, $metrics);
            $success++;
        } catch (\Exception $e) {
            $util->reportError($e);
        }
    }
    if ($success == $total) {
        $util->setSuccess(\DeliciousBrains\WP_Offload_Media\Gcp\True);
    }
    $util->addMetrics($metrics);
    $util->outputMetrics();
    if ($success != $total) {
        # TODO: exit system
        exit(1);
    }
}
function main()
{
    $args = getArgs();
    executeProbes($args['api']);
}
main();