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.batch.implementation;

import com.microsoft.azure.management.apigeneration.LangDefinition;
import com.microsoft.azure.management.batch.Application;
import com.microsoft.azure.management.batch.ApplicationPackage;
import com.microsoft.azure.management.batch.BatchAccount;
import com.microsoft.azure.management.resources.fluentcore.arm.models.implementation.ExternalChildResourceImpl;
import com.microsoft.azure.management.resources.fluentcore.utils.RXMapper;

import rx.Observable;
import rx.functions.Func1;

import java.util.Map;
import java.util.List;

Implementation for BatchAccount Application and its parent interfaces.
/**
 * Implementation for BatchAccount Application and its parent interfaces.
 */
@LangDefinition
public class ApplicationImpl
        extends ExternalChildResourceImpl<Application,
        ApplicationInner,
        BatchAccountImpl,
        BatchAccount>
        implements Application,
        Application.Definition<BatchAccount.DefinitionStages.WithApplicationAndStorage>,
        Application.UpdateDefinition<BatchAccount.Update>,
        Application.Update {
    private final ApplicationPackagesImpl applicationPackages;

    protected ApplicationImpl(
            String name,
            BatchAccountImpl batchAccount,
            ApplicationInner inner) {
        super(name, batchAccount, inner);
        applicationPackages = new ApplicationPackagesImpl(this);
        applicationPackages.enableCommitMode();
    }

    @Override
    public String id() {
        return this.inner().id();
    }

    @Override
    public String displayName() {
        return this.inner().displayName();
    }

    @Override
    public Map<String, ApplicationPackage> applicationPackages() {
        return this.applicationPackages.asMap();
    }

    @Override
    public boolean updatesAllowed() {
        return this.inner().allowUpdates();
    }

    @Override
    public String defaultVersion() {
        return this.inner().defaultVersion();
    }

    @Override
    public Observable<Application> createResourceAsync() {
        final ApplicationImpl self = this;

        return this.parent().manager().inner().applications().createAsync(
                this.parent().resourceGroupName(),
                this.parent().name(),
                this.name(), this.inner())
                .map(new Func1<ApplicationInner, Application>() {
                    @Override
                    public Application call(ApplicationInner inner) {
                        self.setInner(inner);
                        return self;
                    }
                })
                .flatMap(new Func1<Application, Observable<? extends Application>>() {
                    @Override
                    public Observable<? extends Application> call(Application application) {
                        return self.applicationPackages.commitAndGetAllAsync()
                                .map(new Func1<List<ApplicationPackageImpl>, Application>() {
                                    @Override
                                    public Application call(List<ApplicationPackageImpl> applications) {
                                        return self;
                                    }
                                });
                    }
                });
    }

    @Override
    public Observable<Application> updateResourceAsync() {
        final ApplicationImpl self = this;

        return RXMapper.map(this.parent().manager().inner().applications().updateAsync(
                this.parent().resourceGroupName(),
                this.parent().name(),
                this.name(),
                this.inner()), self)
                .flatMap(new Func1<Application, Observable<? extends Application>>() {
                    @Override
                    public Observable<? extends Application> call(Application application) {
                        return self.applicationPackages.commitAndGetAllAsync()
                                .map(new Func1<List<ApplicationPackageImpl>, Application>() {
                                    @Override
                                    public Application call(List<ApplicationPackageImpl> applications) {
                                        return self;
                                    }
                                });
                    }
                });
    }

    @Override
    public Observable<Void> deleteResourceAsync() {
        return this.parent().manager().inner().applications().deleteAsync(
                this.parent().resourceGroupName(),
                this.parent().name(),
                this.name());
    }

    @Override
    public Observable<Application> refreshAsync() {
        return super.refreshAsync().map(new Func1<Application, Application>() {
            @Override
            public Application call(Application application) {
                ApplicationImpl impl = (ApplicationImpl) application;

                impl.applicationPackages.refresh();
                return impl;
            }
        });
    }

    @Override
    protected Observable<ApplicationInner> getInnerAsync() {
        return this.parent().manager().inner().applications().getAsync(
                this.parent().resourceGroupName(), this.parent().name(), this.inner().name());
    }

    @Override
    public BatchAccountImpl attach() {
        return this.parent().withApplication(this);
    }

    @Override
    public ApplicationImpl withAllowUpdates(boolean allowUpdates) {
        this.inner().withAllowUpdates(allowUpdates);
        return this;
    }

    @Override
    public ApplicationImpl withDisplayName(String displayName) {
        this.inner().withDisplayName(displayName);
        return this;
    }

    protected static ApplicationImpl newApplication(
            String name,
            BatchAccountImpl parent) {
        ApplicationInner inner = new ApplicationInner();
        inner.withDisplayName(name);
        ApplicationImpl applicationImpl = new ApplicationImpl(name, parent, inner);
        return applicationImpl;
    }

    @Override
    public Update withoutApplicationPackage(String applicationPackageName) {
        this.applicationPackages.remove(applicationPackageName);
        return this;

    }

    ApplicationImpl withApplicationPackage(ApplicationPackageImpl applicationPackage) {
        this.applicationPackages.addApplicationPackage(applicationPackage);
        return this;
    }

    @Override
    public ApplicationImpl defineNewApplicationPackage(String applicationPackageName) {
        this.withApplicationPackage(this.applicationPackages.define(applicationPackageName));
        return this;
    }
}