cloudflare/pint

package parser

import (
	"bufio"
	"io"
	"strings"

	"github.com/cloudflare/pint/internal/comments"
	"github.com/cloudflare/pint/internal/diags"
)

type skipMode uint8

const (
	skipNone skipMode = iota
	skipNextLine
	skipBegin
	skipEnd
	skipCurrentLine
	skipFile
)

func newContentReader(r io.Reader) *ContentReader {
	return &ContentReader{
		src:         bufio.NewReader(r),
		buf:         nil,
		lines:       make([]string, 0, 100),
		comments:    nil,
		diagnostics: nil,
		lineno:      0,
		skipAll:     false,
		skipNext:    false,
		autoReset:   false,
		inBegin:     false,
	}
}

type ContentReader struct {
	src         *bufio.Reader
	buf         []byte
	comments    []comments.Comment
	diagnostics []diags.Diagnostic
	lines       []string
	lineno      int

	skipAll   bool
	skipNext  bool
	autoReset bool
	inBegin   bool
}

func (r *ContentReader) Read(b []byte) (got int, err error) {
	for got < cap(b) && err == nil {
		if len(r.buf) == 0 {
			err = r.readNextLine()
		}
		n := copy(b[got:], r.buf)
		r.buf = r.buf[n:]
		got += n
	}
	if err != nil && len(r.buf) > 0 {
		err = nil
	}
	return got, err
}

func (r *ContentReader) readNextLine() (err error) {
	r.buf, err = r.src.ReadBytes('\n')
	if len(r.buf) == 0 {
		return err
	}

	r.lineno++
	r.parseComments()
	r.lines = append(r.lines, strings.TrimRight(string(r.buf), "\r\n"))
	return err
}

func (r *ContentReader) parseComments() {
	lineComments := comments.Parse(r.lineno, string(r.buf), 0)

	if r.skipAll {
		r.emptyCurrentLine(lineComments)
		return
	}

	var found bool
	var skip skipMode
	for _, comment := range lineComments {
		// nolint:exhaustive
		switch comment.Type {
		case comments.IgnoreFileType:
			skip = skipFile
			found = true
			r.diagnostics = append(r.diagnostics, diags.Diagnostic{
				Message: "This file was excluded from pint checks.",
				Pos: diags.PositionRanges{
					{
						Line:        r.lineno,
						FirstColumn: comment.Offset + 1,
						LastColumn:  len(r.buf) - 1,
					},
				},
				Expr:        nil,
				FirstColumn: comment.Offset + 1,
				LastColumn:  len(r.buf) - 1,
				Kind:        diags.Issue,
			})
		case comments.IgnoreLineType:
			skip = skipCurrentLine
			found = true
		case comments.IgnoreBeginType:
			skip = skipBegin
			found = true
		case comments.IgnoreEndType:
			skip = skipEnd
			found = true
		case comments.IgnoreNextLineType:
			skip = skipNextLine
			found = true
		case comments.FileOwnerType:
			r.comments = append(r.comments, comment)
		case comments.RuleOwnerType:
			// pass
		case comments.FileDisableType:
			r.comments = append(r.comments, comment)
		case comments.DisableType:
			// pass
		case comments.FileSnoozeType:
			r.comments = append(r.comments, comment)
		case comments.SnoozeType:
			// pass
		case comments.RuleSetType:
			// pass
		case comments.InvalidComment:
			r.comments = append(r.comments, comment)
		}
	}
	switch {
	case found:
		switch skip { // nolint: exhaustive
		case skipFile:
			r.emptyCurrentLine(lineComments)
			r.skipNext = true
			r.autoReset = false
			r.skipAll = true
		case skipCurrentLine:
			r.emptyCurrentLine(lineComments)
			if !r.inBegin {
				r.skipNext = false
				r.autoReset = true
			}
		case skipNextLine:
			r.skipNext = true
			r.autoReset = true
		case skipBegin:
			r.skipNext = true
			r.autoReset = false
			r.inBegin = true
		case skipEnd:
			r.skipNext = false
			r.autoReset = true
			r.inBegin = false
		}
	case r.skipNext:
		r.emptyCurrentLine(lineComments)
		if r.autoReset {
			r.skipNext = false
		}
	}
}

func (r *ContentReader) emptyCurrentLine(comments []comments.Comment) {
	offset := len(r.buf)
	for _, c := range comments {
		offset = c.Offset
		break
	}
	for i := range r.buf {
		if r.buf[i] == '\n' || r.buf[i] == '\r' {
			continue
		}
		if i < offset || r.inBegin {
			r.buf[i] = ' '
		}
	}
}