Thread: Set string pool thread

Author: adazem009

include/scratchcpp/thread.h

@@ -20,6 +20,7 @@ class LIBSCRATCHCPP_EXPORT Thread
     public:
         Thread(Target *target, IEngine *engine, Script *script);
         Thread(const Thread &) = delete;
+        ~Thread();
 
         Target *target() const;
         IEngine *engine() const;

src/engine/thread.cpp

@@ -6,13 +6,16 @@
 #include <scratchcpp/executioncontext.h>
 
 #include "thread_p.h"
+#include "scratch/string_pool_p.h"
 
 using namespace libscratchcpp;
 
 /*! Constructs Thread. */
 Thread::Thread(Target *target, IEngine *engine, Script *script) :
     impl(spimpl::make_unique_impl<ThreadPrivate>(target, engine, script))
 {
+    string_pool_add_thread(this);
+
     if (impl->script) {
         impl->code = impl->script->code();
 
@@ -21,6 +24,12 @@ Thread::Thread(Target *target, IEngine *engine, Script *script) :
     }
 }
 
+/*! Destroys Thread. */
+Thread::~Thread()
+{
+    string_pool_remove_thread(this);
+}
+
 /*! Returns the Target of the script. */
 Target *Thread::target() const
 {
@@ -42,7 +51,9 @@ Script *Thread::script() const
 /*! Runs the script until it finishes or yields. */
 void Thread::run()
 {
+    string_pool_set_thread(this);
     impl->code->run(impl->executionContext.get());
+    string_pool_set_thread(nullptr);
 }
 
 /*! Stops the script. */