From 190122a669e16f9d41e82eb97c9555af7d37c71c Mon Sep 17 00:00:00 2001
From: philippe-ftx <62417741+philippe-ftx@users.noreply.github.com>
Date: Mon, 5 Oct 2020 13:26:34 +0200
Subject: [PATCH] Auto settle (#24)

* Auto settle funds function

* Auto settle

* Add settings panel

* Cleanup

* Use OpenOrders.findForOwner and support custom markets

* Some touch ups

Co-authored-by: Nathaniel Parke <nathaniel.parke@gmail.com>
---
 src/App.js                  |  9 ++--
 src/components/Settings.jsx | 26 +++++++++++
 src/components/TopBar.js    | 22 +++++++++-
 src/utils/markets.js        | 11 +++--
 src/utils/preferences.js    | 55 +++++++++++++++++++++++
 src/utils/send.js           | 87 +++++++++++++++++++++++++++++++++++++
 src/utils/wallet.js         |  4 ++
 7 files changed, 206 insertions(+), 8 deletions(-)
 create mode 100644 src/components/Settings.jsx
 create mode 100644 src/utils/preferences.js

diff --git a/src/App.js b/src/App.js
index a6b15b9..cdf6679 100644
--- a/src/App.js
+++ b/src/App.js
@@ -7,6 +7,7 @@ import { GlobalStyle } from './global_style';
 import { Spin } from 'antd';
 import ErrorBoundary from './components/ErrorBoundary';
 import { Routes } from './routes';
+import { PreferencesProvider } from './utils/preferences';
 
 export default function App() {
   return (
@@ -16,9 +17,11 @@ export default function App() {
         <ConnectionProvider>
           <MarketProvider>
             <WalletProvider>
-              <Suspense fallback={() => <Spin size="large" />}>
-                <Routes />
-              </Suspense>
+              <PreferencesProvider>
+                <Suspense fallback={() => <Spin size="large" />}>
+                  <Routes />
+                </Suspense>
+              </PreferencesProvider>
             </WalletProvider>
           </MarketProvider>
         </ConnectionProvider>
diff --git a/src/components/Settings.jsx b/src/components/Settings.jsx
new file mode 100644
index 0000000..e262d57
--- /dev/null
+++ b/src/components/Settings.jsx
@@ -0,0 +1,26 @@
+import React from 'react';
+import { Switch, Typography } from 'antd';
+import { usePreferences } from '../utils/preferences';
+
+const { Paragraph } = Typography;
+
+export default function Settings({ autoApprove }) {
+  const { autoSettleEnabled, setAutoSettleEnabled } = usePreferences();
+
+  return (
+    <div>
+      <Switch
+        style={{ marginRight: 10 }}
+        disabled={!autoApprove}
+        checked={autoApprove && autoSettleEnabled}
+        onChange={setAutoSettleEnabled}
+      />{' '}
+      Auto settle
+      {!autoApprove && (
+        <Paragraph style={{ color: 'rgba(255,255,255,0.5)', marginTop: 10 }}>
+          To use auto settle, first enable auto approval in your wallet
+        </Paragraph>
+      )}
+    </div>
+  );
+}
diff --git a/src/components/TopBar.js b/src/components/TopBar.js
index 839eb07..5544533 100644
--- a/src/components/TopBar.js
+++ b/src/components/TopBar.js
@@ -1,4 +1,8 @@
-import { InfoCircleOutlined, UserOutlined } from '@ant-design/icons';
+import {
+  InfoCircleOutlined,
+  SettingOutlined,
+  UserOutlined,
+} from '@ant-design/icons';
 import { Button, Menu, Popover, Select } from 'antd';
 import React, { useCallback } from 'react';
 import { useHistory, useLocation } from 'react-router-dom';
@@ -7,6 +11,7 @@ import styled from 'styled-components';
 import { useWallet, WALLET_PROVIDERS } from '../utils/wallet';
 import { ENDPOINTS, useConnectionConfig } from '../utils/connection';
 import LinkAddress from './LinkAddress';
+import Settings from './Settings';
 
 const Wrapper = styled.div`
   background-color: #0d1017;
@@ -63,6 +68,21 @@ export default function TopBar() {
       >
         <Menu.Item key="/">TRADE</Menu.Item>
       </Menu>
+      {connected && (
+        <div>
+          <Popover
+            content={<Settings autoApprove={wallet?.autoApprove} />}
+            placement="bottomRight"
+            title="Settings"
+            trigger="click"
+          >
+            <Button style={{ marginRight: 8 }}>
+              <SettingOutlined />
+              Settings
+            </Button>
+          </Popover>
+        </div>
+      )}
       <div>
         <Select
           onSelect={setEndpoint}
diff --git a/src/utils/markets.js b/src/utils/markets.js
index e397b1f..804f94d 100644
--- a/src/utils/markets.js
+++ b/src/utils/markets.js
@@ -29,15 +29,14 @@ export function useMarketsList() {
   return USE_MARKETS.filter(({ deprecated }) => !deprecated);
 }
 
-export function useAllMarkets() {
+export function useAllMarkets(customMarkets) {
   const connection = useConnection();
   const [markets, setMarkets] = useState([]);
 
   useEffect(() => {
     const getAllMarkets = async () => {
       const markets = [];
-      let marketInfo;
-      for (marketInfo of USE_MARKETS) {
+      for (let marketInfo of getMarketInfos(customMarkets)) {
         try {
           const market = await Market.load(
             connection,
@@ -45,7 +44,11 @@ export function useAllMarkets() {
             {},
             marketInfo.programId,
           );
-          markets.push({ market, marketName: marketInfo.name });
+          markets.push({
+            market,
+            marketName: marketInfo.name,
+            programId: marketInfo.programId,
+          });
         } catch (e) {
           notify({
             message: 'Error loading all market',
diff --git a/src/utils/preferences.js b/src/utils/preferences.js
new file mode 100644
index 0000000..8f4edcd
--- /dev/null
+++ b/src/utils/preferences.js
@@ -0,0 +1,55 @@
+import React, { useContext } from 'react';
+import { useLocalStorageState } from './utils';
+import { useInterval } from './useInterval';
+import { useConnection } from './connection';
+import { useWallet } from './wallet';
+import { useAllMarkets, useTokenAccounts, useMarket } from './markets';
+import { settleAllFunds } from './send';
+
+const PreferencesContext = React.createContext(null);
+
+export function PreferencesProvider({ children }) {
+  const [autoSettleEnabled, setAutoSettleEnabled] = useLocalStorageState(
+    'autoSettleEnabled',
+    false,
+  );
+
+  const [tokenAccounts] = useTokenAccounts();
+  const { connected, wallet } = useWallet();
+  const { customMarkets } = useMarket();
+  const marketList = useAllMarkets(customMarkets);
+  const connection = useConnection();
+
+  useInterval(() => {
+    const autoSettle = async () => {
+      const markets = marketList.map((m) => m.market);
+      try {
+        console.log('Auto settling');
+        await settleAllFunds({ connection, wallet, tokenAccounts, markets });
+      } catch (e) {
+        console.log('Error auto settling funds: ' + e.message);
+      }
+    };
+
+    connected && wallet?.autoApprove && autoSettleEnabled && autoSettle();
+  }, 10000);
+
+  return (
+    <PreferencesContext.Provider
+      value={{
+        autoSettleEnabled,
+        setAutoSettleEnabled,
+      }}
+    >
+      {children}
+    </PreferencesContext.Provider>
+  );
+}
+
+export function usePreferences() {
+  const context = useContext(PreferencesContext);
+  return {
+    autoSettleEnabled: context.autoSettleEnabled,
+    setAutoSettleEnabled: context.setAutoSettleEnabled,
+  };
+}
diff --git a/src/utils/send.js b/src/utils/send.js
index e08c2a6..6d703a5 100644
--- a/src/utils/send.js
+++ b/src/utils/send.js
@@ -1,5 +1,6 @@
 import { notify } from './notifications';
 import { getDecimalCount, sleep } from './utils';
+import { getSelectedTokenAccountForMint } from './markets';
 import {
   Account,
   PublicKey,
@@ -12,6 +13,7 @@ import {
   Market,
   TOKEN_MINTS,
   TokenInstructions,
+  OpenOrders,
 } from '@project-serum/serum';
 
 export async function createTokenAccountTransaction({
@@ -135,6 +137,91 @@ export async function settleFunds({
   });
 }
 
+export async function settleAllFunds({
+  connection,
+  wallet,
+  tokenAccounts,
+  markets,
+}) {
+  if (!markets || !wallet || !connection || !tokenAccounts) {
+    return;
+  }
+
+  const programIds = [];
+  markets
+    .reduce((cumulative, m) => {
+      cumulative.push(m._programId);
+      return cumulative;
+    }, [])
+    .forEach((programId) => {
+      if (!programIds.find((p) => p.equals(programId))) {
+        programIds.push(programId);
+      }
+    });
+
+  const getOpenOrdersAccountsForProgramId = async (programId) => {
+    const openOrdersAccounts = await OpenOrders.findForOwner(
+      connection,
+      wallet.publicKey,
+      programId,
+    );
+    return openOrdersAccounts.filter(
+      (openOrders) =>
+        openOrders.baseTokenTotal.toNumber() ||
+        openOrders.quoteTokenTotal.toNumber(),
+    );
+  };
+
+  const openOrdersAccountsForProgramIds = await Promise.all(
+    programIds.map((programId) => getOpenOrdersAccountsForProgramId(programId)),
+  );
+  const openOrdersAccounts = openOrdersAccountsForProgramIds.reduce(
+    (accounts, current) => accounts.concat(current),
+    [],
+  );
+
+  const settleTransactions = await Promise.all(
+    openOrdersAccounts.map((openOrdersAccount) => {
+      const market = markets.find((m) =>
+        m._decoded?.ownAddress?.equals(openOrdersAccount.market),
+      );
+      return (
+        market &&
+        market.makeSettleFundsTransaction(
+          connection,
+          openOrdersAccount,
+          getSelectedTokenAccountForMint(tokenAccounts, market?.baseMintAddress)
+            ?.pubkey,
+          getSelectedTokenAccountForMint(
+            tokenAccounts,
+            market?.quoteMintAddress,
+          )?.pubkey,
+        )
+      );
+    }),
+  );
+  if (!settleTransactions || settleTransactions.length === 0) return;
+
+  const transactions = settleTransactions.slice(0, 4).map((t) => t.transaction);
+  const signers = [];
+  settleTransactions
+    .reduce((cumulative, t) => cumulative.concat(t.signers), [])
+    .forEach((signer) => {
+      if (!signers.find((s) => s.constructor.name === signer.constructor.name && s.equals(signer))) {
+        signers.push(signer);
+      }
+    });
+
+  const transaction = mergeTransactions(transactions);
+
+  return await sendTransaction({
+    transaction,
+    signers,
+    wallet,
+    connection,
+  });
+}
+
 export async function cancelOrder(params) {
   return cancelOrders({ ...params, orders: [params.order] });
 }
diff --git a/src/utils/wallet.js b/src/utils/wallet.js
index 0497b28..c986174 100644
--- a/src/utils/wallet.js
+++ b/src/utils/wallet.js
@@ -12,16 +12,19 @@ const WalletContext = React.createContext(null);
 
 export function WalletProvider({ children }) {
   const { endpoint } = useConnectionConfig();
+
   const [providerUrl, setProviderUrl] = useLocalStorageState(
     'walletProvider',
     'https://www.sollet.io',
   );
+
   const wallet = useMemo(() => new Wallet(providerUrl, endpoint), [
     providerUrl,
     endpoint,
   ]);
 
   const [connected, setConnected] = useState(false);
+
   useEffect(() => {
     console.log('trying to connect');
     wallet.on('connect', () => {
@@ -52,6 +55,7 @@ export function WalletProvider({ children }) {
       setConnected(false);
     };
   }, [wallet]);
+
   return (
     <WalletContext.Provider
       value={{