import socket
import json

def sendAs(username):
    # Connect to the bester daemon
    d=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    d.connect(("127.0.0.1",2223))

    # First do it and authenticate
    bys=json.dumps({"header":{"authentication":{"username":username, "password":"passwd"}, "scope":"client"},"payload":{"data":{"bruhMsg":input("Enter message naaier: ")},"type":"type2"}})
    print(len(bys), bys)
    d.send(len(bys).to_bytes(4, "little"))
    d.send(bys.encode())

    # Loop prompt and sending of message to tbk
    while True:
        bys=json.dumps({"header":{"authentication":{"username":"ddd", "password":"passwd"}, "scope":"client"},"payload":{"data":{"bruhMsg":input("Enter message naaier: ")},"type":"type2"}})
        print(len(bys), bys)
        d.send(len(bys).to_bytes(4, "little"))
        d.send(bys.encode())



    # Now we can do it again (without authentication)
    #bys=json.dumps({"header":{},"payload":{"data":"POES","type":"type1"}})
    #print(len(bys), bys)
    #d.send(len(bys).to_bytes(4, "little"))
    #d.send(bys.encode())

    while True: pass
    d.close()

def commandBuiltInClose():
    d=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    d.connect(("127.0.0.1",2221))
    bys=json.dumps({"header": {"scope" : "client"},"payload": {
    "data": {
        "command" : {"type" : "close", "args" : None}
    },"type":"builtin"}})
    print(bys)
    d.send(len(bys).to_bytes(4, "little"))
    d.send(bys.encode())
    d.close()

def commandAuthTest():
    d=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    d.connect(("127.0.0.1",2221))
    bys=json.dumps({"header": {"scope" : "client"},"payload": {
    "data": {
        "command" : {"type" : "login", "args" : {"username" :"1", "password":"2"}}
    },"type":"builtin"}})
    print(bys)
    d.send(len(bys).to_bytes(4, "little"))
    d.send(bys.encode())
    d.close()



def simpleTest():
    d=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    d.connect(("127.0.0.1",2224))
    bys=json.dumps({"name":1})
    print(bys)
    d.send(len(bys).to_bytes(4, "little"))
    d.send(bys.encode())
    while True: pass
    d.close()


def runTests():
    # Get the username to send as
    sendAs(input("Enter username to send as: "))
    #simpleTest()
    #commandBuiltInClose()
    #commandAuthTest()
    #basicTest()

runTests()