KeywordTeardownModelOperation.java

/*
 * Copyright 2016 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.presenter.update.keywords;

import java.util.List;

import org.rf.ide.core.testdata.model.AModelElement;
import org.rf.ide.core.testdata.model.ModelType;
import org.rf.ide.core.testdata.model.presenter.update.IExecutablesStepsHolderElementOperation;
import org.rf.ide.core.testdata.model.table.keywords.KeywordTeardown;
import org.rf.ide.core.testdata.model.table.keywords.UserKeyword;
import org.rf.ide.core.testdata.text.read.IRobotTokenType;
import org.rf.ide.core.testdata.text.read.recognizer.RobotToken;
import org.rf.ide.core.testdata.text.read.recognizer.RobotTokenType;

public class KeywordTeardownModelOperation implements IExecutablesStepsHolderElementOperation<UserKeyword> {

    @Override
    public boolean isApplicable(final ModelType elementType) {
        return elementType == ModelType.USER_KEYWORD_TEARDOWN;
    }
    
    @Override
    public boolean isApplicable(final IRobotTokenType elementType) {
        return elementType == RobotTokenType.KEYWORD_SETTING_TEARDOWN;
    }

    @Override
    public AModelElement<?> create(final UserKeyword userKeyword, final String settingName, final List<String> args,
            final String comment) {
        final KeywordTeardown keywordTeardown = userKeyword.newTeardown();
        keywordTeardown.getDeclaration().setText(settingName);
        keywordTeardown.getDeclaration().setRaw(settingName);

        if (!args.isEmpty()) {
            keywordTeardown.setKeywordName(args.get(0));
            for (int i = 1; i < args.size(); i++) {
                keywordTeardown.addArgument(args.get(i));
            }
        }
        if (comment != null && !comment.isEmpty()) {
            keywordTeardown.setComment(comment);
        }
        return keywordTeardown;
    }

    @Override
    public AModelElement<?> insert(final UserKeyword userKeyword, final int index,
            final AModelElement<?> modelElement) {
        userKeyword.addTeardown(0, (KeywordTeardown) modelElement);
        return modelElement;
    }

    @Override
    public void update(final AModelElement<?> modelElement, final int index, final String value) {
        final KeywordTeardown keywordTeardown = (KeywordTeardown) modelElement;

        if (index == 0) {
            keywordTeardown.setKeywordName(value != null ? value : "");
        } else if (index > 0) {
            if (value != null) {
                keywordTeardown.setArgument(index - 1, value);
            } else {
                keywordTeardown.removeElementToken(index - 1);
            }
        }
    }

    @Override
    public void update(final AModelElement<?> modelElement, final List<String> newArguments) {
        final KeywordTeardown teardown = (KeywordTeardown) modelElement;

        teardown.setKeywordName(newArguments.isEmpty() ? null : RobotToken.create(newArguments.get(0)));
        int elementsToRemove = teardown.getArguments().size();
        for (int i = 0; i < elementsToRemove; i++) {
            teardown.removeElementToken(0);
        }
        for (int i = 1; i < newArguments.size(); i++) {
            teardown.setArgument(i - 1, newArguments.get(i));
        }
    }

    @SuppressWarnings("unchecked")
    @Override
    public void remove(final UserKeyword userKeyword, final AModelElement<?> modelElement) {
        userKeyword.removeUnitSettings((AModelElement<UserKeyword>) modelElement);
    }
}