Add support for reading YAML data from remote URLs

awslabs · May 14, 2024 · d13870c · d13870c
1 parent 0a20489
commit d13870c
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 8 deletions.
diff --git a/internal/ctl/create.go b/internal/ctl/create.go
@@ -9,6 +9,7 @@ import (
 	"image/color"
 	"image/png"
 	"os"
+	"strings"
 
 	"github.com/awslabs/diagram-as-code/internal/cache"
 	"github.com/awslabs/diagram-as-code/internal/definition"
@@ -277,3 +278,10 @@ func loadLinks(template *TemplateStruct, resources map[string]types.Node) {
 	}
 
 }
+
+func IsURL(str string) bool {
+	if strings.HasPrefix(str, "http://") || strings.HasPrefix(str, "https://") {
+		return true
+	}
+	return false
+}
diff --git a/internal/ctl/yaml.go b/internal/ctl/yaml.go
@@ -4,6 +4,8 @@
 package ctl
 
 import (
+	"io"
+	"net/http"
 	"os"
 
 	"github.com/awslabs/diagram-as-code/internal/definition"
@@ -14,17 +16,38 @@ import (
 
 func CreateDiagramFromYAML(inputfile string, outputfile *string) {
 
-	log.Infof("input file: %s\n", inputfile)
-	data, err := os.ReadFile(inputfile)
-	if err != nil {
-		log.Fatal(err)
-	}
+	log.Infof("input file path: %s\n", inputfile)
 
 	var template TemplateStruct
 
-	err = yaml.Unmarshal([]byte(data), &template)
-	if err != nil {
-		log.Fatal(err)
+	if IsURL(inputfile) {
+		// URL from remote
+		resp, err := http.Get(inputfile)
+		if err != nil {
+			log.Fatal(err)
+		}
+		defer resp.Body.Close()
+
+		data, err := io.ReadAll(resp.Body)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		err = yaml.Unmarshal(data, &template)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+	} else {
+		// Local file
+		data, err := os.ReadFile(inputfile)
+		if err != nil {
+			log.Fatal(err)
+		}
+		err = yaml.Unmarshal([]byte(data), &template)
+		if err != nil {
+			log.Fatal(err)
+		}
 	}
 
 	var ds definition.DefinitionStructure