import * as $ from '@manahippo/move-to-ts';
import { AptosDataCache, AptosParserRepo, DummyCache, AptosLocalCache } from '@manahippo/move-to-ts';
import { U8, U64, U128 } from '@manahippo/move-to-ts';
import { u8, u64, u128 } from '@manahippo/move-to-ts';
import { TypeParamDeclType, FieldDeclType } from '@manahippo/move-to-ts';
import { AtomicTypeTag, StructTag, TypeTag, VectorTag, SimpleStructTag } from '@manahippo/move-to-ts';
import { OptionTransaction } from '@manahippo/move-to-ts';
import { HexString, AptosClient, AptosAccount, TxnBuilderTypes, Types } from 'aptos';
import * as Error from './error';
import * as Signer from './signer';
import * as Vector from './vector';
export const packageName = 'AptosStdlib';
export const moduleAddress = new HexString('0x1');
export const moduleName = 'capability';

export const ECAPABILITY_ALREADY_EXISTS: U64 = u64('1');
export const ECAPABILITY_NOT_FOUND: U64 = u64('2');
export const EDELEGATE: U64 = u64('3');

export class Cap {
  static moduleAddress = moduleAddress;
  static moduleName = moduleName;
  __app: $.AppType | null = null;
  static structName: string = 'Cap';
  static typeParameters: TypeParamDeclType[] = [{ name: 'Feature', isPhantom: true }];
  static fields: FieldDeclType[] = [{ name: 'root', typeTag: AtomicTypeTag.Address }];

  root: HexString;

  constructor(proto: any, public typeTag: TypeTag) {
    this.root = proto['root'] as HexString;
  }

  static CapParser(data: any, typeTag: TypeTag, repo: AptosParserRepo): Cap {
    const proto = $.parseStructProto(data, typeTag, repo, Cap);
    return new Cap(proto, typeTag);
  }

  static makeTag($p: TypeTag[]): StructTag {
    return new StructTag(moduleAddress, moduleName, 'Cap', $p);
  }
  async loadFullState(app: $.AppType) {
    this.__app = app;
  }
}

export class CapDelegateState {
  static moduleAddress = moduleAddress;
  static moduleName = moduleName;
  __app: $.AppType | null = null;
  static structName: string = 'CapDelegateState';
  static typeParameters: TypeParamDeclType[] = [{ name: 'Feature', isPhantom: true }];
  static fields: FieldDeclType[] = [{ name: 'root', typeTag: AtomicTypeTag.Address }];

  root: HexString;

  constructor(proto: any, public typeTag: TypeTag) {
    this.root = proto['root'] as HexString;
  }

  static CapDelegateStateParser(data: any, typeTag: TypeTag, repo: AptosParserRepo): CapDelegateState {
    const proto = $.parseStructProto(data, typeTag, repo, CapDelegateState);
    return new CapDelegateState(proto, typeTag);
  }

  static async load(repo: AptosParserRepo, client: AptosClient, address: HexString, typeParams: TypeTag[]) {
    const result = await repo.loadResource(client, address, CapDelegateState, typeParams);
    return result as unknown as CapDelegateState;
  }
  static async loadByApp(app: $.AppType, address: HexString, typeParams: TypeTag[]) {
    const result = await app.repo.loadResource(app.client, address, CapDelegateState, typeParams);
    await result.loadFullState(app);
    return result as unknown as CapDelegateState;
  }
  static makeTag($p: TypeTag[]): StructTag {
    return new StructTag(moduleAddress, moduleName, 'CapDelegateState', $p);
  }
  async loadFullState(app: $.AppType) {
    this.__app = app;
  }
}

export class CapState {
  static moduleAddress = moduleAddress;
  static moduleName = moduleName;
  __app: $.AppType | null = null;
  static structName: string = 'CapState';
  static typeParameters: TypeParamDeclType[] = [{ name: 'Feature', isPhantom: true }];
  static fields: FieldDeclType[] = [{ name: 'delegates', typeTag: new VectorTag(AtomicTypeTag.Address) }];

  delegates: HexString[];

  constructor(proto: any, public typeTag: TypeTag) {
    this.delegates = proto['delegates'] as HexString[];
  }

  static CapStateParser(data: any, typeTag: TypeTag, repo: AptosParserRepo): CapState {
    const proto = $.parseStructProto(data, typeTag, repo, CapState);
    return new CapState(proto, typeTag);
  }

