Function parseFromStdin

cmd/gosqlx/cmd/parse.go:115–169

parseFromStdin handles parsing from stdin input

(cmd *cobra.Command)

// parseFromStdin handles parsing from stdin input
func parseFromStdin(cmd *cobra.Command) error {
	// Read from stdin
	content, err := ReadFromStdin()
	if err != nil {
		return fmt.Errorf("failed to read from stdin: %w", err)
	}

	// Validate stdin content
	if err := ValidateStdinInput(content); err != nil {
		return fmt.Errorf("stdin validation failed: %w", err)
	}

	// Load configuration
	cfg, err := config.LoadDefault()
	if err != nil {
		cfg = config.DefaultConfig()
	}

	// Track which flags were explicitly set
	flagsChanged := make(map[string]bool)
	cmd.Flags().Visit(func(f *pflag.Flag) {
		flagsChanged[f.Name] = true
	})
	if cmd.Parent() != nil && cmd.Parent().PersistentFlags() != nil {
		cmd.Parent().PersistentFlags().Visit(func(f *pflag.Flag) {
			flagsChanged[f.Name] = true
		})
	}

	// Create parser options
	opts := ParserOptionsFromConfig(cfg, flagsChanged, ParserFlags{
		ShowAST:    parseShowAST,
		ShowTokens: parseShowTokens,
		TreeView:   parseTreeView,
		Format:     format,
		Verbose:    verbose,
	})

	// Create parser
	parser := NewParser(cmd.OutOrStdout(), cmd.ErrOrStderr(), opts)

	// Parse the stdin content (Parse accepts string input directly)
	result, err := parser.Parse(string(content))
	if err != nil {
		return err
	}

	// CRITICAL: Always release AST if it was created
	if result.AST != nil {
		defer ast.ReleaseAST(result.AST)
	}

	// Display the result
	return parser.Display(result)
}

func init() {
	rootCmd.AddCommand(parseCmd)