package zk

import (
	"encoding/hex"

	"github.com/consensys/gnark-crypto/ecc"
	"github.com/consensys/gnark/backend/groth16"
	"github.com/consensys/gnark/frontend"
	"github.com/consensys/gnark/frontend/cs/r1cs"
	"github.com/consensys/gnark/std/hash/mimc"
)

type Circuit struct {
	Secret frontend.Variable
	Hash   frontend.Variable
}

func (circuit *Circuit) Define(api frontend.API) error {
	mimc, _ := mimc.NewMiMC(api)
	mimc.Write(circuit.Secret)

	api.AssertIsEqual(circuit.Hash, mimc.Sum())

	return nil
}

func POC() {
	var circuit Circuit

	r1cs, err := frontend.Compile(ecc.BW6_633.ScalarField(), r1cs.NewBuilder, &circuit)
	if err != nil {
		panic(err)
	}

	var assignment Circuit
	assignment.Secret = "0x" + hex.EncodeToString([]byte("Hello, World!"))
	assignment.Hash = "27353532363483121849779111558069699868988271024434984431742151050156752877051628698253914937683"

	witness, _ := frontend.NewWitness(&assignment, ecc.BW6_633.ScalarField())
	pk, vk, _ := groth16.Setup(r1cs)
	proof, _ := groth16.Prove(r1cs, pk, witness)
	publicWitness, _ := witness.Public()
	_ = groth16.Verify(proof, vk, publicWitness)
}