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 Comparator from './comparator';
import * as Error from './error';
import * as Option from './option';
import * as Vector from './vector';
export const packageName = 'AptosStdlib';
export const moduleAddress = new HexString('0x1');
export const moduleName = 'simple_map';

export const EKEY_ALREADY_EXISTS: U64 = u64('1');
export const EKEY_NOT_FOUND: U64 = u64('2');

export class Element {
  static moduleAddress = moduleAddress;
  static moduleName = moduleName;
  __app: $.AppType | null = null;
  static structName: string = 'Element';
  static typeParameters: TypeParamDeclType[] = [
    { name: 'Key', isPhantom: false },
    { name: 'Value', isPhantom: false }
  ];
  static fields: FieldDeclType[] = [
    { name: 'key', typeTag: new $.TypeParamIdx(0) },
    { name: 'value', typeTag: new $.TypeParamIdx(1) }
  ];

  key: any;
  value: any;

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

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

  static makeTag($p: TypeTag[]): StructTag {
    return new StructTag(moduleAddress, moduleName, 'Element', $p);
  }
  async loadFullState(app: $.AppType) {
    if (this.key.typeTag instanceof StructTag) {
      await this.key.loadFullState(app);
    }
    if (this.value.typeTag instanceof StructTag) {
      await this.value.loadFullState(app);
    }
    this.__app = app;
  }
}

export class SimpleMap {
  static moduleAddress = moduleAddress;
  static moduleName = moduleName;
  __app: $.AppType | null = null;
  static structName: string = 'SimpleMap';
  static typeParameters: TypeParamDeclType[] = [
    { name: 'Key', isPhantom: false },
    { name: 'Value', isPhantom: false }
  ];
  static fields: FieldDeclType[] = [
    {
      name: 'data',
      typeTag: new VectorTag(
        new StructTag(new HexString('0x1'), 'simple_map', 'Element', [new $.TypeParamIdx(0), new $.TypeParamIdx(1)])
      )
    }
  ];

  data: Element[];

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

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

  static makeTag($p: TypeTag[]): StructTag {
    return new StructTag(moduleAddress, moduleName, 'SimpleMap', $p);
  }
  async loadFullState(app: $.AppType) {
    this.__app = app;
  }
}
export function add_(map: SimpleMap, key: any, value: any, $c: AptosDataCache, $p: TypeTag[] /* <Key, Value>*/): void {
  let temp$1, temp$2, end, maybe_idx, maybe_placement, placement;
  [temp$1, temp$2] = [map, key];
  [maybe_idx, maybe_placement] = find_(temp$1, temp$2, $c, [$p[0], $p[1]]);
  if (!Option.is_none_(maybe_idx, $c, [AtomicTypeTag.U64])) {
    throw $.abortCode(Error.invalid_argument_($.copy(EKEY_ALREADY_EXISTS), $c));
  }
  Vector.push_back_(
    map.data,
    new Element({ key: key, value: value }, new SimpleStructTag(Element, [$p[0], $p[1]])),
    $c,
    [new SimpleStructTag(Element, [$p[0], $p[1]])]
  );
  placement = Option.extract_(maybe_placement, $c, [AtomicTypeTag.U64]);
  end = Vector.length_(map.data, $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]).sub(u64('1'));
  while ($.copy(placement).lt($.copy(end))) {
    {
      Vector.swap_(map.data, $.copy(placement), $.copy(end), $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]);
      placement = $.copy(placement).add(u64('1'));
    }
  }
  return;
}

export function borrow_(map: SimpleMap, key: any, $c: AptosDataCache, $p: TypeTag[] /* <Key, Value>*/): any {
  let idx, maybe_idx;
  [maybe_idx] = find_(map, key, $c, [$p[0], $p[1]]);
  if (!Option.is_some_(maybe_idx, $c, [AtomicTypeTag.U64])) {
    throw $.abortCode(Error.invalid_argument_($.copy(EKEY_NOT_FOUND), $c));
  }
  idx = Option.extract_(maybe_idx, $c, [AtomicTypeTag.U64]);
  return Vector.borrow_(map.data, $.copy(idx), $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]).value;
}

export function borrow_mut_(map: SimpleMap, key: any, $c: AptosDataCache, $p: TypeTag[] /* <Key, Value>*/): any {
  let temp$1, temp$2, idx, maybe_idx;
  [temp$1, temp$2] = [map, key];
  [maybe_idx] = find_(temp$1, temp$2, $c, [$p[0], $p[1]]);
  if (!Option.is_some_(maybe_idx, $c, [AtomicTypeTag.U64])) {
    throw $.abortCode(Error.invalid_argument_($.copy(EKEY_NOT_FOUND), $c));
  }
  idx = Option.extract_(maybe_idx, $c, [AtomicTypeTag.U64]);
  return Vector.borrow_mut_(map.data, $.copy(idx), $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]).value;
}

