com.liferay.ide.xml.search.ui.AddResourceBundleFileMarkerResolution.java Source code

Introduction

Here is the source code for com.liferay.ide.xml.search.ui.AddResourceBundleFileMarkerResolution.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.xml.search.ui;

import com.liferay.ide.core.ILiferayProject;
import com.liferay.ide.core.LiferayCore;
import com.liferay.ide.core.util.CoreUtil;
import com.liferay.ide.portlet.core.dd.PortletDescriptorHelper;

import java.io.ByteArrayInputStream;
import java.net.URL;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IMarker;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.swt.graphics.Image;

/**
 * @author Terry Jia
 */
public class AddResourceBundleFileMarkerResolution extends AbstractResourceBundleMarkerResolution {

    private String resourceBundlePackage = "content";

    private String resourceBundleName = "Language";

    private String portletName = "";

    public AddResourceBundleFileMarkerResolution(IMarker marker, String portletName) {
        super(marker);

        this.portletName = portletName;
    }

    private void checkResourceBundleElement(IProject project) {
        PortletDescriptorHelper portletDescriptorHelper = new PortletDescriptorHelper(project);

        String[] resouceBundles = portletDescriptorHelper.getAllResourceBundles();

        if (resouceBundles.length == 0) {
            portletDescriptorHelper.addResourceBundle(resourceBundlePackage + "." + resourceBundleName,
                    portletName);
        } else {
            for (String resouceBundle : resouceBundles) {
                if (!CoreUtil.isNullOrEmpty(resouceBundle)) {
                    String[] paths = resouceBundle.split("\\.");

                    if (paths.length > 2) {
                        StringBuffer sb = new StringBuffer();

                        for (int i = 0; i < (paths.length - 1); i++) {
                            sb.append(paths[i]);
                            sb.append("/");
                        }

                        resourceBundlePackage = sb.toString();
                        resourceBundleName = paths[paths.length - 1];
                    } else if (paths.length == 2) {
                        resourceBundlePackage = paths[0];
                        resourceBundleName = paths[1];
                    } else if (paths.length == 1) {
                        resourceBundlePackage = "";
                        resourceBundleName = paths[0];
                    }

                    break;
                }
            }
        }
    }

    @Override
    public String getLabel() {
        return "Create a new default resource bundle add it to " + portletName + " portlet";
    }

    public Image getImage() {
        final URL url = LiferayXMLSearchUI.getDefault().getBundle().getEntry("/icons/resource-bundle-new.png");

        return ImageDescriptor.createFromURL(url).createImage();
    }

    @Override
    protected void resolve(IMarker marker) {
        final IProject project = marker.getResource().getProject();

        if (getResourceKey(marker) == null || project == null) {
            return;
        }

        try {
            checkResourceBundleElement(project);

            final ILiferayProject liferayProject = LiferayCore.create(project);

            if (liferayProject == null) {
                return;
            }

            final IFolder folder = liferayProject.getSourceFolder("resources").getFolder(resourceBundlePackage);

            if (!folder.exists()) {
                CoreUtil.makeFolders(folder);
            }

            final IFile resourceBundle = folder.getFile(resourceBundleName + ".properties");

            String resourceKey = getResourceKey(marker);

            if (CoreUtil.isNullOrEmpty(resourceKey)) {
                return;
            }

            String resourceValue = getDefaultResourceValue(resourceKey);
            String resourcePropertyLine = resourceKey + "=" + resourceValue + "\n";

            int contentOffset = 0;
            int resourcePropertyLineOffset = resourcePropertyLine.getBytes().length;

            if (!resourceBundle.exists()) {
                IFolder parent = (IFolder) resourceBundle.getParent();

                CoreUtil.prepareFolder(parent);

                resourceBundle.create(new ByteArrayInputStream(resourcePropertyLine.getBytes("UTF-8")),
                        IResource.FORCE, null);

                contentOffset = resourcePropertyLineOffset;
            } else {
                String contents = CoreUtil.readStreamToString(resourceBundle.getContents());

                StringBuffer sb = new StringBuffer();

                sb.append(contents);
                sb.append(resourcePropertyLine);

                byte[] bytes = sb.toString().trim().getBytes("UTF-8");
                contentOffset = bytes.length;

                resourceBundle.setContents(new ByteArrayInputStream(bytes), IResource.FORCE,
                        new NullProgressMonitor());
            }

            openEditor(resourceBundle, contentOffset - resourcePropertyLineOffset, contentOffset - 1);
        } catch (Exception e) {
            LiferayXMLSearchUI.logError(e);
        }
    }

}