Add 'template add-link' command

Author: stuartleeks

cmd/devcontainer/template.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"io/ioutil"
 	"os"
 	"sort"
 
@@ -18,6 +19,7 @@ func createTemplateCommand() *cobra.Command {
 	}
 	cmd.AddCommand(createTemplateListCommand())
 	cmd.AddCommand(createTemplateAddCommand())
+	cmd.AddCommand(createTemplateAddLinkCommand())
 	return cmd
 }
 
@@ -100,3 +102,67 @@ func createTemplateAddCommand() *cobra.Command {
 	}
 	return cmd
 }
+
+func createTemplateAddLinkCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "add-link TEMPLATE_NAME",
+		Short: "add-link devcontainer from template",
+		Long:  "Symlink a devcontainer definition to the current folder using the specified template",
+		Run: func(cmd *cobra.Command, args []string) {
+
+			if len(args) != 1 {
+				cmd.Usage()
+				os.Exit(1)
+			}
+			name := args[0]
+
+			template, err := devcontainers.GetTemplateByName(name)
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(1)
+			}
+			if template == nil {
+				fmt.Printf("Template '%s' not found\n", name)
+			}
+
+			info, err := os.Stat("./.devcontainer")
+			if info != nil && err == nil {
+				fmt.Println("Current folder already contains a .devcontainer folder - exiting")
+				os.Exit(1)
+			}
+
+			currentDirectory, err := os.Getwd()
+			if err != nil {
+				fmt.Printf("Error reading current directory: %s\n", err)
+			}
+			if err = ioutil2.LinkFolder(template.Path, currentDirectory+"/.devcontainer"); err != nil {
+				fmt.Printf("Error linking folder: %s\n", err)
+				os.Exit(1)
+			}
+
+			content := []byte("*\n")
+			if err := ioutil.WriteFile(currentDirectory+"/.devcontainer/.gitignore", content, 0644); err != nil { // -rw-r--r--
+				fmt.Printf("Error writing .gitignore: %s\n", err)
+				os.Exit(1)
+			}
+		},
+		ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+			// only completing the first arg  (template name)
+			if len(args) != 0 {
+				return nil, cobra.ShellCompDirectiveNoFileComp
+			}
+			templates, err := devcontainers.GetTemplates()
+			if err != nil {
+				fmt.Printf("Error: %v", err)
+				os.Exit(1)
+			}
+			names := []string{}
+			for _, template := range templates {
+				names = append(names, template.Name)
+			}
+			sort.Strings(names)
+			return names, cobra.ShellCompDirectiveNoFileComp
+		},
+	}
+	return cmd
+}

internal/pkg/ioutil/files.go

@@ -10,6 +10,19 @@ import (
 // TODO - recursive copy folder (also add recursive symlink)
 
 func CopyFolder(source string, target string) error {
+	copy := func(sourceFolder string, targetFolder string, item os.FileInfo) error {
+		return CopyFile(sourceFolder+"/"+item.Name(), targetFolder+"/"+item.Name(), item.Mode())
+	}
+	return processFolder(source, target, copy)
+}
+func LinkFolder(source string, target string) error {
+	symlink := func(sourceFolder string, targetFolder string, item os.FileInfo) error {
+		return os.Symlink(sourceFolder+"/"+item.Name(), targetFolder+"/"+item.Name())
+	}
+	return processFolder(source, target, symlink)
+}
+
+func processFolder(source string, target string, fileHandler func(sourceFolder string, targetFolder string, item os.FileInfo) error) error {
 	sourceItem, err := os.Stat(source)
 	if err != nil {
 		return fmt.Errorf("Error reading source folder: %s\n", err)
@@ -25,13 +38,18 @@ func CopyFolder(source string, target string) error {
 
 	for _, sourceSubItem := range sourceSubItems {
 		if sourceSubItem.IsDir() {
-			CopyFolder(source+"/"+sourceSubItem.Name(), target+"/"+sourceSubItem.Name())
+			if err = processFolder(source+"/"+sourceSubItem.Name(), target+"/"+sourceSubItem.Name(), fileHandler); err != nil {
+				return err
+			}
 		} else {
-			CopyFile(source+"/"+sourceSubItem.Name(), target+"/"+sourceSubItem.Name(), sourceSubItem.Mode())
+			if err = fileHandler(source, target, sourceSubItem); err != nil {
+				return err
+			}
 		}
 	}
 	return nil
 }
+
 func CopyFile(source string, target string, perm os.FileMode) error {
 	sourceFile, err := os.Open(source)
 	if err != nil {