opsu-dance/src/yugecin/opsudance/OpsuDance.java

/*
 * opsu!dance - fork of opsu! with cursordance auto
 * Copyright (C) 2017 yugecin
 *
 * opsu!dance 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 3 of the License, or
 * (at your option) any later version.
 *
 * opsu!dance 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 opsu!dance.  If not, see <http://www.gnu.org/licenses/>.
 */
package yugecin.opsudance;

import itdelatrisu.opsu.Utils;
import itdelatrisu.opsu.beatmap.BeatmapWatchService;
import itdelatrisu.opsu.db.DBController;
import itdelatrisu.opsu.downloads.DownloadList;
import org.newdawn.slick.util.Log;

import java.io.IOException;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.UnknownHostException;

import static yugecin.opsudance.core.errorhandling.ErrorHandler.*;
import static yugecin.opsudance.core.Entrypoint.sout;
import static yugecin.opsudance.core.InstanceContainer.*;
import static yugecin.opsudance.options.Options.*;

/*
 * loosely based on itdelatrisu.opsu.Opsu
 */
public class OpsuDance {

	private ServerSocket singleInstanceSocket;

	public void start(String[] args) {
		try {
			sout("initialized");

			optionservice.loadOptions();
			ensureSingleInstance();
			sout("prechecks done and options parsed");

			initDatabase();
			initUpdater(args);
			sout("database & updater initialized");

			displayContainer.init(splashState);
		} catch (Exception e) {
			errorAndExit("startup failure", e);
		}

		while (rungame());
		displayContainer.teardownAL();

		optionservice.saveOptions();
		closeSingleInstanceSocket();
		DBController.closeConnections();
		DownloadList.get().cancelAllDownloads();
		Utils.deleteDirectory(config.TEMP_DIR);
		if (!OPTION_ENABLE_WATCH_SERVICE.state) {
			BeatmapWatchService.destroy();
		}
	}

	private boolean rungame() {
		try {
			displayContainer.setup();
			displayContainer.resume();
		} catch (Exception e) {
			explode("could not initialize GL", e, ALLOW_TERMINATE | PREVENT_CONTINUE);
			return false;
		}
		Exception caughtException = null;
		try {
			displayContainer.run();
		} catch (Exception e) {
			caughtException = e;
		}
		displayContainer.teardown();
		displayContainer.pause();
		return caughtException != null && explode("update/render error", caughtException, ALLOW_TERMINATE);
	}

	private void initDatabase() {
		try {
			DBController.init();
		} catch (UnsatisfiedLinkError e) {
			errorAndExit("Could not initialize database.", e);
		}
	}

	private void initUpdater(String[] args) {
		// check if just updated
		if (args.length >= 2) {
			updater.setUpdateInfo(args[0], args[1]);
		}

		// check for updates
		if (OPTION_DISABLE_UPDATER.state) {
			return;
		}
		new Thread() {
			@Override
			public void run() {
				try {
					updater.checkForUpdates();
				} catch (IOException e) {
					Log.warn("updatecheck failed.", e);
				}
			}
		}.start();
	}

	private void ensureSingleInstance() {
		if (OPTION_NOSINGLEINSTANCE.state) {
			return;
		}
		try {
			singleInstanceSocket = new ServerSocket(OPTION_PORT.val, 1, InetAddress.getLocalHost());
		} catch (UnknownHostException e) {
			// shouldn't happen
		} catch (IOException e) {
			errorAndExit(String.format(
					"Could not launch. Either opsu! is already running or a different program uses port %d.\n" +
					"You can change the port opsu! uses by editing the 'Port' field in the .opsu.cfg configuration file.\n" +
					"If that still does not resolve the problem, you can set 'NoSingleInstance' to 'true', but this is not recommended.", OPTION_PORT.val), e);
		}
	}

	private void closeSingleInstanceSocket() {
		if (singleInstanceSocket == null) {
			return;
		}
		try {
			singleInstanceSocket.close();
		} catch (IOException e) {
			Log.error("Single instance socket was not closed!", e);
		}
	}

	private void errorAndExit(String errstr, Throwable cause) {
		explode(errstr, cause, PREVENT_CONTINUE);
		System.exit(1);
	}

}