main.go

package main

import (
	"fmt"
	"os"

	"github.com/spf13/cobra"
)

func main() {
	rootCmd := &cobra.Command{
		Use:   "mtls",
		Short: "mTLS Certificate Management Tool",
		Long:  `A CLI tool for creating and managing mTLS certificates including Root CAs and server certificates.`,
		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
			// Ensure certs directory exists
			if err := os.MkdirAll("./certs", 0755); err != nil {
				return err
			}
			// Initialize DB
			_, err := InitDB("./certs/registry.db")
			return err
		},
	}

	// CA commands
	caCmd := &cobra.Command{
		Use:   "ca",
		Short: "Manage Root Certificate Authorities",
	}
	caCmd.AddCommand(createCACmd())
	caCmd.AddCommand(listCACmd())
	caCmd.AddCommand(signCSRCmd())
	caCmd.AddCommand(revokeCmd())
	caCmd.AddCommand(genCRLCmd())

	// Certificate commands
	certCmd := &cobra.Command{
		Use:   "cert",
		Short: "Manage server and client certificates",
	}
	certCmd.AddCommand(createServerCertCmd())
	certCmd.AddCommand(listServerCertsCmd())
	certCmd.AddCommand(createClientCertCmd())
	certCmd.AddCommand(listClientCertsCmd())
	certCmd.AddCommand(inspectCmd())
	certCmd.AddCommand(verifyCmd())

	// Add commands to root
	rootCmd.AddCommand(caCmd)
	rootCmd.AddCommand(certCmd)
	rootCmd.AddCommand(createTreeCmd())
	rootCmd.AddCommand(dbCmd())
	rootCmd.AddCommand(serverCmd())

	// Version command
	versionCmd := &cobra.Command{
		Use:   "version",
		Short: "Print version information",
		Run: func(cmd *cobra.Command, args []string) {
			fmt.Println("mtls version 1.0.0")
		},
	}
	rootCmd.AddCommand(versionCmd)

	if err := rootCmd.Execute(); err != nil {
		os.Exit(1)
	}
}