package commands import ( "context" "fmt" "os" "sync" "time" "github.com/spf13/cobra" "github.com/MichaelMure/git-bug/bridge" "github.com/MichaelMure/git-bug/bridge/core" "github.com/MichaelMure/git-bug/cache" "github.com/MichaelMure/git-bug/util/interrupt" ) func runBridgePush(cmd *cobra.Command, args []string) error { backend, err := cache.NewRepoCache(repo) if err != nil { return err } defer backend.Close() interrupt.RegisterCleaner(backend.Close) var b *core.Bridge if len(args) == 0 { b, err = bridge.DefaultBridge(backend) } else { b, err = bridge.LoadBridge(backend, args[0]) } if err != nil { return err } parentCtx := context.Background() ctx, cancel := context.WithCancel(parentCtx) defer cancel() done := make(chan struct{}, 1) var mu sync.Mutex interruptCount := 0 interrupt.RegisterCleaner(func() error { mu.Lock() if interruptCount > 0 { fmt.Println("Received another interrupt before graceful stop, terminating...") os.Exit(0) } interruptCount++ mu.Unlock() fmt.Println("Received interrupt signal, stopping the import...\n(Hit ctrl-c again to kill the process.)") // send signal to stop the importer cancel() // block until importer gracefully shutdown <-done return nil }) err = b.ExportAll(ctx, time.Time{}) if err != nil { return err } // send done signal close(done) return nil } var bridgePushCmd = &cobra.Command{ Use: "push []", Short: "Push updates.", PreRunE: loadRepo, RunE: runBridgePush, Args: cobra.MaximumNArgs(1), } func init() { bridgeCmd.AddCommand(bridgePushCmd) }