feat: support new operation drainDelegate for Lima

re #2065
ecadlabs · Nov 8, 2022 · 984e3a5 · 984e3a5
1 parent e0126e5
commit 984e3a5
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 1 deletion.
diff --git a/packages/taquito-local-forging/src/constants.ts b/packages/taquito-local-forging/src/constants.ts
@@ -50,6 +50,7 @@ export enum CODEC {
   TX_ROLLUP_BATCH_CONTENT = 'tx_rollup_batch_content',
   OP_INCREASE_PAID_STORAGE = 'increase_paid_storage',
   OP_UPDATE_CONSENSUS_KEY = 'update_consensus_key',
+  OP_DRAIN_DELEGATE = 'drain_delegate',
 }
 
 // See https://tezos.gitlab.io/whitedoc/michelson.html#full-grammar
@@ -233,6 +234,7 @@ export const kindMapping: { [key: number]: string } = {
   0x9e: 'transfer_ticket',
   0x71: 'increase_paid_storage',
   0x72: 'update_consensus_key',
+  0x09: 'drain_delegate',
 };
 
 export const kindMappingReverse = (() => {

diff --git a/packages/taquito-local-forging/src/decoder.ts b/packages/taquito-local-forging/src/decoder.ts
@@ -30,6 +30,7 @@ import {
   EndorsementSchema,
   IncreasePaidStorageSchema,
   UpdateConsensusKeySchema,
+  DrainDelegateSchema,
   ManagerOperationSchema,
   operationDecoder,
   OriginationSchema,
@@ -104,4 +105,6 @@ decoders[CODEC.OP_INCREASE_PAID_STORAGE] = (val: Uint8ArrayConsumer) =>
   schemaDecoder(decoders)(IncreasePaidStorageSchema)(val);
 decoders[CODEC.OP_UPDATE_CONSENSUS_KEY] = (val: Uint8ArrayConsumer) =>
   schemaDecoder(decoders)(UpdateConsensusKeySchema)(val);
+decoders[CODEC.OP_DRAIN_DELEGATE] = (val: Uint8ArrayConsumer) =>
+  schemaDecoder(decoders)(DrainDelegateSchema)(val);
 decoders[CODEC.MANAGER] = schemaDecoder(decoders)(ManagerOperationSchema);
diff --git a/packages/taquito-local-forging/src/encoder.ts b/packages/taquito-local-forging/src/encoder.ts
@@ -30,6 +30,7 @@ import {
   EndorsementSchema,
   IncreasePaidStorageSchema,
   UpdateConsensusKeySchema,
+  DrainDelegateSchema,
   ManagerOperationSchema,
   operationEncoder,
   OriginationSchema,
@@ -96,4 +97,5 @@ encoders[CODEC.OP_INCREASE_PAID_STORAGE] = (val: any) =>
   schemaEncoder(encoders)(IncreasePaidStorageSchema)(val);
 encoders[CODEC.OP_UPDATE_CONSENSUS_KEY] = (val: any) =>
   schemaEncoder(encoders)(UpdateConsensusKeySchema)(val);
+encoders[CODEC.OP_DRAIN_DELEGATE] = (val: any) => schemaEncoder(encoders)(DrainDelegateSchema)(val);
 encoders[CODEC.MANAGER] = schemaEncoder(encoders)(ManagerOperationSchema);
diff --git a/packages/taquito-local-forging/src/schema/operation.ts b/packages/taquito-local-forging/src/schema/operation.ts
@@ -145,6 +145,12 @@ export const UpdateConsensusKeySchema = {
   pk: CODEC.PUBLIC_KEY,
 };
 
+export const DrainDelegateSchema = {
+  consensus_key: CODEC.PKH,
+  delegate: CODEC.PKH,
+  destination: CODEC.PKH,
+};
+
 export const operationEncoder =
   (encoders: { [key: string]: (val: object) => string }) => (operation: { kind: string }) => {
     if (!(operation.kind in encoders) || !(operation.kind in kindMappingReverse)) {

diff --git a/packages/taquito-local-forging/src/validator.ts b/packages/taquito-local-forging/src/validator.ts
@@ -16,6 +16,7 @@ import {
   TxRollupSubmitBatchSchema,
   IncreasePaidStorageSchema,
   UpdateConsensusKeySchema,
+  DrainDelegateSchema,
 } from './schema/operation';
 
 type OperationKind =
@@ -33,7 +34,8 @@ type OperationKind =
   | OpKind.TX_ROLLUP_ORIGINATION
   | OpKind.TX_ROLLUP_SUBMIT_BATCH
   | OpKind.INCREASE_PAID_STORAGE
-  | OpKind.UPDATE_CONSENSUS_KEY;
+  | OpKind.UPDATE_CONSENSUS_KEY
+  | OpKind.DRAIN_DELEGATE;
 
 const OperationKindMapping = {
   activate_account: ActivationSchema,
@@ -51,6 +53,7 @@ const OperationKindMapping = {
   tx_rollup_submit_batch: TxRollupSubmitBatchSchema,
   increase_paid_storage: IncreasePaidStorageSchema,
   update_consensus_key: UpdateConsensusKeySchema,
+  drain_delegate: DrainDelegateSchema,
 };
 
 // Asymmetric difference: only account for things in arr2 that are not present in arr1, not vice versa