/*
 * This file is part of Flicklib.
 *
 * Copyright (C) Francis De Brabandere
 *
 * 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 com.flicklib.service;

import java.io.IOException;
import java.io.InputStream;
import java.net.URLConnection;
import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.flicklib.tools.IOTools;

/**
 * Loads a page source file from the class path
 * @author francisdb
 */
public class FileSourceLoader implements SourceLoader {
    private static final Logger LOGGER = LoggerFactory.getLogger(FileSourceLoader.class);
    
    private final String charsetName;
    
    public FileSourceLoader(final String charsetname) {
		this.charsetName = charsetname;
	}
    
    public FileSourceLoader() {
		this(null);
	}
    
	/** {@inheritDoc} */
    @Override
    public Source loadSource(String url) throws IOException {
        return loadSource(url, true);
    }
    
    /** {@inheritDoc} */
    @Override
    public Source post(String url, Map<String, String> parameters, Map<String, String> headers) throws IOException {
        return loadSource(url, true);
    }
	
    @Override
    public Source loadSource(String url, boolean useCache) throws IOException {
    	String contentType = null;
    	contentType = URLConnection.guessContentTypeFromName(url);
    	LOGGER.trace("Content-Type: "+contentType);
        return new Source(url, getOrPost(url, charsetName), contentType, url);
    }
    
    
    private String getOrPost(String url, String charsetName) throws IOException {
        String source = null;
        InputStream fis = null;
        try {
            fis = FileSourceLoader.class.getClassLoader().getResourceAsStream(url);
            if (fis==null) {
                throw new IOException("File not found : "+url);
            }
            source = IOTools.inputSreamToString(fis, charsetName);
        } finally {
            if (fis != null) {
                fis.close();
            }
        }
        return source;
    }
    
    @Override
    public RestBuilder url(String url) {
    	throw new UnsupportedOperationException("not implemented");
    }
}