org.hisp.dhis.rules.RuleExpression Maven / Gradle / Ivy

Go to download
package org.hisp.dhis.rules;

import com.google.auto.value.AutoValue;
import org.hisp.dhis.parser.expression.antlr.ExpressionParser;

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

@AutoValue
public abstract class RuleExpression
{
    /* This method should probably be removed creating a new prefix for program rule variables that is
     *  not shared with indicators.*/
    @Nonnull
    public static String getProgramRuleVariable( ExpressionParser.ExprContext ctx )
    {
        if ( isProgramRuleVariable( ctx ) )
        {
            return getProgramRuleVariableText( ctx );
        }
        else if ( ctx.programVariable() != null )
        {
            return ctx.programVariable().getText();
        }
        else
        {
            return ctx.uid0.getText() + secondPart( ctx ) + thirdPart( ctx );
        }
    }

    private static String getProgramRuleVariableText( ExpressionParser.ExprContext ctx )
    {
        if ( ctx.programRuleVariableName() != null ) {
            return ctx.programRuleVariableName().getText();
        }
        else
        {
            return ctx.programRuleStringVariableName().getText().replaceAll( "\'", "" );
        }
    }

    private static boolean isProgramRuleVariable( ExpressionParser.ExprContext ctx )
    {
        return ctx.programRuleVariableName() != null || ( ctx.programRuleStringVariableName() != null );
    }

    private static String secondPart( ExpressionParser.ExprContext ctx )
    {
        if ( ctx.uid1 != null )
        {
            return "." + ctx.uid1.getText();
        }
        else if ( ctx.wild1 != null )
        {
            return ctx.wild1.getText();
        }
        return "";
    }

    private static String thirdPart( ExpressionParser.ExprContext ctx )
    {
        if ( ctx.uid2 != null && ctx.uid1 == null )
        {
            return ".*." + ctx.uid2.getText();
        }
        else if ( ctx.uid2 != null )
        {
            return "." + ctx.uid2.getText();
        }
        else if ( ctx.wild2 != null )
        {
            return ctx.wild2.getText();
        }
        return "";
    }

    @Nonnull
    public abstract Set variables();
}