com.leonarduk.bookkeeper.web.AbstractWebConfig Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of robotbookkeeper Show documentation

Robot Bookkeeper

There is a newer version: 1.8

/**
 * AbstractWebConfig
 *
 * @author ${author}
 * @since 01-Jul-2016
 */
package com.leonarduk.bookkeeper.web;

import java.io.File;
import java.io.IOException;

import org.apache.log4j.Logger;
import org.openqa.selenium.WebDriver;

import com.leonarduk.web.SeleniumUtils;
import com.leonarduk.webscraper.core.FileUtils;
import com.leonarduk.webscraper.core.config.Config;

public class AbstractWebConfig {
	private static final Logger	_logger	= Logger.getLogger(AbstractWebConfig.class);
	private final Config		config;
	private final File			tempDir;

	private WebDriver downloadCapableBrowser;

	public AbstractWebConfig(final Config config2) throws IOException {
		this.config = config2;
		this.tempDir = FileUtils.createTempDir();
	}

	protected Config getConfig() {
		return this.config;
	}

	public File getDownloadDir() {
		return this.tempDir;
	}

	public String getMandatoryField(final String fieldName) {
		final String property = this.getConfig().getProperty(fieldName);
		if (property == null) {
			AbstractWebConfig._logger.error("Cannot find " + fieldName);
		}
		AbstractWebConfig._logger.info(fieldName + "=" + property);
		return property;
	}

	public WebDriver getWebDriver() throws IOException {
		if (null == this.downloadCapableBrowser) {
			this.downloadCapableBrowser = SeleniumUtils.getDownloadCapableBrowser(this.tempDir);
		}
		return this.downloadCapableBrowser;
	}

}