1use std::hash::{Hash, Hasher};
17
18use nautilus_core::{
19 Params, UnixNanos,
20 correctness::{
21 CorrectnessResult, check_equal_u8, check_valid_string_ascii,
22 check_valid_string_ascii_optional,
23 },
24};
25use rust_decimal::Decimal;
26use serde::{Deserialize, Serialize};
27use ustr::Ustr;
28
29use super::{Instrument, any::InstrumentAny, tick_scheme::check_tick_scheme};
30use crate::{
31 enums::{AssetClass, InstrumentClass, OptionKind},
32 identifiers::{InstrumentId, Symbol},
33 types::{
34 currency::Currency,
35 money::Money,
36 price::{Price, check_positive_price},
37 quantity::{Quantity, check_positive_quantity},
38 },
39};
40
41#[repr(C)]
43#[derive(Clone, Debug, Serialize, Deserialize)]
44#[cfg_attr(
45 feature = "python",
46 pyo3::pyclass(module = "nautilus_trader.model", from_py_object)
47)]
48#[cfg_attr(
49 feature = "python",
50 pyo3_stub_gen::derive::gen_stub_pyclass(module = "nautilus_trader.model")
51)]
52pub struct OptionSpread {
53 pub id: InstrumentId,
55 pub raw_symbol: Symbol,
57 pub asset_class: AssetClass,
59 pub exchange: Option<Ustr>,
61 pub underlying: Ustr,
63 pub strategy_type: Ustr,
65 pub activation_ns: UnixNanos,
67 pub expiration_ns: UnixNanos,
69 pub currency: Currency,
71 pub price_precision: u8,
73 pub price_increment: Price,
75 pub size_increment: Quantity,
77 pub size_precision: u8,
79 pub multiplier: Quantity,
81 pub lot_size: Quantity,
83 pub margin_init: Decimal,
85 pub margin_maint: Decimal,
87 pub max_quantity: Option<Quantity>,
89 pub min_quantity: Option<Quantity>,
91 pub max_price: Option<Price>,
93 pub min_price: Option<Price>,
95 pub tick_scheme: Option<Ustr>,
97 pub info: Option<Params>,
99 pub ts_event: UnixNanos,
101 pub ts_init: UnixNanos,
103}
104
105#[bon::bon]
106impl OptionSpread {
107 #[expect(clippy::too_many_arguments)]
108 fn new_checked(
109 instrument_id: InstrumentId,
110 raw_symbol: Symbol,
111 asset_class: AssetClass,
112 exchange: Option<Ustr>,
113 underlying: Ustr,
114 strategy_type: Ustr,
115 activation_ns: UnixNanos,
116 expiration_ns: UnixNanos,
117 currency: Currency,
118 price_precision: u8,
119 price_increment: Price,
120 multiplier: Quantity,
121 lot_size: Quantity,
122 max_quantity: Option<Quantity>,
123 min_quantity: Option<Quantity>,
124 max_price: Option<Price>,
125 min_price: Option<Price>,
126 margin_init: Option<Decimal>,
127 margin_maint: Option<Decimal>,
128 tick_scheme: Option<Ustr>,
129 info: Option<Params>,
130 ts_event: UnixNanos,
131 ts_init: UnixNanos,
132 ) -> CorrectnessResult<Self> {
133 check_valid_string_ascii_optional(exchange, stringify!(exchange))?;
134 check_valid_string_ascii(strategy_type, stringify!(strategy_type))?;
135 check_equal_u8(
136 price_precision,
137 price_increment.precision,
138 stringify!(price_precision),
139 stringify!(price_increment.precision),
140 )?;
141 check_positive_price(price_increment, stringify!(price_increment))?;
142 check_tick_scheme(tick_scheme)?;
143 check_positive_quantity(multiplier, stringify!(multiplier))?;
144 check_positive_quantity(lot_size, stringify!(lot_size))?;
145
146 Ok(Self {
147 id: instrument_id,
148 raw_symbol,
149 asset_class,
150 exchange,
151 underlying,
152 strategy_type,
153 activation_ns,
154 expiration_ns,
155 currency,
156 price_precision,
157 price_increment,
158 size_precision: 0,
159 size_increment: Quantity::from("1"),
160 multiplier,
161 lot_size,
162 margin_init: margin_init.unwrap_or_default(),
163 margin_maint: margin_maint.unwrap_or_default(),
164 max_quantity,
165 min_quantity: Some(min_quantity.unwrap_or(1.into())),
166 max_price,
167 min_price,
168 tick_scheme,
169 info,
170 ts_event,
171 ts_init,
172 })
173 }
174
175 #[builder(start_fn = builder, finish_fn = build)]
184 pub fn build_checked(
185 instrument_id: InstrumentId,
186 raw_symbol: Symbol,
187 asset_class: AssetClass,
188 exchange: Option<Ustr>,
189 underlying: Ustr,
190 strategy_type: Ustr,
191 activation_ns: UnixNanos,
192 expiration_ns: UnixNanos,
193 currency: Currency,
194 price_precision: u8,
195 price_increment: Price,
196 multiplier: Quantity,
197 lot_size: Quantity,
198 max_quantity: Option<Quantity>,
199 min_quantity: Option<Quantity>,
200 max_price: Option<Price>,
201 min_price: Option<Price>,
202 margin_init: Option<Decimal>,
203 margin_maint: Option<Decimal>,
204 tick_scheme: Option<Ustr>,
205 info: Option<Params>,
206 ts_event: UnixNanos,
207 ts_init: UnixNanos,
208 ) -> CorrectnessResult<Self> {
209 Self::new_checked(
210 instrument_id,
211 raw_symbol,
212 asset_class,
213 exchange,
214 underlying,
215 strategy_type,
216 activation_ns,
217 expiration_ns,
218 currency,
219 price_precision,
220 price_increment,
221 multiplier,
222 lot_size,
223 max_quantity,
224 min_quantity,
225 max_price,
226 min_price,
227 margin_init,
228 margin_maint,
229 tick_scheme,
230 info,
231 ts_event,
232 ts_init,
233 )
234 }
235}
236
237impl PartialEq<Self> for OptionSpread {
238 fn eq(&self, other: &Self) -> bool {
239 self.id == other.id
240 }
241}
242
243impl Eq for OptionSpread {}
244
245impl Hash for OptionSpread {
246 fn hash<H: Hasher>(&self, state: &mut H) {
247 self.id.hash(state);
248 }
249}
250
251impl Instrument for OptionSpread {
252 fn into_any(self) -> InstrumentAny {
253 InstrumentAny::OptionSpread(self)
254 }
255
256 fn id(&self) -> InstrumentId {
257 self.id
258 }
259
260 fn raw_symbol(&self) -> Symbol {
261 self.raw_symbol
262 }
263
264 fn asset_class(&self) -> AssetClass {
265 self.asset_class
266 }
267
268 fn instrument_class(&self) -> InstrumentClass {
269 InstrumentClass::OptionSpread
270 }
271 fn underlying(&self) -> Option<Ustr> {
272 Some(self.underlying)
273 }
274
275 fn base_currency(&self) -> Option<Currency> {
276 None
277 }
278
279 fn quote_currency(&self) -> Currency {
280 self.currency
281 }
282
283 fn settlement_currency(&self) -> Currency {
284 self.currency
285 }
286
287 fn isin(&self) -> Option<Ustr> {
288 None
289 }
290
291 fn option_kind(&self) -> Option<OptionKind> {
292 None
293 }
294
295 fn exchange(&self) -> Option<Ustr> {
296 self.exchange
297 }
298
299 fn strike_price(&self) -> Option<Price> {
300 None
301 }
302
303 fn strategy_type(&self) -> Option<Ustr> {
304 Some(self.strategy_type)
305 }
306
307 fn activation_ns(&self) -> Option<UnixNanos> {
308 Some(self.activation_ns)
309 }
310
311 fn expiration_ns(&self) -> Option<UnixNanos> {
312 Some(self.expiration_ns)
313 }
314
315 fn is_inverse(&self) -> bool {
316 false
317 }
318
319 fn price_precision(&self) -> u8 {
320 self.price_precision
321 }
322
323 fn size_precision(&self) -> u8 {
324 0 }
326
327 fn price_increment(&self) -> Price {
328 self.price_increment
329 }
330
331 fn size_increment(&self) -> Quantity {
332 Quantity::from(1)
333 }
334
335 fn multiplier(&self) -> Quantity {
336 self.multiplier
337 }
338
339 fn lot_size(&self) -> Option<Quantity> {
340 Some(self.lot_size)
341 }
342
343 fn max_quantity(&self) -> Option<Quantity> {
344 self.max_quantity
345 }
346
347 fn min_quantity(&self) -> Option<Quantity> {
348 self.min_quantity
349 }
350
351 fn max_notional(&self) -> Option<Money> {
352 None
353 }
354
355 fn min_notional(&self) -> Option<Money> {
356 None
357 }
358
359 fn max_price(&self) -> Option<Price> {
360 self.max_price
361 }
362
363 fn min_price(&self) -> Option<Price> {
364 self.min_price
365 }
366
367 fn tick_scheme(&self) -> Option<Ustr> {
368 self.tick_scheme
369 }
370
371 fn info(&self) -> Option<&Params> {
372 self.info.as_ref()
373 }
374
375 fn ts_event(&self) -> UnixNanos {
376 self.ts_event
377 }
378
379 fn ts_init(&self) -> UnixNanos {
380 self.ts_init
381 }
382
383 fn margin_init(&self) -> Decimal {
384 self.margin_init
385 }
386
387 fn margin_maint(&self) -> Decimal {
388 self.margin_maint
389 }
390}
391
392#[cfg(test)]
393mod tests {
394 use rstest::rstest;
395 use rust_decimal_macros::dec;
396 use ustr::Ustr;
397
398 use crate::{
399 enums::{AssetClass, InstrumentClass},
400 identifiers::{InstrumentId, Symbol},
401 instruments::{Instrument, OptionSpread, stubs::*},
402 types::{Currency, Price, Quantity},
403 };
404
405 #[rstest]
406 fn test_trait_accessors(option_spread: OptionSpread) {
407 assert_eq!(
408 option_spread.id(),
409 InstrumentId::from("UD:U$: GN 2534559.GLBX")
410 );
411 assert_eq!(option_spread.asset_class(), AssetClass::FX);
412 assert_eq!(
413 option_spread.instrument_class(),
414 InstrumentClass::OptionSpread
415 );
416 assert_eq!(option_spread.quote_currency(), Currency::USD());
417 assert!(!option_spread.is_inverse());
418 assert_eq!(option_spread.exchange(), Some(Ustr::from("XCME")));
419 assert_eq!(option_spread.size_precision(), 0);
420 assert_eq!(option_spread.size_increment(), Quantity::from("1"));
421 assert_eq!(option_spread.min_quantity(), Some(Quantity::from("1")));
422 }
423
424 #[rstest]
425 fn test_new_checked_price_precision_mismatch() {
426 let result = OptionSpread::new_checked(
427 InstrumentId::from("TEST.GLBX"),
428 Symbol::from("TEST"),
429 AssetClass::FX,
430 Some(Ustr::from("XCME")),
431 Ustr::from("SR3"),
432 Ustr::from("GN"),
433 0.into(),
434 0.into(),
435 Currency::USD(),
436 4, Price::from("0.01"),
438 Quantity::from(1),
439 Quantity::from(1),
440 None,
441 None,
442 None,
443 None,
444 None,
445 None,
446 None,
447 None,
448 0.into(),
449 0.into(),
450 );
451 assert!(result.is_err());
452 }
453
454 #[rstest]
455 fn test_serialization_roundtrip(option_spread: OptionSpread) {
456 let json = serde_json::to_string(&option_spread).unwrap();
457 let deserialized: OptionSpread = serde_json::from_str(&json).unwrap();
458 assert_eq!(json, serde_json::to_string(&deserialized).unwrap());
459 }
460
461 #[rstest]
462 fn test_builder_matches_new_checked() {
463 let positional = OptionSpread::new_checked(
464 InstrumentId::from("UD:U$: GN 2534559.GLBX"),
465 Symbol::from("UD:U$: GN 2534559"),
466 AssetClass::FX,
467 Some(Ustr::from("XCME")),
468 Ustr::from("SR3"),
469 Ustr::from("GN"),
470 1.into(),
471 2.into(),
472 Currency::USD(),
473 2,
474 Price::from("0.01"),
475 Quantity::from(10),
476 Quantity::from(5),
477 Some(Quantity::from("100")),
478 Some(Quantity::from("1")),
479 Some(Price::from("999.0")),
480 Some(Price::from("1.0")),
481 Some(dec!(0.01)),
482 Some(dec!(0.02)),
483 None,
484 None,
485 3.into(),
486 4.into(),
487 )
488 .unwrap();
489
490 let built = OptionSpread::builder()
491 .instrument_id(InstrumentId::from("UD:U$: GN 2534559.GLBX"))
492 .raw_symbol(Symbol::from("UD:U$: GN 2534559"))
493 .asset_class(AssetClass::FX)
494 .exchange(Ustr::from("XCME"))
495 .underlying(Ustr::from("SR3"))
496 .strategy_type(Ustr::from("GN"))
497 .activation_ns(1.into())
498 .expiration_ns(2.into())
499 .currency(Currency::USD())
500 .price_precision(2)
501 .price_increment(Price::from("0.01"))
502 .multiplier(Quantity::from(10))
503 .lot_size(Quantity::from(5))
504 .max_quantity(Quantity::from("100"))
505 .min_quantity(Quantity::from("1"))
506 .max_price(Price::from("999.0"))
507 .min_price(Price::from("1.0"))
508 .margin_init(dec!(0.01))
509 .margin_maint(dec!(0.02))
510 .ts_event(3.into())
511 .ts_init(4.into())
512 .build()
513 .unwrap();
514
515 assert_eq!(
516 serde_json::to_value(&positional).unwrap(),
517 serde_json::to_value(&built).unwrap(),
518 );
519 }
520}