gignore/internal/filewriter/filewriter.go

package filewriter

import (
	"bytes"
	"io"
	"os"
)

type FileWriter struct {
}

func New() *FileWriter {
	return &FileWriter{}
}

func (fw *FileWriter) writeContent(content []byte, dst io.Writer) (int64, error) {
	r := bytes.NewReader(content)

	n, err := io.Copy(dst, r)
	if err != nil {
		return 0, err
	}

	return n, nil
}

func (fw *FileWriter) Write(content []byte) (int, error) {
	f, err := os.Create(".gitignore")
	if err != nil {
		return 0, err
	}
	defer f.Close()

	const header = `# Auto-generated .gitignore by gignore: github.com/onyx-and-iris/gignore`
	const footer = `# End of gignore: github.com/onyx-and-iris/gignore`

	var sz int64
	n, err := fw.writeContent([]byte(header), f)
	if err != nil {
		return 0, err
	}
	sz += n
	n, err = fw.writeContent(content, f)
	if err != nil {
		return 0, err
	}
	sz += n
	n, err = fw.writeContent([]byte(footer), f)
	if err != nil {
		return 0, err
	}
	return int(sz + n), nil
}