Plugin Development

package main

import (
    "encoding/json"
    "fmt"
    "github.com/WuKongIM/go-pdk"
)

// Define plugin configuration struct
type Config struct {
    Name string `json:"name" label:"AI Name"` // json is config item name, label is display name in WuKongIM backend
}

// Plugin struct
type AIExample struct {
    Config Config // Plugin configuration, name must be Config
}

// Plugin initialization
func (a *AIExample) Init() error {
    pdk.Log.Info("AI plugin initialized", pdk.Any("config", a.Config))
    return nil
}

// Plugin information
func (a *AIExample) Info() pdk.PluginInfo {
    return pdk.PluginInfo{
        Uid:         "ai_example",           // Plugin unique ID
        Name:        "AI Example Plugin",    // Plugin name
        Description: "AI chat example plugin", // Plugin description
        Version:     "1.0.0",               // Plugin version
        Type:        pdk.PluginTypeUser,    // Plugin type: user plugin
    }
}

// Receive message processing
func (a *AIExample) Receive(message pdk.Message) error {
    // Parse message content
    var content map[string]interface{}
    if err := json.Unmarshal([]byte(message.Payload), &content); err != nil {
        return err
    }
    
    // Get message text
    text, ok := content["content"].(string)
    if !ok {
        return fmt.Errorf("invalid message content")
    }
    
    // AI processing logic (simplified example)
    response := fmt.Sprintf("AI %s replies: I received your message '%s'", a.Config.Name, text)
    
    // Send reply message
    replyContent := map[string]interface{}{
        "type":    "text",
        "content": response,
    }
    
    replyPayload, _ := json.Marshal(replyContent)
    
    return pdk.SendMessage(pdk.SendMessageReq{
        ChannelId:   message.ChannelId,
        ChannelType: message.ChannelType,
        Payload:     string(replyPayload),
    })
}

// Plugin entry point
func main() {
    plugin := &AIExample{}
    pdk.Run(plugin)
}