WafLogParserFactory.java

////////////////////////////////////////////////////////////////////////
//
//     Copyright (c) 2009-2013 Denim Group, Ltd.
//
//     The contents of this file are subject to the Mozilla Public 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.mozilla.org/MPL/
//
//     Software distributed under the License is distributed on an "AS IS"
//     basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//     License for the specific language governing rights and limitations
//     under the License.
//
//     The Original Code is ThreadFix.
//
//     The Initial Developer of the Original Code is Denim Group, Ltd.
//     Portions created by Denim Group, Ltd. are Copyright (C)
//     Denim Group, Ltd. All Rights Reserved.
//
//     Contributor(s): Denim Group, Ltd.
//
////////////////////////////////////////////////////////////////////////
package com.denimgroup.threadfix.service.waflog;

import com.denimgroup.threadfix.data.dao.SecurityEventDao;
import com.denimgroup.threadfix.data.dao.WafRuleDao;
import com.denimgroup.threadfix.data.entities.WafType;
import com.denimgroup.threadfix.service.SanitizedLogger;

/**
 * @author mcollins
 * 
 */
public class WafLogParserFactory {

	private final SanitizedLogger log = new SanitizedLogger(WafLogParserFactory.class);
	private WafRuleDao wafRuleDao;
	private SecurityEventDao securityEventDao;
	
	public WafLogParserFactory(WafRuleDao wafRuleDao, SecurityEventDao securityEventDao) {
		this.wafRuleDao = wafRuleDao;
		this.securityEventDao = securityEventDao;
	}
	
	/**
	 * @param wafName
	 * @return
	 */
	public WafLogParser getTracker(String wafName) {
		if (wafName == null || wafName.trim().equals(""))
			return null;

		if (wafName.equals(WafType.MOD_SECURITY)) {
			return new ModSecurityLogParser(wafRuleDao, securityEventDao);
		} else if (wafName.equals(WafType.SNORT)) {
			return new SnortLogParser(wafRuleDao, securityEventDao);
		} else {
			log.warn("Invalid WAF type name '"
					+ wafName
					+ "'  Unable to find suitable WafLogParser implementation class.  Returning null");
			return null;
		}
	}
}