Refactor port in address and protocol

Author: polldo

arduino/cli/commander.go

@@ -78,13 +78,13 @@ func (c *commander) BoardList() ([]*rpc.DetectedPort, error) {
 
 // UploadBin executes the 'arduino-cli upload -i' command
 // and returns its result.
-func (c *commander) UploadBin(fqbn, bin, port string) error {
+func (c *commander) UploadBin(fqbn, bin, address, protocol string) error {
 	req := &rpc.UploadRequest{
 		Instance:   c.Instance,
 		Fqbn:       fqbn,
 		SketchPath: filepath.Dir(bin),
 		ImportFile: bin,
-		Port:       &rpc.Port{Address: port},
+		Port:       &rpc.Port{Address: address, Protocol: protocol},
 		Verbose:    false,
 	}
 

arduino/commander.go

@@ -25,6 +25,6 @@ import (
 // the arduino-cli commands in a programmatic way.
 type Commander interface {
 	BoardList() ([]*rpc.DetectedPort, error)
-	UploadBin(fqbn, bin, port string) error
+	UploadBin(fqbn, bin, address, protocol string) error
 	//Compile() error
 }

arduino/grpc/compile.go

@@ -38,14 +38,14 @@ func (c compileHandler) Compile() error {
 
 // Upload executes the 'arduino-cli upload -i' command
 // and returns its result.
-func (c compileHandler) UploadBin(fqbn, bin, port string) error {
+func (c compileHandler) UploadBin(fqbn, bin, address, protocol string) error {
 	stream, err := c.serviceClient.Upload(context.Background(),
 		&rpc.UploadRequest{
 			Instance:   c.instance,
 			Fqbn:       fqbn,
 			SketchPath: filepath.Dir(bin),
 			ImportFile: bin,
-			Port:       &rpc.Port{Address: port},
+			Port:       &rpc.Port{Address: address, Protocol: protocol},
 			Verbose:    true,
 		})
 

command/device/board.go

@@ -41,10 +41,11 @@ var (
 
 // board contains details of a physical arduino board.
 type board struct {
-	fqbn   string
-	serial string
-	dType  string
-	port   string
+	fqbn     string
+	serial   string
+	dType    string
+	address  string
+	protocol string
 }
 
 // isCrypto checks if the board is a valid arduino board with a
@@ -78,10 +79,11 @@ func boardFromPorts(ports []*rpc.DetectedPort, params *CreateParams) *board {
 		boardFound := boardFilter(port.MatchingBoards, params)
 		if boardFound != nil {
 			b := &board{
-				fqbn:   boardFound.Fqbn,
-				serial: port.Port.Properties["serialNumber"],
-				dType:  strings.Split(boardFound.Fqbn, ":")[2],
-				port:   port.Port.Address,
+				fqbn:     boardFound.Fqbn,
+				serial:   port.Port.Properties["serialNumber"],
+				dType:    strings.Split(boardFound.Fqbn, ":")[2],
+				address:  port.Port.Address,
+				protocol: port.Port.Protocol,
 			}
 			return b
 		}

command/device/board_test.go

@@ -68,14 +68,14 @@ func TestBoardFromPorts(t *testing.T) {
 			name:   "port-filter",
 			filter: &CreateParams{FQBN: nil, Port: stringPointer("ACM1")},
 			ports:  portsTwoBoards,
-			want:   &board{fqbn: "arduino:avr:uno", port: "ACM1"},
+			want:   &board{fqbn: "arduino:avr:uno", address: "ACM1"},
 		},
 
 		{
 			name:   "fqbn-filter",
 			filter: &CreateParams{FQBN: stringPointer("arduino:avr:uno"), Port: nil},
 			ports:  portsTwoBoards,
-			want:   &board{fqbn: "arduino:avr:uno", port: "ACM1"},
+			want:   &board{fqbn: "arduino:avr:uno", address: "ACM1"},
 		},
 
 		{
@@ -90,7 +90,7 @@ func TestBoardFromPorts(t *testing.T) {
 			filter: &CreateParams{FQBN: nil, Port: nil},
 			ports:  portsTwoBoards,
 			// first board found is selected
-			want: &board{fqbn: "arduino:samd:nano_33_iot", port: "ACM0"},
+			want: &board{fqbn: "arduino:samd:nano_33_iot", address: "ACM0"},
 		},
 
 		{
@@ -104,7 +104,7 @@ func TestBoardFromPorts(t *testing.T) {
 			name:   "both-filter-found",
 			filter: &CreateParams{FQBN: stringPointer("arduino:avr:uno"), Port: stringPointer("ACM1")},
 			ports:  portsTwoBoards,
-			want:   &board{fqbn: "arduino:avr:uno", port: "ACM1"},
+			want:   &board{fqbn: "arduino:avr:uno", address: "ACM1"},
 		},
 
 		{
@@ -123,14 +123,14 @@ func TestBoardFromPorts(t *testing.T) {
 				return
 
 			} else if got != nil && tt.want == nil {
-				t.Errorf("Expected nil board, received not nil board with port %s and fqbn %s", got.port, got.fqbn)
+				t.Errorf("Expected nil board, received not nil board with port %s and fqbn %s", got.address, got.fqbn)
 
 			} else if got == nil && tt.want != nil {
-				t.Errorf("Expected not nil board with port %s and fqbn %s, received a nil board", tt.want.port, tt.want.fqbn)
+				t.Errorf("Expected not nil board with port %s and fqbn %s, received a nil board", tt.want.address, tt.want.fqbn)
 
-			} else if got.port != tt.want.port || got.fqbn != tt.want.fqbn {
+			} else if got.address != tt.want.address || got.fqbn != tt.want.fqbn {
 				t.Errorf("Expected board with port %s and fqbn %s, received board with port %s and fqbn %s",
-					tt.want.port, tt.want.fqbn, got.port, got.fqbn)
+					tt.want.address, tt.want.fqbn, got.address, got.fqbn)
 			}
 		})
 	}

command/device/create.go

@@ -59,7 +59,7 @@ func Create(params *CreateParams) (*DeviceInfo, error) {
 				"Try the 'create-lora' command instead if it's a LoRa device"+
 				" or 'create-generic' otherwise",
 			board.fqbn,
-			board.port,
+			board.address,
 		)
 	}
 

command/device/createlora.go

@@ -84,7 +84,7 @@ func CreateLora(params *CreateLoraParams) (*DeviceLoraInfo, error) {
 				" Try the 'create' command instead if it's a device with a supported crypto-chip"+
 				" or 'create-generic' otherwise",
 			board.fqbn,
-			board.port,
+			board.address,
 		)
 	}
 
@@ -96,13 +96,13 @@ func CreateLora(params *CreateLoraParams) (*DeviceLoraInfo, error) {
 	logrus.Infof("%s", "Uploading deveui sketch on the LoRa board")
 	errMsg := "Error while uploading the LoRa provisioning binary"
 	err = retry(deveuiUploadAttempts, deveuiUploadWait*time.Millisecond, errMsg, func() error {
-		return comm.UploadBin(board.fqbn, bin, board.port)
+		return comm.UploadBin(board.fqbn, bin, board.address, board.protocol)
 	})
 	if err != nil {
 		return nil, fmt.Errorf("failed to upload LoRa provisioning binary: %w", err)
 	}
 
-	eui, err := extractEUI(board.port)
+	eui, err := extractEUI(board.address)
 	if err != nil {
 		return nil, err
 	}

command/device/provision.go

@@ -86,7 +86,8 @@ func (p provision) run() error {
 	errMsg := "Error while uploading the provisioning sketch"
 	err = retry(5, time.Millisecond*1000, errMsg, func() error {
 		//serialutils.Reset(dev.port, true, nil)
-		return p.UploadBin(p.board.fqbn, bin, p.board.port)
+		fmt.Println(p.board.fqbn, bin, p.board.address)
+		return p.UploadBin(p.board.fqbn, bin, p.board.address, p.board.protocol)
 	})
 	if err != nil {
 		return err
@@ -98,7 +99,7 @@ func (p provision) run() error {
 	p.ser = serial.NewSerial()
 	errMsg = "Error while connecting to the board"
 	err = retry(5, time.Millisecond*1000, errMsg, func() error {
-		return p.ser.Connect(p.board.port)
+		return p.ser.Connect(p.board.address)
 	})
 	if err != nil {
 		return err