AO2XP/AO2XP.py

import subprocess, sys, thread, time, os, platform, __builtin__
from os.path import exists, abspath

from PyQt4 import QtGui, QtCore

import audio as AUDIO
import ini

__builtin__.audio = AUDIO
del AUDIO

app = QtGui.QApplication(sys.argv)
osname = platform.system()

path = sys.argv[0]
if osname == "Darwin" and ".app" in path: # bundle
    path = "/".join(path.split("/")[:-1]) # to Resources folder
    os.chdir(path)

debugmode = len(sys.argv) > 1 and sys.argv[1] == "debug"
if not debugmode:
    fakebass = len(sys.argv) > 1 and sys.argv[1] == "ignorebass"
    missing = audio.checkAvailable()
    if missing or fakebass:
        QtGui.QMessageBox.critical(None, "Unable to launch game", "Couldn't find the file %s on the client folder.\nAO2XP needs this file in order to play sounds and music.\nThe file is included in the client's zip file, make sure it's in the same folder as the AO2XP exe.\n\nAdditional info:\n%s\n%s" % (missing, sys.argv, os.getcwd()))
        os._exit(-2)

class gamewindow(QtGui.QMainWindow):
	def __init__(self):
		super(gamewindow, self).__init__()
		self.stackwidget = QtGui.QStackedWidget(self)
		self.widget = mainmenu.lobby(self)
		self.gamewidget = gameview.gui(self)
		self.stackwidget.addWidget(self.widget)
		self.stackwidget.addWidget(self.gamewidget)
		self.setCentralWidget(self.stackwidget)
		self.stackwidget.setCurrentWidget(self.widget)
		self.setFixedSize(self.widget.lobbyimg.size().width(), self.widget.lobbyimg.size().height())
		self.center()
		self.setWindowTitle("AO2XP")
		self.setWindowFlags(QtCore.Qt.WindowMinimizeButtonHint)
		
		self.settingsgui = options.Settings()
	
	def center(self):
		frameGm = self.frameGeometry()
		centerPoint = QtGui.QDesktopWidget().availableGeometry().center()
		frameGm.moveCenter(centerPoint)
		self.move(frameGm.topLeft())
	
	def showGame(self, tcp, charlist, musiclist, background, evidence, areas, features=[], oocjoin=[], hplist=[], webAO_bucket=""):
		self.gamewidget.disconnectnow = False
		self.gamewidget.startGame(tcp, charlist, musiclist, background, evidence, areas, features, oocjoin, hplist, webAO_bucket)
		self.stackwidget.setCurrentWidget(self.gamewidget)
	
	def returnToMenu(self):
		self.gamewidget.disconnectnow = True
		self.setFixedSize(self.widget.lobbyimg.size().width(), self.widget.lobbyimg.size().height())
		self.widget.onClicked_cancelconnect()
		self.stackwidget.setCurrentWidget(self.widget)
		self.setWindowTitle("AO2XP")
	
	def showSettings(self):
		self.settingsgui.showSettings()

if not debugmode:
    # Vanilla downloader
    force_downloader = len(sys.argv) > 1 and sys.argv[1] == "download"
    if force_downloader or (not exists("base/background") and not exists("base/characters") and not exists("base/sounds") and not exists("base/evidence")):
        jm = QtGui.QMessageBox.information(None, "Warning", "You seem to be missing the included Attorney Online content.\nWould you like to download them automatically?", QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
        if jm == QtGui.QMessageBox.Yes:
            import basedownloader
            code = basedownloader.downloadVanilla()
            if code != 0: os._exit(code)
        else:
            os._exit(-3)

    # AO2XP update checker
    can_update = ini.read_ini_bool("AO2XP.ini", "General", "install updates", True)
    force_update = "forceupdate" in sys.argv[1:]
    if can_update or force_update:
        import updater
        code = updater.checkForUpdates(force_update)
        if code == 0:
            subprocess.Popen(["./AO2XPupdat"])
            os._exit(0)
        elif osname == "Darwin" and ".app" in sys.argv[0]: # bundle
            os.chdir(path) # return to Resources folder


import gameview, mainmenu, options

audio.init()
shit = gamewindow()
shit.show()
returnc = app.exec_()
audio.free()
sys.exit(returnc)