com.denimgroup.threadfix.framework.impl.dotNet.Action.java Source code

Introduction

Here is the source code for com.denimgroup.threadfix.framework.impl.dotNet.Action.java

Source

////////////////////////////////////////////////////////////////////////
//
//     Copyright (c) 2009-2015 Denim Group, Ltd.
//
//     The contents of this file are subject to the Mozilla Public 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.mozilla.org/MPL/
//
//     Software distributed under the License is distributed on an "AS IS"
//     basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//     License for the specific language governing rights and limitations
//     under the License.
//
//     The Original Code is ThreadFix.
//
//     The Initial Developer of the Original Code is Denim Group, Ltd.
//     Portions created by Denim Group, Ltd. are Copyright (C)
//     Denim Group, Ltd. All Rights Reserved.
//
//     Contributor(s): Denim Group, Ltd.
//
////////////////////////////////////////////////////////////////////////
package com.denimgroup.threadfix.framework.impl.dotNet;

import com.denimgroup.threadfix.framework.impl.model.ModelField;
import org.apache.commons.lang3.StringUtils;

import javax.annotation.Nonnull;
import java.util.Set;

import static com.denimgroup.threadfix.CollectionUtils.set;

/**
 * Created by mac on 6/26/14.
 */
class Action {
    @Nonnull
    String name;
    @Nonnull
    Set<String> attributes;
    @Nonnull
    Integer lineNumber;
    @Nonnull
    Integer endLineNumber;
    @Nonnull
    Set<String> parameters = set();
    @Nonnull
    Set<ModelField> parametersWithTypes;

    String getMethod() {
        return attributes.contains("HttpPost") ? "POST" : "GET";
    }

    static Action action(@Nonnull String name, @Nonnull Set<String> attributes, @Nonnull Integer lineNumber,
            @Nonnull Integer endLineNumber, @Nonnull Set<String> parameters,
            @Nonnull Set<ModelField> parametersWithTypes) {
        Action action = new Action();
        action.name = name;
        action.attributes = attributes;
        action.lineNumber = lineNumber;
        action.parametersWithTypes = parametersWithTypes;
        action.endLineNumber = endLineNumber;
        action.parameters = parameters;

        for (ModelField field : parametersWithTypes) {
            if (field.getType().equals("Include")) {
                for (String s : StringUtils.split(field.getParameterKey(), ',')) {
                    action.parameters.add(s.trim());
                }
            } else {
                action.parameters.add(field.getParameterKey());
            }
        }

        return action;
    }

    @Override
    public String toString() {
        return name + ": " + getMethod() + parameters;
    }

}