UpdateChecker.java

/*
 * This file is part of FTB Launcher.
 *
 * Copyright © 2012-2016, FTB Launcher Contributors <https://github.com/Slowpoke101/FTBLaunch/>
 * FTB Launcher is licensed under the Apache 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.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.ftb.updater;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.net.URLDecoder;
import java.util.List;

import javax.swing.SwingWorker;

import org.w3c.dom.Document;
import org.w3c.dom.NamedNodeMap;

import net.feed_the_beast.launcher.json.JsonFactory;
import net.feed_the_beast.launcher.json.launcher.Channel;
import net.feed_the_beast.launcher.json.launcher.Update;
import net.ftb.data.Settings;
import net.ftb.download.Locations;
import net.ftb.gui.LaunchFrame;
import net.ftb.log.Logger;
import net.ftb.util.AppUtils;
import net.ftb.util.DownloadUtils;
import net.ftb.util.FTBFileUtils;
import net.ftb.util.OSUtils;

public class UpdateChecker extends SwingWorker<Boolean, Void>
{
	private int version;
	private int buildJenk;
	private int beta;// beta release target
	private int betaJenk;// beta CI build #
	private int relJenk;// release CI build #

	private int latest;
	private int minUsable;// TODO fix this!!
	public static String verString = "";
	public static String betaStr = "";
	private String downloadAddress = "";
	private String betaAddress = "";
	private boolean allowBeta;
	private boolean useBeta;
	private List<String> betaHash;
	public static String UCString;

	public UpdateChecker (int version, int minUsable, int buildJenk)
	{
		this.version = version;
		this.minUsable = minUsable;
		this.allowBeta = Settings.getSettings().isBetaChannel();
		this.buildJenk = buildJenk;
		if (buildJenk == 9999999)
		{
			this.allowBeta = false;
		}
		String path = null;
		try
		{
			path = new File(LaunchFrame.class.getProtectionDomain().getCodeSource().getLocation().getPath()).getCanonicalPath();
			path = URLDecoder.decode(path, "UTF-8");
			Logger.logDebug("Launcher Install path: " + path);// we need this to make sure that the app behaves correctly when updating
		}
		catch (IOException e)
		{
			Logger.logError("Couldn't get path to current launcher jar/exe", e);
		}

	}

	@Override
	protected Boolean doInBackground ()
	{
		loadInfo();
		try
		{
			FTBFileUtils.delete(new File(OSUtils.getCacheStorageLocation(), "updatetemp"));
		}
		catch (Exception ignored)
		{}
		return this.shouldUpdate();
	}

	private void loadInfo ()
	{
		try
		{
			Document doc = AppUtils.downloadXML(new URL(DownloadUtils.getStaticCreeperhostLink("newversion.xml")));
			Update upd = JsonFactory.getUpdate("net.ftb:launcher:beta@json", Locations.FTBMAVENFULL);
			if (upd.getPrimary().equals("beta"))
			{
				Channel beta = upd.getBeta();
				this.betaJenk = beta.getJenkins();
				int beta_ = this.beta = beta.getVersion();
				betaStr += beta_ / (100 * 100) + ".";
				beta_ = beta_ % (100 * 100);
				betaStr += beta_ / (100) + ".";
				beta_ = beta_ % 100;
				betaStr += beta_;
				betaAddress = beta.getFile().getUrl() + beta.getFile().getPath();
				if (beta.getFile().checksums != null)
				{
					betaHash = beta.getFile().checksums;
				}
				if (upd.getRelease() != null)
				{
					// TODO add code here to handle if the releases are in maven!!
				}
			}

			if (doc == null)
			{
				return;
			}
			NamedNodeMap updateAttributes = doc.getDocumentElement().getAttributes();
			int latest_ = latest = Integer.parseInt(updateAttributes.getNamedItem("currentBuild").getTextContent());
			verString += latest_ / (100 * 100) + ".";
			latest_ = latest_ % (100 * 100);
			verString += latest_ / (100) + ".";
			latest_ = latest_ % 100;
			verString += latest_;
			downloadAddress = updateAttributes.getNamedItem("downloadURL").getTextContent();
			if (updateAttributes.getNamedItem("releaseJenkins") != null)
			{
				relJenk = Integer.parseInt(updateAttributes.getNamedItem("releaseJenkins").getTextContent());
			}
			else
			{
				Logger.logInfo("Beta channel hasn't been activated yet!");
			}
		}
		catch (Exception e)
		{
			Logger.logError("Error while loading launcher update info", e);
		}
	}

	public boolean shouldUpdate ()
	{
		Logger.logDebug("updater: buildjenk " + buildJenk + " < betajenk " + betaJenk + "|| version " + version + " < " + beta);
		Logger.logDebug("latest = " + latest);
		if (allowBeta && (buildJenk < betaJenk || version < beta))
		{
			Logger.logInfo("New beta version found. version: " + version + "-" + buildJenk + ", latest: " + beta + "-" + betaJenk);
			UCString = "BETA version " + betaStr + "-" + betaJenk;
			useBeta = true;
			return true;
		}
		else if (version == latest && buildJenk < relJenk)
		{
			Logger.logInfo("Release version found. version: " + version + "-" + buildJenk + ", latest: " + latest);
			useBeta = false;
			UCString = "Version " + verString;
			return true;
		}
		else if (version < latest)
		{
			Logger.logInfo("New version found. version: " + version + ", latest: " + latest);
			useBeta = false;
			UCString = "Version " + verString;
			return true;
		}
		else
		{
			return false;
		}
	}

	public void update ()
	{
		String path = null;
		try
		{
			path = new File(LaunchFrame.class.getProtectionDomain().getCodeSource().getLocation().getPath()).getCanonicalPath();
			path = URLDecoder.decode(path, "UTF-8");
			Logger.logDebug("Launcher Install path: " + path);// we need this to make sure that the app behaves correctly when updating
		}
		catch (IOException e)
		{
			Logger.logError("Couldn't get path to current launcher jar/exe", e);
		}
		String temporaryUpdatePath = OSUtils.getCacheStorageLocation() + File.separator + "updatetemp" + "/" + path.substring(path.lastIndexOf(File.separator) + 1);
		String extension = path.substring(path.lastIndexOf('.') + 1);
		extension = "exe".equalsIgnoreCase(extension) ? extension : "jar";
		try
		{
			URL updateURL = new URL(!useBeta ? DownloadUtils.getCreeperhostLink(downloadAddress + "." + extension) : betaAddress.replace("${ext}", extension).replace("${jenkins}", Integer.toString(betaJenk)).replace("${version}", betaStr));
			File temporaryUpdate = new File(temporaryUpdatePath);
			temporaryUpdate.getParentFile().mkdir();
			DownloadUtils.downloadToFile(updateURL, temporaryUpdate);// TODO hash check this !!!!
			if (useBeta && betaHash != null)
			{
				String sha = DownloadUtils.fileSHA(temporaryUpdate);
				if (betaHash.contains(sha))
				{
					SelfUpdate.runUpdate(path, temporaryUpdate.getCanonicalPath());
				}
				else
				{
					Logger.logDebug("TempPath" + temporaryUpdatePath);
					throw new IOException("Update Download failed hash check please try again! -- fileSha " + sha);
				}
			}
			else
			{
				SelfUpdate.runUpdate(path, temporaryUpdate.getCanonicalPath());
			}
		}
		catch (Exception e)
		{
			Logger.logError("Error while updating launcher", e);
		}
	}
}