package com.microsoft.azure.management.resources.implementation;
import com.microsoft.azure.PagedList;
import com.microsoft.azure.management.resources.PolicyAssignment;
import com.microsoft.azure.management.resources.PolicyAssignments;
import com.microsoft.azure.management.resources.ResourceGroups;
import com.microsoft.azure.management.resources.fluentcore.arm.ResourceUtils;
import com.microsoft.azure.management.resources.fluentcore.arm.collection.implementation.CreatableWrappersImpl;
import com.microsoft.rest.ServiceCallback;
import com.microsoft.rest.ServiceFuture;
import rx.Completable;
import rx.Observable;
import rx.functions.Func1;
final class PolicyAssignmentsImpl
extends CreatableWrappersImpl<PolicyAssignment, PolicyAssignmentImpl, PolicyAssignmentInner>
implements PolicyAssignments {
private final PolicyAssignmentsInner client;
PolicyAssignmentsImpl(final PolicyAssignmentsInner innerClient) {
this.client = innerClient;
}
@Override
public PagedList<PolicyAssignment> list() {
return wrapList(client.list());
}
@Override
public Completable deleteByIdAsync(String id) {
return client.deleteByIdAsync(id).toCompletable();
}
@Override
public PolicyAssignmentImpl define(String name) {
return wrapModel(name);
}
@Override
protected PolicyAssignmentImpl wrapModel(String name) {
return new PolicyAssignmentImpl(name,
new PolicyAssignmentInner().withDisplayName(name),
client);
}
@Override
protected PolicyAssignmentImpl wrapModel(PolicyAssignmentInner inner) {
if (inner == null) {
return null;
}
return new PolicyAssignmentImpl(inner.name(), inner, client);
}
@Override
public PagedList<PolicyAssignment> listByResource(String resourceId) {
return wrapList(client.listForResource(
ResourceUtils.groupFromResourceId(resourceId),
ResourceUtils.resourceProviderFromResourceId(resourceId),
ResourceUtils.relativePathFromResourceId(ResourceUtils.parentResourceIdFromResourceId(resourceId)),
ResourceUtils.resourceTypeFromResourceId(resourceId),
ResourceUtils.nameFromResourceId(resourceId)
));
}
@Override
public PolicyAssignment getById(String id) {
return getByIdAsync(id).toBlocking().last();
}
@Override
public Observable<PolicyAssignment> getByIdAsync(String id) {
return client.getByIdAsync(id).map(new Func1<PolicyAssignmentInner, PolicyAssignment>() {
@Override
public PolicyAssignment call(PolicyAssignmentInner policyAssignmentInner) {
return wrapModel(policyAssignmentInner);
}
});
}
@Override
public ServiceFuture<PolicyAssignment> getByIdAsync(String id, ServiceCallback<PolicyAssignment> callback) {
return ServiceFuture.fromBody(getByIdAsync(id), callback);
}
@Override
public PagedList<PolicyAssignment> listByResourceGroup(String resourceGroupName) {
return wrapList(client.listByResourceGroup(resourceGroupName));
}
@Override
public Observable<PolicyAssignment> listAsync() {
return wrapPageAsync(this.client.listAsync());
}
@Override
public Observable<PolicyAssignment> listByResourceGroupAsync(String resourceGroupName) {
return wrapPageAsync(this.client.listByResourceGroupAsync(resourceGroupName));
}
}