From 9aa7a8b859bfc7622558b8424d3259c43757c2ff Mon Sep 17 00:00:00 2001
From: a <a@tuxpa.in>
Date: Mon, 26 Jun 2023 04:04:30 -0500
Subject: [PATCH] cleaner

---
 cloudflare/socket.go | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/cloudflare/socket.go b/cloudflare/socket.go
index 70876de..425c1d5 100644
--- a/cloudflare/socket.go
+++ b/cloudflare/socket.go
@@ -50,16 +50,16 @@ func (d *Dialer) Dial(ctx context.Context, network, addr string) (net.Conn, erro
 	sock := &TCPSocket{}
 	sock.socket = d.connect.Invoke(addr, optionsObj)
 	sock.options = d.opts
-	sock.ctx, sock.cn = context.WithCancel(ctx)
-	sock.SetDeadline(time.Now().Add(999999 * time.Hour))
 	sock.init(d.ctx)
 	return sock, nil
 }
 
 func (sock *TCPSocket) init(ctx context.Context) {
+	sock.SetDeadline(time.Now().Add(999999 * time.Hour))
 	sock.writer = sock.socket.Get("writable").Call("getWriter")
 	sock.reader = sock.socket.Get("readable").Call("getReader")
 	sock.rd = jsutil.ConvertReadableStreamToReader(sock.reader)
+	sock.ctx, sock.cn = context.WithCancel(ctx)
 	return
 }
 
@@ -128,9 +128,11 @@ func (t *TCPSocket) Write(b []byte) (n int, err error) {
 
 // StartTls will call startTls on the socket
 func (t *TCPSocket) StartTls() *TCPSocket {
-	t.socket = t.socket.Call("startTls")
-	t.init(t.ctx)
-	return t
+	sock := &TCPSocket{}
+	sock.socket = t.socket.Call("startTls")
+	sock.options = t.options
+	sock.init(t.ctx)
+	return sock
 }
 
 // Close closes the connection.