payload.go

package go_bagit

import (
	"fmt"
	"io/fs"
	"os"
	"path/filepath"
	"regexp"
)

type Payload map[string]os.FileInfo

type PayloadMatch struct {
	Path     string
	FileInfo os.FileInfo
}

func loadPayload(bag *Bag) error {
	dataDir := filepath.Join(bag.Path, "data")
	err := filepath.Walk(dataDir, func(path string, info fs.FileInfo, err error) error {
		if path != dataDir {
			bag.Payload[path] = info
		}
		return nil
	})

	if err != nil {
		return err
	}

	return nil
}

func (p Payload) GetFileInPayload(filename string) (PayloadMatch, error) {
	for path, fi := range p {
		if fi.Name() == filename && !fi.IsDir() {
			return PayloadMatch{path, fi}, nil
		}
	}
	return PayloadMatch{}, fmt.Errorf("Payload did not match %s", filename)
}

func (p Payload) GetDirInPayload(dirName string) (PayloadMatch, error) {
	for path, fi := range p {
		if fi.Name() == dirName && fi.IsDir() {
			return PayloadMatch{path, fi}, nil
		}
	}
	return PayloadMatch{}, fmt.Errorf("Payload did not match %s", dirName)
}

func (p Payload) FindFilesInPayload(matcher *regexp.Regexp) []PayloadMatch {
	payloadMatch := []PayloadMatch{}
	for path, fi := range p {
		if matcher.MatchString(path) && !fi.IsDir() {
			payloadMatch = append(payloadMatch, PayloadMatch{path, fi})
		}
	}
	return payloadMatch
}

func (p Payload) FindDirsInPayload(matcher *regexp.Regexp) []PayloadMatch {
	payloadMatch := []PayloadMatch{}
	for path, fi := range p {
		if matcher.MatchString(path) && fi.IsDir() {
			payloadMatch = append(payloadMatch, PayloadMatch{path, fi})
		}
	}
	return payloadMatch
}

func (p Payload) FindAllInPayload(matcher *regexp.Regexp) []PayloadMatch {
	payloadMatches := []PayloadMatch{}
	for path, fi := range p {
		if matcher.MatchString(path) {
			payloadMatches = append(payloadMatches, PayloadMatch{path, fi})
		}
	}
	return payloadMatches
}