import {
  state,
  State,
  method,
  UInt64,
  PrivateKey,
  SmartContract,
  Mina,
  AccountUpdate,
  TransactionPromise,
} from 'o1js';

class SimpleZkapp extends SmartContract {
  @state(UInt64) x = State<UInt64>();

  init() {
    super.init();
    this.x.set(new UInt64(0));
  }

  @method
  async increment() {
    let x = this.x.get();
    this.x.requireEquals(x);
    let newX = x.add(1);
    this.x.set(newX);
  }
}

let Local = await Mina.LocalBlockchain({ proofsEnabled: false });
Mina.setActiveInstance(Local);

// a test account that pays all the fees, and puts additional funds into the zkapp
let [sender] = Local.testAccounts;

// the zkapp account
let zkappKey = PrivateKey.random();
let zkappAddress = zkappKey.toPublicKey();

let initialBalance = 10_000_000_000;
let zkapp = new SimpleZkapp(zkappAddress);
await SimpleZkapp.analyzeMethods();

await Mina.transaction(sender, async () => {
  let senderUpdate = AccountUpdate.fundNewAccount(sender);
  senderUpdate.send({ to: zkappAddress, amount: initialBalance });
  await zkapp.deploy();
})
  .sign([sender.key, zkappKey])
  .send()
  .wait();

console.log('initial state: ' + zkapp.x.get());

console.log('increment');

await Mina.transaction(sender, async () => {
  await zkapp.increment();
})
  .prove()
  .sign([sender.key])
  .send()
  .wait();

console.log('final state: ' + zkapp.x.get());

const a = Mina.transaction(sender, async () => {
  await zkapp.increment();
});
a satisfies TransactionPromise<false, false>;
const b = a.prove() satisfies TransactionPromise<true, false>;
const c = b.sign([sender.key]) satisfies TransactionPromise<true, true>;
await c.send().wait();