From acc9a6f3a6df2961c3ae44352216d915cb9b5315 Mon Sep 17 00:00:00 2001 From: Michael Muré Date: Sat, 10 Sep 2022 11:09:19 +0200 Subject: commands: reorg into different packages --- commands/bridge/bridge_push.go | 99 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 99 insertions(+) create mode 100644 commands/bridge/bridge_push.go (limited to 'commands/bridge/bridge_push.go') diff --git a/commands/bridge/bridge_push.go b/commands/bridge/bridge_push.go new file mode 100644 index 00000000..51baed4d --- /dev/null +++ b/commands/bridge/bridge_push.go @@ -0,0 +1,99 @@ +package bridgecmd + +import ( + "context" + "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/commands/completion" + "github.com/MichaelMure/git-bug/commands/execenv" + "github.com/MichaelMure/git-bug/util/interrupt" +) + +func newBridgePushCommand() *cobra.Command { + env := execenv.NewEnv() + + cmd := &cobra.Command{ + Use: "push [NAME]", + Short: "Push updates to remote bug tracker", + PreRunE: execenv.LoadBackendEnsureUser(env), + RunE: execenv.CloseBackend(env, func(cmd *cobra.Command, args []string) error { + return runBridgePush(env, args) + }), + Args: cobra.MaximumNArgs(1), + ValidArgsFunction: completion.Bridge(env), + } + + return cmd +} + +func runBridgePush(env *execenv.Env, args []string) error { + var b *core.Bridge + var err error + + if len(args) == 0 { + b, err = bridge.DefaultBridge(env.Backend) + } else { + b, err = bridge.LoadBridge(env.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 { + env.Err.Println("Received another interrupt before graceful stop, terminating...") + os.Exit(0) + } + + interruptCount++ + mu.Unlock() + + env.Err.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 + }) + + events, err := b.ExportAll(ctx, time.Time{}) + if err != nil { + return err + } + + exportedIssues := 0 + for result := range events { + if result.Event != core.ExportEventNothing { + env.Out.Println(result.String()) + } + + switch result.Event { + case core.ExportEventBug: + exportedIssues++ + } + } + + env.Out.Printf("exported %d issues with %s bridge\n", exportedIssues, b.Name) + + // send done signal + close(done) + return nil +} -- cgit