pkg/data_git_ignore.go

package pkg

import (
	"bufio"
	"github.com/Azure/golden"
	"github.com/spf13/afero"
	"strings"
)

var _ Data = &GitIgnoreDatasource{}

type GitIgnoreDatasource struct {
	*golden.BaseBlock
	*BaseData
	Records []string `attribute:"records"`
}

func (g *GitIgnoreDatasource) ExecuteDuringPlan() error {
	fs := FsFactory()
	gitIgnoreFile := "./.gitignore"
	exists, err := afero.Exists(fs, gitIgnoreFile)
	if err != nil {
		return err
	}
	if !exists {
		g.Records = []string{}
		return nil
	}
	f, err := fs.Open(gitIgnoreFile)
	if err != nil {
		return err
	}

	scanner := bufio.NewScanner(f)
	for scanner.Scan() {
		line := scanner.Text()
		line = strings.TrimFunc(line, func(r rune) bool {
			return r == '\t' || r == ' '
		})
		if strings.HasPrefix(line, "#") || line == "" {
			continue
		}
		g.Records = append(g.Records, line)
	}
	return scanner.Err()
}

func (g *GitIgnoreDatasource) Type() string {
	return "git_ignore"
}