io.github.bonigarcia.wdm.PhantomJsDriverManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of webdrivermanager Show documentation
Automated driver management and other helper features for Selenium WebDriver in Java
There is a newer version: 5.9.2
/*
 * (C) Copyright 2016 Boni Garcia (http://bonigarcia.github.io/)
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public License
 * (LGPL) version 2.1 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl-2.1.html
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 */
package io.github.bonigarcia.wdm;

import java.io.File;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.io.FileUtils;

/**
 * Manager for PhantomJs.
 *
 * @author Boni Garcia ([email protected])
 * @since 1.4.0
 */
public class PhantomJsDriverManager extends BrowserManager {

	private static PhantomJsDriverManager instance;

	public PhantomJsDriverManager() {
	}

	public static synchronized PhantomJsDriverManager getInstance() {
		if (instance == null) {
			instance = new PhantomJsDriverManager();
		}
		return instance;
	}

	@Override
	protected List getDrivers() throws Exception {
		return getDriversFromMirror(getDriverUrl());
	}

	@Override
	protected String getExportParameter() {
		return WdmConfig.getString("wdm.phantomjsDriverExport");
	}

	@Override
	protected String getDriverVersionKey() {
		return "wdm.phantomjsDriverVersion";
	}

	@Override
	protected String getDriverUrlKey() {
		return "wdm.phantomjsDriverUrl";
	}

	@Override
	protected List getDriverName() {
		return Arrays.asList("phantomjs");
	}

	@Override
	protected String getCurrentVersion(URL url, String driverName)
			throws MalformedURLException {
		String file = url.getFile();
		file = url.getFile().substring(file.lastIndexOf(SEPARATOR),
				file.length());
		final int matchIndex = file.indexOf(driverName);
		String currentVersion = file
				.substring(matchIndex + driverName.length() + 1, file.length());
		final int dashIndex = currentVersion.indexOf('-');
		currentVersion = currentVersion.substring(0, dashIndex);
		return currentVersion;
	}

	@Override
	protected String preDownload(String target, String version)
			throws IOException {
		int iSeparator = target.indexOf(version) - 1;
		int iDash = target.lastIndexOf(version) + version.length();
		int iPoint = target.lastIndexOf(".tar") != -1
				? target.lastIndexOf(".tar") : target.lastIndexOf(".zip");
		target = target.substring(0, iSeparator + 1)
				+ target.substring(iDash + 1, iPoint)
				+ target.substring(iSeparator);
		target = target.replace("beta-", "");
		return target;
	}

	@Override
	protected File postDownload(File archive) throws IOException {
		log.trace("PhatomJS package name: {}", archive);

		File extractFolder = archive.getParentFile().listFiles()[0];
		log.trace("PhatomJS extract folder (to be deleted): {}", extractFolder);

		File binFolder = new File(
				extractFolder.getAbsoluteFile() + File.separator + "bin");
		// Exception for older version of PhantomJS
		int binaryIndex = 0;
		if (!binFolder.exists()) {
			binFolder = extractFolder;
			binaryIndex = 3;
		}

		log.trace("PhatomJS bin folder: {} (index {})", binFolder, binaryIndex);

		File phantomjs = binFolder.listFiles()[binaryIndex];
		log.trace("PhatomJS binary: {}", phantomjs);

		File target = new File(archive.getParentFile().getAbsolutePath()
				+ File.separator + phantomjs.getName());
		log.trace("PhatomJS target: {}", target);

		phantomjs.renameTo(target);
		FileUtils.deleteDirectory(extractFolder);
		return target;
	}

	public BrowserManager useTaobaoMirror() {
		System.setProperty("wdm.phantomjsDriverUrl",
				"http://npm.taobao.org/mirrors/phantomjs");
		return instance;
	}
}