Skip to content

Commit

Permalink
remove formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
0xNeshi committed Feb 25, 2025
1 parent 9dfa23a commit f7530ec
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 28 deletions.
3 changes: 1 addition & 2 deletions package.json
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,5 @@
},
"devDependencies": {
"wsrun": "^5.2.4"
},
"packageManager": "yarn@1.22.22+sha512.a6b2f7906b721bba3d67d4aff083df04dad64c399707841b7acf00f6b133b7ac24255f2652fa22ae3534329dc6180534e98d17432037ff6fd140556e2bb3137e"
}
}
40 changes: 14 additions & 26 deletions packages/core/stylus/src/erc20.ts
Original file line number Diff line number Diff line change
@@ -1,11 +1,7 @@
import { BaseImplementedTrait, Contract, ContractBuilder } from './contract';
import { addPausable } from './add-pausable';
import { defineFunctions } from './utils/define-functions';
import {
CommonContractOptions,
withCommonContractDefaults,
getSelfArg,
} from './common-options';
import { CommonContractOptions, withCommonContractDefaults, getSelfArg } from './common-options';
import { contractDefaults as commonDefaults } from './common-options';
import { printContract } from './print';
import { setAccessControl } from './set-access-control';
Expand Down Expand Up @@ -116,21 +112,15 @@ function addPermit(c: ContractBuilder, pausable: boolean): BaseImplementedTrait
c.addUseClause('alloy_primitives', 'U256');
c.addUseClause('alloy_primitives', 'B256');

c.addFunctionCodeBefore(
erc20PermitTrait,
functions(erc20PermitTrait).permit,
['self.pausable.when_not_paused()?;']
);
c.addFunctionCodeBefore(
erc20PermitTrait,
functions(erc20PermitTrait).transfer,
['self.pausable.when_not_paused()?;']
);
c.addFunctionCodeBefore(
erc20PermitTrait,
functions(erc20PermitTrait).transfer_from,
['self.pausable.when_not_paused()?;']
);
c.addFunctionCodeBefore(erc20PermitTrait, functions(erc20PermitTrait).permit, [
'self.pausable.when_not_paused()?;',
]);
c.addFunctionCodeBefore(erc20PermitTrait, functions(erc20PermitTrait).transfer, [
'self.pausable.when_not_paused()?;',
]);
c.addFunctionCodeBefore(erc20PermitTrait, functions(erc20PermitTrait).transfer_from, [
'self.pausable.when_not_paused()?;',
]);
}

return erc20PermitTrait;
Expand Down Expand Up @@ -175,14 +165,12 @@ function addFlashMint(c: ContractBuilder, pausable: boolean, baseTrait: BaseImpl

c.addImplementedTrait(flashMintTrait);

const fns = functions(baseTrait);

c.addFunction(flashMintTrait, fns.max_flash_loan);
c.addFunction(flashMintTrait, fns.flash_fee);
c.addFunction(flashMintTrait, fns.flash_loan);
c.addFunction(flashMintTrait, functions(baseTrait).max_flash_loan);
c.addFunction(flashMintTrait, functions(baseTrait).flash_fee);
c.addFunction(flashMintTrait, functions(baseTrait).flash_loan);

if (pausable) {
c.addFunctionCodeBefore(flashMintTrait, fns.flash_loan, [
c.addFunctionCodeBefore(flashMintTrait, functions(baseTrait).flash_loan, [
'self.pausable.when_not_paused()?;',
]);
}
Expand Down

0 comments on commit f7530ec

Please sign in to comment.