Implemented http/https remote archive fetching

Author: Knetic

RunSettings.go

@@ -55,7 +55,7 @@ func FindRunSettings() (RunSettings, error) {
 
 	flag.StringVar(&parameterGroup, "p", "", "Semicolon-separated list of parameters in k=v form.")
 	flag.BoolVar(&ret.inspectionRun, "i", false, "Whether or not to show a list of available parameters for the skeleton")
-	flag.BoolVar(&ret.addRegistry, "a", false, "Whether or not to register the template at the given path")
+	flag.BoolVar(&ret.addRegistry, "a", false, "Whether or not to register the template at the given path (can be http/https URLs)")
 	flag.BoolVar(&ret.showRegistry, "l", false, "Whether or not to show a list of all available registered templates")
 	flag.BoolVar(&forceGenerate, "f", false, "Whether or not to overwrite existing files during generation")
 	flag.BoolVar(&forceDelete, "d", false, "Whether or not to delete every existing file in the output path first (only valid when -f is specified)")

TemplateRegistry.go

@@ -6,7 +6,10 @@ import (
 	"github.com/jhoonb/archivex"
 	"github.com/mitchellh/go-homedir"
 	"io/ioutil"
+	"net/http"
 	"os"
+	"net/url"
+	"io"
 	"path/filepath"
 	"strings"
 )
@@ -112,6 +115,15 @@ func (this *TemplateRegistry) RegisterTemplate(path string) (string, error) {
 	var targetPath, name string
 	var err error
 
+	// if this is a remote path, download as a zip to a temporary directory before trying to register.
+	if strings.HasPrefix(path, "http://") || strings.HasPrefix(path, "https://") {
+
+		path, err = downloadArchive(path)
+		if(err != nil) {
+			return "", err
+		}
+	}
+
 	// if the given path is a directory (not a zip file),
 	// archive it and prepare for registration.
 	if !strings.HasSuffix(path, archiveMarker) {
@@ -129,6 +141,57 @@ func (this *TemplateRegistry) RegisterTemplate(path string) (string, error) {
 	return name, err
 }
 
+/*
+	Downloads the given [url] to a temporary directory (as a .zip).
+	Returns the temporary path to the downloaded archive, or an error if it encountered one.
+*/
+func downloadArchive(targetURL string) (string, error) {
+
+	var response *http.Response
+	var outputFile *os.File
+	var parsedURL *url.URL
+	var outputPath string
+	var baseName string
+	var err error
+
+	outputPath, err = ioutil.TempDir("", "wiskRemoteArchive")
+	if(err != nil) {
+		return "", err
+	}
+
+	parsedURL, err = url.Parse(targetURL)
+	if(err != nil) {
+		return "", err
+	}
+
+	baseName = filepath.Base(parsedURL.Path)
+	baseName = strings.TrimSuffix(baseName, filepath.Ext(baseName))
+	outputPath = filepath.Join(outputPath, baseName + ".zip")
+	outputPath, _ = filepath.Abs(outputPath)
+
+	outputFile, err = os.Create(outputPath)
+	if(err != nil) {
+		return "", err
+	}
+	defer outputFile.Close()
+
+	response, err = http.Get(targetURL)
+	if(err != nil) {
+		return "", err
+	}
+
+	if(response.Status != "200") {
+		errorMsg := fmt.Sprintf("Unable to download remote archive: HTTP %s\n", response.Status)
+		return "", errors.New(errorMsg)
+	}
+
+	_, err = io.Copy(outputFile, response.Body)
+	if(err != nil) {
+		return "", err
+	}
+	return outputPath, nil
+}
+
 func archivePath(path string) (string, error) {
 
 	var zip archivex.ZipFile

docs/wisk.7

@@ -22,7 +22,7 @@ Causes the program to list all registered templates by name, then immediately ex
 .IP -i
 Causes the program to inspect the given skeleton (by positional argument 1), print out all of that skeleton's possible parameters, then exit.
 .IP -a
-Causes the program to register the given path (by positional argument 1) as a template. If the path is a *.zip file, it is added to the user's skeleton registry as-is. If the path is a directory, that directory is zipped and added to the user's skeleton registry. In no case will this command ever modify the given directory.
+Causes the program to register the given path (by positional argument 1) as a template. If the path is a *.zip file, it is added to the user's skeleton registry as-is. If the path is a directory, that directory is zipped and added to the user's skeleton registry. If the path is prefixed with 'http://' or 'https://', the remote archive will be downloaded and registered. In no case will this command ever modify the given directory.
 .IP -p
 Specifies parameters to be used when populating the target project. See "Parameters" for details.
 .IP -f