FilteredStory.java

package org.jbehave.core.embedder;

import java.util.HashMap;
import java.util.Map;

import org.jbehave.core.model.Meta;
import org.jbehave.core.model.Scenario;
import org.jbehave.core.model.Story;

public class FilteredStory {

	private boolean alwaysAllowed;
    private boolean storyAllowed;
    private Map<Scenario, Boolean> scenariosAllowed;

    public FilteredStory(MetaFilter filter, Story story, StoryControls storyControls){
    	this(filter, story, storyControls, false);
    }
    
    public FilteredStory(MetaFilter filter, Story story, StoryControls storyControls, boolean givenStory) {
    	if (  givenStory && storyControls.ignoreMetaFiltersIfGivenStory() ){
    		alwaysAllowed = true;
    	}
        String storyMetaPrefix = storyControls.storyMetaPrefix();
        String scenarioMetaPrefix = storyControls.scenarioMetaPrefix();
        Meta storyMeta = story.getMeta().inheritFrom(story.asMeta(storyMetaPrefix));
        storyAllowed = filter.allow(storyMeta);
        scenariosAllowed = new HashMap<Scenario, Boolean>();
        for (Scenario scenario : story.getScenarios()) {
            boolean scenarioAllowed;
            if (scenario.getExamplesTable().getRowCount() > 0 && metaByRow(scenario, storyControls)) {
            	// allow filtering on meta by row 
                scenarioAllowed = true;
            } else {
                Meta scenarioMeta = scenario.getMeta().inheritFrom(
                        scenario.asMeta(scenarioMetaPrefix).inheritFrom(storyMeta));
                scenarioAllowed = filter.allow(scenarioMeta);
            }
            scenariosAllowed.put(scenario, scenarioAllowed);
        }
    }

    public boolean allowed() {
    	if ( alwaysAllowed ) return true;
        return storyAllowed || scenariosAllowed.values().contains(true);
    }

    public boolean allowed(Scenario scenario) {
    	if ( alwaysAllowed ) return true;
        return scenariosAllowed.get(scenario);
    }

    private boolean metaByRow(Scenario scenario, StoryControls storyControls) {
        if (scenario.getExamplesTable().getProperties().containsKey("metaByRow")) {
            return scenario.getExamplesTable().metaByRow();
        }

        return storyControls.metaByRow();
    }
}