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 System_addresses from './system_addresses';
export const packageName = 'AptosFramework';
export const moduleAddress = new HexString('0x1');
export const moduleName = 'staking_config';

export const EINVALID_REWARDS_RATE: U64 = u64('5');
export const EINVALID_STAKE_RANGE: U64 = u64('3');
export const EINVALID_VOTING_POWER_INCREASE_LIMIT: U64 = u64('4');
export const EZERO_LOCKUP_DURATION: U64 = u64('1');
export const EZERO_REWARDS_RATE_DENOMINATOR: U64 = u64('2');
export const MAX_REWARDS_RATE: U64 = u64('1000000');

export class StakingConfig {
  static moduleAddress = moduleAddress;
  static moduleName = moduleName;
  __app: $.AppType | null = null;
  static structName: string = 'StakingConfig';
  static typeParameters: TypeParamDeclType[] = [];
  static fields: FieldDeclType[] = [
    { name: 'minimum_stake', typeTag: AtomicTypeTag.U64 },
    { name: 'maximum_stake', typeTag: AtomicTypeTag.U64 },
    { name: 'recurring_lockup_duration_secs', typeTag: AtomicTypeTag.U64 },
    { name: 'allow_validator_set_change', typeTag: AtomicTypeTag.Bool },
    { name: 'rewards_rate', typeTag: AtomicTypeTag.U64 },
    { name: 'rewards_rate_denominator', typeTag: AtomicTypeTag.U64 },
    { name: 'voting_power_increase_limit', typeTag: AtomicTypeTag.U64 }
  ];

  minimum_stake: U64;
  maximum_stake: U64;
  recurring_lockup_duration_secs: U64;
  allow_validator_set_change: boolean;
  rewards_rate: U64;
  rewards_rate_denominator: U64;
  voting_power_increase_limit: U64;

  constructor(proto: any, public typeTag: TypeTag) {
    this.minimum_stake = proto['minimum_stake'] as U64;
    this.maximum_stake = proto['maximum_stake'] as U64;
    this.recurring_lockup_duration_secs = proto['recurring_lockup_duration_secs'] as U64;
    this.allow_validator_set_change = proto['allow_validator_set_change'] as boolean;
    this.rewards_rate = proto['rewards_rate'] as U64;
    this.rewards_rate_denominator = proto['rewards_rate_denominator'] as U64;
    this.voting_power_increase_limit = proto['voting_power_increase_limit'] as U64;
  }

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

  static async load(repo: AptosParserRepo, client: AptosClient, address: HexString, typeParams: TypeTag[]) {
    const result = await repo.loadResource(client, address, StakingConfig, typeParams);
    return result as unknown as StakingConfig;
  }
  static async loadByApp(app: $.AppType, address: HexString, typeParams: TypeTag[]) {
    const result = await app.repo.loadResource(app.client, address, StakingConfig, typeParams);
    await result.loadFullState(app);
    return result as unknown as StakingConfig;
  }
  static getTag(): StructTag {
    return new StructTag(moduleAddress, moduleName, 'StakingConfig', []);
  }
  async loadFullState(app: $.AppType) {
    this.__app = app;
  }
}
export function get_($c: AptosDataCache): StakingConfig {
  return $.copy($c.borrow_global<StakingConfig>(new SimpleStructTag(StakingConfig), new HexString('0x1')));
}

export function get_allow_validator_set_change_(config: StakingConfig, $c: AptosDataCache): boolean {
  return $.copy(config.allow_validator_set_change);
}

export function get_recurring_lockup_duration_(config: StakingConfig, $c: AptosDataCache): U64 {
  return $.copy(config.recurring_lockup_duration_secs);
}

export function get_required_stake_(config: StakingConfig, $c: AptosDataCache): [U64, U64] {
  return [$.copy(config.minimum_stake), $.copy(config.maximum_stake)];
}

export function get_reward_rate_(config: StakingConfig, $c: AptosDataCache): [U64, U64] {
  return [$.copy(config.rewards_rate), $.copy(config.rewards_rate_denominator)];
}

export function get_voting_power_increase_limit_(config: StakingConfig, $c: AptosDataCache): U64 {
  return $.copy(config.voting_power_increase_limit);
}

export function initialize_(
  aptos_framework: HexString,
  minimum_stake: U64,
  maximum_stake: U64,
  recurring_lockup_duration_secs: U64,
  allow_validator_set_change: boolean,
  rewards_rate: U64,
  rewards_rate_denominator: U64,
  voting_power_increase_limit: U64,
  $c: AptosDataCache
): void {
  let temp$1;
  System_addresses.assert_aptos_framework_(aptos_framework, $c);
  validate_required_stake_($.copy(minimum_stake), $.copy(maximum_stake), $c);
  if (!$.copy(recurring_lockup_duration_secs).gt(u64('0'))) {
    throw $.abortCode(Error.invalid_argument_($.copy(EZERO_LOCKUP_DURATION), $c));
  }
  if (!$.copy(rewards_rate_denominator).gt(u64('0'))) {
    throw $.abortCode(Error.invalid_argument_($.copy(EZERO_REWARDS_RATE_DENOMINATOR), $c));
  }
  if ($.copy(voting_power_increase_limit).gt(u64('0'))) {
    temp$1 = $.copy(voting_power_increase_limit).le(u64('50'));
  } else {
    temp$1 = false;
  }
  if (!temp$1) {
    throw $.abortCode(Error.invalid_argument_($.copy(EINVALID_VOTING_POWER_INCREASE_LIMIT), $c));
  }
  if (!$.copy(rewards_rate).le($.copy(MAX_REWARDS_RATE))) {
    throw $.abortCode(Error.invalid_argument_($.copy(EINVALID_REWARDS_RATE), $c));
  }
  if (!$.copy(rewards_rate).le($.copy(rewards_rate_denominator))) {
    throw $.abortCode(Error.invalid_argument_($.copy(EINVALID_REWARDS_RATE), $c));
  }
  $c.move_to(
    new SimpleStructTag(StakingConfig),
    aptos_framework,
    new StakingConfig(
      {
        minimum_stake: $.copy(minimum_stake),
        maximum_stake: $.copy(maximum_stake),
        recurring_lockup_duration_secs: $.copy(recurring_lockup_duration_secs),
        allow_validator_set_change: allow_validator_set_change,
        rewards_rate: $.copy(rewards_rate),
        rewards_rate_denominator: $.copy(rewards_rate_denominator),
        voting_power_increase_limit: $.copy(voting_power_increase_limit)
      },
      new SimpleStructTag(StakingConfig)
    )
  );
  return;
}

