software.amazon.awssdk.services.codegurusecurity.endpoints.internal.Fn Maven / Gradle / Ivy

Go to download
/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with
 * the License. A copy of the License is located at
 * 
 * http://aws.amazon.com/apache2.0
 * 
 * or in the "license" file accompanying this file. This file 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 software.amazon.awssdk.services.codegurusecurity.endpoints.internal;

import java.util.List;
import java.util.stream.Collectors;
import software.amazon.awssdk.annotations.SdkInternalApi;
import software.amazon.awssdk.utils.Pair;

@SdkInternalApi
public abstract class Fn extends Expr implements Into {
    protected FnNode fnNode;

    public Fn(FnNode fnNode) {
        this.fnNode = fnNode;
    }

    /**
     * Convert this fn into a condition
     */
    public Condition condition() {
        return new Condition.Builder().fn(this).build();
    }

    public Condition condition(String result) {
        return new Condition.Builder().fn(this).result(result).build();
    }

    public abstract  T acceptFnVisitor(FnVisitor visitor);

    public  R accept(ExprVisitor visitor) {
        return visitor.visitFn(this);
    }

    /**
     * Returns the name of this function, eg. {@code isSet}, {@code parseUrl}
     * 
     * @return The name
     */
    public String getName() {
        return fnNode.getId();
    }

    /**
     * @return The arguments to this function
     */
    public List getArgv() {
        return fnNode.getArgv();
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }
        if (!super.equals(o)) {
            return false;
        }

        Fn fn = (Fn) o;

        return fnNode != null ? fnNode.equals(fn.fnNode) : fn.fnNode == null;
    }

    @Override
    public int hashCode() {
        return fnNode != null ? fnNode.hashCode() : 0;
    }

    @Override
    public String toString() {
        return String.format("%s(%s)", fnNode.getId(),
                fnNode.getArgv().stream().map(Expr::toString).collect(Collectors.joining(", ")));
    }

    protected Expr expectOneArg() {
        List argv = this.fnNode.getArgv();
        if (argv.size() == 1) {
            return argv.get(0);
        } else {
            throw RuleError.builder()
                    .cause(SourceException.builder().message("expected 1 argument but found " + argv.size()).build()).build();
        }
    }

    protected Pair expectTwoArgs() {
        List argv = this.fnNode.getArgv();
        if (argv.size() == 2) {
            return Pair.of(argv.get(0), argv.get(1));
        } else {
            throw RuleError.builder()
                    .cause(SourceException.builder().message("expected 2 arguments but found " + argv.size()).build()).build();
        }

    }

    protected List expectVariableArgs(int expectedNumberArgs) {
        List argv = this.fnNode.getArgv();
        if (argv.size() == expectedNumberArgs) {
            return argv;
        } else {
            throw RuleError
                    .builder()
                    .cause(SourceException.builder()
                            .message(String.format("expected %d arguments but found %d", expectedNumberArgs, argv.size()))
                            .build()).build();
        }

    }

    @Override
    public Condition into() {
        return this.condition();
    }
}