522 lines
13 KiB
Go
522 lines
13 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"git.t-juice.club/torjus/labmcp/internal/database"
|
|
"git.t-juice.club/torjus/labmcp/internal/homemanager"
|
|
"git.t-juice.club/torjus/labmcp/internal/mcp"
|
|
"git.t-juice.club/torjus/labmcp/internal/options"
|
|
)
|
|
|
|
const (
|
|
defaultDatabase = "sqlite://hm-options.db"
|
|
version = "0.1.2"
|
|
)
|
|
|
|
func main() {
|
|
app := &cli.App{
|
|
Name: "hm-options",
|
|
Usage: "MCP server for Home Manager options search and query",
|
|
Version: version,
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "database",
|
|
Aliases: []string{"d"},
|
|
Usage: "Database connection string (postgres://... or sqlite://...)",
|
|
EnvVars: []string{"HM_OPTIONS_DATABASE"},
|
|
Value: defaultDatabase,
|
|
},
|
|
},
|
|
Commands: []*cli.Command{
|
|
{
|
|
Name: "serve",
|
|
Usage: "Run MCP server",
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "transport",
|
|
Aliases: []string{"t"},
|
|
Usage: "Transport type: 'stdio' or 'http'",
|
|
Value: "stdio",
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "http-address",
|
|
Usage: "HTTP listen address",
|
|
Value: "127.0.0.1:8080",
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "http-endpoint",
|
|
Usage: "HTTP endpoint path",
|
|
Value: "/mcp",
|
|
},
|
|
&cli.StringSliceFlag{
|
|
Name: "allowed-origins",
|
|
Usage: "Allowed Origin headers for CORS (can be specified multiple times)",
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "tls-cert",
|
|
Usage: "TLS certificate file",
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "tls-key",
|
|
Usage: "TLS key file",
|
|
},
|
|
&cli.DurationFlag{
|
|
Name: "session-ttl",
|
|
Usage: "Session TTL for HTTP transport",
|
|
Value: 30 * time.Minute,
|
|
},
|
|
},
|
|
Action: func(c *cli.Context) error {
|
|
return runServe(c)
|
|
},
|
|
},
|
|
{
|
|
Name: "index",
|
|
Usage: "Index a home-manager revision",
|
|
ArgsUsage: "<revision>",
|
|
Flags: []cli.Flag{
|
|
&cli.BoolFlag{
|
|
Name: "no-files",
|
|
Usage: "Skip indexing file contents (faster, disables get_file tool)",
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "force",
|
|
Aliases: []string{"f"},
|
|
Usage: "Force re-indexing even if revision already exists",
|
|
},
|
|
},
|
|
Action: func(c *cli.Context) error {
|
|
if c.NArg() < 1 {
|
|
return fmt.Errorf("revision argument required")
|
|
}
|
|
return runIndex(c, c.Args().First(), !c.Bool("no-files"), c.Bool("force"))
|
|
},
|
|
},
|
|
{
|
|
Name: "list",
|
|
Usage: "List indexed revisions",
|
|
Action: func(c *cli.Context) error {
|
|
return runList(c)
|
|
},
|
|
},
|
|
{
|
|
Name: "search",
|
|
Usage: "Search for options",
|
|
ArgsUsage: "<query>",
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "revision",
|
|
Aliases: []string{"r"},
|
|
Usage: "Revision to search (default: most recent)",
|
|
},
|
|
&cli.IntFlag{
|
|
Name: "limit",
|
|
Aliases: []string{"n"},
|
|
Usage: "Maximum number of results",
|
|
Value: 20,
|
|
},
|
|
},
|
|
Action: func(c *cli.Context) error {
|
|
if c.NArg() < 1 {
|
|
return fmt.Errorf("query argument required")
|
|
}
|
|
return runSearch(c, c.Args().First())
|
|
},
|
|
},
|
|
{
|
|
Name: "get",
|
|
Usage: "Get details for a specific option",
|
|
ArgsUsage: "<option-name>",
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "revision",
|
|
Aliases: []string{"r"},
|
|
Usage: "Revision to search (default: most recent)",
|
|
},
|
|
},
|
|
Action: func(c *cli.Context) error {
|
|
if c.NArg() < 1 {
|
|
return fmt.Errorf("option name required")
|
|
}
|
|
return runGet(c, c.Args().First())
|
|
},
|
|
},
|
|
{
|
|
Name: "delete",
|
|
Usage: "Delete an indexed revision",
|
|
ArgsUsage: "<revision>",
|
|
Action: func(c *cli.Context) error {
|
|
if c.NArg() < 1 {
|
|
return fmt.Errorf("revision argument required")
|
|
}
|
|
return runDelete(c, c.Args().First())
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
if err := app.Run(os.Args); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
// openStore opens a database store based on the connection string.
|
|
func openStore(connStr string) (database.Store, error) {
|
|
if strings.HasPrefix(connStr, "sqlite://") {
|
|
path := strings.TrimPrefix(connStr, "sqlite://")
|
|
return database.NewSQLiteStore(path)
|
|
}
|
|
if strings.HasPrefix(connStr, "postgres://") || strings.HasPrefix(connStr, "postgresql://") {
|
|
return database.NewPostgresStore(connStr)
|
|
}
|
|
// Default to SQLite with the connection string as path
|
|
return database.NewSQLiteStore(connStr)
|
|
}
|
|
|
|
func runServe(c *cli.Context) error {
|
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer cancel()
|
|
|
|
store, err := openStore(c.String("database"))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
defer store.Close()
|
|
|
|
if err := store.Initialize(ctx); err != nil {
|
|
return fmt.Errorf("failed to initialize database: %w", err)
|
|
}
|
|
|
|
logger := log.New(os.Stderr, "[mcp] ", log.LstdFlags)
|
|
config := mcp.DefaultHomeManagerConfig()
|
|
server := mcp.NewServer(store, logger, config)
|
|
|
|
indexer := homemanager.NewIndexer(store)
|
|
server.RegisterHandlers(indexer)
|
|
|
|
transport := c.String("transport")
|
|
switch transport {
|
|
case "stdio":
|
|
logger.Println("Starting MCP server on stdio...")
|
|
return server.Run(ctx, os.Stdin, os.Stdout)
|
|
|
|
case "http":
|
|
config := mcp.HTTPConfig{
|
|
Address: c.String("http-address"),
|
|
Endpoint: c.String("http-endpoint"),
|
|
AllowedOrigins: c.StringSlice("allowed-origins"),
|
|
SessionTTL: c.Duration("session-ttl"),
|
|
TLSCertFile: c.String("tls-cert"),
|
|
TLSKeyFile: c.String("tls-key"),
|
|
}
|
|
httpTransport := mcp.NewHTTPTransport(server, config)
|
|
return httpTransport.Run(ctx)
|
|
|
|
default:
|
|
return fmt.Errorf("unknown transport: %s (use 'stdio' or 'http')", transport)
|
|
}
|
|
}
|
|
|
|
func runIndex(c *cli.Context, revision string, indexFiles bool, force bool) error {
|
|
ctx := context.Background()
|
|
|
|
store, err := openStore(c.String("database"))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
defer store.Close()
|
|
|
|
if err := store.Initialize(ctx); err != nil {
|
|
return fmt.Errorf("failed to initialize database: %w", err)
|
|
}
|
|
|
|
indexer := homemanager.NewIndexer(store)
|
|
|
|
fmt.Printf("Indexing revision: %s\n", revision)
|
|
|
|
var result *options.IndexResult
|
|
if force {
|
|
result, err = indexer.ReindexRevision(ctx, revision)
|
|
} else {
|
|
result, err = indexer.IndexRevision(ctx, revision)
|
|
}
|
|
if err != nil {
|
|
return fmt.Errorf("indexing failed: %w", err)
|
|
}
|
|
|
|
if result.AlreadyIndexed {
|
|
fmt.Printf("Revision already indexed (%d options). Use --force to re-index.\n", result.OptionCount)
|
|
return nil
|
|
}
|
|
|
|
if dur, ok := result.Duration.(time.Duration); ok {
|
|
fmt.Printf("Indexed %d options in %s\n", result.OptionCount, dur)
|
|
} else {
|
|
fmt.Printf("Indexed %d options\n", result.OptionCount)
|
|
}
|
|
fmt.Printf("Git hash: %s\n", result.Revision.GitHash)
|
|
if result.Revision.ChannelName != "" {
|
|
fmt.Printf("Channel: %s\n", result.Revision.ChannelName)
|
|
}
|
|
|
|
if indexFiles {
|
|
fmt.Println("Indexing files...")
|
|
fileCount, err := indexer.IndexFiles(ctx, result.Revision.ID, result.Revision.GitHash)
|
|
if err != nil {
|
|
return fmt.Errorf("file indexing failed: %w", err)
|
|
}
|
|
fmt.Printf("Indexed %d files\n", fileCount)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func runList(c *cli.Context) error {
|
|
ctx := context.Background()
|
|
|
|
store, err := openStore(c.String("database"))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
defer store.Close()
|
|
|
|
if err := store.Initialize(ctx); err != nil {
|
|
return fmt.Errorf("failed to initialize database: %w", err)
|
|
}
|
|
|
|
revisions, err := store.ListRevisions(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to list revisions: %w", err)
|
|
}
|
|
|
|
if len(revisions) == 0 {
|
|
fmt.Println("No revisions indexed.")
|
|
fmt.Println("Use 'hm-options index <revision>' to index a home-manager version.")
|
|
return nil
|
|
}
|
|
|
|
fmt.Printf("Indexed revisions (%d):\n\n", len(revisions))
|
|
for _, rev := range revisions {
|
|
fmt.Printf(" %s", rev.GitHash[:12])
|
|
if rev.ChannelName != "" {
|
|
fmt.Printf(" (%s)", rev.ChannelName)
|
|
}
|
|
fmt.Printf("\n Options: %d, Indexed: %s\n",
|
|
rev.OptionCount, rev.IndexedAt.Format("2006-01-02 15:04"))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func runSearch(c *cli.Context, query string) error {
|
|
ctx := context.Background()
|
|
|
|
store, err := openStore(c.String("database"))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
defer store.Close()
|
|
|
|
if err := store.Initialize(ctx); err != nil {
|
|
return fmt.Errorf("failed to initialize database: %w", err)
|
|
}
|
|
|
|
// Find revision
|
|
var rev *database.Revision
|
|
revisionArg := c.String("revision")
|
|
if revisionArg != "" {
|
|
rev, err = store.GetRevision(ctx, revisionArg)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get revision: %w", err)
|
|
}
|
|
if rev == nil {
|
|
rev, err = store.GetRevisionByChannel(ctx, revisionArg)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get revision: %w", err)
|
|
}
|
|
}
|
|
} else {
|
|
revisions, err := store.ListRevisions(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to list revisions: %w", err)
|
|
}
|
|
if len(revisions) > 0 {
|
|
rev = revisions[0]
|
|
}
|
|
}
|
|
|
|
if rev == nil {
|
|
return fmt.Errorf("no indexed revision found")
|
|
}
|
|
|
|
filters := database.SearchFilters{
|
|
Limit: c.Int("limit"),
|
|
}
|
|
|
|
options, err := store.SearchOptions(ctx, rev.ID, query, filters)
|
|
if err != nil {
|
|
return fmt.Errorf("search failed: %w", err)
|
|
}
|
|
|
|
if len(options) == 0 {
|
|
fmt.Printf("No options found matching '%s'\n", query)
|
|
return nil
|
|
}
|
|
|
|
fmt.Printf("Found %d options matching '%s':\n\n", len(options), query)
|
|
for _, opt := range options {
|
|
fmt.Printf(" %s\n", opt.Name)
|
|
fmt.Printf(" Type: %s\n", opt.Type)
|
|
if opt.Description != "" {
|
|
desc := opt.Description
|
|
if len(desc) > 100 {
|
|
desc = desc[:100] + "..."
|
|
}
|
|
fmt.Printf(" %s\n", desc)
|
|
}
|
|
fmt.Println()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func runGet(c *cli.Context, name string) error {
|
|
ctx := context.Background()
|
|
|
|
store, err := openStore(c.String("database"))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
defer store.Close()
|
|
|
|
if err := store.Initialize(ctx); err != nil {
|
|
return fmt.Errorf("failed to initialize database: %w", err)
|
|
}
|
|
|
|
// Find revision
|
|
var rev *database.Revision
|
|
revisionArg := c.String("revision")
|
|
if revisionArg != "" {
|
|
rev, err = store.GetRevision(ctx, revisionArg)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get revision: %w", err)
|
|
}
|
|
if rev == nil {
|
|
rev, err = store.GetRevisionByChannel(ctx, revisionArg)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get revision: %w", err)
|
|
}
|
|
}
|
|
} else {
|
|
revisions, err := store.ListRevisions(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to list revisions: %w", err)
|
|
}
|
|
if len(revisions) > 0 {
|
|
rev = revisions[0]
|
|
}
|
|
}
|
|
|
|
if rev == nil {
|
|
return fmt.Errorf("no indexed revision found")
|
|
}
|
|
|
|
opt, err := store.GetOption(ctx, rev.ID, name)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get option: %w", err)
|
|
}
|
|
if opt == nil {
|
|
return fmt.Errorf("option '%s' not found", name)
|
|
}
|
|
|
|
fmt.Printf("%s\n", opt.Name)
|
|
fmt.Printf(" Type: %s\n", opt.Type)
|
|
if opt.Description != "" {
|
|
fmt.Printf(" Description: %s\n", opt.Description)
|
|
}
|
|
if opt.DefaultValue != "" && opt.DefaultValue != "null" {
|
|
fmt.Printf(" Default: %s\n", opt.DefaultValue)
|
|
}
|
|
if opt.Example != "" && opt.Example != "null" {
|
|
fmt.Printf(" Example: %s\n", opt.Example)
|
|
}
|
|
if opt.ReadOnly {
|
|
fmt.Println(" Read-only: yes")
|
|
}
|
|
|
|
// Get declarations
|
|
declarations, err := store.GetDeclarations(ctx, opt.ID)
|
|
if err == nil && len(declarations) > 0 {
|
|
fmt.Println(" Declared in:")
|
|
for _, decl := range declarations {
|
|
if decl.Line > 0 {
|
|
fmt.Printf(" - %s:%d\n", decl.FilePath, decl.Line)
|
|
} else {
|
|
fmt.Printf(" - %s\n", decl.FilePath)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Get children
|
|
children, err := store.GetChildren(ctx, rev.ID, opt.Name)
|
|
if err == nil && len(children) > 0 {
|
|
fmt.Println(" Sub-options:")
|
|
for _, child := range children {
|
|
shortName := child.Name
|
|
if strings.HasPrefix(child.Name, opt.Name+".") {
|
|
shortName = child.Name[len(opt.Name)+1:]
|
|
}
|
|
fmt.Printf(" - %s (%s)\n", shortName, child.Type)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func runDelete(c *cli.Context, revision string) error {
|
|
ctx := context.Background()
|
|
|
|
store, err := openStore(c.String("database"))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
defer store.Close()
|
|
|
|
if err := store.Initialize(ctx); err != nil {
|
|
return fmt.Errorf("failed to initialize database: %w", err)
|
|
}
|
|
|
|
// Find revision
|
|
rev, err := store.GetRevision(ctx, revision)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get revision: %w", err)
|
|
}
|
|
if rev == nil {
|
|
rev, err = store.GetRevisionByChannel(ctx, revision)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get revision: %w", err)
|
|
}
|
|
}
|
|
|
|
if rev == nil {
|
|
return fmt.Errorf("revision '%s' not found", revision)
|
|
}
|
|
|
|
if err := store.DeleteRevision(ctx, rev.ID); err != nil {
|
|
return fmt.Errorf("failed to delete revision: %w", err)
|
|
}
|
|
|
|
fmt.Printf("Deleted revision %s\n", rev.GitHash)
|
|
return nil
|
|
}
|