source: default/v2/trunk/de.ugoe.cs.swe.bnftools.ebnf.ui/src/de/ugoe/cs/swe/bnftools/ui/quickfix/EbnfQuickfixProvider.xtend @ 64

/*
* generated by Xtext
*/
package de.ugoe.cs.swe.bnftools.ui.quickfix

import org.eclipse.xtext.ui.editor.quickfix.Fix
import org.eclipse.xtext.ui.editor.quickfix.IssueResolutionAcceptor
import org.eclipse.xtext.validation.Issue
import de.ugoe.cs.swe.bnftools.validation.EbnfValidator
import org.eclipse.xtext.ui.editor.model.IXtextDocument
import org.eclipse.emf.ecore.EObject
import de.ugoe.cs.swe.bnftools.ebnf.Rule
import de.ugoe.cs.swe.bnftools.ebnf.EbnfFactory
import org.eclipse.xtext.nodemodel.util.NodeModelUtils
import org.eclipse.xtext.nodemodel.INode
import de.ugoe.cs.swe.bnftools.validation.EbnfAnalysisUtils
import java.util.List
import de.ugoe.cs.swe.bnftools.ebnf.RuleReference
import org.eclipse.xtext.nodemodel.ICompositeNode
import de.ugoe.cs.swe.bnftools.ebnf.EtsiBnf
import java.util.ArrayList

/**
 * Custom quickfixes.
 *
 * see http://www.eclipse.org/Xtext/documentation.html#quickfixes
 */
class EbnfQuickfixProvider extends org.eclipse.xtext.ui.editor.quickfix.DefaultQuickfixProvider {
        //works
        @Fix(EbnfValidator.ruleReferencedOneDescription)
        def void fixInlineRuleReferencedOnce(Issue issue, IssueResolutionAcceptor acceptor) {
                acceptor.accept(
                        issue,
                        "Inline the rule",
                        "Delete the rule and replace its single reference by the rule's right side",
                        "upcase.png"
                ) [ element, context |
                        var Rule rule = element as Rule;
                        var IXtextDocument xtextDocument = context.getXtextDocument();
                        var ICompositeNode node = NodeModelUtils.findActualNodeFor(rule);
                        var String nodeText = node.text;
                        var int textLength = nodeText.length - 2;
                        var int offset = node.textRegion.offset;
                        //rename references
                        var ICompositeNode dList = NodeModelUtils.findActualNodeFor(rule.definitionList);
                        var String refText = "(" + dList.text + ")";
                        var int refLength = rule.name.length;
                        var List<RuleReference> references = EbnfAnalysisUtils.findReferences(rule);
                        //is the reference in the rule itself?
                        var boolean foundRule = false;
                        var EObject r = references.get(0) as EObject;
                        var Rule containingRule=null;
                        while (!foundRule) {
                                r = r.eContainer;
                                if (r instanceof Rule) {
                                        containingRule = r as Rule;
                                        foundRule = true;
                                }
                        }
                        if (!rule.equals(containingRule)) {
                                for (ruleRef : references) {
                                        var ICompositeNode refNode = NodeModelUtils.findActualNodeFor(ruleRef);
                                        var int refOffset = refNode.textRegion.offset;

                                        xtextDocument.replace(refOffset, refLength, refText);
                                        if (refOffset < offset) {
                                                offset += refText.length - refLength;
                                        }
                                }
                        }
                

                // delete rule
                xtextDocument.replace(offset, textLength, "");
                ]

        }

        //works
//              @Fix(EbnfValidator.passthroughRuleDescription)
//      def void fixRemovePassthroughRule(Issue issue, IssueResolutionAcceptor acceptor) {
//              acceptor.accept(
//                      issue,
//                      "Remove passthrough rule",
//                      "Delete the passthrough rule and replace its references with its right side",
//                      "upcase.png",
//                      [ element, context |
//                              var Rule rule = element as Rule;
//                              var IXtextDocument xtextDocument = context.getXtextDocument();
//                              var ICompositeNode node = NodeModelUtils.findActualNodeFor(rule);
//                              var String nodeText = node.text;
//                              var int textLength = nodeText.length - 2;
//                              var int offset = node.textRegion.offset;
//                              var ICompositeNode dList = NodeModelUtils.findActualNodeFor(rule.definitionList);
//                              var String refText = "(" + dList.text + ")";
//                              var int refLength = rule.name.length;
//                              var List<RuleReference> references = EbnfAnalysisUtils.findReferences(rule);
//                              //sort references by offset
//                              var int i = 0;
//                              var List<RuleReference> referenceHleper = new ArrayList<RuleReference>();
//                              while(i<references.length){
//                                      
//                                      i++;
//                              }
//                              for (ruleRef : references) {
//                                      var ICompositeNode refNode = NodeModelUtils.findActualNodeFor(ruleRef);
//                                      var int refOffset = refNode.textRegion.offset;
//
//                                      xtextDocument.replace(refOffset, refLength, refText);
//                                      if (refOffset < offset) {
//                                              offset += refText.length - refLength;
//                                      }
//                              }
//                      ]
//              );
//      }

                @Fix(EbnfValidator.unusedRuleDescription)
        def void fixUnusedRule(Issue issue, IssueResolutionAcceptor acceptor) {

                acceptor.accept(issue, "Remove unused rule", "Delete the unused rule", "upcase.png",
                        [ element, context |
                                var Rule rule = element as Rule;
                                var IXtextDocument xtextDocument = context.getXtextDocument();
                                var ICompositeNode node = NodeModelUtils.findActualNodeFor(rule);
                                var int offset = node.textRegion.offset;
                                var String nodeText = node.text;
                                var int textLength = nodeText.length - 2;
                                xtextDocument.replace(offset, textLength, "");
                        ])
        }
}