Create sketch from template if template directory is set

Author: gsingh93

commands/sketch/new.go

@@ -55,19 +55,32 @@ func NewSketch(ctx context.Context, req *rpc.NewSketchRequest) (*rpc.NewSketchRe
 		return nil, err
 	}
 
+	templateDir := configuration.Settings.GetString("directories.template")
 	sketchDirPath := paths.New(sketchesDir).Join(req.SketchName)
-	if err := sketchDirPath.MkdirAll(); err != nil {
-		return nil, &arduino.CantCreateSketchError{Cause: err}
-	}
-	sketchName := sketchDirPath.Base()
-	sketchMainFilePath := sketchDirPath.Join(sketchName + globals.MainFileValidExtension)
-	if !req.Overwrite {
-		if sketchMainFilePath.Exist() {
-			return nil, &arduino.CantCreateSketchError{Cause: errors.New(tr(".ino file already exists"))}
+
+	var sketchMainFilePath *paths.Path
+	if templateDir != "" {
+		templateDirPath := paths.New(templateDir)
+		if err := templateDirPath.CopyDirTo(sketchDirPath); err != nil {
+			return nil, &arduino.CantCreateSketchError{Cause: err}
+		}
+		// TODO: Make this customizable?
+		sketchMainFilePath = sketchDirPath.Join("main.ino")
+	} else {
+		if err := sketchDirPath.MkdirAll(); err != nil {
+			return nil, &arduino.CantCreateSketchError{Cause: err}
+		}
+		sketchName := sketchDirPath.Base()
+		sketchMainFilePath = sketchDirPath.Join(sketchName + globals.MainFileValidExtension)
+		if !req.Overwrite {
+			if sketchMainFilePath.Exist() {
+				return nil, &arduino.CantCreateSketchError{Cause: errors.New(tr(".ino file already exists"))}
+			}
+		}
+
+		if err := sketchMainFilePath.WriteFile(emptySketch); err != nil {
+			return nil, &arduino.CantCreateSketchError{Cause: err}
 		}
-	}
-	if err := sketchMainFilePath.WriteFile(emptySketch); err != nil {
-		return nil, &arduino.CantCreateSketchError{Cause: err}
 	}
 
 	return &rpc.NewSketchResponse{MainFile: sketchMainFilePath.String()}, nil