import { arweave } from '../chain/arweave';
import { cosmos } from '../chain/cosmos';
import { ethereum } from '../chain/ethereum';
import { near } from '../chain/near';
import { Bytes, Entity, Result, TypedMap, TypedMapEntry, Wrapped } from '../common/collections';
import { BigDecimal } from '../common/numbers';
import { JSONValue, Value } from '../common/value';

/**
 * Contains type IDs and their discriminants for every blockchain supported by Graph-Node.
 *
 * Each variant corresponds to the unique ID of an AssemblyScript concrete class used by
 * Graph-Node's runtime.
 *
 * # Rules for updating this enum
 *
 * 1. The discriminants must have the same value as their counterparts in `IndexForAscTypeId` enum
 *    from `graph-node`'s `graph::runtime` module. If not, the runtime will fail to determine the
 *    correct class during allocation.
 * 2. Each supported blockchain has a reserved space of 1,000 * contiguous variants.
 * 3. Once defined, items and their discriminants cannot be changed, as this would break running
 *    subgraphs compiled in previous versions of this representation.
 */
export enum TypeId {
  String = 0,
  ArrayBuffer = 1,
  Int8Array = 2,
  Int16Array = 3,
  Int32Array = 4,
  Int64Array = 5,
  Uint8Array = 6,
  Uint16Array = 7,
  Uint32Array = 8,
  Uint64Array = 9,
  Float32Array = 10,
  Float64Array = 11,
  BigDecimal = 12,
  ArrayBool = 13,
  ArrayUint8Array = 14,
  ArrayEthereumValue = 15,
  ArrayStoreValue = 16,
  ArrayJsonValue = 17,
  ArrayString = 18,
  ArrayEventParam = 19,
  ArrayTypedMapEntryStringJsonValue = 20,
  ArrayTypedMapEntryStringStoreValue = 21,
  SmartContractCall = 22,
  EventParam = 23,
  EthereumTransaction = 24,
  EthereumBlock = 25,
  EthereumCall = 26,
  WrappedTypedMapStringJsonValue = 27,
  WrappedBool = 28,
  WrappedJsonValue = 29,
  EthereumValue = 30,
  StoreValue = 31,
  JsonValue = 32,
  EthereumEvent = 33,
  TypedMapEntryStringStoreValue = 34,
  TypedMapEntryStringJsonValue = 35,
  TypedMapStringStoreValue = 36,
  TypedMapStringJsonValue = 37,
  TypedMapStringTypedMapStringJsonValue = 38,
  ResultTypedMapStringJsonValueBool = 39,
  ResultJsonValueBool = 40,
  ArrayU8 = 41,
  ArrayU16 = 42,
  ArrayU32 = 43,
  ArrayU64 = 44,
  ArrayI8 = 45,
  ArrayI16 = 46,
  ArrayI32 = 47,
  ArrayI64 = 48,
  ArrayF32 = 49,
  ArrayF64 = 50,
  ArrayBigDecimal = 51,

  // Near types
  NearArrayDataReceiver = 52,
  NearArrayCryptoHash = 53,
  NearArrayActionValue = 54,
  NearMerklePath = 55, // or NearArrayMerklePathItem
  NearArrayValidatorStake = 56,
  NearArraySlashedValidator = 57,
  NearArraySignature = 58,
  NearArrayChunkHeader = 59,
  NearAccessKeyPermissionValue = 60,
  NearActionValue = 61,
  NearDirection = 62, // not used in graph-node anymore. Can be ignored.
  NearPublicKey = 63,
  NearSignature = 64,
  NearFunctionCallPermission = 65,
  NearFullAccessPermission = 66,
  NearAccessKey = 67,
  NearDataReceiver = 68,
  NearCreateAccountAction = 69,
  NearDeployContractAction = 70,
  NearFunctionCallAction = 71,
  NearTransferAction = 72,
  NearStakeAction = 73,
  NearAddKeyAction = 74,
  NearDeleteKeyAction = 75,
  NearDeleteAccountAction = 76,
  NearActionReceipt = 77,
  NearSuccessStatus = 78,
  NearMerklePathItem = 79,
  NearExecutionOutcome = 80,
  NearSlashedValidator = 81,
  NearBlockHeader = 82,
  NearValidatorStake = 83,
  NearChunkHeader = 84,
  NearBlock = 85,
  NearReceiptWithOutcome = 86,
  /*
  Reserved discriminant space for more Near type IDs: [87, 999]:
  Continue to add more Near type IDs here. e.g.:
  ```
  NextNearType = 87,
  AnotherNearType = 88,
  ...
  LastNearType = 999,
  ```
  */

