l2j-server-game/L2_GameServer_T1/java/net/sf/l2j/gameserver/cache/HtmCache.java

/*
 * 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 net.sf.l2j.gameserver.cache;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileFilter;
import java.io.FileInputStream;
import java.util.logging.Logger;

import javolution.util.FastMap;
import net.sf.l2j.Config;
import net.sf.l2j.gameserver.util.Util;

/**
 * @author Layane
 *
 */
public class HtmCache
{
    private static Logger _log = Logger.getLogger(HtmCache.class.getName());
    private static HtmCache _instance;

    private FastMap<Integer, String> _cache;

    private int _loadedFiles;
    private long _bytesBuffLen;

    public static HtmCache getInstance()
    {
        if (_instance == null)
            _instance = new HtmCache();

        return _instance;
    }

    public HtmCache()
    {
        _cache = new FastMap<Integer, String>();
        reload();
    }

    public void reload()
    {
        reload(Config.DATAPACK_ROOT);
    }

    public void reload(File f)
    {
        if (!Config.LAZY_CACHE)
        {
        	_log.info("Html cache start...");
            parseDir(f);
            _log.info("Cache[HTML]: " + String.format("%.3f",getMemoryUsage())  + " megabytes on " + getLoadedFiles() + " files loaded");
        }
        else
        {
        	_cache.clear();
        	_loadedFiles = 0;
        	_bytesBuffLen = 0;
            _log.info("Cache[HTML]: Running lazy cache");
        }
    }

    public void reloadPath(File f)
    {
    	parseDir(f);
    	_log.info("Cache[HTML]: Reloaded specified path.");
    }

    public double getMemoryUsage()
    {
    	return ((float)_bytesBuffLen/1048576);
    }

    public int getLoadedFiles()
    {
        return _loadedFiles;
    }

    class HtmFilter implements FileFilter
    {
        public boolean accept(File file)
        {
            if (!file.isDirectory())
            {
                return (file.getName().endsWith(".htm") || file.getName().endsWith(".html"));
            }
            return true;
        }
    }

    private void parseDir(File dir)
    {
        FileFilter filter = new HtmFilter();
        File[] files = dir.listFiles(filter);

        for (File file : files)
        {
            if (!file.isDirectory())
                loadFile(file);
            else
                parseDir(file);
        }
    }

    public String loadFile(File file)
    {
        HtmFilter filter = new HtmFilter();

        if (file.exists() && filter.accept(file) && !file.isDirectory())
        {
            String content;
            FileInputStream fis = null;

            try
            {
                fis = new FileInputStream(file);
                BufferedInputStream bis = new BufferedInputStream(fis);
                int bytes = bis.available();
                byte[] raw = new byte[bytes];

                bis.read(raw);
                content = new String(raw, "UTF-8");
                content = content.replaceAll("\r\n","\n");

                String relpath = Util.getRelativePath(Config.DATAPACK_ROOT,file);
                int hashcode = relpath.hashCode();

                String oldContent = _cache.get(hashcode);

                if (oldContent == null)
                {
                    _bytesBuffLen += bytes;
                    _loadedFiles++;
                }
                else
                {
                    _bytesBuffLen = _bytesBuffLen - oldContent.length() + bytes;
                }

                _cache.put(hashcode,content);

                return content;
            }
            catch (Exception e)
            {
                _log.warning("problem with htm file " + e);
            }
            finally
            {
                try { fis.close(); } catch (Exception e1) { }
            }
        }

        return null;
    }

    public String getHtmForce(String path)
    {
        String content = getHtm(path);

        if (content == null)
        {
            content = "<html><body>My text is missing:<br>" + path + "</body></html>";
            _log.warning("Cache[HTML]: Missing HTML page: " + path);
        }

        return content;
    }

    public String getHtm(String path)
    {
        String content = _cache.get(path.hashCode());

        if (Config.LAZY_CACHE && content == null)
            content = loadFile(new File(Config.DATAPACK_ROOT,path));

        return content;
    }

    public boolean contains(String path)
    {
        return _cache.containsKey(path.hashCode());
    }

    /**
     * Check if an HTM exists and can be loaded
     * @param
     * path The path to the HTM
     * */
    public boolean isLoadable(String path)
    {
    	File file = new File(path);
        HtmFilter filter = new HtmFilter();

        if (file.exists() && filter.accept(file) && !file.isDirectory())
	        return true;

    	return false;
    }
}