SurroundWithTextModification.java

/**
 * Copyright (c) 2011, 2012 Cloudsmith Inc. and other contributors, as listed below.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *   Cloudsmith
 * 
 */
package org.cloudsmith.geppetto.pp.dsl.ui.quickfix;

import org.eclipse.jface.text.BadLocationException;
import org.eclipse.xtext.ui.editor.model.IXtextDocument;
import org.eclipse.xtext.ui.editor.model.edit.IModificationContext;

class SurroundWithTextModification extends ReplacingModification {
	private final String suffix;

	/**
	 * @param offset
	 * @param length
	 * @param text
	 *            text used before and after the replaced text
	 */
	SurroundWithTextModification(int offset, int length, String text) {
		super(offset, length, text);
		suffix = text;
	}

	/**
	 * Surrounds text with prefix, suffix
	 * 
	 * @param offset
	 *            start of section to surround
	 * @param length
	 *            length of section to surround
	 * @param prefix
	 *            text before the section
	 * @param suffix
	 *            text after the section
	 */
	SurroundWithTextModification(int offset, int length, String prefix, String suffix) {
		super(offset, length, prefix);
		this.suffix = suffix;
	}

	@Override
	public void apply(IModificationContext context) throws BadLocationException {
		IXtextDocument xtextDocument = context.getXtextDocument();
		String tmp = text + xtextDocument.get(offset, length) + suffix;
		xtextDocument.replace(offset, length, tmp);
	}

}