Force shared accounts for Jupiter swaps (#145)

Author: jeffyanta

pkg/code/server/grpc/transaction/v2/swap.go

@@ -176,6 +176,7 @@ func (s *transactionServer) Swap(streamer transactionpb.Transaction_SwapServer)
 		50,   // todo: configurable slippage or something based on liquidity?
 		true, // Direct routes for now since we're using legacy instructions
 		16,   // Max accounts limited due to the use of legacy instructions
+		true, // Use shared accounts so we don't have to manage ATAs
 		true, // Force legacy instructions
 	)
 	if err != nil {

pkg/jupiter/client.go

@@ -44,6 +44,7 @@ func NewClient(baseUrl string) *Client {
 type Quote struct {
 	jsonString            string
 	estimatedSwapAmount   uint64
+	useSharedAccounts     bool
 	useLegacyInstructions bool
 }
 
@@ -60,13 +61,14 @@ func (c *Client) GetQuote(
 	slippageBps uint32,
 	forceDirectRoute bool,
 	maxAccounts uint8,
+	useSharedAccounts bool,
 	useLegacyInstruction bool,
 ) (*Quote, error) {
 	tracer := metrics.TraceMethodCall(ctx, metricsStructName, "GetQuote")
 	defer tracer.End()
 
 	url := fmt.Sprintf(
-		"%s%s?inputMint=%s&outputMint=%s&amount=%d&slippageBps=%d&onlyDirectRoutes=%v&maxAccounts=%d&asLegacyTransaction=%v",
+		"%s%s?inputMint=%s&outputMint=%s&amount=%d&slippageBps=%d&onlyDirectRoutes=%v&maxAccounts=%d&useSharedAccounts=%v&asLegacyTransaction=%v",
 		c.baseUrl,
 		quoteEndpointName,
 		inputMint,
@@ -75,6 +77,7 @@ func (c *Client) GetQuote(
 		slippageBps,
 		forceDirectRoute,
 		maxAccounts,
+		useSharedAccounts,
 		useLegacyInstruction,
 	)
 
@@ -107,6 +110,7 @@ func (c *Client) GetQuote(
 	return &Quote{
 		jsonString:            string(respBody),
 		estimatedSwapAmount:   estimatedSwapAmount,
+		useSharedAccounts:     useSharedAccounts,
 		useLegacyInstructions: useLegacyInstruction,
 	}, nil
 }
@@ -136,10 +140,11 @@ func (c *Client) GetSwapInstructions(
 
 	// todo: struct this
 	reqBody := fmt.Sprintf(
-		`{"quoteResponse": %s, "userPublicKey": "%s", "destinationTokenAccount": "%s", "prioritizationFeeLamports": "auto", "asLegacyTransaction": %v}`,
+		`{"quoteResponse": %s, "userPublicKey": "%s", "destinationTokenAccount": "%s", "prioritizationFeeLamports": "auto", "useSharedAccounts": %v, "asLegacyTransaction": %v}`,
 		quote.jsonString,
 		owner,
 		destinationTokenAccount,
+		quote.useSharedAccounts,
 		quote.useLegacyInstructions,
 	)