org.thymeleaf.testing.templateengine.util.ResourceUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of thymeleaf-testing Show documentation
XML/XHTML/HTML5 template engine for Java
There is a newer version: 3.1.0.M2
/*
 * =============================================================================
 * 
 *   Copyright (c) 2011-2016, The THYMELEAF team (http://www.thymeleaf.org)
 * 
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 * 
 *       http://www.apache.org/licenses/LICENSE-2.0
 * 
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 * 
 * =============================================================================
 */
package org.thymeleaf.testing.templateengine.util;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.Reader;
import java.io.StringReader;
import java.io.UnsupportedEncodingException;
import java.io.Writer;

import org.thymeleaf.testing.templateengine.exception.TestEngineExecutionException;
import org.thymeleaf.util.Validate;






public final class ResourceUtils {

    
    
    public static File createTempFile(final String prefix, 
            final String suffix, final String contents, final String characterEncoding) {

        Validate.notNull(contents, "Contents cannot be null");
        Validate.notNull(characterEncoding, "Character encoding cannot be null");
        
        try {

            final String totalPrefix = "thymeleaf-testing" + (prefix != null? "-" + prefix : "");
            final File tempFile = File.createTempFile(totalPrefix, suffix);
            tempFile.deleteOnExit();
            
            Writer writer = null;
            try {
                final FileOutputStream os = new FileOutputStream(tempFile, false);
                writer = new OutputStreamWriter(os, characterEncoding);
                writer.write(contents);
            } catch (final Throwable t) {
                throw new TestEngineExecutionException( 
                        "Could not write contents of temporary file \"" + tempFile.getAbsolutePath() + "\"", t);
            } finally {
                try {
                    if (writer != null) {
                        writer.close();
                    }
                } catch (final Throwable ignored) {
                    // ignored
                }
            }
            
            return tempFile;
            
        } catch (final TestEngineExecutionException e) {
            throw e;
        } catch (final Throwable t) {
            throw new TestEngineExecutionException( 
                    "Could not create temporary file", t);
        }
        
    }
    
    
    
    
    
    public static String read(final InputStream is, final String characterEncoding) 
            throws UnsupportedEncodingException, IOException {
        return read(is, characterEncoding, false);
    }

    
    
    public static String read(final InputStream is, final String characterEncoding, final boolean normalize) 
            throws UnsupportedEncodingException, IOException {
        return read(new InputStreamReader(is, characterEncoding), normalize);
    }

    
    
    public static String read(final Reader reader, final boolean normalize) throws IOException {

        BufferedReader bufferedReader = null;
        try {
            
            bufferedReader = new BufferedReader(reader);
            final StringBuilder strBuilder = new StringBuilder();
            String line = bufferedReader.readLine();
            if (line != null) {
                strBuilder.append(normalize? normalizeLine(line) : line);
                while ((line = bufferedReader.readLine()) != null) {
                    strBuilder.append('\n');
                    strBuilder.append(normalize? normalizeLine(line) : line);
                }
            }

            if (normalize) {
                normalizeFileEnd(strBuilder);
            }
            return strBuilder.toString();
            
        } finally {
            try {
                if (bufferedReader != null) {
                    bufferedReader.close();
                }
            } catch (final Throwable ignored) {
                // ignored
            }
        }

    }
    

    
    
    public static String normalize(final String text) {
        try {
            return read(new StringReader(text), true);
        } catch (final IOException e) {
            // Should never happen
            throw new RuntimeException(e);
        }
    }
    
    
    private static String normalizeLine(final String text) {
        
        final StringBuilder strBuilder = new StringBuilder();
        final int textLen = text.length();
        
        for (int i = 0; i < textLen; i++) {
            final char c = text.charAt(i);
            if (c != 13) {
                strBuilder.append(c);
            }
        }
        
        return strBuilder.toString();
        
    }
    

    
    private static void normalizeFileEnd(final StringBuilder strBuilder) {
        
        if (strBuilder.length() == 0) {
            return;
        }
        
        int i = strBuilder.length() - 1;
        char c = strBuilder.charAt(i);
        while (Character.isWhitespace(c)) {
            strBuilder.deleteCharAt(i);
            if (i == 0) {
                break;
            }
            c = strBuilder.charAt(--i);
        }
        
    }
    
    
    
    private ResourceUtils() {
        super();
    }
    
    
    
}