Add some methods to WorkflowContext

Author: sandrokeil

src/Workflow/WorkflowContext.php

@@ -18,6 +18,14 @@ interface WorkflowContext
      */
     public function get(string $slotName);
 
+    public function has(string $slotName): bool;
+
+    /**
+     * @param string $slotName
+     * @param string|null $type Checks also slot type
+     */
+    public function assertSlot(string $slotName, string $type = null): void;
+
     /**
      * Returns input data for given description
      *
@@ -31,4 +39,9 @@ public function getByDescription(DescriptionWithInputSlot $description): array;
      * @param mixed $slotValue
      */
     public function put(string $slotName, $slotValue): void;
+
+    /**
+     * @return string[]
+     */
+    public function getSlotNames(): array;
 }

src/Workflow/WorkflowContextMap.php

@@ -10,6 +10,9 @@
 
 namespace OpenCodeModeling\CodeGenerator\Workflow;
 
+use OpenCodeModeling\CodeGenerator\Exception\MissingSlotName;
+use OpenCodeModeling\CodeGenerator\Exception\WrongSlotType;
+
 final class WorkflowContextMap implements WorkflowContext
 {
     /**
@@ -27,6 +30,53 @@ public function put(string $slotName, $slotValue): void
         $this->map[$slotName] = $slotValue;
     }
 
+    public function has(string $slotName): bool
+    {
+        return isset($this->map[$slotName]);
+    }
+
+    public function assertSlot(string $slotName, string $type = null): void
+    {
+        if (! $this->has($slotName)) {
+            throw new MissingSlotName($slotName);
+        }
+        if ($type !== null) {
+            $slot = $this->get($slotName);
+
+            switch ($type) {
+                case 'int':
+                case 'integer':
+                    $isType = \is_int($slot);
+                    break;
+                case 'string':
+                    $isType = \is_string($slot);
+                    break;
+                case 'bool':
+                case 'boolean':
+                    $isType = \is_bool($slot);
+                    break;
+                case 'array':
+                    $isType = \is_array($slot);
+                    break;
+                default:
+                    $isType = $slot instanceof $type;
+                    break;
+            }
+
+            if ($isType === false) {
+                throw WrongSlotType::withSlotName($slotName, $type, (\is_object($slot) ? \get_class($slot) : \gettype($slot)));
+            }
+        }
+    }
+
+    /**
+     * @return string[]
+     */
+    public function getSlotNames(): array
+    {
+        return \array_keys($this->map);
+    }
+
     public function getByDescription(DescriptionWithInputSlot $description): array
     {
         $input = [];