Add envdir package (#6148)

pkg/envdir/envdir.go

@@ -0,0 +1,34 @@
+package envdir
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+
+	"github.com/pkg/errors"
+)
+
+func envDir(dir string) ([]string, error) {
+	fis, err := ioutil.ReadDir(dir)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to read env from dir:"+dir)
+	}
+	e := make([]string, 0, len(fis))
+	for _, fi := range fis {
+		if fi.IsDir() {
+			continue
+		}
+		p := filepath.Join(dir, fi.Name())
+		f, err := os.Open(p)
+		if err != nil {
+			return nil, errors.Wrap(err, "failed to read env from dir:"+dir)
+		}
+		c, err := ioutil.ReadAll(f)
+		if err != nil {
+			return nil, errors.Wrap(err, "failed to read env from dir:"+dir)
+		}
+		e = append(e, fmt.Sprintf("%s=%s", fi.Name(), c))
+	}
+	return e, nil
+}

pkg/envdir/envdir_test.go

@@ -0,0 +1,27 @@
+package envdir
+
+import (
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"testing"
+
+	. "gopkg.in/check.v1"
+)
+
+// Hook up gocheck into the "go test" runner.
+func Test(t *testing.T) { TestingT(t) }
+
+type EnvDirSuite struct{}
+
+var _ = Suite(&EnvDirSuite{})
+
+func (s *EnvDirSuite) TestEnvDir(c *C) {
+	d := c.MkDir()
+	p := filepath.Join(d, "FOO")
+	err := ioutil.WriteFile(p, []byte("BAR"), os.ModePerm)
+	c.Assert(err, IsNil)
+	e, err := envDir(d)
+	c.Assert(err, IsNil)
+	c.Assert(e, DeepEquals, []string{"FOO=BAR"})
+}