001// Copyright 2008 The Apache Software Foundation 002// 003// Licensed under the Apache License, Version 2.0 (the "License"); 004// you may not use this file except in compliance with the License. 005// You may obtain a copy of the License at 006// 007// http://www.apache.org/licenses/LICENSE-2.0 008// 009// Unless required by applicable law or agreed to in writing, software 010// distributed under the License is distributed on an "AS IS" BASIS, 011// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 012// See the License for the specific language governing permissions and 013// limitations under the License. 014 015package org.apache.tapestry5.internal.services; 016 017import org.apache.tapestry5.Block; 018import org.apache.tapestry5.internal.structure.Page; 019import org.apache.tapestry5.ioc.internal.util.CollectionFactory; 020import org.apache.tapestry5.services.BeanBlockContribution; 021import org.apache.tapestry5.services.BeanBlockOverrideSource; 022 023import java.util.Collection; 024import java.util.Map; 025 026public class BeanBlockOverrideSourceImpl implements BeanBlockOverrideSource 027{ 028 private final RequestPageCache pageCache; 029 030 private final Map<String, BeanBlockContribution> display = CollectionFactory.newCaseInsensitiveMap(); 031 032 private final Map<String, BeanBlockContribution> edit = CollectionFactory.newCaseInsensitiveMap(); 033 034 public BeanBlockOverrideSourceImpl(RequestPageCache pageCache, 035 Collection<BeanBlockContribution> configuration) 036 { 037 this.pageCache = pageCache; 038 039 for (BeanBlockContribution contribution : configuration) 040 { 041 Map<String, BeanBlockContribution> map = contribution.isEdit() ? edit : display; 042 043 map.put(contribution.getDataType(), contribution); 044 } 045 } 046 047 public boolean hasDisplayBlock(String datatype) 048 { 049 return display.containsKey(datatype); 050 } 051 052 public Block getDisplayBlock(String datatype) 053 { 054 return toBlock(display.get(datatype)); 055 } 056 057 private Block toBlock(BeanBlockContribution contribution) 058 { 059 if (contribution == null) return null; 060 061 Page page = pageCache.get(contribution.getPageName()); 062 063 return page.getRootElement().getBlock(contribution.getBlockId()); 064 } 065 066 public Block getEditBlock(String datatype) 067 { 068 return toBlock(edit.get(datatype)); 069 } 070 071}