package bundleprocessor

import (
	"github.com/iotaledger/goshimmer/packages/errors"
	"github.com/iotaledger/goshimmer/packages/model/value_transaction"
	"github.com/iotaledger/goshimmer/plugins/tangle"
	"github.com/iotaledger/hive.go/daemon"
	"github.com/iotaledger/hive.go/events"
	"github.com/iotaledger/hive.go/logger"
	"github.com/iotaledger/hive.go/node"
)

var PLUGIN = node.NewPlugin("Bundle Processor", node.Enabled, configure, run)
var log = logger.NewLogger("Bundle Processor")

func configure(plugin *node.Plugin) {
	tangle.Events.TransactionSolid.Attach(events.NewClosure(func(tx *value_transaction.ValueTransaction) {
		if tx.IsHead() {
			workerPool.Submit(tx)
		}
	}))

	Events.Error.Attach(events.NewClosure(func(err errors.IdentifiableError) {
		log.Error(err.Error())
	}))

	daemon.Events.Shutdown.Attach(events.NewClosure(func() {
		log.Info("Stopping Bundle Processor ...")

		workerPool.Stop()

		log.Info("Stopping Value Bundle Processor ...")

		valueBundleProcessorWorkerPool.Stop()
	}))
}

func run(plugin *node.Plugin) {
	log.Info("Starting Bundle Processor ...")

	daemon.BackgroundWorker("Bundle Processor", func() {
		log.Info("Starting Bundle Processor ... done")
		workerPool.Run()
		log.Info("Stopping Bundle Processor ... done")
	})

	log.Info("Starting Value Bundle Processor ...")

	daemon.BackgroundWorker("Value Bundle Processor", func() {
		log.Info("Starting Value Bundle Processor ... done")
		valueBundleProcessorWorkerPool.Run()
		log.Info("Stopping Value Bundle Processor ... done")
	})
}