nntpchan/contrib/backends/srndv2/src/srnd/spam.go

package srnd

import (
	"errors"
	"io"
	"net"
)

var ErrSpamFilterNotEnabled = errors.New("spam filter access attempted when disabled")

type SpamFilter struct {
	addr    string
	enabled bool
}

func (sp *SpamFilter) Configure(c SpamConfig) {
	sp.enabled = c.enabled
	sp.addr = c.addr
}

func (sp *SpamFilter) Enabled() bool {
	return sp.enabled
}

func (sp *SpamFilter) Rewrite(msg io.Reader, out io.WriteCloser) error {
	var buff [65636]byte
	if !sp.Enabled() {
		return ErrSpamFilterNotEnabled
	}
	addr, err := net.ResolveTCPAddr("tcp", sp.addr)
	if err != nil {
		return err
	}
	c, err := net.DialTCP("tcp", nil, addr)
	if err != nil {
		return err
	}
	io.WriteString(c, "PROCESS SPAMC/1.5\r\n\r\n")
	io.CopyBuffer(c, msg, buff[:])
	c.CloseWrite()
	_, err = io.CopyBuffer(out, c, buff[:])
	c.Close()
	out.Close()
	return err
}