001/*
002 *  Copyright 2024 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.odf.tree;
017
018import java.util.List;
019import java.util.Map;
020
021import org.apache.avalon.framework.service.ServiceException;
022import org.apache.avalon.framework.service.ServiceManager;
023import org.apache.commons.lang3.StringUtils;
024
025import org.ametys.cms.repository.Content;
026import org.ametys.odf.ProgramItem;
027import org.ametys.odf.orgunit.OrgUnit;
028import org.ametys.runtime.i18n.I18nizableText;
029
030/**
031 * {@link ODFTreeIndicator} to display code of elements
032 *
033 */
034public class ProgramItemCodeIndicator extends AbstractStaticODFTreeIndicator
035{
036    private ODFContentsTreeHelper _odfContentsTreeHelper;
037
038    @Override
039    public void service(ServiceManager smanager) throws ServiceException
040    {
041        super.service(smanager);
042        _odfContentsTreeHelper = (ODFContentsTreeHelper) smanager.lookup(ODFContentsTreeHelper.ROLE);
043    }
044
045    public IndicatorData getIndicatorData(Content content)
046    {
047        if (content instanceof ProgramItem programItem)
048        {
049            String code = _odfContentsTreeHelper.getProgramItemDisplayCode((Content & ProgramItem) programItem);
050            if (StringUtils.isNotEmpty(code))
051            {
052                return new IndicatorData(null, new I18nizableText("plugin.odf", "PLUGINS_ODF_CONTENTS_TREE_INDICATORS_CODE_VALUE", List.of(String.valueOf(code))), null, Map.of());
053            }
054        }
055        else if (content instanceof OrgUnit orgUnit)
056        {
057            return new IndicatorData(null, new I18nizableText("plugin.odf", "PLUGINS_ODF_CONTENTS_TREE_INDICATORS_CODE_VALUE", List.of(orgUnit.getUAICode())), null, Map.of());
058        }
059        
060        return null;
061    }
062}