From 1c00913cb8961188ade99e69accf0bc690c128d2 Mon Sep 17 00:00:00 2001
From: martonp <marci93@gmail.com>
Date: Mon, 21 Oct 2024 18:10:10 +0200
Subject: [PATCH] Update UI

---
 client/webserver/locales/en-us.go             |  6 ++
 .../webserver/site/src/html/mmsettings.tmpl   | 28 +++++++
 client/webserver/site/src/js/mm.ts            | 26 ++++++-
 client/webserver/site/src/js/mmsettings.ts    | 73 +++++++++++++++++--
 client/webserver/site/src/js/registry.ts      |  4 +
 5 files changed, 127 insertions(+), 10 deletions(-)

diff --git a/client/webserver/locales/en-us.go b/client/webserver/locales/en-us.go
index cff6a9089e..1f837b1db4 100644
--- a/client/webserver/locales/en-us.go
+++ b/client/webserver/locales/en-us.go
@@ -653,4 +653,10 @@ var EnUS = map[string]*intl.Translation{
 	"Transaction":                 {T: "Transaction"},
 	"Value":                       {T: "Value"},
 	"Prepaid bond redeemed":       {T: "Prepaid bond redeemed!"},
+	"Top up fee reserves":         {T: "Top up fee reserves"},
+	"top_up_fee_reserves_tooltip": {T: "When enabled, if the bot's fee reserves run too low, it will be allocated additional available funds from the wallet to be able to cover all configured placements. This is only applicable when the fee asset is neither the base nor quote asset."},
+	"Allow external transfers":    {T: "Allow external transfers"},
+	"external_transfers_tooltip":  {T: "When enabled, the bot will be able to transfer funds between the DEX and the CEX."},
+	"Internal transfers only":     {T: "Internal transfers only"},
+	"internal_only_tooltip":       {T: "When enabled, the bot will be able to use any available funds in the wallet to simulate a transfer between the DEX and the CEX, (i.e. increase the bot's DEX balance and decrease the bot's CEX balance when a withdrawal needs to be made) but no actual transfers will be made."},
 }
diff --git a/client/webserver/site/src/html/mmsettings.tmpl b/client/webserver/site/src/html/mmsettings.tmpl
index 2402fef851..7f18f3797f 100644
--- a/client/webserver/site/src/html/mmsettings.tmpl
+++ b/client/webserver/site/src/html/mmsettings.tmpl
@@ -659,6 +659,16 @@
             <span class="ico-lever fs20 pe-2"></span>
             <span class="fs22">Knobs</span>
           </div>
+
+          {{- /* TOP UP FEE RESERVES */ -}}
+          <label id="topUpFeeReserveSettings" for="topUpFeeReservesCheckbox" class="fs16 d-flex align-items-center justify-content-between mt-2 pt-2 border-top">
+            <div class="flex-center">
+              <input type="checkbox" id="topUpFeeReservesCheckbox" class="form-check-input me-2 mt-0" checked>
+              <span>[[[Top up fee reserves]]]</span>
+            </div>
+            <span class="ico-info fs12" data-tooltip="[[[top_up_fee_reserves_tooltip]]]"></span>
+          </label>
+
           {{- /* CEX REBALANCE CHECKBOX */ -}}
           <label id="cexRebalanceSettings" for="cexRebalanceCheckbox" class="fs16 d-flex align-items-center justify-content-between mt-2 pt-2 border-top">
             <div class="flex-center">
@@ -668,6 +678,24 @@
             <span class="ico-info fs12" data-tooltip="[[[enable_rebalance_tooltip]]]"></span>
           </label>
 
