main.go

// SPDX-License-Identifier: MPL-2.0
// Copyright (C) 2025 Arduino SA

package main

import (
	"context"
	"fmt"
	"io"
	"net/http"
	"os"
	"os/exec"
	"path/filepath"
	"strings"

	"github.com/codeclysm/extract/v4"
	cp "github.com/otiai10/copy"
)

func downloadFile(filepath string, url string) (err error) {

	// Create the file
	out, err := os.Create(filepath)
	if err != nil {
		return err
	}
	defer out.Close()

	// Get the data
	resp, err := http.Get(url)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	// Check server response
	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("bad status: %s", resp.Status)
	}

	// Writer the body to file
	_, err = io.Copy(out, resp.Body)
	if err != nil {
		return err
	}

	return nil
}

func main() {

	// Create a temporary folder, download a URL based on a git tag in it
	// and extract the file to the temporary folder
	tmpDir, err := os.MkdirTemp("", "sync-zephyr-artifacts")
	if err != nil {
		fmt.Println("Error creating temp dir:", err)
		return
	}
	defer os.RemoveAll(tmpDir) // Clean up

	// Download the file from http://downloads.arduino.cc/cores/zephyr/ArduinoCore-zephyr-{git_tag}.zip
	gitCorePath := os.Args[1]
	// Force an hash, for debug only
	forceHash := ""
	if len(os.Args) > 2 {
		forceHash = os.Args[2]
	}

	cmd := exec.Command("git", "ls-files", "--exclude-standard", "-dmo", ".")
	stdout, err := cmd.Output()
	if err != nil {
		fmt.Println("Error executing command:", err)
		return
	}

	var lines []string
	var changes []string
	lines = strings.Split(string(stdout), "\n")
	for _, path := range lines {
		if strings.HasPrefix(path, "firmwares/") || strings.HasPrefix(path, "variants/") {
			changes = append(changes, path)
		}
	}

	if len(changes) > 0 {
		fmt.Println("The git repository contains uncommitted files:")
		for _, path := range changes {
			fmt.Println("- ", path)
		}
		fmt.Println("Please commit or stash them before running this script.")
		return
	}

	cmd = exec.Command("git", "describe", "--always", "--abbrev=7")
	stdout, err = cmd.Output()
	if err != nil {
		fmt.Println("Error executing command:", err)
		return
	}

	hash := strings.TrimSpace(string(stdout))
	fmt.Println("Git SHA:", hash)
	if forceHash != "" {
		hash = forceHash
	}

	// Compose download URL from git hash
	filename := fmt.Sprintf("ArduinoCore-zephyr-%s.tar.bz2", hash)
	url := fmt.Sprintf("http://downloads.arduino.cc/cores/zephyr/%s", filename)
	fmt.Println("Download URL:", url)
	// Download the zip file from the URL
	zipFilePath := filepath.Join(tmpDir, filename)
	err = downloadFile(zipFilePath, url)
	if err != nil {
		fmt.Println("Error downloading archive:", err)
		return
	}
	fmt.Println("Downloaded archive to:", zipFilePath)
	// Extract the tar.bz2 file to the temporary folder
	// Use packer/tar and compress/bzip2 to extract the file
	file, err := os.Open(filepath.Join(tmpDir, filename))
	if err != nil {
		fmt.Println("Error opening archive:", err)
		return
	}
	defer file.Close()

	err = extract.Bz2(context.Background(), file, filepath.Join(tmpDir, "extract"), nil)
	if err != nil {
		fmt.Println("Error extracting archive:", err)
		return
	}

	// Remove old firmwares and variants/*/llext-edk files
	os.RemoveAll(filepath.Join(gitCorePath, "firmwares"))
	filepath.WalkDir(filepath.Join(gitCorePath, "variants"), func(path string, d os.DirEntry, err error) error {
		if err != nil {
			fmt.Println("Error:", err)
			return nil
		}

		if d.IsDir() && d.Name() == "llext-edk" {
			os.RemoveAll(path)
		}
		return nil
	})

	// Copy the content of firmware folder to gitCorePath/firmware
	err = cp.Copy(filepath.Join(tmpDir, "extract", "ArduinoCore-zephyr", "firmwares"), filepath.Join(gitCorePath, "firmwares"))
	if err != nil {
		fmt.Println("Error copying firmware folder:", err)
		return
	}
	// Copy the content of variants folder to gitCorePath/variants
	err = cp.Copy(filepath.Join(tmpDir, "extract", "ArduinoCore-zephyr", "variants"), filepath.Join(gitCorePath, "variants"))
	if err != nil {
		fmt.Println("Error copying variants folder:", err)
		return
	}
}