input_spliter.go

package gtree

import (
	"bufio"
	"context"
	"fmt"
	"io"

	md "github.com/ddddddO/gtree/markdown"
)

func split(ctx context.Context, r io.Reader) (<-chan string, <-chan error) {
	sc := bufio.NewScanner(r)
	blockc := make(chan string)
	errc := make(chan error)

	go func() {
		defer func() {
			close(blockc)
			close(errc)
		}()

		block := ""
		for sc.Scan() {
			select {
			case <-ctx.Done():
				return
			default:
				l := sc.Text()
				if isRootBlockBeginning(l) {
					if len(block) != 0 {
						select {
						case <-ctx.Done():
							return
						case blockc <- block:
						}
					}
					block = ""
				}
				block += fmt.Sprintln(l)
			}
		}
		if err := sc.Err(); err != nil {
			errc <- err
			return
		}
		select {
		case <-ctx.Done():
			return
		case blockc <- block: // 最後のRootブロック送出
			return
		}
	}()

	return blockc, errc
}

func isRootBlockBeginning(l string) bool {
	if len(l) == 0 {
		return false
	}
	return md.IsSymbol(l[0:1])
}