ScalarVariable.java

/*
 * Copyright 2015 Nokia Solutions and Networks
 * Licensed under the Apache License, Version 2.0,
 * see license.txt file for details.
 */
package org.rf.ide.core.testdata.model.table.variables;

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

import org.rf.ide.core.testdata.model.presenter.MoveElementHelper;
import org.rf.ide.core.testdata.text.read.recognizer.RobotToken;
import org.rf.ide.core.testdata.text.read.recognizer.RobotTokenType;

public class ScalarVariable extends AVariable {

    private static final long serialVersionUID = 1805466263821781386L;

    private final List<RobotToken> values = new ArrayList<>();

    public ScalarVariable(final String name, final RobotToken declaration, final VariableScope scope) {
        super(VariableType.SCALAR, name, declaration, scope);
    }

    public void addValue(final RobotToken value) {
        fixForTheType(value, RobotTokenType.VARIABLES_VARIABLE_VALUE);
        values.add(value);
    }

    public void addNewValue(final int position, final RobotToken value) {
        fixForTheType(value, RobotTokenType.VARIABLES_VARIABLE_VALUE);
        values.add(position, value);
    }

    public void addValue(final RobotToken value, final int position) {
        updateOrCreateTokenInside(values, position, value, RobotTokenType.VARIABLES_VARIABLE_VALUE);
    }

    public void removeValue(final RobotToken value) {
        values.remove(value);
    }

    public boolean moveLeftValue(final RobotToken value) {
        return MoveElementHelper.moveLeft(values, value);
    }

    public boolean moveRightValue(final RobotToken value) {
        return MoveElementHelper.moveRight(values, value);
    }

    public List<RobotToken> getValues() {
        return Collections.unmodifiableList(values);
    }

    @Override
    public List<RobotToken> getValueTokens() {
        return getValues();
    }

    @Override
    public boolean isPresent() {
        return (getDeclaration() != null);
    }

    @Override
    public VariableType getType() {
        if (values.size() >= 2) {
            this.type = VariableType.SCALAR_AS_LIST;
        } else {
            this.type = VariableType.SCALAR;
        }

        return type;
    }

    @Override
    public List<RobotToken> getElementTokens() {
        final List<RobotToken> tokens = new ArrayList<>();
        if (isPresent()) {
            tokens.add(getDeclaration());
            tokens.addAll(getValues());
            tokens.addAll(getComment());
        }

        return tokens;
    }

    @Override
    public boolean removeElementToken(final int index) {
        return super.removeElementFromList(values, index);
    }
}