/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.jboss.as.subsystem.test;

import static org.jboss.as.controller.client.helpers.ClientConstants.RECURSIVE;
import static org.jboss.as.controller.client.helpers.ClientConstants.RESULT;

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

import org.jboss.as.controller.AttributeDefinition;
import org.jboss.as.controller.OperationDefinition;
import org.jboss.as.controller.client.helpers.Operations;
import org.jboss.dmr.ModelNode;
import org.jboss.dmr.ModelType;

/**
 * @author <a href="mailto:jperkins@redhat.com">James R. Perkins</a>
 */
public class SubsystemOperations extends Operations {

    public static final ModelNode UNDEFINED = new ModelNode();

    static {
        UNDEFINED.protect();
    }

    /**
     * Parses the result and returns the failure description. If the result was successful, an empty string is
     * returned.
     *
     * @param result the result of executing an operation
     *
     * @return the failure message or an empty string
     */
    public static String getFailureDescriptionAsString(final ModelNode result) {
        if (isSuccessfulOutcome(result)) {
            return "";
        }
        return Operations.getFailureDescription(result).asString();
    }

    /**
     * Creates a remove operation.
     *
     * @param address   the address for the operation
     * @param recursive {@code true} if the remove should be recursive
     *
     * @return the operation
     */
    public static ModelNode createRemoveOperation(final ModelNode address, final boolean recursive) {
        final ModelNode op = createRemoveOperation(address);
        op.get(RECURSIVE).set(recursive);
        return op;
    }

    /**
     * Creates an operation to read the attribute represented by the {@code attributeName} parameter.
     *
     * @param address   the address to create the read attribute for
     * @param attribute the attribute to read
     *
     * @return the operation
     */
    public static ModelNode createReadAttributeOperation(final ModelNode address, final AttributeDefinition attribute) {
        return createReadAttributeOperation(address, attribute.getName());
    }

    /**
     * Creates an operation to undefine an attribute value represented by the {@code attribute} parameter.
     *
     * @param address   the address to create the write attribute for
     * @param attribute the attribute to undefine
     *
     * @return the operation
     */
    public static ModelNode createUndefineAttributeOperation(final ModelNode address, final AttributeDefinition attribute) {
        return createUndefineAttributeOperation(address, attribute.getName());
    }

    /**
     * Creates an operation to write an attribute value represented by the {@code attributeName} parameter.
     *
     * @param address   the address to create the write attribute for
     * @param attribute the attribute to write
     * @param value     the value to set the attribute to
     *
     * @return the operation
     */
    public static ModelNode createWriteAttributeOperation(final ModelNode address, final AttributeDefinition attribute, final String value) {
        return createWriteAttributeOperation(address, attribute.getName(), value);
    }

    /**
     * Creates an operation to write an attribute value represented by the {@code attributeName} parameter.
     *
     * @param address   the address to create the write attribute for
     * @param attribute the attribute to write
     * @param value     the value to set the attribute to
     *
     * @return the operation
     */
    public static ModelNode createWriteAttributeOperation(final ModelNode address, final AttributeDefinition attribute, final boolean value) {
        return createWriteAttributeOperation(address, attribute.getName(), value);
    }

    /**
     * Creates an operation to write an attribute value represented by the {@code attributeName} parameter.
     *
     * @param address   the address to create the write attribute for
     * @param attribute the attribute to write
     * @param value     the value to set the attribute to
     *
     * @return the operation
     */
    public static ModelNode createWriteAttributeOperation(final ModelNode address, final AttributeDefinition attribute, final int value) {
        return createWriteAttributeOperation(address, attribute.getName(), value);
    }

    /**
     * Creates an operation to write an attribute value represented by the {@code attributeName} parameter.
     *
     * @param address   the address to create the write attribute for
     * @param attribute the attribute to write
     * @param value     the value to set the attribute to
     *
     * @return the operation
     */
    public static ModelNode createWriteAttributeOperation(final ModelNode address, final AttributeDefinition attribute, final ModelNode value) {
        return createWriteAttributeOperation(address, attribute.getName(), value);
    }

    /**
     * Reads the result of an operation and returns the result as a string. If the operation does not have a {@link
     * org.jboss.as.controller.client.helpers.ClientConstants#RESULT} attribute and empty string is returned.
     *
     * @param result the result of executing an operation
     *
     * @return the result of the operation or an empty string
     */
    public static String readResultAsString(final ModelNode result) {
        return (result.hasDefined(RESULT) ? result.get(RESULT).asString() : "");
    }

    /**
     * Reads the result of an operation and returns the result as a list of strings. If the operation does not have a
     * {@link org.jboss.as.controller.client.helpers.ClientConstants#RESULT} attribute and empty list is returned.
     *
     * @param result the result of executing an operation
     *
     * @return the result of the operation or an empty list
     */
    public static List<String> readResultAsList(final ModelNode result) {
        if (result.hasDefined(RESULT) && result.get(RESULT).getType() == ModelType.LIST) {
            final List<String> list = new ArrayList<String>();
            for (ModelNode n : result.get(RESULT).asList()) list.add(n.asString());
            return list;
        }
        return Collections.emptyList();
    }

    public static class OperationBuilder {
        private final ModelNode op;

        private OperationBuilder(final ModelNode op) {
            this.op = op;
        }

        public static OperationBuilder create(final String operation) {
            return new OperationBuilder(createOperation(operation));
        }

        public static OperationBuilder create(final OperationDefinition operationDefinition) {
            return new OperationBuilder(createOperation(operationDefinition.getName()));
        }

        public static OperationBuilder create(final String operation, final ModelNode address) {
            return new OperationBuilder(createOperation(operation, address));
        }

        public static OperationBuilder create(final OperationDefinition operationDefinition, final ModelNode address) {
            return new OperationBuilder(createOperation(operationDefinition.getName(), address));
        }

        public static OperationBuilder createAddOperation(final ModelNode address) {
            return new OperationBuilder(SubsystemOperations.createAddOperation(address));
        }

        public ModelNode build() {
            return op.clone();
        }

        public OperationBuilder addAttribute(final AttributeDefinition attribute, final String value) {
            op.get(attribute.getName()).set(value);
            return this;
        }

        public OperationBuilder addAttribute(final AttributeDefinition attribute, final boolean value) {
            op.get(attribute.getName()).set(value);
            return this;
        }

        public OperationBuilder addAttribute(final AttributeDefinition attribute, final int value) {
            op.get(attribute.getName()).set(value);
            return this;
        }

        public OperationBuilder addAttribute(final AttributeDefinition attribute, final long value) {
            op.get(attribute.getName()).set(value);
            return this;
        }

        public OperationBuilder addAttribute(final AttributeDefinition attribute, final ModelNode value) {
            op.get(attribute.getName()).set(value);
            return this;
        }
    }
}
