Package org.terasology.logic.console.internal

Source Code of org.terasology.logic.console.internal.CommandInfo

/*
* Copyright 2013 MovingBlocks
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*      http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.terasology.logic.console.internal;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.terasology.entitySystem.entity.EntityRef;
import org.terasology.logic.console.Command;
import org.terasology.logic.console.CommandParam;

import java.lang.annotation.Annotation;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Collection;
import java.util.List;

/**
* Metadata on a command, including the ability to execute it.
*
* @author Immortius
*/
public class CommandInfo {
    private static final Logger logger = LoggerFactory.getLogger(CommandInfo.class);

    private Method method;
    private Object provider;

    private String name;
    private List<String> parameterNames = Lists.newArrayList();
    private List<String> parameterTypes = Lists.newArrayList();
    private String shortDescription;
    private String helpText;
    private boolean clientEntityRequired;
    private boolean runOnServer;

    public CommandInfo(Method method, Object provider) {
        this.method = method;
        this.provider = provider;
        this.name = method.getName();
        Command commandAnnotation = method.getAnnotation(Command.class);
        if (commandAnnotation == null) {
            throw new IllegalArgumentException("Method not annotated with command");
        }
        for (int i = 0; i < method.getParameterTypes().length; ++i) {
            Class<?> parameterType = method.getParameterTypes()[i];
            if (i == method.getParameterTypes().length - 1 && parameterType == EntityRef.class) {
                clientEntityRequired = true;
            } else {
                String paramType = parameterType.getSimpleName().toLowerCase();
                String paramName = getParamName(i);
               
                if (paramName == null) {
                    paramName = "p" + i;
                    logger.warn("Parameter {} in method {} does not have a CommandParam annotation", i, method);
                }

                parameterNames.add(paramName);
                parameterTypes.add(paramType);
            }
        }
        this.runOnServer = commandAnnotation.runOnServer();
        this.shortDescription = commandAnnotation.shortDescription();
        this.helpText = commandAnnotation.helpText();
    }
   
    private String getParamName(int i) {
        // JAVA8: if relevant classes are compiled with the "-parameters" flag,
        // the parameter name can be accessed through reflection like this:
        // Parameter p = method.getParameters()[i]
        // if (p.isNamePresent()) {
        //     return p.getName();
        // }
       
        for (Annotation paramAnnot : method.getParameterAnnotations()[i]) {
            if (paramAnnot instanceof CommandParam) {
                return ((CommandParam) paramAnnot).value();
            }
        }
       
        return null;
    }

    public String getName() {
        return name;
    }

    public Collection<String> getParameterNames() {
        return ImmutableList.copyOf(parameterNames);
    }

    public int getParameterCount() {
        return parameterNames.size();
    }

    public String getShortDescription() {
        return shortDescription;
    }

    public String getHelpText() {
        return helpText;
    }

    public boolean isClientEntityRequired() {
        return clientEntityRequired;
    }

    public boolean isRunOnServer() {
        return runOnServer;
    }

    public String getUsageMessage() {
        StringBuilder builder = new StringBuilder(name);
        for (int i = 0; i < parameterNames.size(); i++) {
            builder.append(" <");
            builder.append(parameterTypes.get(i));
            builder.append(" ");
            builder.append(parameterNames.get(i));
            builder.append(">");
        }

        return builder.toString();
    }

    /**
     * @param params a set of string parameter
     * @param callingClient the responsible client (passed as last parameter)
     * @return the return value of the method call or an empty string, never <code>null</code>.
     * @throws IllegalArgumentException if the method call was not successful
     * @throws Exception if something went seriously wrong
     */
    public String execute(List<String> params, EntityRef callingClient) throws Exception {
        Object[] processedParams = new Object[method.getParameterTypes().length];
        if (isClientEntityRequired()) {
            if (params.size() + 1 != method.getParameterTypes().length) {
                throw new IllegalArgumentException("Incorrect number of parameters, expected " + (method.getParameterTypes().length - 1));
            }

            processedParams[processedParams.length - 1] = callingClient;
        } else if (params.size() != method.getParameterTypes().length) {
            throw new IllegalArgumentException("Incorrect number of parameters, expected " + (method.getParameterTypes().length));
        }
        for (int i = 0; i < params.size(); ++i) {
            try {
                Class<?> type = method.getParameterTypes()[i];
                if (type == Float.TYPE) {
                    processedParams[i] = Float.parseFloat(params.get(i));
                } else if (type == Integer.TYPE) {
                    processedParams[i] = Integer.parseInt(params.get(i));
                } else if (type == String.class) {
                    processedParams[i] = params.get(i);
                }
            } catch (NumberFormatException e) {
                throw new NumberFormatException("Bad argument '" + params.get(i) + "'");
            }
        }
        try {
            Object result = method.invoke(provider, processedParams);
            return (result != null) ? result.toString() : "";
        } catch (InvocationTargetException ite) {
            // we end up here, if the called method throws an exception (which is ok)
            Throwable e = ite.getTargetException();
            throw new IllegalArgumentException(e.getLocalizedMessage(), e);
        }
    }
}
TOP

Related Classes of org.terasology.logic.console.internal.CommandInfo

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.