package generate

import (
	"bytes"
	"fmt"

	"github.com/ledgerwatch/turbo-geth/common/changeset"
	"github.com/ledgerwatch/turbo-geth/common/dbutils"
	"github.com/ledgerwatch/turbo-geth/core"
	"github.com/ledgerwatch/turbo-geth/ethdb"
)

func RegenerateIndex(chaindata string, indexBucket []byte, csBucket []byte) error {
	db, err := ethdb.NewBoltDatabase(chaindata)
	if err != nil {
		return err
	}
	var walker func([]byte) changeset.Walker
	if bytes.Equal(dbutils.AccountChangeSetBucket, csBucket) {
		walker = func(cs []byte) changeset.Walker {
			return changeset.AccountChangeSetBytes(cs)
		}
	}

	if bytes.Equal(dbutils.StorageChangeSetBucket, csBucket) {
		walker = func(cs []byte) changeset.Walker {
			return changeset.StorageChangeSetBytes(cs)
		}
	}

	ig := core.NewIndexGenerator(db)
	err = ig.GenerateIndex(0, csBucket, indexBucket, walker, nil)
	if err != nil {
		return err
	}
	fmt.Println("Index is successfully regenerated")
	return nil
}