001/*
002 *  Copyright 2010 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.web.workflow;
017
018import java.util.HashMap;
019import java.util.Iterator;
020import java.util.Map;
021
022import org.ametys.cms.workflow.DefaultWorkflowStepEnumerator;
023import org.ametys.runtime.i18n.I18nizableText;
024import org.ametys.runtime.parameter.Enumerator;
025
026/**
027 * {@link Enumerator} for the default workflow steps excluding steps of id greater than 9000
028 *
029 */
030public class WorkflowStepsEnumerator extends DefaultWorkflowStepEnumerator
031{
032    @Override
033    public Map<Object, I18nizableText> getEntries() throws Exception
034    {
035        Map<Object, I18nizableText> allEntries = super.getEntries();
036        
037        Map<Object, I18nizableText> entries = new HashMap<>(allEntries);
038        
039        Iterator<Object> itSteps = allEntries.keySet().iterator();
040        while (itSteps.hasNext())
041        {
042            Object key = itSteps.next();
043            if (key instanceof Integer)
044            {
045                int stepId = (Integer) key;
046                if (stepId >= 9000)
047                {
048                    entries.remove(key);
049                }
050            }
051        }
052        
053        return entries;
054    }
055}