  // Reserved discriminant space for more Ethereum type IDs: [1000, 1499]
  TransactionReceipt = 1000,
  Log = 1001,
  ArrayH256 = 1002,
  ArrayLog = 1003,
  /*
  Continue to add more Ethereum type IDs here. e.g.:
  ```
  NextEthereumType = 1004,
  AnotherEthereumType = 1005,
  ...
  LastEthereumType = 1499,
  ```
  */

  // Reserved discriminant space for Cosmos type IDs: [1,500, 2,499]
  CosmosAny = 1500,
  CosmosAnyArray = 1501,
  CosmosBytesArray = 1502,
  CosmosCoinArray = 1503,
  CosmosCommitSigArray = 1504,
  CosmosEventArray = 1505,
  CosmosEventAttributeArray = 1506,
  CosmosEvidenceArray = 1507,
  CosmosModeInfoArray = 1508,
  CosmosSignerInfoArray = 1509,
  CosmosTxResultArray = 1510,
  CosmosValidatorArray = 1511,
  CosmosValidatorUpdateArray = 1512,
  CosmosAuthInfo = 1513,
  CosmosBlock = 1514,
  CosmosBlockId = 1515,
  CosmosBlockIdFlagEnum = 1516,
  CosmosBlockParams = 1517,
  CosmosCoin = 1518,
  CosmosCommit = 1519,
  CosmosCommitSig = 1520,
  CosmosCompactBitArray = 1521,
  CosmosConsensus = 1522,
  CosmosConsensusParams = 1523,
  CosmosDuplicateVoteEvidence = 1524,
  CosmosDuration = 1525,
  CosmosEvent = 1526,
  CosmosEventAttribute = 1527,
  CosmosEventData = 1528,
  CosmosEventVote = 1529,
  CosmosEvidence = 1530,
  CosmosEvidenceList = 1531,
  CosmosEvidenceParams = 1532,
  CosmosFee = 1533,
  CosmosHeader = 1534,
  CosmosHeaderOnlyBlock = 1535,
  CosmosLightBlock = 1536,
  CosmosLightClientAttackEvidence = 1537,
  CosmosModeInfo = 1538,
  CosmosModeInfoMulti = 1539,
  CosmosModeInfoSingle = 1540,
  CosmosPartSetHeader = 1541,
  CosmosPublicKey = 1542,
  CosmosResponseBeginBlock = 1543,
  CosmosResponseDeliverTx = 1544,
  CosmosResponseEndBlock = 1545,
  CosmosSignModeEnum = 1546,
  CosmosSignedHeader = 1547,
  CosmosSignedMsgTypeEnum = 1548,
  CosmosSignerInfo = 1549,
  CosmosTimestamp = 1550,
  CosmosTip = 1551,
  CosmosTransactionData = 1552,
  CosmosTx = 1553,
  CosmosTxBody = 1554,
  CosmosTxResult = 1555,
  CosmosValidator = 1556,
  CosmosValidatorParams = 1557,
  CosmosValidatorSet = 1558,
  CosmosValidatorSetUpdates = 1559,
  CosmosValidatorUpdate = 1560,
  CosmosVersionParams = 1561,
  CosmosMessageData = 1562,
  CosmosTransactionContext = 1563,
  /*
  Continue to add more Cosmos type IDs here. e.g.:
  ```
  NextCosmosType = 1564,
  AnotherCosmosType = 1565,
  ...
  LastCosmosType = 2499,
  ```
  */

