Lindenii Project Forge
Login

go-lindenii-irc

IRC library for the Lindenii Project
Commit info
ID
89db1e89f130e29cfb1b5d8b22821b55621d56b0
Author
Runxi Yu <me@runxiyu.org>
Author date
Sat, 22 Mar 2025 10:57:30 +0800
Committer
Runxi Yu <me@runxiyu.org>
Committer date
Sat, 22 Mar 2025 10:57:30 +0800
Actions
ReadMessage should return the raw line too
package irc

import (
	"bufio"
	"net"
	"slices"
)

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, err error) {
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] == '\r' {
		raw = raw[:len(raw) - 1]
	}

	msg, err = Parse(slices.Clone(raw))
	lineBytes := slices.Clone(raw)
	line = 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(stringToBytes(s))
}