UniversalPage.java

package org.wikibrain.core.model;

import com.google.common.collect.Multimap;
import org.wikibrain.core.WikiBrainException;
import org.wikibrain.core.dao.DaoException;
import org.wikibrain.core.dao.LocalPageDao;
import org.wikibrain.core.lang.Language;
import org.wikibrain.core.lang.LanguageSet;
import org.wikibrain.core.lang.LocalId;

import java.util.ArrayList;
import java.util.Collection;

/**
 * Created with IntelliJ IDEA.
 * User: Brent Hecht
 */
public class UniversalPage extends AbstractUniversalEntity<LocalId> {

    /**
     * The universal id for the universal page. Universal ids are defined within but not across namespaces.
     */
    private final int univId;
    private final NameSpace nameSpace;

    public UniversalPage(int univId, int algorithmId) {
        super(algorithmId);
        this.univId = univId;
        this.nameSpace = null;
    }

    public UniversalPage(int univId, int algorithmId, NameSpace nameSpace, Multimap<Language, LocalId> localPages) {
        super(algorithmId, localPages);
        this.univId = univId;
        this.nameSpace = nameSpace;
    }

    public UniversalPage(int univId, int algorithmId, NameSpace nameSpace, LanguageSet languages) {
        super(algorithmId, languages);
        this.univId = univId;
        this.nameSpace = nameSpace;
    }


    public int getUnivId(){
        return univId;
    }

    public NameSpace getNameSpace() {
        return nameSpace;
    }

    public int getLocalId(Language language) {
        if (isInLanguage(language)) {
            return localEntities.get(language).iterator().next().getId();
        } else {
            return -1;
        }
    }


    public static interface LocalPageChooser {
        public LocalId choose(Collection<LocalId> localPages);
    }

    @Override
    public boolean equals(Object o) {
        if (o instanceof UniversalPage) {
            UniversalPage other = (UniversalPage) o;
            return (this.getUnivId() == other.getUnivId() &&
                    this.getAlgorithmId() == other.getAlgorithmId());
        } else {
            return false;
        }
    }

    //sample code
    public Title getBestEnglishTitle(LocalPageDao lpDao, boolean returnRandomLangIfEnglishNotAvailable) throws WikiBrainException {

        try {
            Language lang = getLanguageSet().getBestAvailableEnglishLang(returnRandomLangIfEnglishNotAvailable);
            LocalPage lp = lpDao.getById(lang, getLocalEntities(lang).iterator().next().getId());
            return lp.getTitle();

        }catch(DaoException e){
            throw new WikiBrainException(e);
        }
    }


}