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
146mod commands;
use dashmap::DashSet;
use poise::{FrameworkOptions, serenity_prelude as serenity};
use scraper::Selector;
use std::{collections::HashSet, env::var, sync::Arc};
use tracing::{error, info};
use crate::commands::Site;
type Error = Box<dyn std::error::Error + Send + Sync>;
type Context<'a> = poise::Context<'a, Data, Error>;
pub struct Data {
subscribers: Arc<DashSet<serenity::UserId>>,
}
#[tokio::main]
async fn main() -> Result<(), Error> {
tracing_subscriber::fmt()
.with_max_level(if cfg!(debug_assertions) {
tracing::Level::DEBUG
} else {
tracing::Level::WARN
})
.init();
let options: FrameworkOptions<Data, Error> = poise::FrameworkOptions {
commands: vec![commands::subscribe(), commands::unsubscribe()],
pre_command: |ctx| {
Box::pin(async move {
info!("Executing command {}...", ctx.command().qualified_name);
})
},
post_command: |ctx| {
Box::pin(async move {
info!("Executed command {}!", ctx.command().qualified_name);
})
},
event_handler: |_ctx, event, _framework, _data| {
Box::pin(async move {
info!(
"Got an event in event handler: {:?}",
event.snake_case_name()
);
Ok(())
})
},
..Default::default()
};
let framework = poise::Framework::builder()
.setup(move |ctx, _ready, framework| {
Box::pin(async move {
info!("Logged in as {}", _ready.user.name);
poise::builtins::register_globally(ctx, &framework.options().commands).await?;
let subscribers: Arc<DashSet<serenity::UserId>> =
Arc::new(match tokio::fs::read_to_string("subscribers.json").await {
Ok(content) => {
match serde_json::from_str::<DashSet<serenity::UserId>>(&content) {
Ok(subscribers) => subscribers,
Err(e) => {
error!("Failed to parse subscribers.json: {}", e);
DashSet::new()
}
}
}
Err(e) => {
error!("Failed to read subscribers.json: {}", e);
DashSet::new()
}
});
let sites = [
Site {
url: "https://global.ippodo-tea.co.jp/collections/matcha",
product_card_selector: Selector::parse("li.m-product-card:not(:has(button.out-of-stock))").unwrap(),
name_selector: Selector::parse(".m-product-card__name a").unwrap(),
href_selector: Selector::parse(".m-product-card__name a").unwrap(),
base_url: "https://global.ippodo-tea.co.jp",
matchas_in_stock: HashSet::new(),
},
Site {
url: "https://global.ippodo-tea.co.jp/collections/utensils",
product_card_selector: Selector::parse("li.m-product-card:not(:has(button.out-of-stock))").unwrap(),
name_selector: Selector::parse(".m-product-card__name a").unwrap(),
href_selector: Selector::parse(".m-product-card__name a").unwrap(),
base_url: "https://global.ippodo-tea.co.jp",
matchas_in_stock: HashSet::new(),
},
Site {
url: "https://www.marukyu-koyamaen.co.jp/english/shop/products/catalog/matcha?viewall=1",
product_card_selector: Selector::parse("li.instock").unwrap(),
name_selector: Selector::parse(".product-name h4").unwrap(),
href_selector: Selector::parse("a.woocommerce-loop-product__link").unwrap(),
base_url: "",
matchas_in_stock: HashSet::new(),
},
Site {
url: "https://www.marukyu-koyamaen.co.jp/english/shop/products/catalog/sweets?viewall=1",
product_card_selector: Selector::parse("li.instock").unwrap(),
name_selector: Selector::parse(".product-name h4").unwrap(),
href_selector: Selector::parse("a.woocommerce-loop-product__link").unwrap(),
base_url: "",
matchas_in_stock: HashSet::new(),
},
Site {
url: "https://www.marukyu-koyamaen.co.jp/english/shop/products/catalog/others?viewall=1",
product_card_selector: Selector::parse("li.instock").unwrap(),
name_selector: Selector::parse(".product-name h4").unwrap(),
href_selector: Selector::parse("a.woocommerce-loop-product__link").unwrap(),
base_url: "",
matchas_in_stock: HashSet::new(),
},
Site {
url: "https://www.sazentea.com/en/products/c22-ceremonial-grade-matcha",
product_card_selector: Selector::parse("div.product").unwrap(),
name_selector: Selector::parse(".product-name a").unwrap(),
href_selector: Selector::parse(".product-name a").unwrap(),
base_url: "https://www.sazentea.com",
matchas_in_stock: HashSet::new(),
}
];
for site in sites {
tokio::spawn(commands::watch_matcha(ctx.clone(), subscribers.clone(), site));
}
Ok(Data {
subscribers: subscribers.clone(),
})
})
})
.options(options)
.build();
let token =
var("DISCORD_TOKEN").map_err(|e| format!("Missing `DISCORD_TOKEN` env var: {}", e))?;
let intents = serenity::GatewayIntents::non_privileged();
let mut client = serenity::ClientBuilder::new(token, intents)
.framework(framework)
.await?;
client.start().await?;
Ok(())
}