export function update_recurring_lockup_duration_secs_(
  aptos_framework: HexString,
  new_recurring_lockup_duration_secs: U64,
  $c: AptosDataCache
): void {
  let staking_config;
  if (!$.copy(new_recurring_lockup_duration_secs).gt(u64('0'))) {
    throw $.abortCode(Error.invalid_argument_($.copy(EZERO_LOCKUP_DURATION), $c));
  }
  System_addresses.assert_aptos_framework_(aptos_framework, $c);
  staking_config = $c.borrow_global_mut<StakingConfig>(new SimpleStructTag(StakingConfig), new HexString('0x1'));
  staking_config.recurring_lockup_duration_secs = $.copy(new_recurring_lockup_duration_secs);
  return;
}

export function update_required_stake_(
  aptos_framework: HexString,
  minimum_stake: U64,
  maximum_stake: U64,
  $c: AptosDataCache
): void {
  let staking_config;
  System_addresses.assert_aptos_framework_(aptos_framework, $c);
  validate_required_stake_($.copy(minimum_stake), $.copy(maximum_stake), $c);
  staking_config = $c.borrow_global_mut<StakingConfig>(new SimpleStructTag(StakingConfig), new HexString('0x1'));
  staking_config.minimum_stake = $.copy(minimum_stake);
  staking_config.maximum_stake = $.copy(maximum_stake);
  return;
}

export function update_rewards_rate_(
  aptos_framework: HexString,
  new_rewards_rate: U64,
  new_rewards_rate_denominator: U64,
  $c: AptosDataCache
): void {
  let staking_config;
  System_addresses.assert_aptos_framework_(aptos_framework, $c);
  if (!$.copy(new_rewards_rate_denominator).gt(u64('0'))) {
    throw $.abortCode(Error.invalid_argument_($.copy(EZERO_REWARDS_RATE_DENOMINATOR), $c));
  }
  if (!$.copy(new_rewards_rate).le($.copy(MAX_REWARDS_RATE))) {
    throw $.abortCode(Error.invalid_argument_($.copy(EINVALID_REWARDS_RATE), $c));
  }
  if (!$.copy(new_rewards_rate).le($.copy(new_rewards_rate_denominator))) {
    throw $.abortCode(Error.invalid_argument_($.copy(EINVALID_REWARDS_RATE), $c));
  }
  staking_config = $c.borrow_global_mut<StakingConfig>(new SimpleStructTag(StakingConfig), new HexString('0x1'));
  staking_config.rewards_rate = $.copy(new_rewards_rate);
  staking_config.rewards_rate_denominator = $.copy(new_rewards_rate_denominator);
  return;
}

export function update_voting_power_increase_limit_(
  aptos_framework: HexString,
  new_voting_power_increase_limit: U64,
  $c: AptosDataCache
): void {
  let temp$1, staking_config;
  System_addresses.assert_aptos_framework_(aptos_framework, $c);
  if ($.copy(new_voting_power_increase_limit).gt(u64('0'))) {
    temp$1 = $.copy(new_voting_power_increase_limit).le(u64('50'));
  } else {
    temp$1 = false;
  }
  if (!temp$1) {
    throw $.abortCode(Error.invalid_argument_($.copy(EINVALID_VOTING_POWER_INCREASE_LIMIT), $c));
  }
  staking_config = $c.borrow_global_mut<StakingConfig>(new SimpleStructTag(StakingConfig), new HexString('0x1'));
  staking_config.voting_power_increase_limit = $.copy(new_voting_power_increase_limit);
  return;
}

export function validate_required_stake_(minimum_stake: U64, maximum_stake: U64, $c: AptosDataCache): void {
  let temp$1;
  if ($.copy(minimum_stake).le($.copy(maximum_stake))) {
    temp$1 = $.copy(maximum_stake).gt(u64('0'));
  } else {
    temp$1 = false;
  }
  if (!temp$1) {
    throw $.abortCode(Error.invalid_argument_($.copy(EINVALID_STAKE_RANGE), $c));
  }
  return;
}

export function loadParsers(repo: AptosParserRepo) {
  repo.addParser('0x1::staking_config::StakingConfig', StakingConfig.StakingConfigParser);
}
export class App {
  constructor(public client: AptosClient, public repo: AptosParserRepo, public cache: AptosLocalCache) {}
  get moduleAddress() {
    {
      return moduleAddress;
    }
  }
  get moduleName() {
    {
      return moduleName;
    }
  }
  get StakingConfig() {
    return StakingConfig;
  }
  async loadStakingConfig(owner: HexString, loadFull = true, fillCache = true) {
    const val = await StakingConfig.load(this.repo, this.client, owner, [] as TypeTag[]);
    if (loadFull) {
      await val.loadFullState(this);
    }
    if (fillCache) {
      this.cache.set(val.typeTag, owner, val);
    }
    return val;
  }
}