root.go•1.76 kB
package commands
import (
"context"
"os"
"os/signal"
"syscall"
"github.com/aquasecurity/trivy-mcp/pkg/flag"
"github.com/aquasecurity/trivy-mcp/pkg/mcpserver"
"github.com/aquasecurity/trivy-mcp/pkg/version"
"github.com/aquasecurity/trivy/pkg/log"
"github.com/spf13/cobra"
)
// NewCmd returns the root command for trivy-mcp with Aqua platform support
func NewRootCommand() *cobra.Command {
cmd := &cobra.Command{
Use: "mcp [flags]",
Aliases: []string{},
Args: cobra.ExactArgs(0),
Short: "[EXPERIMENTAL] Start an MCP Server for Trivy",
Long: `The MCP server is an experimental feature.
It allows you to run Trivy in a server mode and connect to it using the MCP protocol.
This command starts an MCP server that listens for incoming requests and processes them using Trivy.
The server can be configured to use different transports, such as SSE (Server-Sent Events).
`,
PreRunE: func(cmd *cobra.Command, args []string) error {
opts := flag.ToOptions()
log.InitLogger(opts.Debug, opts.Quiet)
return nil
},
RunE: func(cmd *cobra.Command, args []string) error {
opts := flag.ToOptions()
if opts.ShowVersion {
println("Trivy MCP Version: ", version.Version)
println("Trivy Version: ", version.TrivyVersion)
return nil
}
// allow us to gracefully shutdown the server
// when we receive a SIGINT or SIGTERM signal
// Suspend signal isn't handled for now
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
defer cancel()
server := mcpserver.NewMcpServer(opts)
return server.Start(ctx)
},
SilenceErrors: true,
SilenceUsage: true,
}
cmd.AddCommand(NewAuthCommand())
flag.AddBaseFlags(cmd)
flag.AddMcpFlags(cmd)
return cmd
}