package org.apache.airavata.mft.resource.client;

import io.grpc.ManagedChannel;
import org.apache.airavata.mft.resource.service.azure.AzureResourceServiceGrpc;
import org.apache.airavata.mft.resource.service.box.BoxResourceServiceGrpc;
import org.apache.airavata.mft.resource.service.dropbox.DropboxServiceGrpc;
import org.apache.airavata.mft.resource.service.ftp.FTPResourceServiceGrpc;
import org.apache.airavata.mft.resource.service.gcs.GCSResourceServiceGrpc;
import org.apache.airavata.mft.resource.service.local.LocalResourceServiceGrpc;
import org.apache.airavata.mft.resource.service.s3.S3ResourceServiceGrpc;
import org.apache.airavata.mft.resource.service.scp.SCPResourceServiceGrpc;

import java.io.Closeable;
import java.io.IOException;

public class ResourceServiceClient implements Closeable {
    private ManagedChannel channel;

    ResourceServiceClient(ManagedChannel channel) {
        this.channel = channel;
    }

    public SCPResourceServiceGrpc.SCPResourceServiceBlockingStub scp() {
        return SCPResourceServiceGrpc.newBlockingStub(channel);
    }

    public LocalResourceServiceGrpc.LocalResourceServiceBlockingStub local() {
        return LocalResourceServiceGrpc.newBlockingStub(channel);
    }

    public S3ResourceServiceGrpc.S3ResourceServiceBlockingStub s3() {
        return S3ResourceServiceGrpc.newBlockingStub(channel);
    }

    public FTPResourceServiceGrpc.FTPResourceServiceBlockingStub ftp() {
        return FTPResourceServiceGrpc.newBlockingStub(channel);
    }

    public AzureResourceServiceGrpc.AzureResourceServiceBlockingStub azure() {
        return AzureResourceServiceGrpc.newBlockingStub(channel);
    }

    public GCSResourceServiceGrpc.GCSResourceServiceBlockingStub gcs() {
        return GCSResourceServiceGrpc.newBlockingStub(channel);
    }

    public BoxResourceServiceGrpc.BoxResourceServiceBlockingStub box() {
        return BoxResourceServiceGrpc.newBlockingStub(channel);
    }

    public DropboxServiceGrpc.DropboxServiceBlockingStub dropbox() {
        return DropboxServiceGrpc.newBlockingStub(channel);
    }

    @Override
    public void close() throws IOException {
        this.channel.shutdown();
    }
}