feat: implement flash functionality

internal/flash/flash.go

@@ -0,0 +1,278 @@
+package flash
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/goplus/llgo/internal/crosscompile"
+	"github.com/goplus/llgo/internal/env"
+)
+
+// Flash flashes firmware to a device based on the crosscompile configuration
+func Flash(crossCompile crosscompile.Export, app string, port string, verbose bool) error {
+	if verbose {
+		fmt.Fprintf(os.Stderr, "Flashing %s to port %s\n", app, port)
+	}
+
+	method := crossCompile.Flash.Method
+	if method == "" {
+		method = "command"
+	}
+
+	if verbose {
+		fmt.Fprintf(os.Stderr, "Flashing %s using method: %s\n", app, method)
+	}
+
+	switch method {
+	case "command":
+		return flashCommand(crossCompile.Flash, app, port, verbose)
+	case "openocd":
+		return flashOpenOCD(crossCompile.OpenOCD, app, verbose)
+	case "msd":
+		return flashMSD(crossCompile.MSD, app, verbose)
+	case "bmp":
+		return flashBMP(app, verbose)
+	default:
+		return fmt.Errorf("unsupported flash method: %s", method)
+	}
+}
+
+// flashCommand handles command-based flashing
+func flashCommand(flash crosscompile.Flash, app string, port string, verbose bool) error {
+	if flash.Command == "" {
+		return fmt.Errorf("flash command not specified")
+	}
+
+	// Build environment map for template variable expansion
+	envs := buildFlashEnvMap(app, port)
+
+	// Expand template variables in command
+	expandedCommand := expandEnv(flash.Command, envs)
+
+	if verbose {
+		fmt.Fprintf(os.Stderr, "Flash command: %s\n", expandedCommand)
+	}
+
+	// Split command into parts for exec
+	parts := strings.Fields(expandedCommand)
+	if len(parts) == 0 {
+		return fmt.Errorf("empty flash command after expansion")
+	}
+
+	// Handle 1200bps reset if required
+	if flash.Flash1200BpsReset && port != "" {
+		if err := reset1200bps(port, verbose); err != nil {
+			if verbose {
+				fmt.Fprintf(os.Stderr, "Warning: 1200bps reset failed: %v\n", err)
+			}
+		}
+		// Wait for bootloader
+		time.Sleep(2 * time.Second)
+	}
+
+	// Execute flash command
+	cmd := exec.Command(parts[0], parts[1:]...)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	return cmd.Run()
+}
+
+// flashOpenOCD handles OpenOCD-based flashing
+func flashOpenOCD(openocd crosscompile.OpenOCD, app string, verbose bool) error {
+	if openocd.Interface == "" {
+		return fmt.Errorf("OpenOCD interface not specified")
+	}
+
+	args := []string{
+		"-f", "interface/" + openocd.Interface + ".cfg",
+	}
+
+	if openocd.Transport != "" {
+		args = append(args, "-c", "transport select "+openocd.Transport)
+	}
+
+	if openocd.Target != "" {
+		args = append(args, "-f", "target/"+openocd.Target+".cfg")
+	}
+
+	// Add programming commands
+	args = append(args,
+		"-c", "init",
+		"-c", "reset init",
+		"-c", fmt.Sprintf("flash write_image erase %s", app),
+		"-c", "reset",
+		"-c", "shutdown",
+	)
+
+	if verbose {
+		fmt.Fprintf(os.Stderr, "OpenOCD command: openocd %s\n", strings.Join(args, " "))
+	}
+
+	cmd := exec.Command("openocd", args...)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	return cmd.Run()
+}
+
+// flashMSD handles Mass Storage Device flashing
+func flashMSD(msd crosscompile.MSD, app string, verbose bool) error {
+	if len(msd.VolumeName) == 0 {
+		return fmt.Errorf("MSD volume names not specified")
+	}
+
+	if msd.FirmwareName == "" {
+		return fmt.Errorf("MSD firmware name not specified")
+	}
+
+	// Find the MSD volume
+	var mountPoint string
+	for _, volumeName := range msd.VolumeName {
+		// Try common mount points on different platforms
+		candidates := []string{
+			filepath.Join("/Volumes", volumeName),                  // macOS
+			filepath.Join("/media", os.Getenv("USER"), volumeName), // Linux
+			filepath.Join("/mnt", volumeName),                      // Linux alternative
+			volumeName + ":",                                       // Windows (assuming drive letter)
+		}
+
+		for _, candidate := range candidates {
+			if _, err := os.Stat(candidate); err == nil {
+				mountPoint = candidate
+				break
+			}
+		}
+		if mountPoint != "" {
+			break
+		}
+	}
+
+	if mountPoint == "" {
+		return fmt.Errorf("MSD volume not found. Expected volumes: %v", msd.VolumeName)
+	}
+
+	// Copy firmware to MSD
+	destPath := filepath.Join(mountPoint, msd.FirmwareName)
+
+	if verbose {
+		fmt.Fprintf(os.Stderr, "Copying %s to %s\n", app, destPath)
+	}
+
+	return copyFile(app, destPath)
+}
+
+// flashBMP handles Black Magic Probe flashing
+func flashBMP(app string, verbose bool) error {
+	// BMP typically uses GDB for flashing
+	args := []string{
+		"-ex", "target extended-remote /dev/ttyACM0", // Default BMP port
+		"-ex", "monitor swdp_scan",
+		"-ex", "attach 1",
+		"-ex", "load",
+		"-ex", "compare-sections",
+		"-ex", "kill",
+		"-ex", "quit",
+		app,
+	}
+
+	if verbose {
+		fmt.Fprintf(os.Stderr, "BMP command: arm-none-eabi-gdb %s\n", strings.Join(args, " "))
+	}
+
+	cmd := exec.Command("arm-none-eabi-gdb", args...)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	return cmd.Run()
+}
+
+// buildFlashEnvMap creates environment map for template expansion
+func buildFlashEnvMap(app string, port string) map[string]string {
+	envs := make(map[string]string)
+
+	// Basic paths
+	envs["root"] = env.LLGoROOT()
+	envs["tmpDir"] = os.TempDir()
+
+	// Port information
+	if port != "" {
+		envs["port"] = port
+	}
+
+	// File paths based on extension
+	ext := strings.ToLower(filepath.Ext(app))
+	switch ext {
+	case ".hex":
+		envs["hex"] = app
+	case ".bin":
+		envs["bin"] = app
+	case ".elf":
+		envs["elf"] = app
+	case ".uf2":
+		envs["uf2"] = app
+	case ".zip":
+		envs["zip"] = app
+	case ".img":
+		envs["img"] = app
+	default:
+		// Default to binary for unknown extensions
+		envs["bin"] = app
+	}
+
+	return envs
+}
+
+// expandEnv expands template variables in a string
+// Supports variables like {port}, {hex}, {bin}, {root}, {tmpDir}, etc.
+func expandEnv(template string, envs map[string]string) string {
+	if template == "" {
+		return ""
+	}
+
+	result := template
+
+	// Replace named variables
+	for key, value := range envs {
+		if key != "" {
+			result = strings.ReplaceAll(result, "{"+key+"}", value)
+		}
+	}
+
+	return result
+}
+
+// reset1200bps performs 1200bps reset for Arduino-compatible boards
+func reset1200bps(port string, verbose bool) error {
+	if verbose {
+		fmt.Fprintf(os.Stderr, "Performing 1200bps reset on %s\n", port)
+	}
+
+	// This is a simplified implementation
+	// In practice, this would need platform-specific serial port handling
+	// For now, just try to touch the port to trigger reset
+	_, err := os.Stat(port)
+	return err
+}
+
+// copyFile copies a file from src to dst
+func copyFile(src, dst string) error {
+	sourceFile, err := os.Open(src)
+	if err != nil {
+		return err
+	}
+	defer sourceFile.Close()
+
+	destFile, err := os.Create(dst)
+	if err != nil {
+		return err
+	}
+	defer destFile.Close()
+
+	_, err = destFile.ReadFrom(sourceFile)
+	return err
+}