Copyright (c) Microsoft Corporation. All rights reserved.
Licensed under the MIT License. See License.txt in the project root for
license information.
/**
 * Copyright (c) Microsoft Corporation. All rights reserved.
 * Licensed under the MIT License. See License.txt in the project root for
 * license information.
 */

package com.microsoft.azure.management.servicebus.implementation;

import com.microsoft.azure.management.apigeneration.LangDefinition;
import com.microsoft.azure.management.resources.fluentcore.arm.models.HasResourceGroup;
import com.microsoft.azure.management.resources.fluentcore.arm.models.IndependentChildResource;
import com.microsoft.azure.management.resources.fluentcore.arm.models.Resource;
import com.microsoft.azure.management.resources.fluentcore.arm.models.implementation.IndependentChildResourceImpl;
import com.microsoft.azure.management.servicebus.AccessRights;
import com.microsoft.azure.management.servicebus.AuthorizationKeys;
import com.microsoft.azure.management.servicebus.Policykey;
import rx.Observable;
import rx.functions.Func1;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

Base type for various entity specific authorization rules.
Type parameters:
<FluentModelT> – the rule fluent interface
<FluentParentModelT> – the rule parent fluent interface
<InnerModelT> – the rule inner model
<FluentModelImplT> – the parent fluent implementation
<ManagerT> – the manager/**
 * Base type for various entity specific authorization rules.
 *
 * @param <FluentModelT> the rule fluent interface
 * @param <FluentParentModelT> the rule parent fluent interface
 * @param <InnerModelT> the rule inner model
 * @param <FluentModelImplT> the parent fluent implementation
 * @param <ManagerT> the manager
 */
@LangDefinition
abstract class AuthorizationRuleBaseImpl<
        FluentModelT extends IndependentChildResource<ManagerT, InnerModelT>,
        FluentParentModelT extends Resource & HasResourceGroup,
        InnerModelT extends SharedAccessAuthorizationRuleInner,
        FluentModelImplT extends IndependentChildResourceImpl<FluentModelT, FluentParentModelT, InnerModelT, FluentModelImplT, ManagerT>,
        ManagerT> extends IndependentChildResourceImpl<FluentModelT,
        FluentParentModelT,
        InnerModelT,
        FluentModelImplT,
        ManagerT> {
    protected AuthorizationRuleBaseImpl(String name, InnerModelT innerObject, ManagerT manager) {
        super(name, innerObject, manager);
    }

    
Returns:
stream that emits primary, secondary keys and connection strings/**
     * @return stream that emits primary, secondary keys and connection strings
     */
    public Observable<AuthorizationKeys> getKeysAsync() {
        return this.getKeysInnerAsync()
                .map(new Func1<ResourceListKeysInner, AuthorizationKeys>() {
                    @Override
                    public AuthorizationKeys call(ResourceListKeysInner inner) {
                        return new AuthorizationKeysImpl(inner);
                    }
                });
    }

    
Returns:
primary, secondary keys and connection strings/**
     * @return primary, secondary keys and connection strings
     */
    public AuthorizationKeys getKeys() {
        return getKeysAsync().toBlocking().last();
    }

    
Regenerates primary or secondary keys.
Params:
policykey – the key to regenerate
Returns:
stream that emits primary, secondary keys and connection strings/**
     * Regenerates primary or secondary keys.
     *
     * @param policykey the key to regenerate
     * @return stream that emits primary, secondary keys and connection strings
     */
    public Observable<AuthorizationKeys> regenerateKeyAsync(Policykey policykey) {
        return this.regenerateKeysInnerAsync(policykey)
                .map(new Func1<ResourceListKeysInner, AuthorizationKeys>() {
                    @Override
                    public AuthorizationKeys call(ResourceListKeysInner inner) {
                        return new AuthorizationKeysImpl(inner);
                    }
                });
    }

    
Regenerates primary or secondary keys.
Params:
policykey – the key to regenerate
Returns:
primary, secondary keys and connection strings/**
     * Regenerates primary or secondary keys.
     *
     * @param policykey the key to regenerate
     * @return primary, secondary keys and connection strings
     */
    public AuthorizationKeys regenerateKey(Policykey policykey) {
        return regenerateKeyAsync(policykey).toBlocking().last();
    }

    public List<AccessRights> rights() {
        if (this.inner().rights() == null) {
            return Collections.unmodifiableList(new ArrayList<AccessRights>());
        }
        return Collections.unmodifiableList(this.inner().rights());
    }


    @SuppressWarnings("unchecked")
    public FluentModelImplT withListeningEnabled() {
        if (this.inner().rights() == null) {
            this.inner().withRights(new ArrayList<AccessRights>());
        }
        if (!this.inner().rights().contains(AccessRights.LISTEN)) {
            this.inner().rights().add(AccessRights.LISTEN);
        }
        return (FluentModelImplT) this;
    }

    @SuppressWarnings("unchecked")
    public FluentModelImplT withSendingEnabled() {
        if (this.inner().rights() == null) {
            this.inner().withRights(new ArrayList<AccessRights>());
        }
        if (!this.inner().rights().contains(AccessRights.SEND)) {
            this.inner().rights().add(AccessRights.SEND);
        }
        return (FluentModelImplT) this;
    }

    @SuppressWarnings("unchecked")
    public FluentModelImplT withManagementEnabled() {
        withListeningEnabled();
        withSendingEnabled();
        if (!this.inner().rights().contains(AccessRights.MANAGE)) {
            this.inner().rights().add(AccessRights.MANAGE);
        }
        return (FluentModelImplT) this;
    }

    protected abstract Observable<ResourceListKeysInner> getKeysInnerAsync();
    protected abstract Observable<ResourceListKeysInner> regenerateKeysInnerAsync(Policykey policykey);
}