1+ import { Action , ActionExample , composeContext , Content , elizaLogger , generateObject , HandlerCallback , IAgentRuntime , Memory , ModelClass , settings , State } from "@elizaos/core" ;
2+ import { address , Address , createSolanaRpc } from "@solana/web3.js" ;
3+ import { fetchPositionsForOwner , HydratedPosition } from "@orca-so/whirlpools"
4+ import { loadWallet } from "../../utils/loadWallet" ;
5+ import { fetchWhirlpool , Whirlpool } from "@orca-so/whirlpools-client" ;
6+ import { sqrtPriceToPrice , tickIndexToPrice } from "@orca-so/whirlpools-core" ;
7+ import { fetchMint , Mint } from "@solana-program/token-2022"
8+
9+ export interface fetchPositionsByOwnerContent extends Content {
10+ owner : Address | null ;
11+ }
12+
13+ interface FetchedPositionResponse {
14+ whirlpoolAddress : Address ;
15+ positionMint : Address ;
16+ inRange : boolean ;
17+ distanceCenterPositionFromPoolPriceBps : number ;
18+ positionWidthBps : number ;
19+ }
20+
21+ function isPositionOwnerContent (
22+ content : any
23+ ) : content is fetchPositionsByOwnerContent {
24+ return ( typeof content . owner === "string" ) || ( content . owner === null ) ;
25+ }
26+
27+ export default {
28+ name : "FETCH_POSITIONS_BY_OWNER" ,
29+ similes : [
30+ "FETCH_POSITIONS" ,
31+ "FETCH_ORCA_POSITIONS" ,
32+ "FETCH_ORCA_POSITIONS_BY_OWNER" ,
33+ "FETCH_ORCA_POSITIONS_BY_WALLET" ,
34+ ] ,
35+ validate : async ( runtime : IAgentRuntime , message : Memory ) => {
36+ console . log ( "Validating transfer from user:" , message . userId ) ;
37+ return true ;
38+ } ,
39+ description : "Fetch all positions on Orca for the agent's wallet" ,
40+ handler : async (
41+ runtime : IAgentRuntime ,
42+ message : Memory ,
43+ state : State ,
44+ _options : { [ key : string ] : unknown } ,
45+ callback ?: HandlerCallback
46+ ) : Promise < boolean > => {
47+ elizaLogger . log ( "Fetching positions from Orca..." ) ;
48+
49+ // Initialize or update state
50+ if ( ! state ) {
51+ state = ( await runtime . composeState ( message ) ) as State ;
52+ } else {
53+ state = await runtime . updateRecentMessageState ( state ) ;
54+ }
55+
56+ // Compose fetch positions context
57+ const fetchPositionsContext = composeContext ( {
58+ state,
59+ template : `Respond with a JSON markdown block containing only the extracted values. Use null for any values that cannot be determined.
60+
61+ Example response:
62+ \`\`\`json
63+ {
64+ owner: "BieefG47jAHCGZBxi2q87RDuHyGZyYC3vAzxpyu8pump"
65+ }
66+ \`\`\`
67+ ` ,
68+ } ) ;
69+
70+ // Generate fetch positions content
71+ const content = await generateObject ( {
72+ runtime,
73+ context : fetchPositionsContext ,
74+ modelClass : ModelClass . LARGE ,
75+ } ) ;
76+
77+ if ( ! isPositionOwnerContent ( content ) ) {
78+ if ( callback ) {
79+ callback ( {
80+ text : "Unable to process transfer request. Invalid content provided." ,
81+ content : { error : "Invalid transfer content" } ,
82+ } ) ;
83+ }
84+ return false ;
85+ }
86+
87+ try {
88+ let ownerAddress : Address ;
89+ if ( content . owner ) {
90+ ownerAddress = address ( content . owner ) ;
91+ } else {
92+ const { address } = await loadWallet (
93+ runtime ,
94+ true
95+ ) ;
96+ ownerAddress = address ;
97+ }
98+
99+
100+ const rpc = createSolanaRpc ( settings . RPC_URL ! ) ;
101+ const positions = await fetchPositionsForOwner ( rpc , ownerAddress ) ;
102+
103+ const fetchedWhirlpools : Map < string , Whirlpool > = new Map ( ) ;
104+ const fetchedMints : Map < string , Mint > = new Map ( ) ;
105+ const positionContent : FetchedPositionResponse [ ] = await Promise . all ( positions . map ( async ( position ) => {
106+ const positionData = ( position as HydratedPosition ) . data ;
107+ const positionMint = positionData . positionMint
108+ const whirlpoolAddress = positionData . whirlpool ;
109+ if ( ! fetchedWhirlpools . has ( whirlpoolAddress ) ) {
110+ const whirlpool = await fetchWhirlpool ( rpc , whirlpoolAddress ) ;
111+ if ( whirlpool ) {
112+ fetchedWhirlpools . set ( whirlpoolAddress , whirlpool . data ) ;
113+ }
114+ }
115+ const whirlpool = fetchedWhirlpools . get ( whirlpoolAddress ) ;
116+ const { tokenMintA, tokenMintB } = whirlpool ;
117+ if ( ! fetchedMints . has ( tokenMintA ) ) {
118+ const mintA = await fetchMint ( rpc , tokenMintA ) ;
119+ fetchedMints . set ( tokenMintA , mintA . data ) ;
120+ }
121+ if ( ! fetchedMints . has ( tokenMintB ) ) {
122+ const mintB = await fetchMint ( rpc , tokenMintB ) ;
123+ fetchedMints . set ( tokenMintB , mintB . data ) ;
124+ }
125+ const mintA = fetchedMints . get ( tokenMintA ) ;
126+ const mintB = fetchedMints . get ( tokenMintB ) ;
127+ const currentPrice = sqrtPriceToPrice ( whirlpool . sqrtPrice , mintA . decimals , mintB . decimals ) ;
128+ const positionLowerPrice = tickIndexToPrice ( positionData . tickLowerIndex , mintA . decimals , mintB . decimals ) ;
129+ const positionUpperPrice = tickIndexToPrice ( positionData . tickUpperIndex , mintA . decimals , mintB . decimals ) ;
130+
131+ const inRange = currentPrice >= positionLowerPrice && currentPrice <= positionUpperPrice ;
132+ const positionCenterPrice = ( positionLowerPrice + positionUpperPrice ) / 2 ;
133+ const distanceCenterPositionFromPoolPriceBps = Math . abs ( currentPrice - positionCenterPrice ) / currentPrice * 10000 ;
134+ const positionWidthBps = ( positionUpperPrice - positionLowerPrice ) / positionCenterPrice * 10000 ;
135+
136+ return {
137+ whirlpoolAddress,
138+ positionMint : positionMint ,
139+ inRange,
140+ distanceCenterPositionFromPoolPriceBps,
141+ positionWidthBps,
142+ } as FetchedPositionResponse ;
143+ } ) ) ;
144+
145+ if ( callback ) {
146+ callback ( {
147+ text : JSON . stringify ( positionContent , null , 2 ) ,
148+ } ) ;
149+ }
150+
151+ return true ;
152+ } catch ( error ) {
153+ console . error ( "Error during feching positions" , error ) ;
154+ if ( callback ) {
155+ callback ( {
156+ text : `Error transferring tokens: ${ error . message } ,
157+ content : { error : error . message } ,
158+ } ) ;
159+ }
160+ return false ;
161+ }
162+ } ,
163+
164+ examples : [
165+ [
166+ {
167+ user : "{{user1}}" ,
168+ content : {
169+ text : "Fetch all positions on Orca for the agent's wallet" ,
170+ } ,
171+ } ,
172+ {
173+ user : "{{user2}}" ,
174+ content : {
175+ text : "What are my positions on Orca?" ,
176+ action : "FETCH_ORCA_POSITIONS_BY_WALLET" ,
177+ } ,
178+ } ,
179+ {
180+ user : "{{user2}}" ,
181+ content : {
182+ text : "What are the position for owner BieefG47jAHCGZBxi2q87RDuHyGZyYC3vAzxpyu8pump?" ,
183+ action : "FETCH_ORCA_POSITIONS_BY_OWNER" ,
184+ } ,
185+ } ,
186+ ] ,
187+ ] as ActionExample [ ] [ ] ,
188+ } as Action ;
0 commit comments