org.xmind.core.internal.dom.WorkbookStyleRefCounter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xmindjbehaveplugin Show documentation
Plugin that manages JBehave stories storage in XMind mindmap file, which allows design jBehave tests right after the brainstorm
The newest version!
/* ******************************************************************************
 * Copyright (c) 2006-2012 XMind Ltd. and others.
 * 
 * This file is a part of XMind 3. XMind releases 3 and
 * above are dual-licensed under the Eclipse Public License (EPL),
 * which is available at http://www.eclipse.org/legal/epl-v10.html
 * and the GNU Lesser General Public License (LGPL), 
 * which is available at http://www.gnu.org/licenses/lgpl.html
 * See http://www.xmind.net/license.html for details.
 * 
 * Contributors:
 *     XMind Ltd. - initial API and implementation
 *******************************************************************************/
package org.xmind.core.internal.dom;

import static org.xmind.core.internal.zip.ArchiveConstants.STYLES_XML;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import org.xmind.core.IFileEntry;
import org.xmind.core.internal.AbstractRefCounter;
import org.xmind.core.style.IStyle;
import org.xmind.core.util.HyperlinkUtils;
import org.xmind.core.util.IStyleRefCounter;
import org.xmind.core.util.Property;

public class WorkbookStyleRefCounter extends AbstractRefCounter implements
        IStyleRefCounter {

    private StyleSheetImpl sheet;

    private ManifestImpl manifest;

    private Map groupCache = new HashMap();

    WorkbookStyleRefCounter(StyleSheetImpl sheet, ManifestImpl manifest) {
        this.sheet = sheet;
        this.manifest = manifest;
    }

    protected Object findResource(String resourceId) {
        return sheet.findStyle(resourceId);
    }

    protected void postIncreaseRef(String resourceId, Object resource) {
        IStyle style = (IStyle) resource;
        if (sheet.equals(style.getOwnedStyleSheet())) {
            String group = sheet.findOwnedGroup(style);
            if (group != null) {
                groupCache.put(resourceId, group);
            } else {
                String cachedGroup = groupCache.get(resourceId);
                if (cachedGroup != null) {
                    addStyleToGroup(style, cachedGroup);
                }
            }

            Iterator defaultStyles = style.defaultStyles();
            while (defaultStyles.hasNext()) {
                Property ds = defaultStyles.next();
                increaseRef(ds.value);
            }

            IFileEntry entry = manifest.getFileEntry(STYLES_XML);
            if (entry == null) {
                entry = manifest.createFileEntry(STYLES_XML);
            }
            entry.increaseReference();

            Iterator it = style.properties();
            while (it.hasNext()) {
                Property p = it.next();
                if (HyperlinkUtils.isAttachmentURL(p.value)) {
                    String entryPath = HyperlinkUtils.toAttachmentPath(p.value);
                    entry = manifest.getFileEntry(entryPath);
                    if (entry != null) {
                        entry.increaseReference();
                    }
                }
            }

        }
    }

    private void addStyleToGroup(IStyle style, String groupName) {
        sheet.addStyle(style, groupName);
    }

    protected void postDecreaseRef(String resourceId, Object resource) {
        IStyle style = (IStyle) resource;
        if (sheet.equals(style.getOwnedStyleSheet())) {
            String group = sheet.findOwnedGroup(style);
            if (group != null) {
                if (getRefCount(resourceId) <= 0) {
                    sheet.removeStyle(style);
                }
            }

            Iterator defaultStyles = style.defaultStyles();
            while (defaultStyles.hasNext()) {
                Property ds = defaultStyles.next();
                decreaseRef(ds.value);
            }

            if (sheet.isEmpty()) {
                IFileEntry entry = manifest.getFileEntry(STYLES_XML);
                if (entry != null) {
                    entry.decreaseReference();
                }
            }

            Iterator it = style.properties();
            while (it.hasNext()) {
                Property p = it.next();
                if (HyperlinkUtils.isAttachmentURL(p.value)) {
                    String entryPath = HyperlinkUtils.toAttachmentPath(p.value);
                    IFileEntry entry = manifest.getFileEntry(entryPath);
                    if (entry != null) {
                        entry.decreaseReference();
                    }
                }
            }
        }
    }

}