Initial commit
This commit is contained in:
79
cmd/localrun.go
Normal file
79
cmd/localrun.go
Normal file
@@ -0,0 +1,79 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"sync"
|
||||
"syscall"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// localrunCmd represents the localrun command
|
||||
var localrunCmd = &cobra.Command{
|
||||
Use: "localrun",
|
||||
Short: "A brief description of your command",
|
||||
Long: `A longer description that spans multiple lines and likely contains examples
|
||||
and usage of using your command. For example:
|
||||
|
||||
Cobra is a CLI library for Go that empowers applications.
|
||||
This application is a tool to generate the needed files
|
||||
to quickly create a Cobra application.`,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
|
||||
folderPath := "bin"
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
wg := &sync.WaitGroup{}
|
||||
|
||||
// Handle Ctrl+C (SIGINT) and other termination signals
|
||||
signalChannel := make(chan os.Signal, 1)
|
||||
signal.Notify(signalChannel, syscall.SIGINT, syscall.SIGTERM)
|
||||
|
||||
if findCaddyRuntime(folderPath) == "" {
|
||||
log.Println("Caddy runtime not found")
|
||||
downloadLatestCaddy(folderPath)
|
||||
}
|
||||
|
||||
// Start the goroutines to run programs
|
||||
walkFolder(folderPath, ctx, wg)
|
||||
|
||||
go func() {
|
||||
// Wait for a termination signal (like Ctrl+C)
|
||||
<-signalChannel
|
||||
|
||||
log.Println("Received termination signal. Stopping all processes...")
|
||||
|
||||
// Call cancel to stop all the processes gracefully
|
||||
cancel()
|
||||
|
||||
// Force kill all processes
|
||||
stopAllProcesses()
|
||||
|
||||
// Exit the program
|
||||
os.Exit(0)
|
||||
}()
|
||||
|
||||
// Wait for all processes to finish
|
||||
wg.Wait()
|
||||
|
||||
log.Println("All processes stopped.")
|
||||
|
||||
},
|
||||
}
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(localrunCmd)
|
||||
|
||||
// Here you will define your flags and configuration settings.
|
||||
|
||||
// Cobra supports Persistent Flags which will work for this command
|
||||
// and all subcommands, e.g.:
|
||||
// localrunCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||
|
||||
// Cobra supports local flags which will only run when this command
|
||||
// is called directly, e.g.:
|
||||
// localrunCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||
}
|
||||
Reference in New Issue
Block a user