#!/usr/bin/env python

import socket

class NConnect:

    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        try:
            self.sock.connect((host, port))
            print("[*] Connected")
        except socket.error:
            print("[!] Unable to connect")


    def read(self):
        msg = self.sock.recv(1024).decode('utf-8').strip()
        print('[>] {}'.format(msg))
        return msg


    def read_until(self, word):
        msg = ''
        while (word not in msg):
            msg += self.sock.recv(1024).decode('utf-8')

        msg = msg.strip()
        print('[>] {}'.format(msg))
        return msg


    def send(self, msg):
        print('[<] {}'.format(msg))
        self.sock.send(str(msg)+"\n")


    def close(self):
        print("[!] Closing connection")
        self.sock.close()


host = "a.b.c.d"
port = 8080

nc = NConnect(host, port)

try:
    while True:
        message = nc.read_until("//")
        print(message)

except KeyboardInterrupt:
    print("[!] ^C Received, closing connection")
    nc.close()

except EOFError:
    print("[!] ^D Received, closing connection")
    nc.close()