001/*
002 *  Copyright 2016 Anyware Services
003 *
004 *  Licensed under the Apache License, Version 2.0 (the "License");
005 *  you may not use this file except in compliance with the License.
006 *  You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 *  Unless required by applicable law or agreed to in writing, software
011 *  distributed under the License is distributed on an "AS IS" BASIS,
012 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 *  See the License for the specific language governing permissions and
014 *  limitations under the License.
015 */
016package org.ametys.plugins.contentio.synchronize;
017
018import java.util.HashMap;
019import java.util.Map;
020
021import org.apache.avalon.framework.service.ServiceException;
022import org.apache.avalon.framework.service.ServiceManager;
023import org.apache.avalon.framework.service.Serviceable;
024
025import org.ametys.runtime.i18n.I18nizableText;
026import org.ametys.runtime.model.Enumerator;
027
028/**
029 * Collections enumerator
030 */
031public class CollectionEnumerator implements Enumerator<String>, Serviceable
032{
033    private ServiceManager _serviceManager;
034    private SynchronizableContentsCollectionDAO _synchronizableContentsCollectionDAO;
035    
036    @Override
037    public void service(ServiceManager manager) throws ServiceException
038    {
039        _serviceManager = manager;
040    }
041    
042    public I18nizableText getEntry(String value) throws Exception
043    {
044        SynchronizableContentsCollection collection = _getSynchronizableContentsCollectionDAO().getSynchronizableContentsCollection(value);
045        return collection == null ? new I18nizableText("") : collection.getLabel();
046    }
047
048    public Map<String, I18nizableText> getEntries() throws Exception
049    {
050        Map<String, I18nizableText> entries = new HashMap<>();
051        
052        for (SynchronizableContentsCollection collection : _getSynchronizableContentsCollectionDAO().getSynchronizableContentsCollections())
053        {
054            entries.put(collection.getId(), collection.getLabel());
055        }
056        return entries;
057    }
058    
059    /**
060     * Retrieves the synchronizable contents collection DAO
061     * @return the synchronizable contents collection DAO
062     */
063    protected SynchronizableContentsCollectionDAO _getSynchronizableContentsCollectionDAO()
064    {
065        if (_synchronizableContentsCollectionDAO == null)
066        {
067            try
068            {
069                _synchronizableContentsCollectionDAO = (SynchronizableContentsCollectionDAO) _serviceManager.lookup(SynchronizableContentsCollectionDAO.ROLE);
070            }
071            catch (ServiceException e)
072            {
073                throw new RuntimeException("Unable to lookup after the synchronizable contents collection DAO", e);
074            }
075        }
076        
077        return _synchronizableContentsCollectionDAO;
078    }
079}