export function contains_key_(map: SimpleMap, key: any, $c: AptosDataCache, $p: TypeTag[] /* <Key, Value>*/): boolean {
  let maybe_idx;
  [maybe_idx] = find_(map, key, $c, [$p[0], $p[1]]);
  return Option.is_some_(maybe_idx, $c, [AtomicTypeTag.U64]);
}

export function create_($c: AptosDataCache, $p: TypeTag[] /* <Key, Value>*/): SimpleMap {
  return new SimpleMap(
    { data: Vector.empty_($c, [new SimpleStructTag(Element, [$p[0], $p[1]])]) },
    new SimpleStructTag(SimpleMap, [$p[0], $p[1]])
  );
}

export function destroy_empty_(map: SimpleMap, $c: AptosDataCache, $p: TypeTag[] /* <Key, Value>*/): void {
  let { data: data } = map;
  Vector.destroy_empty_(data, $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]);
  return;
}

export function find_(
  map: SimpleMap,
  key: any,
  $c: AptosDataCache,
  $p: TypeTag[] /* <Key, Value>*/
): [Option.Option, Option.Option] {
  let temp$1, temp$2, temp$3, temp$4, left, length, mid, potential_key, right;
  length = Vector.length_(map.data, $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]);
  if ($.copy(length).eq(u64('0'))) {
    return [Option.none_($c, [AtomicTypeTag.U64]), Option.some_(u64('0'), $c, [AtomicTypeTag.U64])];
  } else {
  }
  left = u64('0');
  right = $.copy(length);
  while ($.copy(left).neq($.copy(right))) {
    {
      mid = $.copy(left).add($.copy(right).sub($.copy(left)).div(u64('2')));
      potential_key = Vector.borrow_(map.data, $.copy(mid), $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]).key;
      temp$1 = Comparator.compare_(potential_key, key, $c, [$p[0]]);
      if (Comparator.is_smaller_than_(temp$1, $c)) {
        left = $.copy(mid).add(u64('1'));
      } else {
        right = $.copy(mid);
      }
    }
  }
  if ($.copy(left).neq($.copy(length))) {
    temp$2 = $.dyn_eq(
      $p[0],
      key,
      Vector.borrow_(map.data, $.copy(left), $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]).key
    );
  } else {
    temp$2 = false;
  }
  if (temp$2) {
    [temp$3, temp$4] = [Option.some_($.copy(left), $c, [AtomicTypeTag.U64]), Option.none_($c, [AtomicTypeTag.U64])];
  } else {
    [temp$3, temp$4] = [Option.none_($c, [AtomicTypeTag.U64]), Option.some_($.copy(left), $c, [AtomicTypeTag.U64])];
  }
  return [temp$3, temp$4];
}

export function length_(map: SimpleMap, $c: AptosDataCache, $p: TypeTag[] /* <Key, Value>*/): U64 {
  return Vector.length_(map.data, $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]);
}

export function remove_(map: SimpleMap, key: any, $c: AptosDataCache, $p: TypeTag[] /* <Key, Value>*/): [any, any] {
  let temp$1, temp$2, end, maybe_idx, placement;
  [temp$1, temp$2] = [map, key];
  [maybe_idx] = find_(temp$1, temp$2, $c, [$p[0], $p[1]]);
  if (!Option.is_some_(maybe_idx, $c, [AtomicTypeTag.U64])) {
    throw $.abortCode(Error.invalid_argument_($.copy(EKEY_NOT_FOUND), $c));
  }
  placement = Option.extract_(maybe_idx, $c, [AtomicTypeTag.U64]);
  end = Vector.length_(map.data, $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]).sub(u64('1'));
  while ($.copy(placement).lt($.copy(end))) {
    {
      Vector.swap_(map.data, $.copy(placement), $.copy(placement).add(u64('1')), $c, [
        new SimpleStructTag(Element, [$p[0], $p[1]])
      ]);
      placement = $.copy(placement).add(u64('1'));
    }
  }
  let { key: key__3, value: value } = Vector.pop_back_(map.data, $c, [new SimpleStructTag(Element, [$p[0], $p[1]])]);
  return [key__3, value];
}

export function loadParsers(repo: AptosParserRepo) {
  repo.addParser('0x1::simple_map::Element', Element.ElementParser);
  repo.addParser('0x1::simple_map::SimpleMap', SimpleMap.SimpleMapParser);
}
export class App {
  constructor(public client: AptosClient, public repo: AptosParserRepo, public cache: AptosLocalCache) {}
  get moduleAddress() {
    {
      return moduleAddress;
    }
  }
  get moduleName() {
    {
      return moduleName;
    }
  }
  get Element() {
    return Element;
  }
  get SimpleMap() {
    return SimpleMap;
  }
}