import socket
import json

# Server configuration
SERVER_HOST = '127.0.0.1'
SERVER_PORT = 65432

# Function to send DNS update requests
def update_dns_record(domain, record_type, record_data):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
        client_socket.connect((SERVER_HOST, SERVER_PORT))
        request_data = {
            "action": "update",
            "domain": domain,
            "type": record_type,
            "data": record_data
        }
        client_socket.sendall(json.dumps(request_data).encode())
        response = client_socket.recv(1024)
        return response.decode()

# Main client function
def run_update_client():
    domain = input("Enter domain name: ")
    record_type = input("Enter record type (A/AAAA/MX/NS/SRV/TXT/CNAME): ")
    record_data = input("Enter record data: ")
    response = update_dns_record(domain, record_type, record_data)
    print("Response from server:", response)

if __name__ == "__main__":
    run_update_client()
