10719161702

Committed 05 Sep 2024 11:07AM UTC coverage: 46.163% (+0.3%) from 45.889%

Build # 10719161702

Build Type

Pull #1895

github

Committed by

web-flow

Commit Message

Update src/routes/transactions/mappers/common/native-staking.mapper.ts

Co-authored-by: Aaron Cook <aaron@safe.global>

Pull Request Pull Request #1895: Factorize Native Staking helpers and mappers common logic

Run Details

499 of 3129 branches covered (15.95%)

Branch coverage included in aggregate %.

26 of 83 new or added lines in 11 files covered. (31.33%)

185 existing lines in 8 files now uncovered.

4837 of 8430 relevant lines covered (57.38%)

12.18 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

23.68

/src/routes/transactions/helpers/kiln-native-staking.helper.ts

import { IStakingRepository } from '@/domain/staking/staking.repository.interface';
import { StakingRepositoryModule } from '@/domain/staking/staking.repository.module';
import {
  TransactionFinder,
  TransactionFinderModule,
} from '@/routes/transactions/helpers/transaction-finder.helper';
import { Inject, Injectable, Module } from '@nestjs/common';
import { toFunctionSelector } from 'viem';

@Injectable()
export class KilnNativeStakingHelper {
  constructor(
    private readonly transactionFinder: TransactionFinder,
    @Inject(IStakingRepository)
    private readonly stakingRepository: IStakingRepository,
  ) {}

  public async findDeposit(args: {
    chainId: string;
    to?: `0x${string}`;
    data: `0x${string}`;
  }): Promise<{
    to: `0x${string}`;
    data: `0x${string}`;
  } | null> {
    return this.findTransactionAndCheckDeployment({
      ...args,
      selector: toFunctionSelector('function deposit() external payable'),
    });
  }

  public async findValidatorsExit(args: {
    chainId: string;
    to?: `0x${string}`;
    data: `0x${string}`;
  }): Promise<{
    to: `0x${string}`;
    data: `0x${string}`;
  } | null> {
    return this.findTransactionAndCheckDeployment({
      ...args,
      selector: toFunctionSelector(
        'function requestValidatorsExit(bytes) external',
      ),
    });
  }

  public async findWithdraw(args: {
    chainId: string;
    to?: `0x${string}`;
    data: `0x${string}`;
  }): Promise<{
    to: `0x${string}`;
    data: `0x${string}`;
  } | null> {
    return this.findTransactionAndCheckDeployment({
      ...args,
      selector: toFunctionSelector(
        'function batchWithdrawCLFee(bytes) external',
      ),
    });
  }

  private async findTransactionAndCheckDeployment(args: {
    chainId: string;
    to?: `0x${string}`;
    data: `0x${string}`;
    selector: string;
  }): Promise<{
    to: `0x${string}`;
    data: `0x${string}`;
  } | null> {
    const transaction = this.transactionFinder.findTransaction(
      (transaction) => transaction.data.startsWith(args.selector),
      args,
    );

    if (!transaction?.to) {
      return null;
    }

    const deployment = await this.stakingRepository
      .getDeployment({
        chainId: args.chainId,
        address: transaction.to,
      })
      .catch(() => null);

    if (
      deployment?.product_type !== 'dedicated' &&
      deployment?.chain !== 'unknown'
    ) {
      return null;
    }

    return {
      to: transaction.to,
      data: transaction.data,
    };
  }
}

@Module({
  imports: [TransactionFinderModule, StakingRepositoryModule],
  providers: [KilnNativeStakingHelper],
  exports: [KilnNativeStakingHelper],
})
export class KilnNativeStakingHelperModule {}

1	import { IStakingRepository } from '@/domain/staking/staking.repository.interface';	16✔
2	import { StakingRepositoryModule } from '@/domain/staking/staking.repository.module';	16✔
3	import {	16✔
4	TransactionFinder,
5	TransactionFinderModule,
6	} from '@/routes/transactions/helpers/transaction-finder.helper';
7	import { Inject, Injectable, Module } from '@nestjs/common';	16✔
8	import { toFunctionSelector } from 'viem';	16✔
9
10	@Injectable()
11	export class KilnNativeStakingHelper {	16✔
12	constructor(
13	private readonly transactionFinder: TransactionFinder,	16✔
14	@Inject(IStakingRepository)
15	private readonly stakingRepository: IStakingRepository,	16✔
16	) {}
17
18	public async findDeposit(args: {
19	chainId: string;
20	to?: `0x${string}`;
21	data: `0x${string}`;
22	}): Promise<{
23	to: `0x${string}`;
24	data: `0x${string}`;
25	} \| null> {
NEW 26	return this.findTransactionAndCheckDeployment({	×
27	...args,
28	selector: toFunctionSelector('function deposit() external payable'),
29	});
30	}
31
32	public async findValidatorsExit(args: {
33	chainId: string;
34	to?: `0x${string}`;
35	data: `0x${string}`;
36	}): Promise<{
37	to: `0x${string}`;
38	data: `0x${string}`;
39	} \| null> {
NEW 40	return this.findTransactionAndCheckDeployment({	×
41	...args,
42	selector: toFunctionSelector(
43	'function requestValidatorsExit(bytes) external',
44	),
45	});
46	}
47
48	public async findWithdraw(args: {
49	chainId: string;
50	to?: `0x${string}`;
51	data: `0x${string}`;
52	}): Promise<{
53	to: `0x${string}`;
54	data: `0x${string}`;
55	} \| null> {
NEW 56	return this.findTransactionAndCheckDeployment({	×
57	...args,
58	selector: toFunctionSelector(
59	'function batchWithdrawCLFee(bytes) external',
60	),
61	});
62	}
63
64	private async findTransactionAndCheckDeployment(args: {
65	chainId: string;
66	to?: `0x${string}`;
67	data: `0x${string}`;
68	selector: string;
69	}): Promise<{
70	to: `0x${string}`;
71	data: `0x${string}`;
72	} \| null> {
UNCOV 73	const transaction = this.transactionFinder.findTransaction(	×
NEW 74	(transaction) => transaction.data.startsWith(args.selector),	×
75	args,
76	);
77
UNCOV 78	if (!transaction?.to) {	×
UNCOV 79	return null;	×
80	}
81
82	const deployment = await this.stakingRepository	×
83	.getDeployment({
84	chainId: args.chainId,
85	address: transaction.to,
86	})
UNCOV 87	.catch(() => null);	×
88
NEW 89	if (	×
90	deployment?.product_type !== 'dedicated' &&	×
91	deployment?.chain !== 'unknown'	×
92	) {
UNCOV 93	return null;	×
94	}
95
UNCOV 96	return {	×
97	to: transaction.to,
98	data: transaction.data,
99	};
100	}
101	}
102
103	@Module({
104	imports: [TransactionFinderModule, StakingRepositoryModule],
105	providers: [KilnNativeStakingHelper],
106	exports: [KilnNativeStakingHelper],
107	})
108	export class KilnNativeStakingHelperModule {}	16✔

safe-global / safe-client-gateway / 10719161702

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous