Implement control_stop

Author: adazem009

src/dev/blocks/controlblocks.cpp

@@ -5,6 +5,9 @@
 #include <scratchcpp/dev/compilerconstant.h>
 #include <scratchcpp/value.h>
 #include <scratchcpp/input.h>
+#include <scratchcpp/field.h>
+#include <scratchcpp/dev/executioncontext.h>
+#include <scratchcpp/thread.h>
 
 #include "controlblocks.h"
 
@@ -26,6 +29,7 @@ void ControlBlocks::registerBlocks(IEngine *engine)
     engine->addCompileFunction(this, "control_repeat", &compileRepeat);
     engine->addCompileFunction(this, "control_if", &compileIf);
     engine->addCompileFunction(this, "control_if_else", &compileIfElse);
+    engine->addCompileFunction(this, "control_stop", &compileStop);
 }
 
 CompilerValue *ControlBlocks::compileForever(Compiler *compiler)
@@ -57,3 +61,32 @@ CompilerValue *ControlBlocks::compileIfElse(Compiler *compiler)
     compiler->moveToIfElse(compiler->addInput("CONDITION"), substack ? substack->valueBlock() : nullptr, substack2 ? substack2->valueBlock() : nullptr);
     return nullptr;
 }
+
+CompilerValue *ControlBlocks::compileStop(Compiler *compiler)
+{
+    Field *option = compiler->field("STOP_OPTION");
+
+    if (option) {
+        std::string str = option->value().toString();
+
+        if (str == "all")
+            compiler->addFunctionCallWithCtx("control_stop_all", Compiler::StaticType::Void);
+        else if (str == "this script")
+            compiler->createStop();
+        else if (str == "other scripts in sprite" || str == "other scripts in stage")
+            compiler->addFunctionCallWithCtx("control_stop_other_scripts_in_target", Compiler::StaticType::Void);
+    }
+
+    return nullptr;
+}
+
+extern "C" void control_stop_all(ExecutionContext *ctx)
+{
+    ctx->engine()->stop();
+}
+
+extern "C" void control_stop_other_scripts_in_target(ExecutionContext *ctx)
+{
+    Thread *thread = ctx->thread();
+    ctx->engine()->stopTarget(thread->target(), thread);
+}

src/dev/blocks/controlblocks.h

@@ -20,6 +20,7 @@ class ControlBlocks : public IExtension
         static CompilerValue *compileRepeat(Compiler *compiler);
         static CompilerValue *compileIf(Compiler *compiler);
         static CompilerValue *compileIfElse(Compiler *compiler);
+        static CompilerValue *compileStop(Compiler *compiler);
 };
 
 } // namespace libscratchcpp

test/dev/blocks/control_blocks_test.cpp

@@ -4,6 +4,8 @@
 #include <scratchcpp/sprite.h>
 #include <scratchcpp/block.h>
 #include <scratchcpp/input.h>
+#include <scratchcpp/script.h>
+#include <scratchcpp/thread.h>
 #include <enginemock.h>
 
 #include "../common.h"
@@ -250,3 +252,83 @@ TEST_F(ControlBlocksTest, IfElse)
         builder.run();
     }
 }
+
+TEST_F(ControlBlocksTest, Stop)
+{
+    auto target = std::make_shared<Sprite>();
+
+    // Stop all
+    {
+        ScriptBuilder builder(m_extension.get(), m_engine, target);
+
+        builder.addBlock("control_stop");
+        builder.addDropdownField("STOP_OPTION", "all");
+        auto block = builder.currentBlock();
+
+        Compiler compiler(&m_engineMock, target.get());
+        auto code = compiler.compile(block);
+        Script script(target.get(), block, &m_engineMock);
+        script.setCode(code);
+        Thread thread(target.get(), &m_engineMock, &script);
+
+        EXPECT_CALL(m_engineMock, stop());
+        thread.run();
+    }
+
+    m_engine->clear();
+    target = std::make_shared<Sprite>();
+
+    // Stop this script
+    {
+        ScriptBuilder builder(m_extension.get(), m_engine, target);
+
+        builder.addBlock("control_stop");
+        builder.addDropdownField("STOP_OPTION", "this script");
+        builder.addBlock("test_print_test");
+
+        builder.build();
+
+        testing::internal::CaptureStdout();
+        builder.run();
+        ASSERT_TRUE(testing::internal::GetCapturedStdout().empty());
+    }
+
+    m_engine->clear();
+    target = std::make_shared<Sprite>();
+
+    // Stop other scripts in sprite
+    {
+        ScriptBuilder builder(m_extension.get(), m_engine, target);
+
+        builder.addBlock("control_stop");
+        builder.addDropdownField("STOP_OPTION", "other scripts in sprite");
+        auto block = builder.currentBlock();
+
+        Compiler compiler(&m_engineMock, target.get());
+        auto code = compiler.compile(block);
+        Script script(target.get(), block, &m_engineMock);
+        script.setCode(code);
+        Thread thread(target.get(), &m_engineMock, &script);
+
+        EXPECT_CALL(m_engineMock, stopTarget(target.get(), &thread));
+        thread.run();
+    }
+
+    // Stop other scripts in stage
+    {
+        ScriptBuilder builder(m_extension.get(), m_engine, target);
+
+        builder.addBlock("control_stop");
+        builder.addDropdownField("STOP_OPTION", "other scripts in stage");
+        auto block = builder.currentBlock();
+
+        Compiler compiler(&m_engineMock, target.get());
+        auto code = compiler.compile(block);
+        Script script(target.get(), block, &m_engineMock);
+        script.setCode(code);
+        Thread thread(target.get(), &m_engineMock, &script);
+
+        EXPECT_CALL(m_engineMock, stopTarget(target.get(), &thread));
+        thread.run();
+    }
+}