63 lines
1.9 KiB
Go
63 lines
1.9 KiB
Go
package mcpserver
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
|
|
"isphere-ai-bridge/internal/helperclient"
|
|
"isphere-ai-bridge/internal/isphere"
|
|
"isphere-ai-bridge/internal/tools"
|
|
)
|
|
|
|
const (
|
|
ServerName = "isphere-ai-bridge"
|
|
ServerTitle = "iSphere AI Bridge"
|
|
ServerVersion = "0.1.0"
|
|
EnvPacketLogFileKey = "ISPHERE_PACKET_LOG_FILE"
|
|
EnvPacketLogDirKey = "ISPHERE_PACKET_LOG_DIR"
|
|
)
|
|
|
|
func NewServer() *mcp.Server {
|
|
return NewServerWithReceiveSource(isphere.EncryptedPacketLogSource{})
|
|
}
|
|
|
|
func NewServerFromEnv() (*mcp.Server, error) {
|
|
var source tools.ReceiveMessagesSource = isphere.EncryptedPacketLogSource{}
|
|
packetLogFile := strings.TrimSpace(os.Getenv(EnvPacketLogFileKey))
|
|
packetLogDir := strings.TrimSpace(os.Getenv(EnvPacketLogDirKey))
|
|
if packetLogFile != "" && packetLogDir != "" {
|
|
return nil, fmt.Errorf("configure packet log source: set only one of %s or %s", EnvPacketLogFileKey, EnvPacketLogDirKey)
|
|
}
|
|
if packetLogFile != "" {
|
|
loaded, err := isphere.LoadEncryptedPacketLogSourceFromFile(packetLogFile)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("configure %s: %w", EnvPacketLogFileKey, err)
|
|
}
|
|
source = loaded
|
|
} else if packetLogDir != "" {
|
|
loaded, err := isphere.LoadEncryptedPacketLogSourceFromDirectory(packetLogDir)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("configure %s: %w", EnvPacketLogDirKey, err)
|
|
}
|
|
source = loaded
|
|
}
|
|
return NewServerWithReceiveSource(source), nil
|
|
}
|
|
|
|
func NewServerWithReceiveSource(source tools.ReceiveMessagesSource) *mcp.Server {
|
|
server := mcp.NewServer(&mcp.Implementation{
|
|
Name: ServerName,
|
|
Title: ServerTitle,
|
|
Version: ServerVersion,
|
|
}, nil)
|
|
tools.RegisterWinHelperTools(server, helperclient.Client{})
|
|
tools.RegisterISphereReadTools(server, source)
|
|
tools.RegisterISphereContactTools(server, source)
|
|
tools.RegisterISphereGroupTools(server, source)
|
|
tools.RegisterISphereFileTools(server, source)
|
|
return server
|
|
}
|