package morph

import (
	"encoding/json"
	"fmt"
	"os"
	"path"
	"path/filepath"

	"github.com/nspcc-dev/neo-go/cli/input"
	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
	"github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest"
	"github.com/nspcc-dev/neo-go/pkg/util"
	"github.com/nspcc-dev/neo-go/pkg/wallet"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

const contractWalletName = "contract.json"

func initializeContractWallet(walletDir string) (*wallet.Wallet, error) {
	var (
		password string
		err      error
	)

	if key := "credentials.contract"; viper.IsSet(key) {
		password = viper.GetString(key)
	} else {
		prompt := "Password for contract wallet > "
		password, err = input.ReadPassword(prompt)
		if err != nil {
			return nil, err
		}
	}

	w, err := wallet.NewWallet(path.Join(walletDir, contractWalletName))
	if err != nil {
		return nil, err
	}

	acc, err := wallet.NewAccount()
	if err != nil {
		return nil, err
	}

	err = acc.Encrypt(password, keys.NEP2ScryptParams())
	if err != nil {
		return nil, err
	}

	w.AddAccount(acc)
	if err := w.Save(); err != nil {
		return nil, err
	}

	return w, nil
}

func openContractWallet(cmd *cobra.Command, walletDir string) (*wallet.Wallet, error) {
	p := path.Join(walletDir, contractWalletName)
	w, err := wallet.NewWalletFromFile(p)
	if err != nil {
		if !os.IsNotExist(err) {
			return nil, fmt.Errorf("can't open wallet: %w", err)
		}

		cmd.Printf("Contract group wallet is missing, initialize at %s\n",
			filepath.Join(walletDir, contractWalletName))
		w, err = initializeContractWallet(walletDir)
		if err != nil {
			return nil, err
		}
	}

	var password string
	if key := "credentials.contract"; viper.IsSet(key) {
		password = viper.GetString(key)
	} else {
		prompt := "Password for contract wallet > "
		password, err = input.ReadPassword(prompt)
		if err != nil {
			return nil, fmt.Errorf("can't fetch password: %w", err)
		}
	}

	for i := range w.Accounts {
		if err := w.Accounts[i].Decrypt(password, keys.NEP2ScryptParams()); err != nil {
			return nil, fmt.Errorf("can't unlock wallet: %w", err)
		}
	}

	return w, nil
}

func (c *initializeContext) addManifestGroup(h util.Uint160, cs *contractState) error {
	priv := c.ContractWallet.Accounts[0].PrivateKey()
	pub := priv.PublicKey()

	sig := priv.Sign(h.BytesBE())
	found := false

	for i := range cs.Manifest.Groups {
		if cs.Manifest.Groups[i].PublicKey.Equal(pub) {
			cs.Manifest.Groups[i].Signature = sig
			found = true
			break
		}
	}
	if !found {
		cs.Manifest.Groups = append(cs.Manifest.Groups, manifest.Group{
			PublicKey: pub,
			Signature: sig,
		})
	}

	data, err := json.Marshal(cs.Manifest)
	if err != nil {
		return err
	}

	cs.RawManifest = data
	return nil
}