#Pastebin Client
import socket
import sys
sys.path.append("..")
from ColorText import ColorText

class Client(object):
    def __init__(self):
        self.userId = None
        self.s = socket.socket()
        self.text = ""
        self.fileName = "DefaultFileName"
        #Not giving user this option currently
        self.expiryDate = '1 minute'
        self.folderName = "DefaultFolderName"
        self.sizeLimitInBytes = 1024
        self.port = 8

    def displayPage(self):
        ct = ColorText()
        ct.display(4*"\t"+"Pastebin", "red-highlight")

    def getStringSizeInBytes(self, string):
        return len(string.encode('utf-8'))

    def checkSize(self, string):
        stringSize = self.getStringSizeInBytes(string)
        if(int(stringSize) > self.sizeLimitInBytes):
            return False
        return True

    def getText(self):
        self.text = raw_input("Enter text (limit 1024 bytes): \n")
        while(not self.checkSize(self.text)):
            print "Text size too large"
            self.text = raw_input("Enter text (limit 1024 bytes): \n")
            
    def getFileName(self):
        choice = raw_input("Would you like to enter FILE name (y/n): ")
        if(choice.lower() != 'y'):
            return
        self.fileName = raw_input("Enter file name: ")
        while(not self.checkSize(self.fileName)):
            print "Text size too large"
            self.fileName = raw_input("Enter file name: ")

    def getFolderName(self):
        choice = raw_input("Would you like to enter FOLDER name (y/n): ")
        if(choice.lower() != 'y'):
            return
        self.folderName = raw_input("Enter folder name: ")
        while(not self.checkSize(self.folderName)):
            print "Text size too large"
            self.folderName = raw_input("Enter folder name: ")

    def getUserId(self):
        self.userId = raw_input("Enter User Id: ")

    def run(self):
        self.displayPage()
        self.getUserId()
        self.getText()
        self.getFileName()
        self.getFolderName()
        #To connect to port of server
        self.s.connect(('127.0.0.1', self.port))
        self.s.send(self.userId)
        print self.s.recv(self.sizeLimitInBytes)
        self.s.send(self.text)
        print self.s.recv(self.sizeLimitInBytes)
        self.s.send(self.fileName)
        print self.s.recv(self.sizeLimitInBytes)
        self.s.send(self.folderName)
        print self.s.recv(self.sizeLimitInBytes)
        self.s.close()

#Main
obj = Client()
obj.run()