+          {{- /* ALLOW EXTERNAL TRANSFERS */ -}}
+          <label id="externalTransfersSettings" for="externalTransfersRadio" class="fs16 d-flex align-items-center justify-content-between mt-1 pt-1 ms-2">
+            <div class="flex-center">
+              <input type="radio" id="externalTransfersRadio" class="form-check-input me-2 mt-0">
+              <span>[[[Allow external transfers]]]</span>
+            </div>
+            <span class="ico-info fs12" data-tooltip="[[[external_transfers_tooltip]]]"></span>
+          </label>
+
+          {{- /* INTERNAL TRANSFERS ONLY */ -}}
+          <label id="internalOnlySettings" for="internalOnlyRadio" class="fs16 d-flex align-items-center justify-content-between mt-1 pt-1 ms-2">
+            <div class="flex-center">
+              <input type="radio" id="internalOnlyRadio" class="form-check-input me-2 mt-0" checked>
+              <span>[[[Internal transfers only]]]</span>
+            </div>
+            <span class="ico-info fs12" data-tooltip="[[[internal_only_tooltip]]]"></span>
+          </label>
+
           {{- /* DRIFT TOLERANCE */ -}}
           <div id="driftToleranceBox" class="flex-stretch-column">
             <div class="d-flex align-items-center justify-content-between mt-2 pt-2 border-top">
