package cmd

import (
	"context"
	"log/slog"

	"github.com/SigNoz/signoz/pkg/instrumentation"
	"github.com/SigNoz/signoz/pkg/signoz"
	"github.com/SigNoz/signoz/pkg/version"
	"github.com/spf13/cobra"
)

func registerGenerateOpenAPI(parentCmd *cobra.Command) {
	openapiCmd := &cobra.Command{
		Use:   "openapi",
		Short: "Generate OpenAPI schema for SigNoz",
		RunE: func(currCmd *cobra.Command, args []string) error {
			return runGenerateOpenAPI(currCmd.Context())
		},
	}

	parentCmd.AddCommand(openapiCmd)
}

func runGenerateOpenAPI(ctx context.Context) error {
	instrumentation, err := instrumentation.New(ctx, instrumentation.Config{Logs: instrumentation.LogsConfig{Level: slog.LevelInfo}}, version.Info, "signoz")
	if err != nil {
		return err
	}

	openapi, err := signoz.NewOpenAPI(ctx, instrumentation)
	if err != nil {
		return err
	}

	if err := openapi.CreateAndWrite("docs/api/openapi.yml"); err != nil {
		return err
	}

	return nil
}