1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
use borsh::{BorshDeserialize, BorshSerialize};
use phoenix::state::{SelfTradeBehavior, Side};

#[derive(BorshDeserialize, BorshSerialize)]
pub enum OrderPacketEnum {
    PostOnly,
    Limit,
}

#[derive(BorshDeserialize, BorshSerialize)]
struct DeprecatedPostOnlyPacket {
    side: Side,

    /// The price of the order, in ticks
    price_in_ticks: u64,

    /// Number of base lots to place on the book
    num_base_lots: u64,

    /// Client order id used to identify the order in the response to the client
    client_order_id: u128,

    /// Flag for whether or not to reject the order if it would immediately match or amend it to the best non-crossing price
    /// Default value is true
    reject_post_only: bool,

    /// Flag for whether or not the order should only use funds that are already in the account
    /// Using only deposited funds will allow the trader to pass in less accounts per instruction and
    /// save transaction space as well as compute. This is only for traders who have a seat
    use_only_deposited_funds: bool,
}

#[derive(BorshDeserialize, BorshSerialize)]
pub struct PostOnlyPacket {
    pub side: Side,

    /// The price of the order, in ticks
    pub price_in_ticks: u64,

    /// Number of base lots to place on the book
    pub num_base_lots: u64,

    /// Client order id used to identify the order in the response to the client
    pub client_order_id: u128,

    /// Flag for whether or not to reject the order if it would immediately match or amend it to the best non-crossing price
    /// Default value is true
    pub reject_post_only: bool,

    /// Flag for whether or not the order should only use funds that are already in the account
    /// Using only deposited funds will allow the trader to pass in less accounts per instruction and
    /// save transaction space as well as compute. This is only for traders who have a seat
    pub use_only_deposited_funds: bool,

    /// If this is set, the order will be invalid after the specified slot
    pub last_valid_slot: Option<u64>,

    /// If this is set, the order will be invalid after the specified unix timestamp
    pub last_valid_unix_timestamp_in_seconds: Option<u64>,
}

#[derive(BorshDeserialize, BorshSerialize)]
struct DeprecatedLimitPacket {
    side: Side,

    /// The price of the order, in ticks
    price_in_ticks: u64,

    /// Total number of base lots to place on the book or fill at a better price
    num_base_lots: u64,

    /// How the matching engine should handle a self trade
    self_trade_behavior: SelfTradeBehavior,

    /// Number of orders to match against. If this is `None` there is no limit
    match_limit: Option<u64>,

    /// Client order id used to identify the order in the response to the client
    client_order_id: u128,

    /// Flag for whether or not the order should only use funds that are already in the account.
    /// Using only deposited funds will allow the trader to pass in less accounts per instruction and
    /// save transaction space as well as compute. This is only for traders who have a seat
    use_only_deposited_funds: bool,
}

#[derive(BorshDeserialize, BorshSerialize)]
pub struct LimitPacket {
    pub side: Side,

    /// The price of the order, in ticks
    pub price_in_ticks: u64,

    /// Total number of base lots to place on the book or fill at a better price
    pub num_base_lots: u64,

    /// How the matching engine should handle a self trade
    pub self_trade_behavior: SelfTradeBehavior,

    /// Number of orders to match against. If this is `None` there is no limit
    pub match_limit: Option<u64>,

    /// Client order id used to identify the order in the response to the client
    pub client_order_id: u128,

    /// Flag for whether or not the order should only use funds that are already in the account.
    /// Using only deposited funds will allow the trader to pass in less accounts per instruction and
    /// save transaction space as well as compute. This is only for traders who have a seat
    pub use_only_deposited_funds: bool,

    /// If this is set, the order will be invalid after the specified slot
    pub last_valid_slot: Option<u64>,

    /// If this is set, the order will be invalid after the specified unix timestamp
    pub last_valid_unix_timestamp_in_seconds: Option<u64>,
}

pub fn decode_post_only_packet_data(bytes: &[u8]) -> anyhow::Result<PostOnlyPacket> {
    let (tag, bytes) = bytes
        .split_first()
        .ok_or(anyhow::anyhow!("Invalid packet"))?;

    match OrderPacketEnum::try_from_slice(&[*tag])? {
        OrderPacketEnum::PostOnly => {
            let packet = PostOnlyPacket::try_from_slice(bytes)
                .map_err(|_| anyhow::Error::msg("Invalid Post-Only packet"));
            let deprecated_packet_result = DeprecatedPostOnlyPacket::try_from_slice(bytes);
            if packet.is_ok() {
                return packet;
            }
            let deprecated_packet = deprecated_packet_result?;
            let DeprecatedPostOnlyPacket {
                side,
                price_in_ticks,
                num_base_lots,
                client_order_id,
                reject_post_only,
                use_only_deposited_funds,
            } = deprecated_packet;
            Ok(PostOnlyPacket {
                side,
                price_in_ticks,
                num_base_lots,
                client_order_id,
                reject_post_only,
                use_only_deposited_funds,
                last_valid_slot: None,
                last_valid_unix_timestamp_in_seconds: None,
            })
        }
        _ => Err(anyhow::anyhow!("Invalid Post-Only packet")),
    }
}

pub fn decode_limit_packet_data(bytes: &[u8]) -> anyhow::Result<LimitPacket> {
    let (tag, bytes) = bytes
        .split_first()
        .ok_or(anyhow::anyhow!("Invalid packet"))?;

    match OrderPacketEnum::try_from_slice(&[*tag])? {
        OrderPacketEnum::Limit => {
            let packet = LimitPacket::try_from_slice(bytes)
                .map_err(|_| anyhow::Error::msg("Invalid Limit packet"));
            let deprecated_packet_result = DeprecatedLimitPacket::try_from_slice(bytes);
            if packet.is_ok() {
                return packet;
            }
            let deprecated_packet = deprecated_packet_result?;
            let DeprecatedLimitPacket {
                side,
                price_in_ticks,
                self_trade_behavior,
                match_limit,
                num_base_lots,
                client_order_id,
                use_only_deposited_funds,
            } = deprecated_packet;
            Ok(LimitPacket {
                side,
                price_in_ticks,
                self_trade_behavior,
                match_limit,
                num_base_lots,
                client_order_id,
                use_only_deposited_funds,
                last_valid_slot: None,
                last_valid_unix_timestamp_in_seconds: None,
            })
        }
        _ => Err(anyhow::anyhow!("Invalid Post-Only packet")),
    }
}