package org.owasp.webgoat.session;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Properties;
import org.owasp.webgoat.HammerHead;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * *************************************************************************************************
 *
 *
 * This file is part of WebGoat, an Open Web Application Security Project
 * utility. For details, please see http://www.owasp.org/
 *
 * Copyright (c) 2002 - 2007 Bruce Mayhew
 *
 * 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 2 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, write to the Free Software Foundation, Inc., 59 Temple
 * Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 * Getting Source ==============
 *
 * Source for this application is maintained at code.google.com, a repository
 * for free software projects.
 *
 * For details, please see http://code.google.com/p/webgoat/
 */
public class WebgoatProperties extends Properties {

    /**
     *
     */
    private static final long serialVersionUID = 4351681705558227918L;
    final Logger logger = LoggerFactory.getLogger(WebgoatProperties.class);

    public WebgoatProperties(String propertiesFileName) throws IOException {
        if (propertiesFileName == null) {
            throw new IOException("Path to webgoat.properties is null, initialization must have failed");
        }
        File propertiesFile = new File(propertiesFileName);
        if (propertiesFile.exists() == false) {
            throw new IOException("Unable to locate webgoat.properties at: " + propertiesFileName);
        }
        FileInputStream in = new FileInputStream(propertiesFile);
        load(in);
    }

    public int getIntProperty(String key, int defaultValue) {
        int value = defaultValue;

        String s = getProperty(key);
        if (s != null) {
            value = Integer.parseInt(s);
        }

        return value;
    }

    public boolean getBooleanProperty(String key, boolean defaultValue) {
        boolean value = defaultValue;
        key = this.trimLesson(key);

        String s = getProperty(key);
        if (s != null) {
            if (s.equalsIgnoreCase("true")) {
                value = true;
            } else if (s.equalsIgnoreCase("yes")) {
                value = true;
            } else if (s.equalsIgnoreCase("on")) {
                value = true;
            } else if (s.equalsIgnoreCase("false")) {
                value = false;
            } else if (s.equalsIgnoreCase("no")) {
                value = false;
            } else if (s.equalsIgnoreCase("off")) {
                value = false;
            }
        }

        return value;
    }

    private String trimLesson(String lesson) {
        String result = "";

        if (lesson.startsWith("org.owasp.webgoat.lessons.")) {
            result = lesson.substring("org.owasp.webgoat.lessons.".length(), lesson.length());
        } else {
            result = lesson;
        }

        return result;
    }

    public static void main(String[] args) {
        WebgoatProperties properties = null;
        try {
            properties = new WebgoatProperties("C:\\webgoat.properties");
        } catch (IOException e) {
            System.out.println("Error loading properties");
            e.printStackTrace();
        }
        System.out.println(properties.getProperty("CommandInjection.category"));
    }

}