Added upload command

Author: cmaglie

commands/root/root.go

@@ -44,6 +44,7 @@ import (
 	"github.com/bcmi-labs/arduino-cli/commands/login"
 	"github.com/bcmi-labs/arduino-cli/commands/logout"
 	"github.com/bcmi-labs/arduino-cli/commands/sketch"
+	"github.com/bcmi-labs/arduino-cli/commands/upload"
 	"github.com/bcmi-labs/arduino-cli/commands/validate"
 	"github.com/bcmi-labs/arduino-cli/commands/version"
 	"github.com/bcmi-labs/arduino-cli/common/formatter"
@@ -107,6 +108,7 @@ func Init(_isTesting bool) {
 	login.Init(Command)
 	logout.Init(Command)
 	sketch.Init(Command)
+	upload.Init(Command)
 	validate.Init(Command)
 	version.Init(Command)
 }

commands/upload/upload.go

@@ -0,0 +1,193 @@
+/*
+ * This file is part of arduino-cli.
+ *
+ * arduino-cli is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ * As a special exception, you may use this file as part of a free software
+ * library without restriction.  Specifically, if other files instantiate
+ * templates or use macros or inline functions from this file, or you compile
+ * this file and link it with other files to produce an executable, this
+ * file does not by itself cause the resulting executable to be covered by
+ * the GNU General Public License.  This exception does not however
+ * invalidate any other reasons why the executable file might be covered by
+ * the GNU General Public License.
+ *
+ * Copyright 2017-2018 ARDUINO AG (http://www.arduino.cc/)
+ */
+
+package upload
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	properties "github.com/arduino/go-properties-map"
+	"github.com/bcmi-labs/arduino-cli/commands"
+	"github.com/bcmi-labs/arduino-cli/common/formatter"
+	"github.com/bcmi-labs/arduino-cli/cores"
+	"github.com/bcmi-labs/arduino-cli/sketches"
+	"github.com/spf13/cobra"
+)
+
+// Init prepares the command.
+func Init(rootCommand *cobra.Command) {
+	rootCommand.AddCommand(command)
+	command.Flags().StringVarP(&flags.fqbn, "fqbn", "b", "", "Fully Qualified Board Name, e.g.: arduino:avr:uno")
+	command.Flags().StringVarP(&flags.port, "port", "p", "", "Upload port, e.g.: COM10 or /dev/ttyACM0")
+	command.Flags().BoolVarP(&flags.verbose, "verbose", "v", false, "Optional, turns on verbose mode.")
+}
+
+var flags struct {
+	fqbn    string
+	port    string
+	verbose bool
+}
+
+var command = &cobra.Command{
+	Use:     "upload",
+	Short:   "Upload Arduino sketches.",
+	Long:    "Upload Arduino sketches.",
+	Example: "arduino upload [sketchName]",
+	Args:    cobra.ExactArgs(0),
+	Run:     run,
+}
+
+func run(cmd *cobra.Command, args []string) {
+	// FIXME: factorize a general way to determine current sketch
+	var sketchName string
+	if len(args) == 0 {
+		sketchName, err := os.Getwd()
+		if err != nil {
+			formatter.PrintError(err, "Could not determine current working directory")
+			os.Exit(commands.ErrGeneric)
+		}
+		sketchName = filepath.Base(sketchName)
+	} else {
+		sketchName = args[0]
+	}
+
+	// FIXME: make a specification on how a port is specified via command line
+	port := flags.port
+	if port == "" {
+		formatter.PrintErrorMessage("No port provided.")
+		os.Exit(commands.ErrBadCall)
+	}
+
+	fqbn := flags.fqbn
+	if fqbn == "" {
+		sketch, err := sketches.GetSketch(sketchName)
+		if err == nil && sketch != nil {
+			fqbn = sketch.Metadata.CPU.Fqbn
+		}
+	}
+	if fqbn == "" {
+		formatter.PrintErrorMessage("No Fully Qualified Board Name provided.")
+		os.Exit(commands.ErrBadCall)
+	}
+	fqbnParts := strings.Split(fqbn, ":")
+	if len(fqbnParts) < 3 || len(fqbnParts) > 4 {
+		formatter.PrintErrorMessage("Fully Qualified Board Name has incorrect format.")
+		os.Exit(commands.ErrBadCall)
+	}
+	packageName := fqbnParts[0]
+	coreName := fqbnParts[1]
+	boardName := fqbnParts[2]
+
+	pm := commands.InitPackageManager()
+	if err := pm.LoadHardware(); err != nil {
+		fmt.Printf("Error loading hardware: %s", err)
+		os.Exit(commands.ErrCoreConfig)
+	}
+
+	// Find target board
+	// TODO: Make a packagemanager function to do this
+	targetPackage := pm.GetPackages().Packages[packageName]
+	if targetPackage == nil {
+		formatter.PrintErrorMessage("Unknown package " + packageName + ".")
+		os.Exit(commands.ErrBadCall)
+	}
+	platform := targetPackage.Platforms[coreName]
+	if platform == nil {
+		formatter.PrintErrorMessage("Unknown platform " + packageName + ":" + coreName + ".")
+		os.Exit(commands.ErrBadCall)
+	}
+	platformRelease := platform.GetInstalled()
+	if platformRelease == nil {
+		formatter.PrintErrorMessage("Platform " + packageName + ":" + coreName + " is not installed.")
+		os.Exit(commands.ErrBadCall)
+	}
+	board := platformRelease.Boards[boardName]
+	if board == nil {
+		formatter.PrintErrorMessage("Unknown board " + packageName + ":" + coreName + ":" + boardName + ".")
+		os.Exit(commands.ErrBadCall)
+	}
+
+	// Create board configuration
+	var boardProperties properties.Map
+	if len(fqbnParts) == 3 {
+		boardProperties = board.Properties
+	} else {
+		if props, err := board.GeneratePropertiesForConfiguration(fqbnParts[3]); err != nil {
+			formatter.PrintError(err, "Invalid FQBN.")
+			os.Exit(commands.ErrBadCall)
+		} else {
+			boardProperties = props
+		}
+	}
+
+	// Load programmer tool
+	var referencedPackage *cores.Package
+	var referencedPlatform *cores.Platform
+	var referencedPlatformRelease *cores.PlatformRelease
+	uploadToolID, have := boardProperties["upload.tool"]
+	if !have || uploadToolID == "" {
+		formatter.PrintErrorMessage("The board defines an invalid 'upload.tool': " + uploadToolID)
+		os.Exit(commands.ErrGeneric)
+	}
+	var uploadTool *cores.Tool
+	if split := strings.Split(uploadToolID, ":"); len(split) == 1 {
+		uploadTool = targetPackage.Tools[uploadToolID]
+	} else if len(split) == 2 {
+		referencedPackage = pm.GetPackages().Packages[split[0]]
+		if referencedPackage == nil {
+			formatter.PrintErrorMessage("The board requires a tool from package '" + split[0] + "' that is not installed: " + uploadToolID)
+			os.Exit(commands.ErrGeneric)
+		}
+		uploadTool = referencedPackage.Tools[split[1]]
+
+		referencedPlatform = referencedPackage.Platforms[coreName]
+		if referencedPlatform != nil {
+			referencedPlatformRelease = referencedPlatform.GetInstalled()
+		}
+	} else {
+		formatter.PrintErrorMessage("The board defines an invalid 'upload.tool': " + uploadToolID)
+		os.Exit(commands.ErrGeneric)
+	}
+	if uploadTool == nil {
+		formatter.PrintErrorMessage("Upload tool '" + uploadToolID + "' not found.")
+		os.Exit(commands.ErrGeneric)
+	}
+	// FIXME: Look into index if the platform requires a specific version
+	uploadToolInstance := uploadTool.GetLatestInstalled()
+	if uploadToolInstance == nil {
+		formatter.PrintErrorMessage("Upload tool '" + uploadToolID + "' not installed.")
+		os.Exit(commands.ErrGeneric)
+	}
+	fmt.Println(boardProperties.Dump())
+	fmt.Println(uploadToolInstance)
+
+}