go-mtree/hierarchy.go

package mtree

import (
	"io"
	"sort"
)

// DirectoryHierarchy is the mapped structure for an mtree directory hierarchy
// spec
type DirectoryHierarchy struct {
	Entries []Entry
}

// WriteTo simplifies the output of the resulting hierarchy spec
func (dh DirectoryHierarchy) WriteTo(w io.Writer) (n int64, err error) {
	sort.Sort(byPos(dh.Entries))
	var sum int64
	for _, e := range dh.Entries {
		str := e.String()
		i, err := io.WriteString(w, str+"\n")
		if err != nil {
			return sum, err
		}
		sum += int64(i)
	}
	return sum, nil
}

// CollectUsedKeywords collects and returns all the keywords used in a
// a DirectoryHierarchy
func CollectUsedKeywords(dh *DirectoryHierarchy) []string {
	if dh != nil {
		usedkeywords := []string{}
		for _, e := range dh.Entries {
			switch e.Type {
			case FullType, RelativeType, SpecialType:
				if e.Type != SpecialType || e.Name == "/set" {
					kvs := e.Keywords
					for _, kv := range kvs {
						kw := KeyVal(kv).Keyword()
						if !inSlice(kw, usedkeywords) {
							usedkeywords = append(usedkeywords, kw)
						}
					}
				}
			}
		}
		return usedkeywords
	}
	return nil
}