// // Copyright 2026 The InfiniFlow Authors. All Rights Reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. // package main import ( "fmt" "os" "os/signal" "ragflow/internal/common" "syscall" "ragflow/internal/cli" ) func newMain() { parseArgs, err := cli.ParseArgs(os.Args[1:]) if err != nil { return } if parseArgs.ShowHelp { cli.PrintUsage() return } //parseArgs.Print() logLevel := "warn" // Default to warn (quiet mode) if parseArgs.Verbose { logLevel = "info" } if err = common.Init(logLevel, ""); err != nil { fmt.Printf("Warning: Failed to initialize logger: %v\n", err) } client, err := cli.NewCLIWithConfig(parseArgs) if err != nil { fmt.Printf("Failed to create CLI: %v\n", err) os.Exit(1) } sigChan := make(chan os.Signal, 1) signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM) go func() { <-sigChan client.Cleanup() os.Exit(0) }() if parseArgs.Command != nil { if err = client.RunSingleCommand(parseArgs.Command); err != nil { fmt.Printf("Error: %v\n", err) os.Exit(1) } } else { if err = client.NewRun(); err != nil { fmt.Printf("CLI error: %v\n", err) os.Exit(1) } } return } func main() { newMain() //// Parse command line arguments (skip program name) //args, err := cli.ParseConnectionArgs(os.Args[1:]) //if err != nil { // fmt.Printf("Error: %v\n", err) // os.Exit(1) //} // //// Initialize logger with appropriate level //logLevel := "warn" // Default to warn (quiet mode) //if args.Verbose { // logLevel = "info" //} //if err = common.Init(logLevel); err != nil { // fmt.Printf("Warning: Failed to initialize logger: %v\n", err) //} // //// Show help and exit //if args.ShowHelp { // cli.PrintUsage() // os.Exit(0) //} // //// Create CLI instance with parsed arguments //cliApp, err := cli.NewCLIWithArgs(args) //if err != nil { // fmt.Printf("Failed to create CLI: %v\n", err) // os.Exit(1) //} // //// Handle interrupt signal //sigChan := make(chan os.Signal, 1) //signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM) //go func() { // <-sigChan // cliApp.Cleanup() // os.Exit(0) //}() // //// Check if we have a single command to execute //if args.Command != nil { // // Single command mode // if err = cliApp.RunSingleCommand(args.Command); err != nil { // fmt.Printf("Error: %v\n", err) // os.Exit(1) // } //} else { // // Interactive mode // if err = cliApp.Run(); err != nil { // fmt.Printf("CLI error: %v\n", err) // os.Exit(1) // } //} }