ergo/irc/net.go

package irc

import (
	"bufio"
	"io"
	"log"
	"net"
	"strings"
)

// Adapt `net.Conn` to a `chan string`.
func StringReadChan(conn net.Conn) <-chan string {
	ch := make(chan string)
	reader := bufio.NewReader(conn)
	go func() {
		defer close(ch)
		for {
			line, err := reader.ReadString('\n')
			if err != nil {
				if err != io.EOF {
					log.Printf("%s → %s error: %s", conn.RemoteAddr(), conn.LocalAddr(), err)
				}
				break
			}
			if DEBUG_NET {
				log.Printf("%s → %s %s", conn.RemoteAddr(), conn.LocalAddr(), line)
			}

			ch <- strings.TrimSpace(line)
		}
	}()
	return ch
}

func maybeLogWriteError(conn net.Conn, err error) bool {
	if err != nil {
		if err != io.EOF {
			log.Printf("%s ← %s error: %s", conn.RemoteAddr(), conn.LocalAddr(), err)
		}
		return true
	}
	return false
}

func StringWriteChan(conn net.Conn) chan<- string {
	ch := make(chan string)
	writer := bufio.NewWriter(conn)
	go func() {
		defer close(ch)
		for str := range ch {
			if DEBUG_NET {
				log.Printf("%s ← %s %s", conn.RemoteAddr(), conn.LocalAddr(), str)
			}
			if _, err := writer.WriteString(str); maybeLogWriteError(conn, err) {
				break
			}
			if _, err := writer.WriteString(CRLF); maybeLogWriteError(conn, err) {
				break
			}
			if err := writer.Flush(); maybeLogWriteError(conn, err) {
				break
			}
		}
	}()
	return ch
}

func LookupHostname(addr net.Addr) string {
	addrStr := addr.String()
	ipaddr, _, err := net.SplitHostPort(addrStr)
	if err != nil {
		return addrStr
	}
	names, err := net.LookupHost(ipaddr)
	if err != nil {
		return ipaddr
	}
	return names[0]
}