  static async load(repo: AptosParserRepo, client: AptosClient, address: HexString, typeParams: TypeTag[]) {
    const result = await repo.loadResource(client, address, CapState, typeParams);
    return result as unknown as CapState;
  }
  static async loadByApp(app: $.AppType, address: HexString, typeParams: TypeTag[]) {
    const result = await app.repo.loadResource(app.client, address, CapState, typeParams);
    await result.loadFullState(app);
    return result as unknown as CapState;
  }
  static makeTag($p: TypeTag[]): StructTag {
    return new StructTag(moduleAddress, moduleName, 'CapState', $p);
  }
  async loadFullState(app: $.AppType) {
    this.__app = app;
  }
}

export class LinearCap {
  static moduleAddress = moduleAddress;
  static moduleName = moduleName;
  __app: $.AppType | null = null;
  static structName: string = 'LinearCap';
  static typeParameters: TypeParamDeclType[] = [{ name: 'Feature', isPhantom: true }];
  static fields: FieldDeclType[] = [{ name: 'root', typeTag: AtomicTypeTag.Address }];

  root: HexString;

  constructor(proto: any, public typeTag: TypeTag) {
    this.root = proto['root'] as HexString;
  }

  static LinearCapParser(data: any, typeTag: TypeTag, repo: AptosParserRepo): LinearCap {
    const proto = $.parseStructProto(data, typeTag, repo, LinearCap);
    return new LinearCap(proto, typeTag);
  }

  static makeTag($p: TypeTag[]): StructTag {
    return new StructTag(moduleAddress, moduleName, 'LinearCap', $p);
  }
  async loadFullState(app: $.AppType) {
    this.__app = app;
  }
}
export function acquire_(
  requester: HexString,
  _feature_witness: any,
  $c: AptosDataCache,
  $p: TypeTag[] /* <Feature>*/
): Cap {
  return new Cap({ root: validate_acquire_(requester, $c, [$p[0]]) }, new SimpleStructTag(Cap, [$p[0]]));
}

export function acquire_linear_(
  requester: HexString,
  _feature_witness: any,
  $c: AptosDataCache,
  $p: TypeTag[] /* <Feature>*/
): LinearCap {
  return new LinearCap({ root: validate_acquire_(requester, $c, [$p[0]]) }, new SimpleStructTag(LinearCap, [$p[0]]));
}

export function add_element_(v: any[], x: any, $c: AptosDataCache, $p: TypeTag[] /* <E>*/): void {
  let temp$1, temp$2;
  [temp$1, temp$2] = [v, x];
  if (!Vector.contains_(temp$1, temp$2, $c, [$p[0]])) {
    Vector.push_back_(v, x, $c, [$p[0]]);
  } else {
  }
  return;
}

export function create_(
  owner: HexString,
  _feature_witness: any,
  $c: AptosDataCache,
  $p: TypeTag[] /* <Feature>*/
): void {
  let addr;
  addr = Signer.address_of_(owner, $c);
  if (!!$c.exists(new SimpleStructTag(CapState, [$p[0]]), $.copy(addr))) {
    throw $.abortCode(Error.already_exists_($.copy(ECAPABILITY_ALREADY_EXISTS), $c));
  }
  $c.move_to(
    new SimpleStructTag(CapState, [$p[0]]),
    owner,
    new CapState({ delegates: Vector.empty_($c, [AtomicTypeTag.Address]) }, new SimpleStructTag(CapState, [$p[0]]))
  );
  return;
}

export function delegate_(
  cap: Cap,
  _feature_witness: any,
  to: HexString,
  $c: AptosDataCache,
  $p: TypeTag[] /* <Feature>*/
): void {
  let addr;
  addr = Signer.address_of_(to, $c);
  if ($c.exists(new SimpleStructTag(CapDelegateState, [$p[0]]), $.copy(addr))) {
    return;
  } else {
  }
  $c.move_to(
    new SimpleStructTag(CapDelegateState, [$p[0]]),
    to,
    new CapDelegateState({ root: $.copy(cap.root) }, new SimpleStructTag(CapDelegateState, [$p[0]]))
  );
  add_element_(
    $c.borrow_global_mut<CapState>(new SimpleStructTag(CapState, [$p[0]]), $.copy(cap.root)).delegates,
    $.copy(addr),
    $c,
    [AtomicTypeTag.Address]
  );
  return;
}

export function linear_root_addr_(
  cap: LinearCap,
  _feature_witness: any,
  $c: AptosDataCache,
  $p: TypeTag[] /* <Feature>*/
): HexString {
  return $.copy(cap.root);
}

