001    // Copyright 2007, 2008, 2009 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    
015    package org.apache.tapestry5.internal.services;
016    
017    import org.apache.tapestry5.ContentType;
018    import org.apache.tapestry5.MarkupWriter;
019    import org.apache.tapestry5.dom.DefaultMarkupModel;
020    import org.apache.tapestry5.dom.MarkupModel;
021    import org.apache.tapestry5.dom.XMLMarkupModel;
022    import org.apache.tapestry5.internal.structure.Page;
023    import org.apache.tapestry5.services.MarkupWriterFactory;
024    
025    public class MarkupWriterFactoryImpl implements MarkupWriterFactory
026    {
027        private final PageContentTypeAnalyzer analyzer;
028    
029        private final RequestPageCache cache;
030    
031        private final MarkupModel htmlModel = new DefaultMarkupModel();
032    
033        private final MarkupModel xmlModel = new XMLMarkupModel();
034    
035        private final MarkupModel htmlPartialModel = new DefaultMarkupModel(true);
036    
037        private final MarkupModel xmlPartialModel = new XMLMarkupModel(true);
038    
039        public MarkupWriterFactoryImpl(PageContentTypeAnalyzer analyzer, RequestPageCache cache)
040        {
041            this.analyzer = analyzer;
042            this.cache = cache;
043        }
044    
045        public MarkupWriter newMarkupWriter(ContentType contentType)
046        {
047            return newMarkupWriter(contentType, false);
048        }
049    
050        public MarkupWriter newPartialMarkupWriter(ContentType contentType)
051        {
052            return newMarkupWriter(contentType, true);
053        }
054    
055        @SuppressWarnings({"UnusedDeclaration"})
056        private MarkupWriter newMarkupWriter(ContentType contentType, boolean partial)
057        {
058            boolean isHTML = contentType.getMimeType().equalsIgnoreCase("text/html");
059    
060            MarkupModel model = partial
061                                ? (isHTML ? htmlPartialModel : xmlPartialModel)
062                                : (isHTML ? htmlModel : xmlModel);
063    
064            // The charset parameter sets the encoding attribute of the XML declaration, if
065            // not null and if using the XML model.
066    
067            return new MarkupWriterImpl(model, contentType.getCharset());
068        }
069    
070        public MarkupWriter newMarkupWriter(String pageName)
071        {
072            Page page = cache.get(pageName);
073    
074            ContentType contentType = analyzer.findContentType(page);
075    
076            return newMarkupWriter(contentType);
077        }
078    }