1 files changed, 157 insertions, 0 deletions
diff --git a/client/go/cmd/init.go b/client/go/cmd/init.go
new file mode 100644
index 00000000000..64e8b342309
--- /dev/null
+++ b/client/go/cmd/init.go
@@ -0,0 +1,157 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// vespa init command
+// author: bratseth
+
+package cmd
+
+import (
+    "archive/zip"
+    "errors"
+    "path/filepath"
+    "github.com/spf13/cobra"
+    "github.com/vespa-engine/vespa/util"
+    "io"
+    "io/ioutil"
+    "net/http"
+    "net/url"
+    "os"
+    "strings"
+    "time"
+)
+
+// Set this to test without downloading this file from github
+var existingSampleAppsZip string
+
+func init() {
+    existingSampleAppsZip = ""
+    rootCmd.AddCommand(initCmd)
+}
+
+var initCmd = &cobra.Command{
+    // TODO: "application" and "list" subcommands?
+    Use:   "init",
+    Short: "Creates the files and directory structure for a new Vespa application",
+    Long:  `TODO: vespa init applicationName source`,
+    Args: func(cmd *cobra.Command, args []string) error {
+        if len(args) != 2 {
+            return errors.New("vespa init requires a project name and source")
+        }
+        return nil
+    },
+    Run: func(cmd *cobra.Command, args []string) {
+        initApplication(args[0], args[1])
+    },
+}
+
+func initApplication(name string, source string) {
+    zipFile := getSampleAppsZip()
+    if zipFile == nil {
+        return
+    }
+    if existingSampleAppsZip == "" { // Indicates we created a temp file now
+        defer os.Remove(zipFile.Name())
+    }
+
+    createErr := os.Mkdir(name, 0755)
+    if createErr != nil {
+        util.Error("Could not create directory '" + name + "'")
+        util.Detail(createErr.Error())
+        return
+    }
+
+	zipReader, zipOpenError := zip.OpenReader(zipFile.Name())
+	if zipOpenError != nil {
+        util.Error("Could not open sample apps zip '" + zipFile.Name() + "'")
+        util.Detail(zipOpenError.Error())
+	}
+	defer zipReader.Close()
+
+    found := false
+    for _, f := range zipReader.File {
+        zipEntryPrefix := "sample-apps-master/" + source + "/"
+	    if strings.HasPrefix(f.Name, zipEntryPrefix) {
+ 	        found = true
+	        copyError := copy(f, name, zipEntryPrefix)
+	        if copyError != nil {
+                util.Error("Could not copy zip entry '" + f.Name + "' to " + name)
+                util.Detail(copyError.Error())
+                return
+	        }
+        }
+	}
+	if !found {
+	    util.Error("Could not find source application '" + source + "'")
+	} else {
+        util.Success("Created " + name)
+    }
+}
+
+func getSampleAppsZip() *os.File {
+    if existingSampleAppsZip != "" {
+        existing, openExistingError := os.Open(existingSampleAppsZip)
+        if openExistingError != nil {
+            util.Error("Could not open existing sample apps zip file '" + existingSampleAppsZip + "'")
+            util.Detail(openExistingError.Error())
+        }
+        return existing
+    }
+
+    // TODO: Cache it?
+    util.Detail("Downloading sample apps ...") // TODO: Spawn thread to indicate progress
+    zipUrl, _ := url.Parse("https://github.com/vespa-engine/sample-apps/archive/refs/heads/master.zip")
+    request := &http.Request{
+        URL: zipUrl,
+        Method: "GET",
+    }
+    response := util.HttpDo(request, time.Minute * 60, "GitHub")
+    defer response.Body.Close()
+    if response.StatusCode != 200 {
+        util.Error("Could not download sample apps from github")
+        util.Detail(response.Status)
+        return nil
+    }
+
+    destination, tempFileError := ioutil.TempFile("", "prefix")
+    if tempFileError != nil {
+        util.Error("Could not create a temp file to hold sample apps")
+        util.Detail(tempFileError.Error())
+    }
+    // destination, _ := os.Create("./" + name + "/sample-apps.zip")
+    // defer destination.Close()
+    _, err := io.Copy(destination, response.Body)
+    if err != nil {
+        util.Error("Could not download sample apps from GitHub")
+        util.Detail(err.Error())
+        return nil
+    }
+    return destination
+}
+
+func copy(f *zip.File, destinationDir string, zipEntryPrefix string) error {
+    destinationPath := filepath.Join(destinationDir, filepath.FromSlash(strings.TrimPrefix(f.Name, zipEntryPrefix)))
+    if strings.HasSuffix(f.Name, "/") {
+        if f.Name != zipEntryPrefix { // root is already created
+            createError := os.Mkdir(destinationPath, 0755)
+            if createError != nil {
+                return createError
+            }
+        }
+    } else {
+        zipEntry, zipEntryOpenError := f.Open()
+        if zipEntryOpenError != nil {
+            return zipEntryOpenError
+        }
+        defer zipEntry.Close()
+
+        destination, createError := os.Create(destinationPath)
+        if createError != nil {
+            return createError
+        }
+
+        _, copyError := io.Copy(destination, zipEntry)
+        if copyError != nil {
+            return copyError
+        }
+    }
+    return nil
+}