package commands

import (
	"fmt"
	"os"
	"path"
	"time"

	"github.com/ledgerwatch/bolt"
	"github.com/ledgerwatch/turbo-geth/cmd/state/stateless"
	"github.com/spf13/cobra"
)

func init() {
	stateGrowthCmd := &cobra.Command{
		Use:   "stateGrowth",
		Short: "stateGrowth",
		RunE: func(cmd *cobra.Command, args []string) error {
			localDb, err := bolt.Open(file()+"_sg", 0600, &bolt.Options{})
			if err != nil {
				panic(err)
			}
			ctx := getContext()
			remoteDb, err := connectRemoteDb(ctx, remoteDbAddress)
			if err != nil {
				return err
			}

			fmt.Println("Processing started...")
			stateless.NewStateGrowth1Reporter(ctx, remoteDb, localDb).StateGrowth1(ctx)
			stateless.NewStateGrowth2Reporter(ctx, remoteDb, localDb).StateGrowth2(ctx)
			return nil
		},
	}

	withChaindata(stateGrowthCmd)
	withRemoteDb(stateGrowthCmd)
	rootCmd.AddCommand(stateGrowthCmd)
}

// Generate name off the file for snapshot
// Each day has it's own partition
// It means that you can only continue execution of report from last snapshot.Save() checkpoint - read buckets forward from last key
// But not re-read bucket
func file() string {
	dir := path.Join(os.TempDir(), "turbo_geth_reports")
	if err := os.MkdirAll(dir, 0770); err != nil {
		panic(err)
	}
	return path.Join(dir, time.Now().Format("2006-01-02"))
}