planet/planet/main.py

#!/usr/bin/python3
"""


Copyright (C) 2022  Alexey Pavlov
Copyright (C) 2022  Red-Exe-Engineer

This program 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 2 of the License, or
(at your option) any later version.

This program 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 this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

"""


import sys
import os
import random
from datetime import date
import json
import pathlib

absolute_path = pathlib.Path(__file__).parent.absolute()

if str(absolute_path).startswith("/usr/bin"):
    absolute_path = "/usr/lib/planet-launcher/"

sys.path.append(absolute_path)
if os.path.exists("/usr/lib/planet-launcher/"):
    sys.path.append("/usr/lib/planet-launcher/")

import launcher
from splashes import SPLASHES


from PyQt5.QtCore import *
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from PyQt5.QtWebKit import *
from PyQt5.QtWebKitWidgets import *

import qdarktheme
import pypresence

dark_stylesheet = qdarktheme.load_stylesheet()

USER = os.getenv("USER")

if not os.path.exists(f"/home/{USER}/.planet-launcher/mods"):
    os.makedirs(f"/home/{USER}/.planet-launcher/mods")

# if os.path.exists(f"/home/{USER}/.gmcpil.json"):
#    with open(f"/home/{USER}/.gmcpil.json") as f:
#        DEFAULT_FEATURES = json.loads(f.read())["features"]
# else:
# TODO: Add a tab with a button to import features from gMCPIL

class ConfigPluto(QDialog):
    def __init__(self):
        super().__init__()
        self.setWindowFlag(Qt.FramelessWindowHint)
        
        layout = QVBoxLayout()
        titlelayout = QGridLayout()
        
        logopixmap = QPixmap(f"{absolute_path}/assets/logo512.png").scaled(100, 100, Qt.KeepAspectRatio)
        namelabel = QLabel("Pluto Wizard")
        
        logolabel = QLabel()
        logolabel.setPixmap(logopixmap)
        logolabel.setAlignment(Qt.AlignRight)
        
        font = namelabel.font()
        font.setPointSize(30)
        namelabel.setFont(font)
        namelabel.setAlignment(Qt.AlignLeft)
        
        titlelayout.addWidget(logolabel, 0, 0)
        titlelayout.addWidget(namelabel, 0, 1)

        titlewidget = QWidget()
        titlewidget.setLayout(titlelayout)
        
        info_label = QLabel("Please select the executable you downloaded.\nIf you installed a DEB, please select the \"Link\" option")
        
        self.executable_btn = QPushButton("Select executable")
        self.executable_btn.clicked.connect(self.get_appimage)
        
        self.premade_btn = QPushButton("Link /usr/bin/minecraft-pi-reborn-client")
        self.premade_btn.clicked.connect(self.link_appimage)
        
        layout.addWidget(titlewidget)
        layout.addWidget(info_label)
        layout.addWidget(self.executable_btn)
        layout.addWidget(self.premade_btn)
        
        self.setLayout(layout)
    
    def mousePressEvent(self, event):
        if event.button() == Qt.LeftButton:
            self.moveFlag = True
            self.movePosition = event.globalPos() - self.pos()
            self.setCursor(QCursor(Qt.OpenHandCursor))
            event.accept()
    
    def mouseMoveEvent(self, event):
        if Qt.LeftButton and self.moveFlag:
            self.move(event.globalPos() - self.movePosition)
            event.accept()
    
    def mouseReleaseEvent(self, event):
        self.moveFlag = False
        self.setCursor(Qt.ArrowCursor)
        
    def get_appimage(self):
        self.hide()
        self.filename =  QFileDialog.getOpenFileName(self, 'Select executable',  '/',"Executable files (*.AppImage *.bin *.sh *)")
    def link_appimage(self):
        self.hide()
        os.symlink("/usr/bin/minecraft-pi-reborn-client", f"/home/{USER}/.planet-launcher/minecraft.AppImage")
        self.filename = list()
        self.filename.append(False)


