Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 0 additions & 6 deletions cmd/picoclaw/internal/onboard/command.go
Original file line number Diff line number Diff line change
@@ -1,15 +1,9 @@
package onboard

import (
"embed"

"github.com/spf13/cobra"
)

//go:generate cp -r ../../../../workspace .
//go:embed workspace
var embeddedFiles embed.FS

func NewOnboardCommand() *cobra.Command {
cmd := &cobra.Command{
Use: "onboard",
Expand Down
9 changes: 5 additions & 4 deletions cmd/picoclaw/internal/onboard/helpers.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import (
"os"
"path/filepath"

picoclaw "github.com/sipeed/picoclaw"
"github.com/sipeed/picoclaw/cmd/picoclaw/internal"
"github.com/sipeed/picoclaw/pkg/config"
)
Expand Down Expand Up @@ -47,20 +48,20 @@ func onboard() {
}

func createWorkspaceTemplates(workspace string) {
err := copyEmbeddedToTarget(workspace)
err := copyEmbeddedToTarget(picoclaw.EmbeddedWorkspace, workspace)
if err != nil {
fmt.Printf("Error copying workspace templates: %v\n", err)
}
}

func copyEmbeddedToTarget(targetDir string) error {
func copyEmbeddedToTarget(source fs.FS, targetDir string) error {
// Ensure target directory exists
if err := os.MkdirAll(targetDir, 0o755); err != nil {
return fmt.Errorf("Failed to create target directory: %w", err)
}

// Walk through all files in embed.FS
err := fs.WalkDir(embeddedFiles, "workspace", func(path string, d fs.DirEntry, err error) error {
err := fs.WalkDir(source, "workspace", func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
Expand All @@ -71,7 +72,7 @@ func copyEmbeddedToTarget(targetDir string) error {
}

// Read embedded file
data, err := embeddedFiles.ReadFile(path)
data, err := fs.ReadFile(source, path)
if err != nil {
return fmt.Errorf("Failed to read embedded file %s: %w", path, err)
}
Expand Down
4 changes: 3 additions & 1 deletion cmd/picoclaw/internal/onboard/helpers_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,14 @@ import (
"os"
"path/filepath"
"testing"

picoclaw "github.com/sipeed/picoclaw"
)

func TestCopyEmbeddedToTargetUsesAgentsMarkdown(t *testing.T) {
targetDir := t.TempDir()

if err := copyEmbeddedToTarget(targetDir); err != nil {
if err := copyEmbeddedToTarget(picoclaw.EmbeddedWorkspace, targetDir); err != nil {
t.Fatalf("copyEmbeddedToTarget() error = %v", err)
}

Expand Down
8 changes: 8 additions & 0 deletions embedded_workspace.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
package picoclaw

import "embed"

// EmbeddedWorkspace bundles the default workspace templates used by `picoclaw onboard`.
//
//go:embed workspace
var EmbeddedWorkspace embed.FS