dify-plugin-daemon/internal/core/plugin_manager/remote_manager/run.go

package remote_manager

import (
	"time"

	"github.com/langgenius/dify-plugin-daemon/internal/core/plugin_manager/plugin_errors"
	"github.com/langgenius/dify-plugin-daemon/internal/types/entities/plugin_entities"
	"github.com/langgenius/dify-plugin-daemon/internal/utils/log"
	"github.com/langgenius/dify-plugin-daemon/internal/utils/parser"
	"github.com/langgenius/dify-plugin-daemon/internal/utils/routine"
)

func (r *RemotePluginRuntime) InitEnvironment() error {
	return nil
}

func (r *RemotePluginRuntime) Stopped() bool {
	return !r.alive
}

func (r *RemotePluginRuntime) Stop() {
	r.alive = false
	if r.conn == nil {
		return
	}
	r.conn.Close()
}

func (r *RemotePluginRuntime) StartPlugin() error {
	var exit_error error

	// handle heartbeat
	routine.Submit(func() {
		r.last_active_at = time.Now()
		ticker := time.NewTicker(5 * time.Second)
		defer ticker.Stop()

		for {
			select {
			case <-ticker.C:
				if time.Since(r.last_active_at) > 20*time.Second {
					// kill this connection
					r.conn.Close()
					exit_error = plugin_errors.ErrPluginNotActive
					return
				}
			case <-r.shutdown_chan:
				return
			}
		}
	})

	for r.response.Next() {
		data, err := r.response.Read()
		if err != nil {
			return err
		}

		// handle event
		event, err := parser.UnmarshalJsonBytes[plugin_entities.PluginUniversalEvent](data)
		if err != nil {
			continue
		}

		session_id := event.SessionId

		switch event.Event {
		case plugin_entities.PLUGIN_EVENT_LOG:
			if event.Event == plugin_entities.PLUGIN_EVENT_LOG {
				log_event, err := parser.UnmarshalJsonBytes[plugin_entities.PluginLogEvent](
					event.Data,
				)
				if err != nil {
					log.Error("unmarshal json failed: %s", err.Error())
					continue
				}

				log.Info("plugin %s: %s", r.Configuration().Identity(), log_event.Message)
			}
		case plugin_entities.PLUGIN_EVENT_SESSION:
			r.callbacks_lock.RLock()
			listeners := r.callbacks[session_id][:]
			r.callbacks_lock.RUnlock()

			// handle session event
			for _, listener := range listeners {
				listener(event.Data)
			}
		case plugin_entities.PLUGIN_EVENT_ERROR:
			log.Error("plugin %s: %s", r.Configuration().Identity(), event.Data)
		case plugin_entities.PLUGIN_EVENT_HEARTBEAT:
			r.last_active_at = time.Now()
		}
	}

	return exit_error
}

func (r *RemotePluginRuntime) Wait() (<-chan bool, error) {
	return r.shutdown_chan, nil
}