/* * #%L * Alfresco Repository * %% * Copyright (C) 2005 - 2016 Alfresco Software Limited * %% * This file is part of the Alfresco software. * If the software was purchased under a paid Alfresco license, the terms of * the paid license agreement will prevail. Otherwise, the software is * provided under the following open source license terms: * * Alfresco 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 3 of the License, or * (at your option) any later version. * * Alfresco 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. * * You should have received a copy of the GNU Lesser General Public License * along with Alfresco. If not, see <http://www.gnu.org/licenses/>. * #L% */ package org.alfresco.repo.admin.patch.impl; import java.io.InputStream; import java.util.ArrayList; import java.util.List; import org.alfresco.model.ContentModel; import org.alfresco.repo.admin.patch.AbstractPatch; import org.alfresco.repo.content.MimetypeMap; import org.alfresco.service.cmr.model.FileFolderService; import org.alfresco.service.cmr.repository.ContentService; import org.alfresco.service.cmr.repository.ContentWriter; import org.alfresco.service.cmr.repository.NodeRef; /** * Helper generic patch useful when updating email templates. * * @author Roy Wetherall */ public abstract class GenericEMailTemplateUpdatePatch extends AbstractPatch { /** Content service */ protected ContentService contentService; /** File folder service */ protected FileFolderService fileFolderService; /** Indicates whether to update the base file or not */ private boolean updateBaseFile = true; /** Indicates whether to create a sibling if it's missing (rather than just update) */ private boolean createSiblingIfMissing = true; /** * @param contentService content service */ public void setContentService(ContentService contentService) { this.contentService = contentService; } /** * @param fileFolderService file folder service */ public void setFileFolderService(FileFolderService fileFolderService) { this.fileFolderService = fileFolderService; } /** * @param createSiblingIfMissing boolean */ public void setCreateSiblingIfMissing(boolean createSiblingIfMissing) { this.createSiblingIfMissing = createSiblingIfMissing; } /** * @param updateBaseFile boolean */ public void setUpdateBaseFile(boolean updateBaseFile) { this.updateBaseFile = updateBaseFile; } /** * * @throws Exception */ protected void updateTemplates() throws Exception { NodeRef baseTemplate = getBaseTemplate(); if (nodeService.exists(baseTemplate) == true) { if (updateBaseFile == true) { updateContent(baseTemplate, getPath(), getBaseFileName(), false); } for (String siblingFile : getSiblingFiles()) { updateSiblingContent(baseTemplate, getPath(), siblingFile); } } } protected abstract NodeRef getBaseTemplate(); protected abstract String getPath(); protected abstract String getBaseFileName(); protected abstract String[] getLocales(); protected List<String> getSiblingFiles() { List<String> siblingFiles = new ArrayList<String>(getLocales().length); for (String locale : getLocales()) { siblingFiles.add(makeSiblingFileName(getBaseFileName(), locale)); } return siblingFiles; } private String makeSiblingFileName(String baseFileName, String locale) { int index = baseFileName.lastIndexOf("."); StringBuilder builder = new StringBuilder(); builder.append(baseFileName.substring(0, index)) .append("_") .append(locale) .append(baseFileName.substring(index)); return builder.toString(); } private void updateSiblingContent(NodeRef nodeRef, String path, String fileName) { NodeRef parent = nodeService.getPrimaryParent(nodeRef).getParentRef(); if (parent != null) { NodeRef sibling = fileFolderService.searchSimple(parent, fileName); if (sibling != null) { updateContent(sibling, path, fileName, false); } else if (createSiblingIfMissing == true) { sibling = fileFolderService.create(parent, fileName, ContentModel.TYPE_CONTENT).getNodeRef(); updateContent(sibling, path, fileName, true); } } } private void updateContent(NodeRef nodeRef, String path, String fileName, boolean newFile) { // Make versionable nodeService.addAspect(nodeRef, ContentModel.ASPECT_VERSIONABLE, null); // Update content InputStream is = this.getClass().getClassLoader().getResourceAsStream(path + fileName); if (is != null) { ContentWriter contentWriter = contentService.getWriter(nodeRef, ContentModel.PROP_CONTENT, true); if (newFile == true) { contentWriter.setMimetype(MimetypeMap.MIMETYPE_TEXT_PLAIN); contentWriter.setEncoding("UTF-8"); } contentWriter.putContent(is); } } }