mkxp-sunshine/journal/unix/journal.py

# -*- coding: utf-8 -*-

import os, sys, time

from PyQt5.QtCore import Qt, QEvent, QThread, pyqtSignal, QRect, QRectF, QTimer, QPoint
from PyQt5.QtWidgets import QApplication, QWidget, QDesktopWidget, QLabel
from PyQt5.QtGui import QIcon, QPixmap, QPainter

if sys.platform == "win32":
	pipe_path = '\\\\.\\pipe\\oneshot-journal-to-game'
	import ctypes.wintypes
	buff = ctypes.create_unicode_buffer(ctypes.wintypes.MAX_PATH)
	ctypes.windll.shell32.SHGetFolderPathW(None, CSIDL_PERSONAL, None, SHGFP_TYPE_CURRENT, buf)
	documents_path = os.path.join(buf.value, 'My Games')
else:
	pipe_path = '/tmp/oneshot-pipe'
	documents_path = os.path.expanduser('~/Documents')

left_close = False
if sys.platform == "darwin": left_close = True

try: base_path = sys._MEIPASS
except AttributeError: base_path = os.path.abspath('.')

class WatchPipe(QThread):
	change_image = pyqtSignal(str)

	def run(self):
		while True:
			while True:
				if os.path.exists(pipe_path): break
				else: time.sleep(0.1)
			pipe = open(pipe_path, 'r')
			pipe.flush()

			while True:
				if not os.path.exists(pipe_path):
					break # Handle game quitting cleanup
				message = os.read(pipe.fileno(), 256)
				if len(message) > 0:
					self.change_image.emit(message.decode())

				time.sleep(0.05)

class AnimationTimer(QThread):
	next_frame = pyqtSignal()

	def run(self):
		while True:
			self.next_frame.emit()
			time.sleep(1.0 / 60)

class Journal(QWidget):
	def __init__(self, *args, **kwargs):
		self.app = kwargs['app']
		del kwargs['app']

		super().__init__(*args, **kwargs)

		self.mousedown = False
		self.mousedownpos = QPoint(0, 0)

		self.label = QLabel(self)
		self.change_image('default_en')

		self.close_label = QLabel(self)
		self.close_label.setPixmap(QPixmap(os.path.join(base_path, 'images', 'close.png')))
		if not left_close: self.close_label.move(776, 0) # X = 800-24

		if "linux" in sys.platform: self.setWindowFlags(Qt.FramelessWindowHint)
		else: self.setWindowFlags(self.windowFlags() | Qt.FramelessWindowHint | Qt.NoDropShadowWindowHint)
		self.setAttribute(Qt.WA_TranslucentBackground)
		self.setMouseTracking(True)
		self.setWindowTitle(' ')
		self.setMinimumSize(800, 600)
		self.setMaximumSize(800, 600)
		self.setGeometry(0, 0, 800, 600)

		self.show()

	def mousePressEvent(self, event):
		self.mousedown = True
		self.mousedownpos = event.pos()

		if ((left_close and self.mousedownpos.x() <= 24) or (not left_close and self.mousedownpos.x() >= 776)) and self.mousedownpos.y() < 24:
			self.app.quit()

	def mouseReleaseEvent(self, event):
		self.mousedown = False

	def mouseMoveEvent(self, event):
		if event.buttons() == Qt.LeftButton:
			pos = event.pos()
			frameGm = self.frameGeometry()
			self.setGeometry(frameGm.x() + pos.x() - self.mousedownpos.x(), frameGm.y() + pos.y() - self.mousedownpos.y(), 800, 600)

	def change_image(self, image):
		name, lang = image.split('_', 1)
		if lang == 'en':
			img = os.path.join(base_path, 'images', '{}.png'.format(name))
		else:
			img = os.path.join(base_path, 'images', lang.upper(), '{}.png'.format(name))
		if not os.path.exists(img): return
		self.pixmap = QPixmap(img)
		self.label.setPixmap(self.pixmap)

class Niko(QWidget):
	def __init__(self, *args, **kwargs):
		self.x, self.y, self.start_y, self.app, self.thread = kwargs['x'], kwargs['y'], kwargs['y'], kwargs['app'], kwargs['thread']
		self.screen_height = kwargs['screen_height']
		del kwargs['x'], kwargs['y'], kwargs['screen_height'], kwargs['app'], kwargs['thread']

		super().__init__(*args, **kwargs)

		self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint)
		self.setAttribute(Qt.WA_TranslucentBackground)
		self.setMinimumSize(48, 64)
		self.setMaximumSize(48, 64)

		self.label = QLabel(self)
		self.frames = [QPixmap(os.path.join(base_path, 'images', 'niko{}.png'.format(n))) for n in range(1,4)]
		self.label.setPixmap(self.frames[1])

		self.show()

		self.thread.start()

	def getFrame(self):
		if ((self.y - self.start_y) % 32 >= 16): return 1
		if ((self.y - self.start_y) % 64 >= 32): return 0
		else: return 2

	def update(self):
		self.label.setPixmap(self.frames[self.getFrame()])
		self.y += 2
		if self.y > self.screen_height:
			self.app.quit()
			return
		elif self.y > self.screen_height - 64:
			self.setMinimumSize(48, self.screen_height - self.y)
			self.resize(48, self.screen_height - self.y)
		self.move(self.x, self.y)

if __name__ == '__main__':
	app = QApplication(sys.argv)

	if len(sys.argv) == 3:
		# "Niko-leaves-the-screen" mode
		x, y = int(sys.argv[1]), int(sys.argv[2])

		thread = AnimationTimer()

		niko = Niko(x = x, y = y, screen_height = app.primaryScreen().size().height(), app = app, thread = thread)

		thread.next_frame.connect(niko.update)

	else:
		# Author's Journal mode
		journal = Journal(app = app)
		save_path = os.path.join(documents_path, 'Oneshot', 'save_progress.oneshot')
		if os.path.exists(save_path):
			with open(save_path, 'rb') as save:
				save.seek(-8, os.SEEK_END)
				lang = save.read().decode('utf-8')
				lang = lang[lang.find('[') + 1:lang.find(']')]
				if lang == 'en_US': lang = 'en'
				journal.change_image('save_' + lang)
		else:
			thread = WatchPipe()
			thread.change_image.connect(journal.change_image)
			thread.start()

	pipe_file = open(pipe_path, "w+")
	pipe_file.close()

	app.exec_()

	try:
		os.remove(pipe_path)
	except:
		# Most likely due to the file being in use.  Ignore.
		pass