cmd/hub/util/flow.go

// Copyright (c) 2022 EPAM Systems, Inc.
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

package util

import (
	"fmt"
	"log"
	"os"

	"github.com/epam/hubctl/cmd/hub/config"
)

var atDone []func() <-chan struct{}

func MaybeFatalf(format string, v ...interface{}) {
	if config.Force {
		Warn(format, v...)
	} else {
		Done()
		log.Fatalf(format, v...)
	}
}

func MaybeFatalf2(cleanup func(string, bool), format string, v ...interface{}) {
	msg := fmt.Sprintf(format, v...)
	if config.Force {
		Warn("%s", msg)
	} else {
		log.Print(msg)
	}
	if cleanup != nil {
		cleanup(msg, !config.Force)
	}
	if !config.Force {
		Done()
		os.Exit(1)
	}
}

func AtDone(cleanup func() <-chan struct{}) {
	atDone = append(atDone, cleanup)
}

func Done() {
	var chs []<-chan struct{}
	for _, cleanup := range atDone {
		ch := cleanup()
		if ch != nil {
			chs = append(chs, ch)
		}
	}
	atDone = nil
	for _, ch := range chs {
		<-ch
	}
}