dify-plugin-daemon/cmd/commandline/bundle/init.go

package bundle

import (
	"bytes"
	_ "embed"
	"errors"
	"os"
	"path"
	"text/template"
	"time"

	"github.com/langgenius/dify-plugin-daemon/internal/utils/log"
	"github.com/langgenius/dify-plugin-daemon/pkg/entities/bundle_entities"
	"github.com/langgenius/dify-plugin-daemon/pkg/entities/manifest_entities"
	"github.com/langgenius/dify-plugin-daemon/pkg/entities/plugin_entities"

	tea "github.com/charmbracelet/bubbletea"
)

//go:embed templates/icon.svg
var BUNDLE_ICON []byte

//go:embed templates/README.md
var BUNDLE_README []byte

func generateNewBundle() (*bundle_entities.Bundle, error) {
	m := newProfile()
	p := tea.NewProgram(m)
	if result, err := p.Run(); err != nil {
		return nil, err
	} else {
		if _, ok := result.(profile); ok {
			author := m.inputs[1].Value()
			name := m.inputs[0].Value()
			description := m.inputs[2].Value()

			bundle := &bundle_entities.Bundle{
				Name:         name,
				Icon:         "icon.svg",
				Labels:       plugin_entities.NewI18nObject(name),
				Description:  plugin_entities.NewI18nObject(description),
				Version:      "0.0.1",
				Author:       author,
				Type:         manifest_entities.BundleType,
				Dependencies: []bundle_entities.Dependency{},
			}

			return bundle, nil
		} else {
			return nil, errors.New("invalid profile")
		}
	}
}

func InitBundle() {
	bundle, err := generateNewBundle()
	if err != nil {
		log.Error("Failed to generate new bundle: %v", err)
		return
	}

	// create bundle directory
	cwd, err := os.Getwd()
	if err != nil {
		log.Error("Error getting current directory: %v", err)
		return
	}

	bundleDir := path.Join(cwd, bundle.Name)
	if err := os.MkdirAll(bundleDir, 0755); err != nil {
		log.Error("Error creating bundle directory: %v", err)
		return
	}

	success := false
	defer func() {
		if !success {
			os.RemoveAll(bundleDir)
		}
	}()

	// save
	bundleYaml := marshalYamlBytes(bundle)
	if err := os.WriteFile(path.Join(bundleDir, "manifest.yaml"), bundleYaml, 0644); err != nil {
		log.Error("Error saving manifest.yaml: %v", err)
		return
	}

	// create README.md
	tmpl := template.Must(template.New("README").Parse(string(BUNDLE_README)))
	// render the template
	var buf bytes.Buffer
	if err := tmpl.Execute(&buf, map[string]interface{}{
		"Author":  bundle.Author,
		"Version": bundle.Version,
		"Date":    time.Now().Format(time.DateOnly),
	}); err != nil {
		log.Error("Error rendering README template: %v", err)
		return
	}

	// save README.md
	if err := os.WriteFile(path.Join(bundleDir, "README.md"), buf.Bytes(), 0644); err != nil {
		log.Error("Error saving README.md: %v", err)
		return
	}

	// create _assets directory
	if err := os.MkdirAll(path.Join(bundleDir, "_assets"), 0755); err != nil {
		log.Error("Error creating _assets directory: %v", err)
		return
	}

	// create _assets/icon.svg
	if err := os.WriteFile(path.Join(bundleDir, "_assets", "icon.svg"), BUNDLE_ICON, 0644); err != nil {
		log.Error("Error saving icon.svg: %v", err)
		return
	}

	success = true

	log.Info("Bundle created successfully: %s", bundleDir)
}