  // Reserved discriminant space for Tendermint type IDs: [2,500, 3,499]
  ArweaveBlock = 2500,
  ArweaveProofOfAccess = 2501,
  ArweaveTag = 2502,
  ArweaveTagArray = 2503,
  ArweaveTransaction = 2504,
  ArweaveTransactionArray = 2505,
  ArweaveTransactionWithBlockPtr = 2506,
  /*
  Continue to add more Arweave type IDs here. e.g.:
  ```
  NextArweaveType = 2507,
  AnotherArweaveType = 2508,
  ...
  LastArweaveType = 3499,
  ```
  */

  // Reserved discriminant space for a future blockchain type IDs: [3,500, 4,499]
}

export function id_of_type(typeId: TypeId): usize {
  switch (typeId) {
    case TypeId.String:
      return idof<string>();
    case TypeId.ArrayBuffer:
      return idof<ArrayBuffer>();
    case TypeId.Int8Array:
      return idof<Int8Array>();
    case TypeId.Int16Array:
      return idof<Int16Array>();
    case TypeId.Int32Array:
      return idof<Int32Array>();
    case TypeId.Int64Array:
      return idof<Int64Array>();
    case TypeId.Uint8Array:
      return idof<Uint8Array>();
    case TypeId.Uint16Array:
      return idof<Uint16Array>();
    case TypeId.Uint32Array:
      return idof<Uint32Array>();
    case TypeId.Uint64Array:
      return idof<Uint64Array>();
    case TypeId.Float32Array:
      return idof<Float32Array>();
    case TypeId.Float64Array:
      return idof<Float64Array>();
    case TypeId.BigDecimal:
      return idof<BigDecimal>();
    case TypeId.ArrayBool:
      return idof<Array<bool>>();
    case TypeId.ArrayUint8Array:
      return idof<Array<Uint8Array>>();
    case TypeId.ArrayEthereumValue:
      return idof<Array<ethereum.Value>>();
    case TypeId.ArrayStoreValue:
      return idof<Array<Value>>();
    case TypeId.ArrayJsonValue:
      return idof<Array<JSONValue>>();
    case TypeId.ArrayString:
      return idof<Array<string>>();
    case TypeId.ArrayEventParam:
      return idof<Array<ethereum.EventParam>>();
    case TypeId.ArrayTypedMapEntryStringJsonValue:
      return idof<Array<TypedMapEntry<string, JSONValue>>>();
    case TypeId.ArrayTypedMapEntryStringStoreValue:
      return idof<Array<Entity>>();
    case TypeId.WrappedTypedMapStringJsonValue:
      return idof<Wrapped<TypedMapEntry<string, JSONValue>>>();
    case TypeId.WrappedBool:
      return idof<Wrapped<boolean>>();
    case TypeId.WrappedJsonValue:
      return idof<Wrapped<JSONValue>>();
    case TypeId.SmartContractCall:
      return idof<ethereum.SmartContractCall>();
    case TypeId.EventParam:
      return idof<ethereum.EventParam>();
    case TypeId.EthereumTransaction:
      return idof<ethereum.Transaction>();
    case TypeId.EthereumBlock:
      return idof<ethereum.Block>();
    case TypeId.EthereumCall:
      return idof<ethereum.Call>();
    case TypeId.EthereumValue:
      return idof<ethereum.Value>();
    case TypeId.StoreValue:
      return idof<Value>();
    case TypeId.JsonValue:
      return idof<JSONValue>();
    case TypeId.EthereumEvent:
      return idof<ethereum.Event>();
    case TypeId.TypedMapEntryStringStoreValue:
      return idof<Entity>();
    case TypeId.TypedMapEntryStringJsonValue:
      return idof<TypedMap<string, JSONValue>>();
    case TypeId.TypedMapStringStoreValue:
      return idof<TypedMap<string, Value>>();
    case TypeId.TypedMapStringJsonValue:
      return idof<TypedMap<string, JSONValue>>();
    case TypeId.TypedMapStringTypedMapStringJsonValue:
      return idof<TypedMap<string, TypedMap<string, JSONValue>>>();
    case TypeId.ResultTypedMapStringJsonValueBool:
      return idof<Result<TypedMap<string, JSONValue>, boolean>>();
    case TypeId.ResultJsonValueBool:
      return idof<Result<JSONValue, boolean>>();
    case TypeId.ArrayU8:
      return idof<Array<u8>>();
    case TypeId.ArrayU16:
      return idof<Array<u16>>();
    case TypeId.ArrayU32:
      return idof<Array<u32>>();
    case TypeId.ArrayU64:
      return idof<Array<u64>>();
    case TypeId.ArrayI8:
      return idof<Array<i8>>();
    case TypeId.ArrayI16:
      return idof<Array<i16>>();
    case TypeId.ArrayI32:
      return idof<Array<i32>>();
    case TypeId.ArrayI64:
      return idof<Array<i64>>();
    case TypeId.ArrayF32:
      return idof<Array<f32>>();
    case TypeId.ArrayF64:
      return idof<Array<f64>>();
    case TypeId.ArrayBigDecimal:
      return idof<Array<BigDecimal>>();
    case TypeId.NearArrayDataReceiver:
      return idof<Array<near.DataReceiver>>();
    case TypeId.NearArrayCryptoHash:
      return idof<Array<near.CryptoHash>>();
    case TypeId.NearArrayActionValue:
      return idof<Array<near.ActionValue>>();
    case TypeId.NearMerklePath:
      return idof<near.MerklePath>();
    case TypeId.NearArrayValidatorStake:
      return idof<Array<near.ValidatorStake>>();
    case TypeId.NearArraySlashedValidator:
      return idof<Array<near.SlashedValidator>>();
    case TypeId.NearArraySignature:
      return idof<Array<near.Signature>>();
    case TypeId.NearArrayChunkHeader:
      return idof<Array<near.ChunkHeader>>();
    case TypeId.NearAccessKeyPermissionValue:
      return idof<near.AccessKeyPermissionValue>();
    case TypeId.NearActionValue:
      return idof<near.ActionValue>();
    // Commented out because it's an enum, there's no type_id
    // case TypeId.NearDirection:
    //   return idof<near.Direction>()
    case TypeId.NearPublicKey:
      return idof<near.PublicKey>();
    case TypeId.NearSignature:
      return idof<near.Signature>();
    case TypeId.NearFunctionCallPermission:
      return idof<near.FunctionCallPermission>();
    case TypeId.NearFullAccessPermission:
      return idof<near.FullAccessPermission>();
    case TypeId.NearAccessKey:
      return idof<near.AccessKeyPermissionValue>();
    case TypeId.NearDataReceiver:
      return idof<near.DataReceiver>();
    case TypeId.NearCreateAccountAction:
      return idof<near.CreateAccountAction>();
    case TypeId.NearDeployContractAction:
      return idof<near.DeployContractAction>();
    case TypeId.NearFunctionCallAction:
      return idof<near.FunctionCallAction>();
    case TypeId.NearTransferAction:
      return idof<near.TransferAction>();
    case TypeId.NearStakeAction:
      return idof<near.StakeAction>();
    case TypeId.NearAddKeyAction:
      return idof<near.AddKeyAction>();
    case TypeId.NearDeleteKeyAction:
      return idof<near.DeleteKeyAction>();
    case TypeId.NearDeleteAccountAction:
      return idof<near.DeleteAccountAction>();
    case TypeId.NearActionReceipt:
      return idof<near.ActionReceipt>();
    case TypeId.NearSuccessStatus:
      return idof<near.SuccessStatus>();
    case TypeId.NearMerklePathItem:
      return idof<near.MerklePathItem>();
    case TypeId.NearExecutionOutcome:
      return idof<near.ExecutionOutcome>();
    case TypeId.NearSlashedValidator:
      return idof<near.SlashedValidator>();
    case TypeId.NearBlockHeader:
      return idof<near.BlockHeader>();
    case TypeId.NearValidatorStake:
      return idof<near.ValidatorStake>();
    case TypeId.NearChunkHeader:
      return idof<near.ChunkHeader>();
    case TypeId.NearBlock:
      return idof<near.Block>();
    case TypeId.NearReceiptWithOutcome:
      return idof<near.ReceiptWithOutcome>();
    case TypeId.TransactionReceipt:
      return idof<ethereum.TransactionReceipt>();
    case TypeId.Log:
      return idof<ethereum.Log>();
    case TypeId.ArrayH256:
      return idof<Array<Uint8Array>>();
    case TypeId.ArrayLog:
      return idof<Array<ethereum.Log>>();
    case TypeId.CosmosAny:
      return idof<cosmos.Any>();
    case TypeId.CosmosAnyArray:
      return idof<Array<cosmos.Any>>();
    case TypeId.CosmosBytesArray:
      return idof<Array<Bytes>>();
    case TypeId.CosmosCoinArray:
      return idof<Array<cosmos.Coin>>();
    case TypeId.CosmosCommitSigArray:
      return idof<Array<cosmos.CommitSig>>();
    case TypeId.CosmosEventArray:
      return idof<Array<cosmos.Event>>();
    case TypeId.CosmosEventAttributeArray:
      return idof<Array<cosmos.EventAttribute>>();
    case TypeId.CosmosEvidenceArray:
      return idof<Array<cosmos.Evidence>>();
    case TypeId.CosmosModeInfoArray:
      return idof<Array<cosmos.ModeInfo>>();
    case TypeId.CosmosSignerInfoArray:
      return idof<Array<cosmos.SignerInfo>>();
    case TypeId.CosmosTxResultArray:
      return idof<Array<cosmos.TxResult>>();
    case TypeId.CosmosValidatorArray:
      return idof<Array<cosmos.Validator>>();
    case TypeId.CosmosValidatorUpdateArray:
      return idof<Array<cosmos.ValidatorUpdate>>();
    case TypeId.CosmosAuthInfo:
      return idof<cosmos.AuthInfo>();
    case TypeId.CosmosBlock:
      return idof<cosmos.Block>();
    case TypeId.CosmosBlockId:
      return idof<cosmos.BlockID>();
    case TypeId.CosmosBlockIdFlagEnum:
      return idof<Array<cosmos.BlockIDFlag>>();
    case TypeId.CosmosBlockParams:
      return idof<cosmos.BlockParams>();
    case TypeId.CosmosCoin:
      return idof<cosmos.Coin>();
    case TypeId.CosmosCommit:
      return idof<cosmos.Commit>();
    case TypeId.CosmosCommitSig:
      return idof<cosmos.CommitSig>();
    case TypeId.CosmosCompactBitArray:
      return idof<cosmos.CompactBitArray>();
    case TypeId.CosmosConsensus:
      return idof<cosmos.Consensus>();
    case TypeId.CosmosConsensusParams:
      return idof<cosmos.ConsensusParams>();
    case TypeId.CosmosDuplicateVoteEvidence:
      return idof<cosmos.DuplicateVoteEvidence>();
    case TypeId.CosmosDuration:
      return idof<cosmos.Duration>();
    case TypeId.CosmosEvent:
      return idof<cosmos.Event>();
    case TypeId.CosmosEventAttribute:
      return idof<cosmos.EventAttribute>();
    case TypeId.CosmosEventData:
      return idof<cosmos.EventData>();
    case TypeId.CosmosEventVote:
      return idof<cosmos.EventVote>();
    case TypeId.CosmosEvidence:
      return idof<cosmos.Evidence>();
    case TypeId.CosmosEvidenceList:
      return idof<cosmos.EvidenceList>();
    case TypeId.CosmosEvidenceParams:
      return idof<cosmos.EvidenceParams>();
    case TypeId.CosmosFee:
      return idof<cosmos.Fee>();
    case TypeId.CosmosHeader:
      return idof<cosmos.Header>();
    case TypeId.CosmosHeaderOnlyBlock:
      return idof<cosmos.HeaderOnlyBlock>();
    case TypeId.CosmosLightBlock:
      return idof<cosmos.LightBlock>();
    case TypeId.CosmosLightClientAttackEvidence:
      return idof<cosmos.LightClientAttackEvidence>();
    case TypeId.CosmosModeInfo:
      return idof<cosmos.ModeInfo>();
    case TypeId.CosmosModeInfoMulti:
      return idof<cosmos.ModeInfoMulti>();
    case TypeId.CosmosModeInfoSingle:
      return idof<cosmos.ModeInfoSingle>();
    case TypeId.CosmosPartSetHeader:
      return idof<cosmos.PartSetHeader>();
    case TypeId.CosmosPublicKey:
      return idof<cosmos.PublicKey>();
    case TypeId.CosmosResponseBeginBlock:
      return idof<cosmos.ResponseBeginBlock>();
    case TypeId.CosmosResponseDeliverTx:
      return idof<cosmos.ResponseDeliverTx>();
    case TypeId.CosmosResponseEndBlock:
      return idof<cosmos.ResponseEndBlock>();
    case TypeId.CosmosSignModeEnum:
      return idof<Array<cosmos.SignMode>>();
    case TypeId.CosmosSignedHeader:
      return idof<cosmos.SignedHeader>();
    case TypeId.CosmosSignedMsgTypeEnum:
      return idof<Array<cosmos.SignedMsgType>>();
    case TypeId.CosmosSignerInfo:
      return idof<cosmos.SignerInfo>();
    case TypeId.CosmosTimestamp:
      return idof<cosmos.Timestamp>();
    case TypeId.CosmosTip:
      return idof<cosmos.Tip>();
    case TypeId.CosmosTransactionData:
      return idof<cosmos.TransactionData>();
    case TypeId.CosmosTx:
      return idof<cosmos.Tx>();
    case TypeId.CosmosTxBody:
      return idof<cosmos.TxBody>();
    case TypeId.CosmosTxResult:
      return idof<cosmos.TxResult>();
    case TypeId.CosmosValidator:
      return idof<cosmos.Validator>();
    case TypeId.CosmosValidatorParams:
      return idof<cosmos.ValidatorParams>();
    case TypeId.CosmosValidatorSet:
      return idof<cosmos.ValidatorSet>();
    case TypeId.CosmosValidatorSetUpdates:
      return idof<cosmos.ValidatorSetUpdates>();
    case TypeId.CosmosValidatorUpdate:
      return idof<cosmos.ValidatorUpdate>();
    case TypeId.CosmosVersionParams:
      return idof<cosmos.VersionParams>();
    case TypeId.CosmosMessageData:
      return idof<cosmos.MessageData>();
    case TypeId.CosmosTransactionContext:
      return idof<cosmos.TransactionContext>();
    /**
     * Arweave type ids
     */
    case TypeId.ArweaveBlock:
      return idof<arweave.Block>();
    case TypeId.ArweaveProofOfAccess:
      return idof<arweave.ProofOfAccess>();
    case TypeId.ArweaveTag:
      return idof<arweave.Tag>();
    case TypeId.ArweaveTagArray:
      return idof<Array<arweave.Tag>>();
    case TypeId.ArweaveTransaction:
      return idof<arweave.Transaction>();
    case TypeId.ArweaveTransactionArray:
      return idof<Array<arweave.Transaction>>();
    case TypeId.ArweaveTransactionWithBlockPtr:
      return idof<arweave.TransactionWithBlockPtr>();
    default:
      return 0;
  }
}

export function allocate(size: usize): usize {
  // @ts-expect-error We do not want to expose __alloc, hence why we just ignore the error
  return __alloc(size);
}