risq 0.4.1

Re-implementation of Bisq (https://github.com/bisq-network/bisq) in rust
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

use super::{message::*, *};
use crate::{
    bisq::SequencedMessageHash,
    domain::{price_feed::*, CommandResult},
    prelude::*,
};
use std::{collections::HashMap, sync::Arc, time::Duration};

const CHECK_TTL_INTERVAL: Duration = Duration::from_secs(40);

pub struct OfferBook {
    open_offers: Arc<HashMap<SequencedMessageHash, OpenOffer>>,
    price_feed: Addr<PriceFeed>,
    price_data: Arc<HashMap<&'static str, PriceData>>,
}
impl Actor for OfferBook {
    type Context = Context<Self>;
    fn started(&mut self, ctx: &mut Self::Context) {
        ctx.run_interval(CHECK_TTL_INTERVAL, |offer_book, ctx| {
            ctx.spawn(
                fut::wrap_future(offer_book.price_feed.send(GetCurrentPrices)).then(
                    |maybe_data, offer_book: &mut OfferBook, _| {
                        if let Ok(price_data) = maybe_data {
                            offer_book.price_data = price_data;
                        }
                        let open_offers = offer_book
                            .open_offers
                            .iter()
                            .filter_map(|(hash, offer)| {
                                if offer.is_expired() {
                                    None
                                } else {
                                    let mut offer = offer.clone();
                                    offer.update_display_price(&offer_book.price_data);
                                    Some((*hash, offer))
                                }
                            })
                            .collect();
                        offer_book.open_offers = Arc::new(open_offers);
                        fut::ok(())
                    },
                ),
            );
        });
    }
}
impl OfferBook {
    pub fn start(price_feed: Addr<PriceFeed>) -> Addr<OfferBook> {
        OfferBook {
            open_offers: Arc::new(HashMap::new()),
            price_feed,
            price_data: Arc::new(HashMap::new()),
        }
        .start()
    }
}

impl Handler<AddOffer> for OfferBook {
    type Result = MessageResult<AddOffer>;
    fn handle(&mut self, AddOffer(mut offer): AddOffer, _ctx: &mut Self::Context) -> Self::Result {
        if !offer.is_expired() {
            offer.update_display_price(&self.price_data);
            match self.open_offers.get(&offer.bisq_hash) {
                None => {
                    info!("Adding {:?}", offer.id);
                    let offers = Arc::make_mut(&mut self.open_offers);
                    offers.insert(offer.bisq_hash, offer);
                    return MessageResult(CommandResult::Accepted);
                }
                Some(existing) if existing.would_refresh(offer.latest_sequence) => {
                    let offers = Arc::make_mut(&mut self.open_offers);
                    offers.insert(offer.bisq_hash, offer);
                    return MessageResult(CommandResult::Accepted);
                }
                _ => (),
            }
        }
        MessageResult(CommandResult::Ignored)
    }
}
impl Handler<RemoveOffer> for OfferBook {
    type Result = MessageResult<RemoveOffer>;
    fn handle(
        &mut self,
        RemoveOffer(offer): RemoveOffer,
        _ctx: &mut Self::Context,
    ) -> Self::Result {
        match self.open_offers.get(&offer.bisq_hash) {
            None => MessageResult(CommandResult::Ignored),
            Some(_) => {
                info!("Removing {:?}", offer.id);
                let offers = Arc::make_mut(&mut self.open_offers);
                offers.remove(&offer.bisq_hash);
                MessageResult(CommandResult::Accepted)
            }
        }
    }
}
impl Handler<RefreshOffer> for OfferBook {
    type Result = MessageResult<RefreshOffer>;
    fn handle(
        &mut self,
        RefreshOffer {
            bisq_hash,
            sequence,
        }: RefreshOffer,
        _ctx: &mut Self::Context,
    ) -> Self::Result {
        if let Some(offer) = self.open_offers.get(&bisq_hash) {
            if offer.would_refresh(sequence) {
                let offers = Arc::make_mut(&mut self.open_offers);
                let offer = offers.get_mut(&bisq_hash).unwrap();
                if offer.refresh(sequence) {
                    return MessageResult(CommandResult::Accepted);
                }
            }
        }
        MessageResult(CommandResult::Ignored)
    }
}

impl Handler<GetOpenOffers> for OfferBook {
    type Result = MessageResult<GetOpenOffers>;
    fn handle(&mut self, _: GetOpenOffers, _ctx: &mut Self::Context) -> Self::Result {
        MessageResult(Arc::clone(&self.open_offers))
    }
}