com.liferay.ide.ui.snippets.AddModelEntitySnippetInsertion.java Source code

Introduction

Here is the source code for com.liferay.ide.ui.snippets.AddModelEntitySnippetInsertion.java

Source

/*******************************************************************************
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 *
 *******************************************************************************/

package com.liferay.ide.ui.snippets;

import com.liferay.ide.core.util.CoreUtil;
import com.liferay.ide.core.util.StringPool;
import com.liferay.ide.ui.snippets.wizard.AbstractModelWizard;
import com.liferay.ide.ui.snippets.wizard.AddModelEntityWizard;

import org.apache.commons.lang.StringUtils;
import org.eclipse.ui.IEditorPart;

/**
 * @author Greg Amerson
 */
public class AddModelEntitySnippetInsertion extends ModelSnippetInsertion {

    public AddModelEntitySnippetInsertion() {
        super();
    }

    @Override
    protected AbstractModelWizard createModelWizard(IEditorPart fEditorPart) {
        return new AddModelEntityWizard(fEditorPart);
    }

    protected String getPreparedText(AbstractModelWizard wizard) {
        String text = super.getPreparedText(wizard);

        StringBuffer fields = new StringBuffer();
        String[] propColumns = wizard.getPropertyColumns();
        String var = wizard.getVarName();

        if (!CoreUtil.isNullOrEmpty(propColumns)) {
            for (String prop : propColumns) {
                fields.append(var + ".set" + StringUtils.capitalize(prop) + "(" + prop + ");\n"); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
            }
        }

        String fieldsVal = fields.toString();
        text = StringUtils.replace(text, "${fields}", //$NON-NLS-1$
                CoreUtil.isNullOrEmpty(fieldsVal) ? StringPool.EMPTY : fieldsVal);

        return text;
    }

}