package eth

import (
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/consensus"
	"github.com/ethereum/go-ethereum/consensus/ethash"
	"github.com/ethereum/go-ethereum/core/vm"
	"github.com/ethereum/go-ethereum/eth/ethconfig"
	"github.com/ethereum/go-ethereum/ethdb"
	"github.com/ethereum/go-ethereum/log"
	"github.com/ethereum/go-ethereum/node"
	"github.com/ethereum/go-ethereum/params"
	"github.com/ethereum/go-ethereum/plugins"
	"github.com/opoenrelayxyz/plugeth-utils/core"
	"math/big"
	"reflect"
	"time"
)

// func PluginCreateConsensusEngine(pl *plugins.PluginLoader, stack *node.Node, chainConfig *params.ChainConfig, config *ethash.Config, notify []string, noverify bool, db ethdb.Database) consensus.Engine {
//   fnList := pl.Lookup("CreateConsensusEngine", func(item interface{}) bool {
//     _, ok := item.(func(*node.Node, *params.ChainConfig, *ethash.Config, []string, bool, ethdb.Database) consensus.Engine)
//     return ok
//   })
//   for _, fni := range fnList {
//     if fn, ok := fni.(func(*node.Node, *params.ChainConfig, *ethash.Config, []string, bool, ethdb.Database) consensus.Engine); ok {
//       return fn(stack, chainConfig, config, notify, noverify, db)
//     }
//   }
//   return ethconfig.CreateConsensusEngine(stack, chainConfig, config, notify, noverify, db)
// }
//
// func pluginCreateConsensusEngine(stack *node.Node, chainConfig *params.ChainConfig, config *ethash.Config, notify []string, noverify bool, db ethdb.Database) consensus.Engine {
//   if plugins.DefaultPluginLoader == nil {
// 		log.Warn("Attempting CreateConsensusEngine, but default PluginLoader has not been initialized")
// 		return ethconfig.CreateConsensusEngine(stack, chainConfig, config, notify, noverify, db)
// 	}
//   return PluginCreateConsensusEngine(plugins.DefaultPluginLoader, stack, chainConfig, config, notify, noverify, db)
// }

// TODO (philip): Translate to core.Tracer instead of core.Tracer, with appropriate type adjustments (let me know if this one is too hard)
type metaTracer struct {
	tracers []core.Tracer
}

func (mt *metaTracer) CaptureStart(env *vm.EVM, from common.Address, to common.Address, create bool, input []byte, gas uint64, value *big.Int) {
	for _, tracer := range mt.tracers {
		tracer.CaptureStart(env, from, to, create, input, gas, value)
	}
}
func (mt *metaTracer) CaptureState(env *vm.EVM, pc uint64, op vm.OpCode, gas, cost uint64, scope *vm.ScopeContext, rData []byte, depth int, err error) {
	for _, tracer := range mt.tracers {
		tracer.CaptureState(env, pc, op, gas, cost, scope, rData, depth, err)
	}
}
func (mt *metaTracer) CaptureFault(env *vm.EVM, pc uint64, op vm.OpCode, gas, cost uint64, scope *vm.ScopeContext, depth int, err error) {
	for _, tracer := range mt.tracers {
		tracer.CaptureFault(env, pc, op, gas, cost, scope, depth, err)
	}
}
func (mt *metaTracer) CaptureEnd(output []byte, gasUsed uint64, t time.Duration, err error) {
	for _, tracer := range mt.tracers {
		tracer.CaptureEnd(output, gasUsed, t, err)
	}
}

func PluginUpdateBlockchainVMConfig(pl *plugins.PluginLoader, cfg *vm.Config) {
	tracerList := plugins.Lookup("LiveTracer", func(item interface{}) bool {
		_, ok := item.(*core.Tracer)
		log.Info("Item is LiveTracer", "ok", ok, "type", reflect.TypeOf(item))
		return ok
	})
	if len(tracerList) > 0 {
		mt := &metaTracer{tracers: []core.Tracer{}}
		for _, tracer := range tracerList {
			if v, ok := tracer.(*core.Tracer); ok {
				log.Info("LiveTracer registered")
				mt.tracers = append(mt.tracers, *v)
			} else {
				log.Info("Item is not tracer")
			}
		}
		cfg.Debug = true
		cfg.Tracer = mt
	} else {
		log.Warn("Module is not tracer")
	}

	fnList := plugins.Lookup("UpdateBlockchainVMConfig", func(item interface{}) bool {
		_, ok := item.(func(*vm.Config))
		return ok
	})
	for _, fni := range fnList {
		if fn, ok := fni.(func(*vm.Config)); ok {
			fn(cfg)
			return
		}
	}
}

func pluginUpdateBlockchainVMConfig(cfg *vm.Config) {
	if plugins.DefaultPluginLoader == nil {
		log.Warn("Attempting CreateConsensusEngine, but default PluginLoader has not been initialized")
		return
	}
	PluginUpdateBlockchainVMConfig(plugins.DefaultPluginLoader, cfg)
}