1 files changed, 7 insertions, 7 deletions
diff --git a/battleNetApi/include/tokenBuilder.hpp b/battleNetApi/include/tokenBuilder.hpp
index 71286cd..2893728 100644
--- a/battleNetApi/include/tokenBuilder.hpp
+++ b/battleNetApi/include/tokenBuilder.hpp
@@ -1,12 +1,12 @@
+#pragma once
+
+#include <memory>
 #include <token.hpp>
 
 class TokenBuilder {
   public:
-    TokenBuilder();
-    [[nodiscard]]
-    Token build() const;
-    TokenBuilder &allowSourceFileUpdate();
-
-  private:
-    bool mAllowSourceFileUpdate{false};
+    TokenBuilder() = default;
+    virtual ~TokenBuilder() = default;
+    virtual TokenBuilder &buildToken() = 0;
+    virtual std::shared_ptr<Token> getToken() = 0;
 };