com.liferay.knowledgebase.display.selector.KBFolderKBArticleSelector.java Source code

Introduction

Here is the source code for com.liferay.knowledgebase.display.selector.KBFolderKBArticleSelector.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.knowledgebase.display.selector;

import com.liferay.knowledgebase.model.KBArticle;
import com.liferay.knowledgebase.model.KBFolder;
import com.liferay.knowledgebase.model.KBFolderConstants;
import com.liferay.knowledgebase.model.impl.KBFolderImpl;
import com.liferay.knowledgebase.service.KBArticleLocalServiceUtil;
import com.liferay.knowledgebase.service.KBFolderLocalServiceUtil;
import com.liferay.portal.kernel.exception.PortalException;
import com.liferay.portal.kernel.util.Validator;
import com.liferay.portal.kernel.workflow.WorkflowConstants;

/**
 * @author Adolfo Prez
 */
public class KBFolderKBArticleSelector implements KBArticleSelector {

    @Override
    public KBArticle findByResourcePrimKey(long groupId, String preferredKBFolderUrlTitle,
            long ancestorResourcePrimKey, long resourcePrimKey) throws PortalException {

        KBFolder ancestorKBFolder = _rootKBFolder;

        if (ancestorResourcePrimKey != KBFolderConstants.DEFAULT_PARENT_FOLDER_ID) {

            ancestorKBFolder = KBFolderLocalServiceUtil.fetchKBFolder(ancestorResourcePrimKey);

            if (ancestorKBFolder == null) {
                return null;
            }
        }

        KBArticle kbArticle = KBArticleLocalServiceUtil.fetchLatestKBArticle(resourcePrimKey,
                WorkflowConstants.STATUS_APPROVED);

        if ((kbArticle == null) || !isDescendant(kbArticle, ancestorKBFolder)) {
            return findClosestMatchingKBArticle(groupId, ancestorKBFolder, preferredKBFolderUrlTitle);
        }

        return kbArticle;
    }

    @Override
    public KBArticle findByUrlTitle(long groupId, String preferredKBFolderUrlTitle, long ancestorResourcePrimKey,
            String kbFolderUrlTitle, String urlTitle) throws PortalException {

        KBFolder ancestorKBFolder = _rootKBFolder;

        if (ancestorResourcePrimKey != KBFolderConstants.DEFAULT_PARENT_FOLDER_ID) {

            ancestorKBFolder = KBFolderLocalServiceUtil.fetchKBFolder(ancestorResourcePrimKey);

            if (ancestorKBFolder == null) {
                return null;
            }
        }

        KBFolder kbFolder = _rootKBFolder;

        if (Validator.isNotNull(kbFolderUrlTitle)) {
            if (kbFolderUrlTitle.equals(ancestorKBFolder.getUrlTitle())) {
                kbFolder = ancestorKBFolder;
            } else {
                kbFolder = KBFolderLocalServiceUtil.fetchKBFolderByUrlTitle(groupId,
                        ancestorKBFolder.getKbFolderId(), kbFolderUrlTitle);
            }
        }

        KBArticle kbArticle = KBArticleLocalServiceUtil.fetchKBArticleByUrlTitle(groupId, kbFolder.getKbFolderId(),
                urlTitle);

        if ((kbArticle == null) || !isDescendant(kbArticle, ancestorKBFolder)) {
            return findClosestMatchingKBArticle(groupId, ancestorKBFolder, preferredKBFolderUrlTitle,
                    kbFolderUrlTitle, urlTitle);
        }

        return kbArticle;
    }

    protected KBArticle findClosestMatchingKBArticle(long groupId, KBFolder ancestorKBFolder,
            String preferredKBFolderUrlTitle) throws PortalException {

        KBFolder kbFolder = null;

        if (Validator.isNotNull(preferredKBFolderUrlTitle)) {
            kbFolder = KBFolderLocalServiceUtil.fetchKBFolderByUrlTitle(groupId, ancestorKBFolder.getKbFolderId(),
                    preferredKBFolderUrlTitle);
        }

        if (kbFolder == null) {
            kbFolder = KBFolderLocalServiceUtil.fetchFirstChildKBFolder(groupId, ancestorKBFolder.getKbFolderId());
        }

        if (kbFolder == null) {
            kbFolder = ancestorKBFolder;
        }

        return KBArticleLocalServiceUtil.fetchFirstChildKBArticle(groupId, kbFolder.getKbFolderId());
    }

    protected KBArticle findClosestMatchingKBArticle(long groupId, KBFolder ancestorKBFolder,
            String preferredKBFolderUrlTitle, String kbFolderUrlTitle, String urlTitle) throws PortalException {

        KBFolder kbFolder = getCandidateKBFolder(groupId, preferredKBFolderUrlTitle, ancestorKBFolder,
                kbFolderUrlTitle);

        KBArticle kbArticle = KBArticleLocalServiceUtil.fetchKBArticleByUrlTitle(groupId, kbFolder.getKbFolderId(),
                urlTitle);

        if (kbArticle != null) {
            return kbArticle;
        }

        return KBArticleLocalServiceUtil.fetchFirstChildKBArticle(groupId, kbFolder.getKbFolderId());
    }

    protected KBFolder getCandidateKBFolder(long groupId, String preferredKBFolderUrlTitle,
            KBFolder ancestorKBFolder, String kbFolderUrlTitle) throws PortalException {

        KBFolder kbFolder = null;

        if (Validator.isNotNull(kbFolderUrlTitle)) {
            kbFolder = KBFolderLocalServiceUtil.fetchKBFolderByUrlTitle(groupId, ancestorKBFolder.getKbFolderId(),
                    kbFolderUrlTitle);
        }

        if ((kbFolder == null) && Validator.isNotNull(preferredKBFolderUrlTitle)) {

            kbFolder = KBFolderLocalServiceUtil.fetchKBFolderByUrlTitle(groupId, ancestorKBFolder.getKbFolderId(),
                    preferredKBFolderUrlTitle);
        }

        if (kbFolder == null) {
            kbFolder = KBFolderLocalServiceUtil.fetchFirstChildKBFolder(groupId, ancestorKBFolder.getKbFolderId());
        }

        if (kbFolder == null) {
            return ancestorKBFolder;
        }

        return kbFolder;
    }

    protected boolean isDescendant(KBArticle kbArticle, KBFolder ancestorKBFolder) throws PortalException {

        KBFolder parentKBFolder = KBFolderLocalServiceUtil.fetchKBFolder(kbArticle.getKbFolderId());

        while ((parentKBFolder != null) && !parentKBFolder.equals(ancestorKBFolder)) {

            parentKBFolder = KBFolderLocalServiceUtil.fetchKBFolder(parentKBFolder.getParentKBFolderId());
        }

        if (parentKBFolder != null) {
            return true;
        }

        return false;
    }

    private static final KBFolder _rootKBFolder;

    static {
        _rootKBFolder = new KBFolderImpl();

        _rootKBFolder.setKbFolderId(KBFolderConstants.DEFAULT_PARENT_FOLDER_ID);
    }

}