dir_reader.go

package license

import (
	gopath "path"

	bosherr "github.com/cloudfoundry/bosh-utils/errors"
	boshsys "github.com/cloudfoundry/bosh-utils/system"

	. "github.com/cloudfoundry/bosh-cli/release/resource"
)

type DirReaderImpl struct {
	archiveFactory ArchiveFunc
	fs             boshsys.FileSystem
}

func NewDirReaderImpl(archiveFactory ArchiveFunc, fs boshsys.FileSystem) DirReaderImpl {
	return DirReaderImpl{archiveFactory: archiveFactory, fs: fs}
}

func (r DirReaderImpl) Read(path string) (*License, error) {
	files, err := r.collectFiles(path)
	if err != nil {
		return nil, bosherr.WrapErrorf(err, "Collecting license files")
	}

	if len(files) == 0 {
		return nil, nil
	}

	archive := r.archiveFactory(files, nil, nil)

	fp, err := archive.Fingerprint()
	if err != nil {
		return nil, err
	}

	return NewLicense(NewResource("license", fp, archive)), nil
}

func (r DirReaderImpl) collectFiles(path string) ([]File, error) {
	var files []File

	licenseMatches, err := r.fs.Glob(gopath.Join(path, "LICENSE*"))
	if err != nil {
		return nil, err
	}

	noticeMatches, err := r.fs.Glob(gopath.Join(path, "NOTICE*"))
	if err != nil {
		return nil, err
	}

	for _, filePath := range append(licenseMatches, noticeMatches...) {
		file := NewFile(filePath, path)
		file.UseBasename = true
		file.ExcludeMode = true
		files = append(files, file)
	}

	return files, nil
}