egutierrez
cef681ec87
Nuevo sistema de generacion automatica de avatares: - pkg/avatar/ — tipos puros y URL builders para proveedores gratuitos: DiceBear (bottts, pixel-art, etc.), RoboHash (robots, monsters), Multiavatar (multicultural). Sin I/O. - shell/avatar/ — fetcher impuro: descarga imagen por HTTP a temp file. - agentctl auto-avatar <id> — genera, descarga, sube y activa avatar con un solo comando. Soporta --provider, --style, --set, --dry-run. Respeta pure core / impure shell. El seed del avatar es el agent ID, asi cada bot obtiene una imagen unica y determinista. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/enmanuel/agents/pkg/avatar"
|
|
shellavatar "github.com/enmanuel/agents/shell/avatar"
|
|
shellmatrix "github.com/enmanuel/agents/shell/matrix"
|
|
)
|
|
|
|
func autoAvatarCmd() *cobra.Command {
|
|
var (
|
|
provider string
|
|
style string
|
|
set string
|
|
size int
|
|
dryRun bool
|
|
)
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "auto-avatar <agent-id>",
|
|
Short: "Generate and set a random avatar from a free provider",
|
|
Long: `Fetches a unique avatar image from a free provider (dicebear, robohash, multiavatar)
|
|
using the agent ID as seed, uploads it to the Matrix media repo, and sets it as the bot's avatar.
|
|
|
|
Examples:
|
|
agentctl auto-avatar assistant-bot
|
|
agentctl auto-avatar assistant-bot --provider robohash --set set1
|
|
agentctl auto-avatar assistant-bot --provider dicebear --style pixel-art
|
|
agentctl auto-avatar assistant-bot --dry-run # only show the URL`,
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
agentID := args[0]
|
|
|
|
opts := avatar.DefaultOptions()
|
|
if size > 0 {
|
|
opts.Size = size
|
|
}
|
|
if style != "" {
|
|
opts.DiceBearStyle = avatar.DiceBearStyle(style)
|
|
}
|
|
if set != "" {
|
|
opts.RoboHashSet = avatar.RoboHashSet(set)
|
|
}
|
|
|
|
p := avatar.Provider(provider)
|
|
imageURL := avatar.URL(p, agentID, opts)
|
|
|
|
if dryRun {
|
|
fmt.Printf("url %-20s %s\n", agentID, imageURL)
|
|
return nil
|
|
}
|
|
|
|
// Fetch image from provider
|
|
tmpPath, err := shellavatar.Fetch(context.Background(), p, agentID, opts)
|
|
if err != nil {
|
|
return fmt.Errorf("fetch avatar: %w", err)
|
|
}
|
|
defer os.Remove(tmpPath)
|
|
|
|
fmt.Printf("fetch %-20s %s\n", agentID, imageURL)
|
|
|
|
// Upload to Matrix and set as avatar
|
|
cfg, err := loadMatrixCfg(agentID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
client, err := shellmatrix.New(cfg.Matrix)
|
|
if err != nil {
|
|
return fmt.Errorf("matrix client: %w", err)
|
|
}
|
|
|
|
uri, err := client.SetAvatar(context.Background(), tmpPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("ok %-20s avatar → %s\n", agentID, uri)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
cmd.Flags().StringVar(&provider, "provider", "dicebear", "Avatar provider: dicebear, robohash, multiavatar")
|
|
cmd.Flags().StringVar(&style, "style", "", "DiceBear style: bottts, pixel-art, adventurer, shapes, fun-emoji, identicon, thumbs")
|
|
cmd.Flags().StringVar(&set, "set", "", "RoboHash set: set1 (robots), set2 (monsters), set3 (heads), set4 (cats), set5 (humans)")
|
|
cmd.Flags().IntVar(&size, "size", 256, "Image size in pixels (square)")
|
|
cmd.Flags().BoolVar(&dryRun, "dry-run", false, "Only print the image URL without fetching or uploading")
|
|
|
|
return cmd
|
|
}
|