com.microsoft.azure.toolkit.lib.common.exception.AzureToolkitRuntimeException Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of azure-toolkit-common-lib Show documentation
The newest version!
/*
 * 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.toolkit.lib.common.exception;

import com.azure.core.http.HttpResponse;
import com.azure.core.management.exception.ManagementException;
import com.microsoft.azure.toolkit.lib.common.action.Action;
import lombok.Getter;
import org.apache.commons.lang3.exception.ExceptionUtils;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Objects;
import java.util.Optional;

@Getter
public class AzureToolkitRuntimeException extends RuntimeException {
    /**
     * array of action ids or
     * {@link com.microsoft.azure.toolkit.lib.common.action.Action} or
     * {@link com.microsoft.azure.toolkit.lib.common.action.Action.Id}
     */
    private final Object[] actions;
    @Nullable
    private final String tips;

    public AzureToolkitRuntimeException(@Nonnull Throwable cause) {
        this(null, cause);
    }

    public AzureToolkitRuntimeException(@Nonnull String cause) {
        this(cause, (Object[]) null);
    }

    public AzureToolkitRuntimeException(String error, @Nonnull Throwable cause) {
        this(error, cause, (Object[]) null);
    }

    /**
     * @param actions array of action ids or
     *                {@link com.microsoft.azure.toolkit.lib.common.action.Action} or
     *                {@link com.microsoft.azure.toolkit.lib.common.action.Action.Id}
     */
    public AzureToolkitRuntimeException(@Nonnull String cause, Object... actions) {
        super(cause);
        this.actions = actions;
        this.tips = null;
    }

    /**
     * @param actions array of action ids or
     *                {@link com.microsoft.azure.toolkit.lib.common.action.Action} or
     *                {@link com.microsoft.azure.toolkit.lib.common.action.Action.Id}
     */
    public AzureToolkitRuntimeException(String error, @Nonnull Throwable cause, Object... actions) {
        super(error, cause);
        this.actions = actions;
        this.tips = null;
    }

    public AzureToolkitRuntimeException(@Nonnull Throwable cause, @Nonnull String tips) {
        this(null, cause, tips);
    }

    public AzureToolkitRuntimeException(@Nonnull String cause, @Nonnull String tips) {
        this(cause, tips, (Object[]) null);
    }

    public AzureToolkitRuntimeException(String error, @Nonnull Throwable cause, @Nonnull String tips) {
        this(error, cause, tips, (Object[]) null);
    }

    /**
     * @param actions array of action ids or
     *                {@link com.microsoft.azure.toolkit.lib.common.action.Action} or
     *                {@link com.microsoft.azure.toolkit.lib.common.action.Action.Id}
     */
    public AzureToolkitRuntimeException(@Nonnull String cause, @Nonnull String tips, Object... actions) {
        super(cause);
        this.actions = actions;
        this.tips = tips;
    }

    /**
     * @param actions array of action ids or
     *                {@link com.microsoft.azure.toolkit.lib.common.action.Action} or
     *                {@link com.microsoft.azure.toolkit.lib.common.action.Action.Id}
     */
    public AzureToolkitRuntimeException(String error, @Nonnull Throwable cause, @Nonnull String tips, Object... actions) {
        super(error, cause);
        this.actions = actions;
        this.tips = tips;
    }

    public static RuntimeException addDefaultActions(final RuntimeException e) {
        if (!(e instanceof AzureToolkitRuntimeException)) {
            Action.Id[] defaultActions = null;
            final Throwable rootCause = ExceptionUtils.getRootCause(e);
            if (rootCause instanceof ManagementException) {
                final int errorCode = Optional.ofNullable((ManagementException) rootCause).map(ManagementException::getResponse)
                    .map(HttpResponse::getStatusCode).orElse(0);
                if (errorCode == 403) {
                    defaultActions = new Action.Id[]{Action.SELECT_SUBS};
                } else if (errorCode == 401) {
                    defaultActions = new Action.Id[]{Action.AUTHENTICATE};
                }
            }
            if (Objects.nonNull(defaultActions)) {
                throw new AzureToolkitRuntimeException(e.getMessage(), e, (Object[]) defaultActions);
            }
        }
        return e;
    }
}