Improve setup scripts.

package.json

@@ -6,7 +6,8 @@
     "compile": "typechain-compiler --toolchain nightly-2023-08-10",
     "compile:release": "typechain-compiler --release --toolchain nightly-2023-08-10",
     "deploy:local": "source scripts/local_env && ts-node scripts/deploy.ts",
-    "example:local": "source scripts/local_env && ts-node scripts/example_setup.ts",
+    "create-tokens-local": "source scripts/local_env && ts-node scripts/create_tokens.ts",
+    "add-liquidity-local": "source scripts/local_env && ts-node scripts/add_liquidity.ts",
     "format": "prettier --write ."
   },
   "dependencies": {

scripts/add_liquidity.ts

@@ -0,0 +1,91 @@
+import { ApiPromise, WsProvider, Keyring } from '@polkadot/api';
+import { loadAddresses } from './utils';
+import Token from '../types/contracts/psp22_token';
+import Router from '../types/contracts/router_contract';
+import Factory from '../types/contracts/factory_contract';
+import { ONE_THOUSAND_STABLECOIN } from './constants';
+import { addLiquidityNative } from './utils';
+import { parseUnits } from './shared';
+
+// Create a new instance of contract
+const wsProvider = new WsProvider(process.env.WS_NODE);
+// Create a keyring instance
+const keyring = new Keyring({ type: 'sr25519' });
+
+async function main(): Promise<void> {
+  const api = await ApiPromise.create({ provider: wsProvider });
+  const deployer = keyring.addFromUri(process.env.AUTHORITY_SEED);
+
+  const {
+    routerAddress,
+    factoryAddress,
+    wnativeAddress,
+    dogeAddress,
+    usdcAddress,
+    usdtAddress,
+  } = loadAddresses();
+
+  const router = new Router(routerAddress, deployer, api);
+  const factory = new Factory(factoryAddress, deployer, api);
+
+  /// Create tokens
+
+  const doge = new Token(dogeAddress, deployer, api);
+  const usdc = new Token(usdcAddress, deployer, api);
+  const usdt = new Token(usdtAddress, deployer, api);
+
+  /// Add liquidity
+  const dogeAmount = parseUnits(1000, 18).toString();
+
+  await doge.tx.approve(router.address, dogeAmount);
+  console.log('approved 1000 DOGE to spend by router');
+  await addLiquidityNative(
+    router,
+    doge,
+    dogeAmount,
+    dogeAmount,
+    deployer.address,
+  );
+  console.log('added 1000 DOGE liquidity');
+  await usdc.tx.approve(router.address, ONE_THOUSAND_STABLECOIN);
+  console.log('approved 1000 USDC to spend by router');
+  await addLiquidityNative(
+    router,
+    usdc,
+    ONE_THOUSAND_STABLECOIN,
+    ONE_THOUSAND_STABLECOIN,
+    deployer.address,
+  );
+  console.log('added 1000 USDC liquidity');
+  await usdt.tx.approve(router.address, ONE_THOUSAND_STABLECOIN);
+  console.log('approved 1000 USDT to spend by router');
+  await addLiquidityNative(
+    router,
+    usdt,
+    ONE_THOUSAND_STABLECOIN,
+    ONE_THOUSAND_STABLECOIN,
+    deployer.address,
+  );
+  console.log('added 1000 USDT liquidity');
+
+  /// Query pair addresses
+  const {
+    value: { ok: dogeWAzeroAddress },
+  } = await factory.query.getPair(doge.address, wnativeAddress);
+  console.log('dogeWAzeroAddress', dogeWAzeroAddress);
+  const {
+    value: { ok: usdcWAzeroAddress },
+  } = await factory.query.getPair(usdc.address, wnativeAddress);
+  console.log('usdcWAzeroAddress', usdcWAzeroAddress);
+  const {
+    value: { ok: usdtWAzeroAddress },
+  } = await factory.query.getPair(usdt.address, wnativeAddress);
+  console.log('usdtWAzeroAddress', usdtWAzeroAddress);
+
+  await api.disconnect();
+}
+
+main().catch((error) => {
+  console.error(error);
+  process.exitCode = 1;
+});

scripts/constants.ts

@@ -2,7 +2,11 @@ import { parseUnits } from './shared';
 
 export const DEADLINE = '111111111111111111';
 export const ONE_WAZERO = parseUnits(1, 12).toString();
-export const ONE_STABLECOIN = parseUnits(100, 6).toString();
+export const ONE_HUNDRED_WAZERO = parseUnits(100, 12).toString();
+export const ONE_THOUSAND_WAZERO = parseUnits(1_000, 12).toString();
+export const ONE_STABLECOIN = parseUnits(1, 6).toString();
+export const ONE_HUNDRED_STABLECOIN = parseUnits(100, 6).toString();
+export const ONE_THOUSAND_STABLECOIN = parseUnits(1_000, 6).toString();
 export const TOTAL_SUPPLY = parseUnits(1_000_000, 18).toString();
 export const STABLE_TOTAL_SUPPLY = parseUnits(1_000_000, 6).toString();
 export const DUMMY_ADDRESS =

scripts/create_tokens.ts

@@ -0,0 +1,64 @@
+import { ApiPromise, WsProvider, Keyring } from '@polkadot/api';
+import { loadAddresses, storeAddresses } from './utils';
+import Token_factory from '../types/constructors/psp22_token';
+import * as token from './token';
+import { TOTAL_SUPPLY, STABLE_TOTAL_SUPPLY } from './constants';
+
+// Create a new instance of contract
+const wsProvider = new WsProvider(process.env.WS_NODE);
+// Create a keyring instance
+const keyring = new Keyring({ type: 'sr25519' });
+
+async function main(): Promise<void> {
+  const api = await ApiPromise.create({ provider: wsProvider });
+  const deployer = keyring.addFromUri(process.env.AUTHORITY_SEED);
+  const tokenCodeHash = await token.upload(api, deployer);
+  console.log('token code hash:', tokenCodeHash);
+
+  const addresses = loadAddresses();
+
+  const tokenFactory = new Token_factory(api, deployer);
+
+  /// Create tokens
+  const tokenInitGas = await token.estimateInit(api, deployer);
+  const { address: dogeAddress } = await tokenFactory.new(
+    TOTAL_SUPPLY,
+    'Doge Coin',
+    'DOGE',
+    18,
+    { gasLimit: tokenInitGas },
+  );
+  console.log('doge coin address', dogeAddress);
+  const { address: usdcAddress } = await tokenFactory.new(
+    STABLE_TOTAL_SUPPLY,
+    'USD Coin',
+    'USDC',
+    6,
+    { gasLimit: tokenInitGas },
+  );
+  console.log('usdc token address:', usdcAddress);
+  const { address: usdtAddress } = await tokenFactory.new(
+    STABLE_TOTAL_SUPPLY,
+    'Tether USD',
+    'USDT',
+    6,
+    { gasLimit: tokenInitGas },
+  );
+  console.log('usdt token address:', usdtAddress);
+
+  storeAddresses({
+    ...addresses,
+    dogeAddress,
+    usdcAddress,
+    usdtAddress,
+  });
+
+  console.log('psp22 token addresses stored');
+
+  await api.disconnect();
+}
+
+main().catch((error) => {
+  console.error(error);
+  process.exitCode = 1;
+});

scripts/shared.ts

@@ -7,4 +7,7 @@ export type Addresses = {
   factoryAddress: string;
   wnativeAddress: string;
   routerAddress: string;
+  dogeAddress?: string;
+  usdcAddress?: string;
+  usdtAddress?: string;
 };

scripts/token.ts

@@ -19,8 +19,8 @@ export async function estimateInit(
 ): Promise<WeightV2> {
   return estimateContractInit(api, deployer, 'psp22_token.contract', [
     TOTAL_SUPPLY,
-    'Apollo Token',
-    'APLO',
+    'Doge Coin',
+    'DOGE',
     18,
   ]);
 }

scripts/utils.ts

@@ -1,6 +1,6 @@
 import Token from '../types/contracts/psp22_token';
 import Router from '../types/contracts/router_contract';
-import { ONE_WAZERO, DEADLINE } from './constants';
+import { DEADLINE, ONE_THOUSAND_WAZERO } from './constants';
 import { Addresses } from './shared';
 import fs from 'fs';
 
@@ -24,11 +24,11 @@ export const addLiquidityNative = async (
     token.address,
     amountDesired,
     amountMin,
-    ONE_WAZERO,
+    ONE_THOUSAND_WAZERO,
     to,
     DEADLINE,
     {
-      value: ONE_WAZERO,
+      value: ONE_THOUSAND_WAZERO,
     },
   );
 };