package chain

import (
	"context"
	"fmt"
	"os"
	"strings"

	chainconfig "git.cw.tr/mukan-network/mukan-ignite/ignite/config/chain"
)

type InPlaceArgs struct {
	NewChainID         string
	NewOperatorAddress string
	PrvKeyValidator    string
	AccountsToFund     string
}

func (c Chain) TestnetInPlace(ctx context.Context, args InPlaceArgs) error {
	commands, err := c.Commands(ctx)
	if err != nil {
		return err
	}

	// make sure that config.yml exists
	if c.options.ConfigFile != "" {
		if _, err := os.Stat(c.options.ConfigFile); err != nil {
			return err
		}
	} else if _, err := chainconfig.LocateDefault(c.app.Path); err != nil {
		return err
	}

	err = c.InPlace(ctx, commands, args)
	if err != nil {
		return err
	}
	return nil
}

type MultiNodeArgs struct {
	OutputDir             string
	NumValidator          string
	ValidatorsStakeAmount string
	NodeDirPrefix         string
	ListPorts             []uint
}

func (m MultiNodeArgs) ConvertPorts() string {
	var result []string

	for _, port := range m.ListPorts {
		result = append(result, fmt.Sprintf("%d", port))
	}

	return strings.Join(result, ",")
}

// If the app state still exists, TestnetMultiNode will reuse it.
// Otherwise, it will automatically re-initialize the app state from the beginning.
func (c Chain) TestnetMultiNode(ctx context.Context, args MultiNodeArgs) error {
	commands, err := c.Commands(ctx)
	if err != nil {
		return err
	}

	// make sure that config.yml exists
	if c.options.ConfigFile != "" {
		if _, err := os.Stat(c.options.ConfigFile); err != nil {
			return err
		}
	} else if _, err := chainconfig.LocateDefault(c.app.Path); err != nil {
		return err
	}

	return c.MultiNode(ctx, commands, args)
}