Merge pull request #6101 from BitGo/WIN-5398-create-coin-factory-using-custom-coin-map

feat: create coin factory using ams response

Author: veetragjain

modules/bitgo/src/bitgo.ts

@@ -6,17 +6,23 @@
 import pjson = require('../package.json');
 import * as _ from 'lodash';
 
-import GlobalCoinFactory from './v2/coinFactory';
-import { BaseCoin, common } from '@bitgo/sdk-core';
+import { BaseCoin, CoinFactory, common } from '@bitgo/sdk-core';
 import { BitGoAPI, BitGoAPIOptions } from '@bitgo/sdk-api';
+import { createTokenMapUsingTrimmedConfigDetails, TrimmedAmsTokenConfig } from '@bitgo/statics';
+import { GlobalCoinFactory, registerCoinConstructors } from './v2/coinFactory';
 
-export type BitGoOptions = BitGoAPIOptions;
+// constructor params used exclusively for BitGo class
+export type BitGoOptions = BitGoAPIOptions & {
+  useAms?: boolean;
+};
 
 export class BitGo extends BitGoAPI {
+  private _coinFactory: CoinFactory;
+  private _useAms: boolean;
   /**
    * Constructor for BitGo Object
    */
-  constructor(params: BitGoAPIOptions = {}) {
+  constructor(params: BitGoOptions = {}) {
     super(params);
     if (
       !common.validateParams(
@@ -34,7 +40,8 @@ export class BitGo extends BitGoAPI {
           'stellarFederationServerUrl',
         ]
       ) ||
-      (params.useProduction && !_.isBoolean(params.useProduction))
+      (params.useProduction && !_.isBoolean(params.useProduction)) ||
+      (params.useAms && !_.isBoolean(params.useAms))
     ) {
       throw new Error('invalid argument');
     }
@@ -43,15 +50,27 @@ export class BitGo extends BitGoAPI {
       throw new Error('invalid argument - must provide both client id and secret');
     }
 
+    this._useAms = !!params.useAms;
     this._version = pjson.version;
     this._userAgent = params.userAgent || 'BitGoJS/' + this.version();
+    this._coinFactory = new CoinFactory();
+  }
+
+  initCoinFactory(tokenConfigMap: Record<string, TrimmedAmsTokenConfig[]>): void {
+    // TODO(WIN-5057): use AMS endpoint to fetch config details
+    const coinMap = createTokenMapUsingTrimmedConfigDetails(tokenConfigMap);
+    this._coinFactory = new CoinFactory();
+    registerCoinConstructors(this._coinFactory, coinMap);
   }
 
   /**
    * Create a basecoin object
    * @param coinName
    */
   coin(coinName: string): BaseCoin {
+    if (this._useAms) {
+      return this._coinFactory.getInstance(this, coinName);
+    }
     return GlobalCoinFactory.getInstance(this, coinName);
   }
 

modules/bitgo/src/index.ts

@@ -10,7 +10,7 @@ import * as _ from 'lodash';
 import { common } from '@bitgo/sdk-core';
 export * from '@bitgo/sdk-api';
 import * as utxolib from '@bitgo/utxo-lib';
-import GlobalCoinFactory from './v2/coinFactory';
+import { GlobalCoinFactory } from './v2/coinFactory';
 
 export * from './bitgo';