插件开发

package main

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

// 定义插件的配置结构体
type Config struct {
    Name string `json:"name" label:"AI名字"` // json为配置项名称，label为WuKongIM后台显示的配置项名称
}

// 插件结构体
type AIExample struct {
    Config Config // 插件的配置，名字必须为Config
}

// 创建一个插件对象
func New() interface{} {
    return &AIExample{}
}

// 收到消息时调用（用户插件）
func (a *AIExample) Receive(c *pdk.Context) {
    // 获取发送消息内容
    text := a.getText(c)

    // 编码消息
    data, _ := json.Marshal(map[string]interface{}{
        "type":    1,
        "content": fmt.Sprintf("我是%s,收到您的消息：%s", a.Config.Name, text),
    })

    // 回复消息
    c.Reply(data)
}

// 辅助函数：获取消息文本内容
func (a *AIExample) getText(c *pdk.Context) string {
    var payload map[string]interface{}
    json.Unmarshal(c.GetPayload(), &payload)
    
    if content, ok := payload["content"].(string); ok {
        return content
    }
    return ""
}

func main() {
    pdk.RunServer(New, "wk.plugin.ai-example", 
        pdk.WithVersion("0.0.1"), 
        pdk.WithPriority(1))
}

package main

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

type Config struct {
    Name   string `json:"name" label:"AI名字"`
    ApiKey string `json:"api_key" label:"API密钥"`
}

type AIPlugin struct {
    Config Config
}

func New() interface{} {
    return &AIPlugin{}
}

func (a *AIPlugin) Setup() {
    fmt.Printf("AI插件 %s 启动成功\n", a.Config.Name)
}

func (a *AIPlugin) Receive(c *pdk.Context) {
    text := a.getText(c)

    // 调用AI服务获取回复
    reply := a.getAIReply(text)

    // 构造回复消息
    data, _ := json.Marshal(map[string]interface{}{
        "type":    1,
        "content": reply,
    })

    c.Reply(data)
}

func (a *AIPlugin) getText(c *pdk.Context) string {
    var payload map[string]interface{}
    json.Unmarshal(c.GetPayload(), &payload)

    if content, ok := payload["content"].(string); ok {
        return content
    }
    return ""
}

func (a *AIPlugin) getAIReply(text string) string {
    // 这里可以调用实际的AI服务
    // 示例：简单的回复逻辑
    if strings.Contains(text, "你好") {
        return fmt.Sprintf("你好！我是%s，很高兴为您服务！", a.Config.Name)
    }
    return fmt.Sprintf("我是%s，收到您的消息：%s", a.Config.Name, text)
}

func (a *AIPlugin) Route(c *pdk.Route) {
    c.GET("/status", a.getStatus)
}

func (a *AIPlugin) getStatus(c *pdk.Context) {
    c.JSON(200, map[string]interface{}{
        "name":   a.Config.Name,
        "status": "running",
    })
}

func (a *AIPlugin) ConfigUpdate() {
    fmt.Printf("配置已更新: %+v\n", a.Config)
}

func (a *AIPlugin) Stop() {
    fmt.Printf("AI插件 %s 停止\n", a.Config.Name)
}

func main() {
    pdk.RunServer(New, "wk.plugin.ai-chat",
        pdk.WithVersion("1.0.0"),
        pdk.WithPriority(1))
}

package main

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

type Config struct {
    Words []string `json:"words" label:"敏感词列表"`
}

type FilterPlugin struct {
    Config Config
}

func New() interface{} {
    return &FilterPlugin{}
}

func (f *FilterPlugin) Send(c *pdk.Context) {
    var payload map[string]interface{}
    json.Unmarshal(c.GetPayload(), &payload)

    if content, ok := payload["content"].(string); ok {
        // 检查敏感词
        filtered := f.filterContent(content)
        if filtered != content {
            payload["content"] = filtered
            newPayload, _ := json.Marshal(payload)
            c.SetPayload(newPayload)
        }
    }
}

func (f *FilterPlugin) filterContent(content string) string {
    for _, word := range f.Config.Words {
        if strings.Contains(content, word) {
            content = strings.ReplaceAll(content, word, "***")
        }
    }
    return content
}

func main() {
    pdk.RunServer(New, "wk.plugin.filter",
        pdk.WithVersion("1.0.0"),
        pdk.WithPriority(10))
}