1 files changed, 49 insertions, 0 deletions
diff --git a/forged/internal/irc/conn.go b/forged/internal/irc/conn.go
new file mode 100644
index 0000000..b975b72
--- /dev/null
+++ b/forged/internal/irc/conn.go
@@ -0,0 +1,49 @@
+package irc
+
+import (
+	"bufio"
+	"net"
+	"slices"
+
+	"go.lindenii.runxiyu.org/forge/forged/internal/misc"
+)
+
+type Conn struct {
+	netConn   net.Conn
+	bufReader *bufio.Reader
+}
+
+func NewConn(netConn net.Conn) Conn {
+	return Conn{
+		netConn:   netConn,
+		bufReader: bufio.NewReader(netConn),
+	}
+}
+
+func (c *Conn) ReadMessage() (msg Message, line string, err error) {
+	raw, err := c.bufReader.ReadSlice('\n')
+	if err != nil {
+		return
+	}
+
+	if raw[len(raw)-1] == '\n' {
+		raw = raw[:len(raw)-1]
+	}
+	if raw[len(raw)-1] == '\r' {
+		raw = raw[:len(raw)-1]
+	}
+
+	lineBytes := slices.Clone(raw)
+	line = misc.BytesToString(lineBytes)
+	msg, err = Parse(lineBytes)
+
+	return
+}
+
+func (c *Conn) Write(p []byte) (n int, err error) {
+	return c.netConn.Write(p)
+}
+
+func (c *Conn) WriteString(s string) (n int, err error) {
+	return c.netConn.Write(misc.StringToBytes(s))
+}