pkger/pkging/stuffing/stuffing.go

package stuffing

import (
	"io"

	"github.com/markbates/pkger/here"
	"github.com/markbates/pkger/pkging/mem"
	"github.com/markbates/pkger/pkging/stdos"
)

func Stuff(w io.Writer, cur here.Info, paths []here.Path) error {
	disk, err := stdos.New(cur)
	if err != nil {
		return err
	}

	pkg, err := mem.New(cur)
	if err != nil {
		return err
	}

	for _, pt := range paths {
		err = func() error {
			f, err := disk.Open(pt.String())
			if err != nil {
				return err
			}
			defer f.Close()
			if err := pkg.Add(f); err != nil {
				return err
			}

			return nil
		}()
		if err != nil {
			return err
		}
	}

	b, err := pkg.MarshalEmbed()
	if err != nil {
		return err
	}
	_, err = w.Write(b)
	return err
}