From c136af25d0a2bf5235cadd145caab60f72be6cee Mon Sep 17 00:00:00 2001
From: Luiz Filho <luizbafilho@gmail.com>
Date: Wed, 25 Nov 2020 20:09:54 -0300
Subject: [PATCH] Spliting up file and directory creation

---
 lokustctl/cmd/create.go | 32 +++++++++++++++++++++++++++-----
 1 file changed, 27 insertions(+), 5 deletions(-)

diff --git a/lokustctl/cmd/create.go b/lokustctl/cmd/create.go
index 07a9032..c391c08 100644
--- a/lokustctl/cmd/create.go
+++ b/lokustctl/cmd/create.go
@@ -97,14 +97,24 @@ func buildLocustTest(config Config, cm *corev1.ConfigMap) *loadtestsv1beta1.Locu
 }
 
 func buildLocustfile(config Config) *corev1.ConfigMap {
-	file, err := ioutil.ReadFile(config.Locustfile)
+	info, err := os.Stat(config.Locustfile)
 	if err != nil {
-		fmt.Printf("Error reading locustfile. err: %s", err)
+		fmt.Printf("Error reading locustfile: %s. err: %s\n", config.Locustfile, err)
 		os.Exit(1)
 	}
 
-	configMapData := make(map[string]string, 0)
-	configMapData["locustfile.py"] = string(file)
+	var cmData map[string]string
+
+	if info.IsDir() {
+		fmt.Printf("lokust does not support using directories in the moment. \nPlease open an issue on GitHub if you want to see that feature.")
+		os.Exit(1)
+	} else {
+		cmData, err = buildConfigmapFromFile(config.Locustfile)
+		if err != nil {
+			fmt.Printf("Error reading locustfile. err: %s\n", err)
+			os.Exit(1)
+		}
+	}
 
 	configMap := corev1.ConfigMap{
 		TypeMeta: metav1.TypeMeta{
@@ -115,12 +125,24 @@ func buildLocustfile(config Config) *corev1.ConfigMap {
 			Name:      buildResourceName(config.Name, "configmap"),
 			Namespace: config.Namespace,
 		},
-		Data: configMapData,
+		Data: cmData,
 	}
 
 	return &configMap
 }
 
+func buildConfigmapFromFile(filename string) (map[string]string, error) {
+	file, err := ioutil.ReadFile(filename)
+	if err != nil {
+		return nil, err
+	}
+
+	configMapData := make(map[string]string, 0)
+	configMapData["locustfile.py"] = string(file)
+
+	return configMapData, nil
+}
+
 func buildResourceName(testName string, resourceType ...string) string {
 	name := fmt.Sprintf("lokust-%s", testName)