testRead.go

package main

import (
	"bytes"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"regexp"
	"strings"
	"sync/atomic"

	"github.com/Tensai75/nntp"
)

var (
	testReadCounter atomic.Int64
)

func (c *safeConn) testHead(messageId string) (*nntp.Article, error) {
	// for reading error testing
	/*
		counter := testReadCounter.Add(1)
		if counter%150 == 0 {
			return nil, fmt.Errorf("Test Error")
		}
	*/
	return readHeadFromFile(messageId)
}

func (c *safeConn) testBody(messageId string) (io.Reader, error) {
	// for reading error testing
	/*
		counter := testReadCounter.Add(1)
		if counter%150 == 0 {
			return nil, fmt.Errorf("Test Error")
		}
	*/
	return readBodyFromFile(messageId)
}

func (c *safeConn) testArticle(messageId string) (*nntp.Article, error) {
	// for reading error testing
	/*
		counter := testReadCounter.Add(1)
		if counter%150 == 0 {
			return nil, fmt.Errorf("Test Error")
		}
	*/
	return readArticleFromFile(messageId)
}

func readHead(conn *safeConn, messageId string) (*nntp.Article, error) {
	if conf.Test != "" {
		return conn.testHead(messageId)
	}
	return conn.Head(fmt.Sprintf("<%v>", messageId))
}

func readBody(conn *safeConn, messageId string) (io.Reader, error) {
	if conf.Test != "" {
		return conn.testBody(messageId)
	}
	return conn.Body(fmt.Sprintf("<%v>", messageId))
}

func readArticle(conn *safeConn, messageId string) (*nntp.Article, error) {
	if conf.Test != "" {
		return conn.testArticle(messageId)
	}
	return conn.Article(fmt.Sprintf("<%v>", messageId))
}

func readHeadFromFile(messageId string) (*nntp.Article, error) {

	article, err := readArticleFromFile(messageId)
	article.Body = nil
	return article, err

}

func readBodyFromFile(messageId string) (io.Reader, error) {

	article, err := readArticleFromFile(messageId)
	return article.Body, err

}

func readArticleFromFile(messageId string) (*nntp.Article, error) {

	var (
		readFile []byte
		article  nntp.Article
		body     bytes.Buffer
	)

	if readFile, err = os.ReadFile(filepath.Join(conf.Test, messageId+".txt")); err != nil {
		return nil, fmt.Errorf("Unable to load message <%v>", messageId)
	}

	expHeader := regexp.MustCompile(`(Subject|Date|From|Message-ID|Path|Newsgroups|X-Nxg): (.*)[ \n\r]`)
	headers := expHeader.FindAllStringSubmatch(string(readFile), -1)
	article.Header = make(map[string][]string)
	for _, header := range headers {
		headerName := strings.Trim(header[1], " \r\n")
		value := strings.Trim(header[2], " \r\n")
		article.Header[headerName] = append(article.Header[headerName], value)
	}

	expBody := regexp.MustCompile(`=ybegin[\w\W]+`)
	body.Write([]byte(expBody.FindString(string(readFile))))
	article.Body = &body

	return &article, nil

}