diff --git a/client/webserver/site/src/js/mm.ts b/client/webserver/site/src/js/mm.ts
index 6b61611805..d8d8d1a4ce 100644
--- a/client/webserver/site/src/js/mm.ts
+++ b/client/webserver/site/src/js/mm.ts
@@ -8,6 +8,7 @@ import {
   StartConfig,
   OrderPlacement,
   AutoRebalanceConfig,
+  UIConfig,
   CEXNotification
 } from './registry'
 import {
@@ -742,7 +743,7 @@ class Bot extends BotMarket {
   }
 
   async start () {
-    const { page, alloc, baseID, quoteID, host, cexName, cfg: { uiConfig: { cexRebalance } } } = this
+    const { page, alloc, baseID, quoteID, host, cexName, cfg: { uiConfig } } = this
 
     Doc.hide(page.errMsg)
     if (cexName && !app().mmStatus.cexes[cexName]?.connected) {
@@ -762,7 +763,8 @@ class Bot extends BotMarket {
       host: host,
       alloc: alloc
     }
-    if (cexName && cexRebalance) startConfig.autoRebalance = this.autoRebalanceSettings()
+
+    startConfig.autoRebalance = this.autoRebalanceSettings(uiConfig)
 
     try {
       app().log('mm', 'starting mm bot', startConfig)
@@ -776,7 +778,7 @@ class Bot extends BotMarket {
     this.hideAllocationDialog()
   }
 
-  autoRebalanceSettings (): AutoRebalanceConfig {
+  minTransferAmounts (): [number, number] {
     const {
       proj: { bProj, qProj, alloc }, baseFeeID, quoteFeeID, cfg: { uiConfig: { baseConfig, quoteConfig } },
       baseID, quoteID, cexName, mktID
@@ -801,7 +803,23 @@ class Bot extends BotMarket {
     const minBaseTransfer = Math.round(minB + baseConfig.transferFactor * (maxB - minB))
     const [minQ, maxQ] = [mkt.quoteMinWithdraw, Math.max(mkt.quoteMinWithdraw * 2, maxQuote)]
     const minQuoteTransfer = Math.round(minQ + quoteConfig.transferFactor * (maxQ - minQ))
-    return { minBaseTransfer, minQuoteTransfer }
+    return [minBaseTransfer, minQuoteTransfer]
+  }
+
+  autoRebalanceSettings (uiCfg: UIConfig) : AutoRebalanceConfig | undefined {
+    if (!uiCfg.cexRebalance && !uiCfg.internalTransfers && !uiCfg.topUpFeeReserves) return
+    const cfg : AutoRebalanceConfig = {
+      topUpFeeReserves: uiCfg.topUpFeeReserves,
+      internalTransfers: uiCfg.internalTransfers,
+      minBaseTransfer: 0,
+      minQuoteTransfer: 0
+    }
+    if (uiCfg.cexRebalance && this.cexName) {
+      const [minBaseTransfer, minQuoteTransfer] = this.minTransferAmounts()
+      cfg.minBaseTransfer = minBaseTransfer
+      cfg.minQuoteTransfer = minQuoteTransfer
+    }
+    return cfg
   }
 
   reconfigure () {
diff --git a/client/webserver/site/src/js/mmsettings.ts b/client/webserver/site/src/js/mmsettings.ts
index 0a5e0f44e5..753343986c 100644
--- a/client/webserver/site/src/js/mmsettings.ts
+++ b/client/webserver/site/src/js/mmsettings.ts
@@ -179,6 +179,8 @@ interface ConfigState {
   driftTolerance: number
   orderPersistence: number // epochs
   cexRebalance: boolean
+  internalTransfers: boolean
+  topUpFeeReserves: boolean
   disabled: boolean
   buyPlacements: OrderPlacement[]
   sellPlacements: OrderPlacement[]
@@ -295,6 +297,9 @@ export default class MarketMakerSettingsPage extends BasePage {
     Doc.bind(page.marketHeader, 'click', () => { this.showMarketSelectForm() })
     Doc.bind(page.marketFilterInput, 'input', () => { this.sortMarketRows() })
     Doc.bind(page.cexRebalanceCheckbox, 'change', () => { this.autoRebalanceChanged() })
+    Doc.bind(page.topUpFeeReserveSettings, 'change', () => { this.topUpFeeReserveChanged() })
+    Doc.bind(page.internalOnlyRadio, 'change', () => { this.internalOnlyChanged() })
+    Doc.bind(page.externalTransfersRadio, 'change', () => { this.externalTransfersChanged() })
     Doc.bind(page.switchToAdvanced, 'click', () => { this.showAdvancedConfig() })
     Doc.bind(page.switchToQuickConfig, 'click', () => { this.switchToQuickConfig() })
     Doc.bind(page.qcMatchBuffer, 'change', () => { this.matchBufferChanged() })
@@ -594,7 +599,7 @@ export default class MarketMakerSettingsPage extends BasePage {
     }) as ConfigState
 
     if (botCfg) {
-      const { basicMarketMakingConfig: mmCfg, arbMarketMakingConfig: arbMMCfg, simpleArbConfig: arbCfg, uiConfig: { cexRebalance } } = botCfg
+      const { basicMarketMakingConfig: mmCfg, arbMarketMakingConfig: arbMMCfg, simpleArbConfig: arbCfg, uiConfig } = botCfg
       this.creatingNewBot = false
       // This is kinda sloppy, but we'll copy any relevant issues from the
       // old config into the originalConfig.
@@ -605,7 +610,9 @@ export default class MarketMakerSettingsPage extends BasePage {
       oldCfg.quoteConfig = Object.assign({}, defaultBotAssetConfig, botCfg.uiConfig.quoteConfig)
       oldCfg.baseOptions = botCfg.baseWalletOptions || {}
       oldCfg.quoteOptions = botCfg.quoteWalletOptions || {}
-      oldCfg.cexRebalance = cexRebalance
+      oldCfg.cexRebalance = uiConfig.cexRebalance
+      oldCfg.internalTransfers = uiConfig.internalTransfers
+      oldCfg.topUpFeeReserves = uiConfig.topUpFeeReserves
 
       if (mmCfg) {
         oldCfg.buyPlacements = mmCfg.buyPlacements
@@ -648,9 +655,15 @@ export default class MarketMakerSettingsPage extends BasePage {
     setMarketElements(document.body, baseID, quoteID, host)
     Doc.setVis(botType !== botTypeBasicArb, page.driftToleranceBox, page.switchToAdvanced)
     Doc.setVis(Boolean(cexName), ...Doc.applySelector(document.body, '[data-cex-show]'))
-
     Doc.setVis(viewOnly, page.viewOnlyRunning)
+
+    const baseFeeAssetID = baseToken?.parentID ?? baseID
+    const quoteFeeAssetID = quoteToken?.parentID ?? quoteID
+    let showFeeReserveTopUpSetting = baseFeeAssetID !== baseID && baseFeeAssetID !== quoteID
+    showFeeReserveTopUpSetting = showFeeReserveTopUpSetting || (quoteFeeAssetID !== baseID && quoteFeeAssetID !== quoteID)
+    Doc.setVis(showFeeReserveTopUpSetting, page.topUpFeeReserveSettings)
     Doc.setVis(cexName, page.cexRebalanceSettings)
+
     if (cexName) setCexElements(document.body, cexName)
 
     await this.fetchMarketReport()
@@ -1168,12 +1181,55 @@ export default class MarketMakerSettingsPage extends BasePage {
     }
   }
 
+  internalOnlyChanged () {
+    const checked = Boolean(this.page.internalOnlyRadio.checked)
+    this.page.externalTransfersRadio.checked = !checked
+    this.updatedConfig.cexRebalance = !checked
+    this.updatedConfig.internalTransfers = checked
+    this.updateAllocations()
+  }
+
+  externalTransfersChanged () {
+    const checked = Boolean(this.page.externalTransfersRadio.checked)
+    this.page.internalOnlyRadio.checked = !checked
+    this.updatedConfig.cexRebalance = checked
+    this.updatedConfig.internalTransfers = !checked
+    this.updateAllocations()
+  }
+
   autoRebalanceChanged () {
     const { page, updatedConfig: cfg } = this
-    cfg.cexRebalance = page.cexRebalanceCheckbox?.checked ?? false
+    const checked = page.cexRebalanceCheckbox?.checked
+    Doc.setVis(checked, page.internalOnlySettings, page.externalTransfersSettings)
+    if (checked && !cfg.cexRebalance && !cfg.internalTransfers) {
+      // default to external transfers
+      cfg.cexRebalance = true
+      page.externalTransfersRadio.checked = true
+      page.internalOnlyRadio.checked = false
+    } else if (!checked) {
+      cfg.cexRebalance = false
+      cfg.internalTransfers = false
+      page.externalTransfersRadio.checked = false
+      page.internalOnlyRadio.checked = false
+    } else if (cfg.cexRebalance && cfg.internalTransfers) {
+      // should not happen.. set to default
+      cfg.internalTransfers = false
+      page.externalTransfersRadio.checked = true
+      page.internalOnlyRadio.checked = false
+    } else {
+      // set to current values. This case should only be called when the form
+      // is loaded.
+      page.externalTransfersRadio.checked = cfg.cexRebalance
+      page.internalOnlyRadio.checked = cfg.internalTransfers
+    }
+
     this.updateAllocations()
   }
 
+  topUpFeeReserveChanged () {
+    this.updatedConfig.topUpFeeReserves = Boolean(this.page.topUpFeeReservesCheckbox.checked)
+  }
+
   async submitBotType () {
     const loaded = app().loading(this.page.botTypeForm)
     try {
@@ -1620,9 +1676,12 @@ export default class MarketMakerSettingsPage extends BasePage {
     this.qcProfitSlider.setValue((profit - defaultProfit.minV) / defaultProfit.range)
 
     if (cexName) {
-      page.cexRebalanceCheckbox.checked = cfg.cexRebalance
+      page.cexRebalanceCheckbox.checked = cfg.cexRebalance || cfg.internalTransfers
+      page.internalOnlyRadio.checked = cfg.internalTransfers
+      page.externalTransfersRadio.checked = cfg.cexRebalance
       this.autoRebalanceChanged()
     }
+    page.topUpFeeReservesCheckbox.checked = cfg.topUpFeeReserves
 
     // Gap strategy
     if (!page.gapStrategySelect.options) return
@@ -1701,7 +1760,9 @@ export default class MarketMakerSettingsPage extends BasePage {
         simpleArbLots: cfg.simpleArbLots,
         baseConfig: cfg.baseConfig,
         quoteConfig: cfg.quoteConfig,
-        cexRebalance: cfg.cexRebalance
+        cexRebalance: cfg.cexRebalance,
+        internalTransfers: cfg.internalTransfers,
+        topUpFeeReserves: cfg.topUpFeeReserves
       },
       baseWalletOptions: cfg.baseOptions,
       quoteWalletOptions: cfg.quoteOptions
diff --git a/client/webserver/site/src/js/registry.ts b/client/webserver/site/src/js/registry.ts
index 5e8c088bf2..38213ebe26 100644
--- a/client/webserver/site/src/js/registry.ts
+++ b/client/webserver/site/src/js/registry.ts
@@ -757,6 +757,8 @@ export interface OrderPlacement {
 export interface AutoRebalanceConfig {
   minBaseTransfer: number
   minQuoteTransfer: number
+  internalTransfers: boolean
+  topUpFeeReserves: boolean
 }
 
 export interface BasicMarketMakingConfig {
@@ -807,6 +809,8 @@ export interface UIConfig {
   quoteConfig: BotAssetConfig
   simpleArbLots?: number
   cexRebalance: boolean
+  internalTransfers: boolean
+  topUpFeeReserves: boolean
 }
 
 export interface StartConfig extends MarketWithHost {