class Planet(QMainWindow):
    
    launchfeatures = dict()
    env = os.environ.copy()

    def __init__(self):
        super().__init__()
        
        RPC = pypresence.Presence(787496148763541505)
        try: 
            RPC = pypresence.Presence(787496148763541505)
            RPC.connect()
            RPC.update(state="Launched with Planet Launcher",  details="Minecraft Pi Edition: Reborn",  large_image="logo",  small_image=random.choice(["heart",  "portal",  "multiplayer",  "pi"]))
        except:
            print("Unable to initalize Discord RPC. Skipping.")
        
        if not os.path.exists(f"/home/{USER}/.planet-launcher/config.json"):
            
            self.conf = {
                "username": "StevePi", 
                "options": launcher.get_features_dict(f"/home/{USER}/.planet-launcher/minecraft.AppImage"), 
                "hidelauncher": True, 
                "profile": "Modded MCPE", 
                "render_distance": "Short", 
                "theme": "QDarkTheme Light", 
                "discord_rpc": True, 
            }
            
            with open(f"/home/{USER}/.planet-launcher/config.json",  "w") as file:
                file.write(json.dumps(self.conf))
        else:
            with open(f"/home/{USER}/.planet-launcher/config.json") as file:
                self.conf = json.loads(file.read())
        

        self.setWindowTitle("Planet")

        self.setWindowIcon(QIcon(f"{absolute_path}/assets/logo512.png"))

        tabs = QTabWidget()
        tabs.setTabPosition(QTabWidget.West)
        tabs.setMovable(True)

        play_tab = tabs.addTab(self.play_tab(), "Play")
        tabs.setTabIcon(play_tab, QIcon(f"{absolute_path}/assets/logo512.png"))
        features_tab = tabs.addTab(self.features_tab(), "Features")
        tabs.setTabIcon(features_tab, QIcon(f"{absolute_path}/assets/heart512.png"))
        servers_tab = tabs.addTab(self.servers_tab(),  "Servers")
        tabs.setTabIcon(servers_tab, QIcon(f"{absolute_path}/assets/multiplayer512.png"))
        #mods_tab = tabs.addTab(self.custom_mods_tab(), "Mods")
        #tabs.setTabIcon(mods_tab, QIcon(f"{absolute_path}/assets/portal512.png"))
        changelog_tab = tabs.addTab(self.changelog_tab(), "Changelog")
        tabs.setTabIcon(changelog_tab, QIcon(f"{absolute_path}/assets/pi512.png"))

        self.setCentralWidget(tabs)

        self.setGeometry(600, 900, 200, 200)
        
        self.usernameedit.setText(self.conf["username"])
        self.profilebox.setCurrentText(self.conf["profile"])
        self.distancebox.setCurrentText(self.conf["render_distance"])
        
        for feature in self.features:
            if self.conf["options"][feature]:
                self.features[feature].setCheckState(Qt.Checked)
            else:
                self.features[feature].setCheckState(Qt.Unchecked)
        
        self.showlauncher.setChecked(self.conf["hidelauncher"])

        self.set_features()

    def play_tab(self) -> QWidget:
        layout = QGridLayout()

        titlelayout = QGridLayout()

        logopixmap = QPixmap(f"{absolute_path}/assets/logo512.png").scaled(100, 100, Qt.KeepAspectRatio)

        logolabel = QLabel()
        logolabel.setPixmap(logopixmap)
        logolabel.setAlignment(Qt.AlignRight)

        namelabel = QLabel()

        if date.today().month == 4 and date.today().day == 1:
            namelabel.setText("Banana Launcher")
        else:
            if random.randint(1, 100) == 1:
                namelabel.setText("Pluto Launcher")
            else:
                namelabel.setText("Planet Launcher")

        font = namelabel.font()
        font.setPointSize(30)
        namelabel.setFont(font)
        namelabel.setAlignment(Qt.AlignLeft)

        splashlabel = QLabel(f'<font color="gold">{random.choice(SPLASHES)}</font>')
        splashlabel.adjustSize()
        splashlabel.setAlignment(Qt.AlignHCenter)

        usernamelabel = QLabel("Username")

        self.usernameedit = QLineEdit()
        self.usernameedit.setPlaceholderText("StevePi")

        distancelabel = QLabel("Render Distance")

        self.distancebox = QComboBox()
        self.distancebox.addItems(["Far", "Normal", "Short", "Tiny"])
        self.distancebox.setCurrentText("Short")

        profilelabel = QLabel("Profile")

        self.profilebox = QComboBox()
        self.profilebox.addItems(
            ["Vanilla MCPi", "Modded MCPi", "Modded MCPE", "Optimized MCPE", "Custom"]
        )
        self.profilebox.setCurrentText("Modded MCPE")

        self.showlauncher = QRadioButton("Hide Launcher")

        self.playbutton = QPushButton("Play")

        self.playbutton.setCheckable(True)
        self.playbutton.clicked.connect(self.launch)

        titlelayout.addWidget(logolabel, 0, 0)
        titlelayout.addWidget(namelabel, 0, 1)

        titlewidget = QWidget()
        titlewidget.setLayout(titlelayout)

        layout.addWidget(titlewidget, 0, 0, 2, 5)

        layout.addWidget(splashlabel, 2, 0, 1, 6)

        layout.addWidget(usernamelabel, 3, 0)
        layout.addWidget(self.usernameedit, 3, 4, 1, 2)

        layout.addWidget(distancelabel, 4, 0)
        layout.addWidget(self.distancebox, 4, 4, 1, 2)

        layout.addWidget(profilelabel, 5, 0)
        layout.addWidget(self.profilebox, 5, 4, 1, 2)

        layout.addWidget(self.showlauncher, 6, 4, 1, 2)

        layout.addWidget(self.playbutton, 8, 5)

        widget = QWidget()

        widget.setLayout(layout)

        return widget

    def features_tab(self) -> QWidget:

        layout = QVBoxLayout()

        self.features = dict()
        default_features = launcher.get_features_dict(f"/home/{USER}/.planet-launcher/minecraft.AppImage")

        for feature in default_features:
            checkbox = QCheckBox(feature)
            if default_features[feature]:
                checkbox.setCheckState(Qt.Checked)
            else:
                checkbox.setCheckState(Qt.Unchecked)

            checkbox.clicked.connect(self.set_features)

            self.features[feature] = checkbox

            layout.addWidget(checkbox)

        fakewidget = QWidget()
        fakewidget.setLayout(layout)

        scroll = QScrollArea()

        scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
        scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        scroll.setWidgetResizable(True)
        scroll.setWidget(fakewidget)

        fakelayout = QGridLayout()
        fakelayout.addWidget(scroll, 0, 0)

        widget = QWidget()

        widget.setLayout(fakelayout)

        return widget
        
    def servers_tab(self) -> QWidget:
        widget = QWidget()
        layout = QGridLayout()

        self.serversedit = QTextEdit()

        self.serversedit.textChanged.connect(self.save_servers)
        with open(f"/home/{USER}/.minecraft-pi/servers.txt") as servers:
            self.serversedit.setPlainText(servers.read())

        infolabel = QLabel("Servers are stored in the format of <font color=\"gold\">IP: </font><font color=\"blue\">Port</font>")

        layout.addWidget(self.serversedit, 0, 0)
        layout.addWidget(infolabel,  6, 0)

        widget.setLayout(layout)
        return widget

    def custom_mods_tab(self) -> QWidget:
        layout = QVBoxLayout()

        for mod in os.listdir(f"/home/{USER}/.planet-launcher/mods/"):
            checkbox = QCheckBox(mod)
            checkbox.setCheckState(Qt.Unchecked)

            layout.addWidget(checkbox)

        fakewidget = QWidget()
        fakewidget.setLayout(layout)

        scroll = QScrollArea()

        scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
        scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        scroll.setWidgetResizable(True)
        scroll.setWidget(fakewidget)

        fakelayout = QGridLayout()
        fakelayout.addWidget(scroll, 0, 0)

        widget = QWidget()

        widget.setLayout(fakelayout)

        return widget

    def changelog_tab(self):
        web = QWebView()
        web.load(
            QUrl().fromLocalFile(
                f"{absolute_path}/assets/changelog.html"
            )
        )

        return web

    def set_features(self):
        for feature in self.features:
            if self.features[feature].isChecked():
                self.launchfeatures[feature] = True
            else:
                self.launchfeatures[feature] = False
                
    def save_profile(self):
        self.conf["username"] = self.usernameedit.text()
        self.conf["options"] = self.launchfeatures
        self.conf["render_distance"] = self.distancebox.currentText()
        self.conf["hidelauncher"] = self.showlauncher.isChecked()
        
        with open(f"/home/{USER}/.planet-launcher/config.json",  "w") as file:
                file.write(json.dumps(self.conf))
    
    def save_servers(self):
        with open(f"/home/{USER}/.minecraft-pi/servers.txt",  "w") as file:
            file.write(self.serversedit.toPlainText())

    def launch(self):
        self.save_profile()
        self.env = launcher.set_username(self.env, self.usernameedit.text())
        self.env = launcher.set_options(self.env, self.launchfeatures)
        self.env = launcher.set_render_distance(
            self.env, self.distancebox.currentText()
        )

        print(self.env)
        if self.showlauncher.isChecked() == True:
            self.hide()
        launcher.run(self.env,  f"/home/{USER}/.planet-launcher/minecraft.AppImage")
        self.show()


if __name__ == "__main__":
    
    apppath = str()
    
    app = QApplication(sys.argv)
    app.setPalette(qdarktheme.load_palette("dark"))
    

    
    if not os.path.exists(f"/home/{USER}/.planet-launcher/minecraft.AppImage"):
        pluto = ConfigPluto()
        pluto.show()
        pluto.exec()
        if pluto.filename[0] == '':
            sys.exit(-1)
        elif pluto.filename[0] == False:
            print("Using /usr/bin as an executable.")
        else:
            with open(pluto.filename[0],  "rb") as appimage:
                with open(f"/home/{USER}/.planet-launcher/minecraft.AppImage",  "wb") as out:
                    out.write(appimage.read())
                    os.chmod(f"/home/{USER}/.planet-launcher/minecraft.AppImage",   0o777)
    
    
    window = Planet()
    window.show()
    
   
    app.exec()