feat: ✨ add OWNER_ID environment variable

jontitorr · Dec 11, 2023 · 5c12389 · 5c12389
1 parent 7fd4e99
commit 5c12389
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/include/saber/saber.hpp b/include/saber/saber.hpp
@@ -10,7 +10,7 @@
 
 namespace saber {
 struct Saber {
-	explicit Saber(std::string_view token);
+	explicit Saber(std::string_view token, ekizu::Snowflake owner_id);
 
 	void run();
 	void handle_event(ekizu::Event ev);
@@ -20,7 +20,7 @@ struct Saber {
 	ekizu::HttpClient http;
 	std::shared_ptr<spdlog::logger> logger;
 	ekizu::LruCache<ekizu::Snowflake, ekizu::Message> messages_cache{500};
-	ekizu::Snowflake owner_id{155780111197536256};
+	ekizu::Snowflake owner_id;
 	std::string prefix{">"};
 	ekizu::Shard shard;
 	ekizu::CurrentUser user;

diff --git a/src/main.cpp b/src/main.cpp
@@ -1,6 +1,8 @@
 #include <iostream>
 #include <saber/saber.hpp>
 
+static constexpr uint64_t DEFAULT_OWNER_ID{155780111197536256};
+
 int main() {
 	const auto* token = std::getenv("DISCORD_TOKEN");
 
@@ -9,7 +11,11 @@ int main() {
 		return 1;
 	}
 
-	auto saber = saber::Saber{token};
+	const auto* owner_id_str = std::getenv("OWNER_ID");
+	ekizu::Snowflake owner_id{
+		owner_id_str != nullptr ? std::stoull(owner_id_str) : DEFAULT_OWNER_ID};
+
+	auto saber = saber::Saber{token, owner_id};
 
 	saber.run();
 }
diff --git a/src/saber.cpp b/src/saber.cpp
@@ -11,9 +11,10 @@ overload(Func...) -> overload<Func...>;
 }  // namespace
 
 namespace saber {
-Saber::Saber(std::string_view token)
+Saber::Saber(std::string_view token, ekizu::Snowflake owner_id_)
 	: commands{this},
 	  http{token},
+	  owner_id{owner_id_},
 	  shard{ekizu::ShardId::ONE, token, ekizu::Intents::AllIntents} {
 	logger = spdlog::stdout_color_mt("saber");
 	spdlog::set_pattern("[%Y-%m-%d %H:%M:%S.%e] [%^%l%$] %v");