export function remove_element_(v: any[], x: any, $c: AptosDataCache, $p: TypeTag[] /* <E>*/): void {
  let temp$1, temp$2, found, index;
  [temp$1, temp$2] = [v, x];
  [found, index] = Vector.index_of_(temp$1, temp$2, $c, [$p[0]]);
  if (found) {
    Vector.remove_(v, $.copy(index), $c, [$p[0]]);
  } else {
  }
  return;
}

export function revoke_(
  cap: Cap,
  _feature_witness: any,
  from: HexString,
  $c: AptosDataCache,
  $p: TypeTag[] /* <Feature>*/
): void {
  if (!$c.exists(new SimpleStructTag(CapDelegateState, [$p[0]]), $.copy(from))) {
    return;
  } else {
  }
  let { root: _root } = $c.move_from<CapDelegateState>(new SimpleStructTag(CapDelegateState, [$p[0]]), $.copy(from));
  remove_element_(
    $c.borrow_global_mut<CapState>(new SimpleStructTag(CapState, [$p[0]]), $.copy(cap.root)).delegates,
    from,
    $c,
    [AtomicTypeTag.Address]
  );
  return;
}

export function root_addr_(
  cap: Cap,
  _feature_witness: any,
  $c: AptosDataCache,
  $p: TypeTag[] /* <Feature>*/
): HexString {
  return $.copy(cap.root);
}

export function validate_acquire_(requester: HexString, $c: AptosDataCache, $p: TypeTag[] /* <Feature>*/): HexString {
  let temp$1, addr, root_addr;
  addr = Signer.address_of_(requester, $c);
  if ($c.exists(new SimpleStructTag(CapDelegateState, [$p[0]]), $.copy(addr))) {
    root_addr = $.copy(
      $c.borrow_global<CapDelegateState>(new SimpleStructTag(CapDelegateState, [$p[0]]), $.copy(addr)).root
    );
    if (!$c.exists(new SimpleStructTag(CapState, [$p[0]]), $.copy(root_addr))) {
      throw $.abortCode(Error.invalid_state_($.copy(EDELEGATE), $c));
    }
    if (
      !Vector.contains_(
        $c.borrow_global<CapState>(new SimpleStructTag(CapState, [$p[0]]), $.copy(root_addr)).delegates,
        addr,
        $c,
        [AtomicTypeTag.Address]
      )
    ) {
      throw $.abortCode(Error.invalid_state_($.copy(EDELEGATE), $c));
    }
    temp$1 = $.copy(root_addr);
  } else {
    if (!$c.exists(new SimpleStructTag(CapState, [$p[0]]), $.copy(addr))) {
      throw $.abortCode(Error.not_found_($.copy(ECAPABILITY_NOT_FOUND), $c));
    }
    temp$1 = $.copy(addr);
  }
  return temp$1;
}

export function loadParsers(repo: AptosParserRepo) {
  repo.addParser('0x1::capability::Cap', Cap.CapParser);
  repo.addParser('0x1::capability::CapDelegateState', CapDelegateState.CapDelegateStateParser);
  repo.addParser('0x1::capability::CapState', CapState.CapStateParser);
  repo.addParser('0x1::capability::LinearCap', LinearCap.LinearCapParser);
}
export class App {
  constructor(public client: AptosClient, public repo: AptosParserRepo, public cache: AptosLocalCache) {}
  get moduleAddress() {
    {
      return moduleAddress;
    }
  }
  get moduleName() {
    {
      return moduleName;
    }
  }
  get Cap() {
    return Cap;
  }
  get CapDelegateState() {
    return CapDelegateState;
  }
  async loadCapDelegateState(owner: HexString, $p: TypeTag[] /* <Feature> */, loadFull = true, fillCache = true) {
    const val = await CapDelegateState.load(this.repo, this.client, owner, $p);
    if (loadFull) {
      await val.loadFullState(this);
    }
    if (fillCache) {
      this.cache.set(val.typeTag, owner, val);
    }
    return val;
  }
  get CapState() {
    return CapState;
  }
  async loadCapState(owner: HexString, $p: TypeTag[] /* <Feature> */, loadFull = true, fillCache = true) {
    const val = await CapState.load(this.repo, this.client, owner, $p);
    if (loadFull) {
      await val.loadFullState(this);
    }
    if (fillCache) {
      this.cache.set(val.typeTag, owner, val);
    }
    return val;
  }
  get LinearCap() {
    return LinearCap;
  }
}