From f71ef86d4865d0909f1c360c907005f0092832dc Mon Sep 17 00:00:00 2001
From: Matthias Loibl <mail@matthiasloibl.com>
Date: Wed, 24 Apr 2019 18:21:22 +0200
Subject: [PATCH] Create pkg/jsonnetfile to better encapsulate jsonnetfile
 things

---
 pkg/jsonnetfile/jsonnetfile.go      |  66 +++++++++++++
 pkg/jsonnetfile/jsonnetfile_test.go | 147 ++++++++++++++++++++++++++++
 2 files changed, 213 insertions(+)
 create mode 100644 pkg/jsonnetfile/jsonnetfile.go
 create mode 100644 pkg/jsonnetfile/jsonnetfile_test.go

diff --git a/pkg/jsonnetfile/jsonnetfile.go b/pkg/jsonnetfile/jsonnetfile.go
new file mode 100644
index 0000000..2b6d082
--- /dev/null
+++ b/pkg/jsonnetfile/jsonnetfile.go
@@ -0,0 +1,66 @@
+package jsonnetfile
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"os"
+	"path"
+
+	"github.com/jsonnet-bundler/jsonnet-bundler/spec"
+	"github.com/pkg/errors"
+)
+
+const File = "jsonnetfile.json"
+const LockFile = "jsonnetfile.lock.json"
+
+var ErrNoFile = errors.New("no jsonnetfile")
+
+func Choose(dir string) (string, bool, error) {
+	jsonnetfileLock := path.Join(dir, LockFile)
+	jsonnetfile := path.Join(dir, File)
+
+	lockExists, err := fileExists(jsonnetfileLock)
+	if err != nil {
+		return "", false, err
+	}
+	if lockExists {
+		return jsonnetfileLock, true, nil
+	}
+
+	fileExists, err := fileExists(jsonnetfile)
+	if err != nil {
+		return "", false, err
+	}
+	if fileExists {
+		return jsonnetfile, false, nil
+	}
+
+	return "", false, ErrNoFile
+}
+
+func Load(filepath string) (spec.JsonnetFile, error) {
+	m := spec.JsonnetFile{}
+
+	bytes, err := ioutil.ReadFile(filepath)
+	if err != nil {
+		return m, errors.Wrap(err, "failed to read file")
+	}
+
+	if err := json.Unmarshal(bytes, &m); err != nil {
+		return m, errors.Wrap(err, "failed to unmarshal file")
+	}
+
+	return m, nil
+}
+
+func fileExists(path string) (bool, error) {
+	_, err := os.Stat(path)
+	if os.IsNotExist(err) {
+		return false, nil
+	}
+	if err != nil {
+		return false, err
+	}
+
+	return true, nil
+}
diff --git a/pkg/jsonnetfile/jsonnetfile_test.go b/pkg/jsonnetfile/jsonnetfile_test.go
new file mode 100644
index 0000000..e7405b2
--- /dev/null
+++ b/pkg/jsonnetfile/jsonnetfile_test.go
@@ -0,0 +1,147 @@
+package jsonnetfile_test
+
+import (
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/jsonnet-bundler/jsonnet-bundler/pkg/jsonnetfile"
+	"github.com/jsonnet-bundler/jsonnet-bundler/spec"
+	"github.com/stretchr/testify/assert"
+)
+
+const notExist = "/this/does/not/exist"
+
+func TestChoose(t *testing.T) {
+	testcases := []struct {
+		Name             string
+		Jsonnetfile      []byte
+		JsonnetfileLock  []byte
+		ExpectedFilename string
+		ExpectedLock     bool
+		ExpectedError    error
+	}{{
+		Name:             "NoFiles",
+		ExpectedFilename: "",
+		ExpectedLock:     false,
+		ExpectedError:    jsonnetfile.ErrNoFile,
+	}, {
+		Name:             "Jsonnetfile",
+		Jsonnetfile:      []byte(`{}`),
+		ExpectedFilename: jsonnetfile.File,
+		ExpectedLock:     false,
+		ExpectedError:    nil,
+	}, {
+		Name:             "JsonnetfileLock",
+		Jsonnetfile:      []byte(`{}`),
+		JsonnetfileLock:  []byte(`{}`),
+		ExpectedFilename: jsonnetfile.LockFile,
+		ExpectedLock:     true,
+		ExpectedError:    nil,
+	}}
+
+	for _, tc := range testcases {
+		t.Run(tc.Name, func(t *testing.T) {
+			dir, err := ioutil.TempDir("", "jsonnetfile-choose")
+			assert.Nil(t, err)
+			defer os.Remove(dir)
+
+			if tc.Jsonnetfile != nil {
+				err := ioutil.WriteFile(filepath.Join(dir, jsonnetfile.File), tc.Jsonnetfile, os.ModePerm)
+				assert.NoError(t, err)
+			}
+			if tc.JsonnetfileLock != nil {
+				err := ioutil.WriteFile(filepath.Join(dir, jsonnetfile.LockFile), tc.JsonnetfileLock, os.ModePerm)
+				assert.NoError(t, err)
+			}
+
+			filename, isLock, err := jsonnetfile.Choose(dir)
+
+			assert.Equal(t, tc.ExpectedFilename, strings.TrimPrefix(filename, dir+"/"))
+			assert.Equal(t, tc.ExpectedLock, isLock)
+
+			if tc.ExpectedError != nil {
+				assert.EqualError(t, err, tc.ExpectedError.Error())
+			} else {
+				assert.NoError(t, err)
+			}
+		})
+	}
+}
+
+func TestLoad(t *testing.T) {
+	empty := spec.JsonnetFile{}
+
+	jsonnetfileContent := `{
+    "dependencies": [
+        {
+            "name": "foobar",
+            "source": {
+                "git": {
+                    "remote": "https://github.com/foobar/foobar",
+                    "subdir": ""
+                }
+            },
+            "version": "master"
+        }
+    ]
+}
+`
+	jsonnetFileExpected := spec.JsonnetFile{
+		Dependencies: []spec.Dependency{{
+			Name: "foobar",
+			Source: spec.Source{
+				GitSource: &spec.GitSource{
+					Remote: "https://github.com/foobar/foobar",
+					Subdir: "",
+				},
+			},
+			Version:   "master",
+			DepSource: "",
+		}},
+	}
+
+	{
+		jf, err := jsonnetfile.Load(notExist)
+		assert.Equal(t, empty, jf)
+		assert.Error(t, err)
+	}
+	{
+		tempDir, err := ioutil.TempDir("", "jb-load-jsonnetfile")
+		if err != nil {
+			t.Fatal(err)
+		}
+		defer func() {
+			err := os.RemoveAll(tempDir)
+			assert.Nil(t, err)
+		}()
+
+		tempFile := filepath.Join(tempDir, jsonnetfile.File)
+		err = ioutil.WriteFile(tempFile, []byte(`{}`), os.ModePerm)
+		assert.Nil(t, err)
+
+		jf, err := jsonnetfile.Load(tempFile)
+		assert.Nil(t, err)
+		assert.Equal(t, empty, jf)
+	}
+	{
+		tempDir, err := ioutil.TempDir("", "jb-load-jsonnetfile")
+		if err != nil {
+			t.Fatal(err)
+		}
+		defer func() {
+			err := os.RemoveAll(tempDir)
+			assert.Nil(t, err)
+		}()
+
+		tempFile := filepath.Join(tempDir, jsonnetfile.File)
+		err = ioutil.WriteFile(tempFile, []byte(jsonnetfileContent), os.ModePerm)
+		assert.Nil(t, err)
+
+		jf, err := jsonnetfile.Load(tempFile)
+		assert.Nil(t, err)
+		assert.Equal(t, jsonnetFileExpected, jf)
+	}
+}