/*

  *   SONEWS News Server

  *   see AUTHORS for the list of contributors

  *

  *   This program is free software: you can redistribute it and/or modify

  *   it under the terms of the GNU General Public License as published by

  *   the Free Software Foundation, either version 3 of the License, or

  *   (at your option) any later version.

  *

  *   This program is distributed in the hope that it will be useful,

  *   but WITHOUT ANY WARRANTY; without even the implied warranty of

  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  *   GNU General Public License for more details.

  *

  *   You should have received a copy of the GNU General Public License

  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.

  */

 package org.sonews.util.io;

 import java.io.BufferedReader;

 import java.io.File;

 import java.io.FileInputStream;

 import java.io.IOException;

 import java.io.InputStream;

 import java.io.InputStreamReader;

 import java.net.URL;

 import java.nio.charset.Charset;

 /**

  * Provides method for loading of resources.

  * @author Christian Lins

  * @since sonews/0.5.0

  */

 public final class Resource

 {

 	/**

 	 * Loads a resource and returns it as URL reference.

 	 * The Resource's classloader is used to load the resource, not

 	 * the System's ClassLoader so it may be safe to use this method

 	 * in a sandboxed environment.

 	 * @return

 	 */

 	public static URL getAsURL(final String name)

 	{

 		if (name == null) {

 			return null;

 		}

 		return Resource.class.getClassLoader().getResource(name);

 	}

 	/**

 	 * Loads a resource and returns an InputStream to it.

 	 * @param name

 	 * @return

 	 */

 	public static InputStream getAsStream(String name)

 	{

 		try {

 			URL url = getAsURL(name);

 			if (url == null) {

 				File file = new File(name);

 				if(file.exists()) {

 					return new FileInputStream(file);

 				}

 				return null;

 			} else {

 				return url.openStream();

 			}

 		} catch (IOException e) {

 			e.printStackTrace();

 			return null;

 		}

 	}

 	/**

 	 * Loads a plain text resource.

 	 * @param withNewline If false all newlines are removed from the

 	 * return String

 	 */

 	public static String getAsString(String name, boolean withNewline)

 	{

 		if (name == null) {

 			return null;

 		}

 		BufferedReader in = null;

 		try {

 			InputStream ins = getAsStream(name);

 			if (ins == null) {

 				return null;

 			}

 			in = new BufferedReader(

 				new InputStreamReader(ins, Charset.forName("UTF-8")));

 			StringBuffer buf = new StringBuffer();

 			for (;;) {

 				String line = in.readLine();

 				if (line == null) {

 					break;

 				}

 				buf.append(line);

 				if (withNewline) {

 					buf.append('\n');

 				}

 			}

 			return buf.toString();

 		} catch (Exception e) {

 			e.printStackTrace();

 			return null;

 		} finally {

 			try {

 				if (in != null) {

 					in.close();

 				}

 			} catch (IOException ex) {

 				ex.printStackTrace();

 			}

 		}

 	}

 }