#!/usr/bin/env python3 
from tkinter import *
from socket import *
from struct import *
from sys import argv, exit

class LurkReader(Frame):
    def __init__(self, master):
        Frame.__init__(self, master)
        self.pack(fill="both", expand=1)
        self.connect_box = Frame(self)
        self.connect_box.pack(fill="y", expand=1)
        self.hostname_textbox = Text(self.connect_box, height=1, width=50)
        self.hostname_textbox.grid(row=0, column=0)
        self.port_textbox = Text(self.connect_box, height=1, width=7)
        self.port_textbox.grid(row=0, column=1)
        connect_button = Button(self.connect_box, text="Connect to Server", command=self.connect_to_server)
        connect_button.grid(row=0, column=2)
        self.server_info = Text(self)
        self.server_info.pack()

    def connect_to_server(self):
        # Get the hostname and the port from the GUI
        hostname = self.hostname_textbox.get(1.0, "end").strip()
        port = int(self.port_textbox.get(1.0, "end").strip())
        print((hostname, port))

        skt = socket(AF_INET, SOCK_STREAM)
        skt.connect((hostname, port))
        # It'd be a good idea to see if this worked!
        mtype = unpack_from("<B", skt.recv(1))[0]
        server_description = ""
        if(mtype == 14):
            major, minor, elen = unpack_from("<BBH", skt.recv(4))
            server_description += "Server version is:  " + str(major) + "." + str(minor) + "\n"
            mtype = unpack_from("<B", skt.recv(1))[0]
        
        if(mtype == 11):
            game_header = skt.recv(6)
            format_string = "<HHH"
            ip, sl, dl = unpack_from(format_string, game_header)
            description = skt.recv(dl)
            server_description += "Initial Points:  " + str(ip) + "\nStat Limit: " + str(sl) + "\nDescription: " + description.decode("utf-8")
            server_description += "\n"
        else:
            server_description += "Server presented type " + str(mtype) + "\n"
        
        self.server_info.insert("end", server_description)
        
        skt.close()


root = Tk()
# The next two lines set the window size to the screen size (maximize it)
w,h = root.winfo_screenwidth(), root.winfo_screenheight()
root.geometry("%dx%d+0+0" % (w-200,h-200))
root.title("Lurk Game Reader")
lr = LurkReader(root)
root.mainloop()