import {
  InternalMoveFunction,
  InternalMoveFunctionVisibility,
  InternalMoveModule,
  InternalMoveStruct,
  InternalMoveStructField,
  parseMoveType
} from '@typemove/move'
import {
  MoveFunctionVisibility,
  MoveFunction,
  MoveModuleBytecode,
  MoveStruct,
  MoveStructField
} from '@aptos-labs/ts-sdk'

export function toInternalModule(module: MoveModuleBytecode): InternalMoveModule {
  if (!module.abi) {
    throw Error('module with no ABI found')
  }
  const abi = module.abi
  return {
    address: abi.address,
    exposedFunctions: abi.exposed_functions.map(toInternalFunction),
    name: abi.name,
    structs: abi.structs.map(toInternalStruct),
    enums: []
  }
}

export function toInternalFunction(func: MoveFunction): InternalMoveFunction {
  let visibility
  switch (func.visibility) {
    case MoveFunctionVisibility.PRIVATE:
      visibility = InternalMoveFunctionVisibility.PRIVATE
      break
    case MoveFunctionVisibility.PUBLIC:
      visibility = InternalMoveFunctionVisibility.PUBLIC
      break
    case MoveFunctionVisibility.FRIEND:
      visibility = InternalMoveFunctionVisibility.FRIEND
      break
  }
  return {
    typeParams: func.generic_type_params,
    isEntry: func.is_entry,
    isView: func.is_view,
    name: func.name,
    params: func.params.map(parseMoveType),
    return: func.return.map(parseMoveType),
    visibility: visibility
  }
}

export function toInternalStruct(struct: MoveStruct): InternalMoveStruct {
  return {
    abilities: struct.abilities,
    fields: struct.fields.map(toInternalField),
    typeParams: struct.generic_type_params,
    isNative: struct.is_native,
    isEvent: struct.is_event,
    name: struct.name
  }
}

export function toInternalField(module: MoveStructField): InternalMoveStructField {
  return {
    name: module.name,
    type: parseMoveType(module.type)
  }
}