/*
* 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.mapping.hash.comment.tables.keyword;
import java.util.List;
import org.rf.ide.core.testdata.mapping.IHashCommentMapper;
import org.rf.ide.core.testdata.model.RobotFile;
import org.rf.ide.core.testdata.model.table.keywords.KeywordArguments;
import org.rf.ide.core.testdata.model.table.keywords.UserKeyword;
import org.rf.ide.core.testdata.text.read.ParsingState;
import org.rf.ide.core.testdata.text.read.RobotLine;
import org.rf.ide.core.testdata.text.read.recognizer.RobotToken;
public class UserKeywordSettingArgumentsCommentMapper implements IHashCommentMapper {
@Override
public boolean isApplicable(ParsingState state) {
return (state == ParsingState.KEYWORD_SETTING_ARGUMENTS
|| state == ParsingState.KEYWORD_SETTING_ARGUMENTS_ARGUMENT_VALUE);
}
@Override
public void map(final RobotLine currentLine, final RobotToken rt, final ParsingState currentState,
final RobotFile fileModel) {
List<UserKeyword> keywords = fileModel.getKeywordTable().getKeywords();
UserKeyword keyword = keywords.get(keywords.size() - 1);
List<KeywordArguments> arguments = keyword.getArguments();
KeywordArguments keywordArguments = arguments.get(arguments.size() - 1);
keywordArguments.addCommentPart(rt);
}
}