/*
* 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.KeywordReturn;
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 UserKeywordSettingReturnCommentMapper implements IHashCommentMapper {
@Override
public boolean isApplicable(ParsingState state) {
return (state == ParsingState.KEYWORD_SETTING_RETURN || state == ParsingState.KEYWORD_SETTING_RETURN_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<KeywordReturn> returns = keyword.getReturns();
KeywordReturn returned = returns.get(returns.size() - 1);
returned.addCommentPart(rt);
}
}