diff --git a/src/entities/SecurityToken/SecurityToken.ts b/src/entities/SecurityToken/SecurityToken.ts index 98209b1..f315306 100644 --- a/src/entities/SecurityToken/SecurityToken.ts +++ b/src/entities/SecurityToken/SecurityToken.ts @@ -9,7 +9,6 @@ import { Permissions } from './Permissions'; import { Transfers } from './Transfers'; import { PolymathError } from '../../PolymathError'; import { ErrorCode } from '../../types'; -import { Restrictions } from './Restrictions'; import { Documents } from './Documents'; export interface UniqueIdentifiers { @@ -72,8 +71,6 @@ export class SecurityToken extends Entity { public transfers: Transfers; - public restrictions: Restrictions; - public documents: Documents; constructor(params: Params & UniqueIdentifiers, context: Context) { @@ -92,7 +89,6 @@ export class SecurityToken extends Entity { this.offerings = new Offerings(this, context); this.permissions = new Permissions(this, context); this.transfers = new Transfers(this, context); - this.restrictions = new Restrictions(this, context); this.documents = new Documents(this, context); } diff --git a/src/entities/SecurityToken/Restrictions/CountRestrictions.ts b/src/entities/SecurityToken/Transfers/Restrictions/CountRestrictions.ts similarity index 86% rename from src/entities/SecurityToken/Restrictions/CountRestrictions.ts rename to src/entities/SecurityToken/Transfers/Restrictions/CountRestrictions.ts index 9ccd8bb..8cc36a7 100644 --- a/src/entities/SecurityToken/Restrictions/CountRestrictions.ts +++ b/src/entities/SecurityToken/Transfers/Restrictions/CountRestrictions.ts @@ -1,8 +1,8 @@ import { ModuleName } from '@polymathnetwork/contract-wrappers'; -import { ErrorCode } from '../../../types'; -import { ModifyMaxHolderCount } from '../../../procedures'; -import { SubModule } from '../SubModule'; -import { PolymathError } from '../../../PolymathError'; +import { ErrorCode } from '../../../../types'; +import { ModifyMaxHolderCount } from '../../../../procedures'; +import { SubModule } from '../../SubModule'; +import { PolymathError } from '../../../../PolymathError'; export class CountRestrictions extends SubModule { /** diff --git a/src/entities/SecurityToken/Restrictions/PercentageRestrictions.ts b/src/entities/SecurityToken/Transfers/Restrictions/PercentageRestrictions.ts similarity index 95% rename from src/entities/SecurityToken/Restrictions/PercentageRestrictions.ts rename to src/entities/SecurityToken/Transfers/Restrictions/PercentageRestrictions.ts index 54b68bf..51960a0 100644 --- a/src/entities/SecurityToken/Restrictions/PercentageRestrictions.ts +++ b/src/entities/SecurityToken/Transfers/Restrictions/PercentageRestrictions.ts @@ -8,11 +8,11 @@ import { ErrorCode, PercentageWhitelistEntry, ModifyPercentageExemptionsProcedureArgs, -} from '../../../types'; -import { ModifyMaxHolderPercentage, ModifyPercentageExemptions } from '../../../procedures'; -import { SubModule } from '../SubModule'; -import { PolymathError } from '../../../PolymathError'; -import { TransactionQueue } from '../../TransactionQueue'; +} from '../../../../types'; +import { ModifyMaxHolderPercentage, ModifyPercentageExemptions } from '../../../../procedures'; +import { SubModule } from '../../SubModule'; +import { PolymathError } from '../../../../PolymathError'; +import { TransactionQueue } from '../../../TransactionQueue'; interface PercentageWhitelistParams { whitelistEntries: PercentageWhitelistEntry[]; diff --git a/src/entities/SecurityToken/Restrictions/Restrictions.ts b/src/entities/SecurityToken/Transfers/Restrictions/Restrictions.ts similarity index 77% rename from src/entities/SecurityToken/Restrictions/Restrictions.ts rename to src/entities/SecurityToken/Transfers/Restrictions/Restrictions.ts index 560f856..fe8c4b9 100644 --- a/src/entities/SecurityToken/Restrictions/Restrictions.ts +++ b/src/entities/SecurityToken/Transfers/Restrictions/Restrictions.ts @@ -1,8 +1,8 @@ -import { SubModule } from '../SubModule'; +import { SubModule } from '../../SubModule'; import { CountRestrictions } from './CountRestrictions'; import { PercentageRestrictions } from './PercentageRestrictions'; -import { SecurityToken } from '../SecurityToken'; -import { Context } from '../../../Context'; +import { SecurityToken } from '../../SecurityToken'; +import { Context } from '../../../../Context'; export class Restrictions extends SubModule { public count: CountRestrictions; diff --git a/src/entities/SecurityToken/Restrictions/index.ts b/src/entities/SecurityToken/Transfers/Restrictions/index.ts similarity index 100% rename from src/entities/SecurityToken/Restrictions/index.ts rename to src/entities/SecurityToken/Transfers/Restrictions/index.ts diff --git a/src/entities/SecurityToken/Transfers.ts b/src/entities/SecurityToken/Transfers/Transfers.ts similarity index 74% rename from src/entities/SecurityToken/Transfers.ts rename to src/entities/SecurityToken/Transfers/Transfers.ts index 6d8c912..591e4c2 100644 --- a/src/entities/SecurityToken/Transfers.ts +++ b/src/entities/SecurityToken/Transfers/Transfers.ts @@ -1,8 +1,19 @@ import { BigNumber } from '@polymathnetwork/contract-wrappers'; -import { SubModule } from './SubModule'; -import { ControllerTransfer, SetController } from '../../procedures'; +import { SubModule } from '../SubModule'; +import { ControllerTransfer, SetController } from '../../../procedures'; +import { Restrictions } from './Restrictions'; +import { SecurityToken } from '../SecurityToken'; +import { Context } from '../../../Context'; export class Transfers extends SubModule { + public restrictions: Restrictions; + + constructor(securityToken: SecurityToken, context: Context) { + super(securityToken, context); + + this.restrictions = new Restrictions(securityToken, context); + } + /** * Set the address of the Security Token's Controller. The controller may perform forced transfers */ diff --git a/src/entities/SecurityToken/Transfers/index.ts b/src/entities/SecurityToken/Transfers/index.ts new file mode 100644 index 0000000..55da812 --- /dev/null +++ b/src/entities/SecurityToken/Transfers/index.ts @@ -0,0 +1 @@ +export { Transfers } from './Transfers';