Implement sensing_setdragmode block

Author: adazem009

src/blocks/sensingblocks.cpp

@@ -8,6 +8,7 @@
 #include <scratchcpp/executioncontext.h>
 #include <scratchcpp/value.h>
 #include <scratchcpp/input.h>
+#include <scratchcpp/field.h>
 #include <scratchcpp/sprite.h>
 #include <scratchcpp/textbubble.h>
 #include <scratchcpp/stringptr.h>
@@ -46,6 +47,7 @@ void SensingBlocks::registerBlocks(IEngine *engine)
     engine->addCompileFunction(this, "sensing_mousedown", &compileMouseDown);
     engine->addCompileFunction(this, "sensing_mousex", &compileMouseX);
     engine->addCompileFunction(this, "sensing_mousey", &compileMouseY);
+    engine->addCompileFunction(this, "sensing_setdragmode", &compileSetDragMode);
 }
 
 void SensingBlocks::onInit(IEngine *engine)
@@ -185,6 +187,28 @@ CompilerValue *SensingBlocks::compileMouseY(Compiler *compiler)
     return compiler->addFunctionCallWithCtx("sensing_mousey", Compiler::StaticType::Number);
 }
 
+CompilerValue *SensingBlocks::compileSetDragMode(Compiler *compiler)
+{
+    if (compiler->target()->isStage())
+        return nullptr;
+
+    Field *field = compiler->field("DRAG_MODE");
+    assert(field);
+
+    std::string mode = field->value().toString();
+    CompilerValue *draggable = nullptr;
+
+    if (mode == "draggable")
+        draggable = compiler->addConstValue(true);
+    else if (mode == "not draggable")
+        draggable = compiler->addConstValue(false);
+    else
+        return nullptr;
+
+    compiler->addTargetFunctionCall("sensing_setdragmode", Compiler::StaticType::Void, { Compiler::StaticType::Bool }, { draggable });
+    return nullptr;
+}
+
 void SensingBlocks::onAnswer(const std::string &answer)
 {
     // https://github.com/scratchfoundation/scratch-vm/blob/6055823f203a696165084b873e661713806583ec/src/blocks/scratch3_sensing.js#L99-L115
@@ -361,3 +385,8 @@ extern "C" double sensing_mousey(ExecutionContext *ctx)
 {
     return ctx->engine()->mouseY();
 }
+
+extern "C" void sensing_setdragmode(Sprite *sprite, bool draggable)
+{
+    sprite->setDraggable(draggable);
+}

src/blocks/sensingblocks.h

@@ -52,6 +52,7 @@ class SensingBlocks : public IExtension
         static CompilerValue *compileMouseDown(Compiler *compiler);
         static CompilerValue *compileMouseX(Compiler *compiler);
         static CompilerValue *compileMouseY(Compiler *compiler);
+        static CompilerValue *compileSetDragMode(Compiler *compiler);
 
         static void onAnswer(const std::string &answer);
         static void enqueueAsk(const std::string &question, Thread *thread);

test/blocks/sensing_blocks_test.cpp

@@ -1297,3 +1297,68 @@ TEST_F(SensingBlocksTest, MouseY)
     ASSERT_EQ(value_toDouble(&value), -78.21);
     value_free(&value);
 }
+
+TEST_F(SensingBlocksTest, SetDragMode_Draggable)
+{
+    auto sprite = std::make_shared<Sprite>();
+    sprite->setDraggable(false);
+
+    ScriptBuilder builder(m_extension.get(), m_engine, sprite);
+    builder.addBlock("sensing_setdragmode");
+    builder.addDropdownField("DRAG_MODE", "draggable");
+    builder.build();
+
+    builder.run();
+    ASSERT_TRUE(sprite->draggable());
+
+    builder.run();
+    ASSERT_TRUE(sprite->draggable());
+}
+
+TEST_F(SensingBlocksTest, SetDragMode_NotDraggable)
+{
+    auto sprite = std::make_shared<Sprite>();
+    sprite->setDraggable(true);
+
+    ScriptBuilder builder(m_extension.get(), m_engine, sprite);
+    builder.addBlock("sensing_setdragmode");
+    builder.addDropdownField("DRAG_MODE", "not draggable");
+    builder.build();
+
+    builder.run();
+    ASSERT_FALSE(sprite->draggable());
+
+    builder.run();
+    ASSERT_FALSE(sprite->draggable());
+}
+
+TEST_F(SensingBlocksTest, SetDragMode_Invalid)
+{
+    auto sprite = std::make_shared<Sprite>();
+    sprite->setDraggable(true);
+
+    ScriptBuilder builder(m_extension.get(), m_engine, sprite);
+    builder.addBlock("sensing_setdragmode");
+    builder.addDropdownField("DRAG_MODE", "lorem ipsum");
+    builder.build();
+
+    builder.run();
+    ASSERT_TRUE(sprite->draggable());
+
+    sprite->setDraggable(false);
+
+    builder.run();
+    ASSERT_FALSE(sprite->draggable());
+}
+
+TEST_F(SensingBlocksTest, SetDragMode_Stage)
+{
+    auto stage = std::make_shared<Stage>();
+
+    ScriptBuilder builder(m_extension.get(), m_engine, stage);
+    builder.addBlock("sensing_setdragmode");
+    builder.addDropdownField("DRAG_MODE", "draggable");
+    builder.build();
+
+    builder.run();
+}