--

lyrixx · lyrixx · commit 6f8e66c5f23c · 2018-11-08T22:33:05.000+01:00
diff --git a/src/Symfony/Component/Workflow/MarkingStore/MethodSingleStateMarkingStore.php b/src/Symfony/Component/Workflow/MarkingStore/MethodSingleStateMarkingStore.php
@@ -0,0 +1,59 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Workflow\MarkingStore;
+
+use Symfony\Component\PropertyAccess\PropertyAccess;
+use Symfony\Component\PropertyAccess\PropertyAccessorInterface;
+
+/**
+ * SingleStateMarkingStore stores the marking with a method of the subject.
+ *
+ * This store deals with a "single state" Marking. It means a subject can be in
+ * one and only one state at the same time.
+ *
+ * The constructor accept a `property` name.
+ * The `getMarking` method will use `$subject->getProperty()`.
+ * The `setMarking` method will use `$subject->setProperty($places, $context)`.
+ *
+ * @author Grégoire Pineau <lyrixx@lyrixx.info>
+ */
+class MethodSingleStateMarkingStore implements MarkingStoreInterface
+{
+    private $property;
+
+    public function __construct(string $property = 'setMarking')
+    {
+        $this->property = $property;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getMarking($subject)
+    {
+        $placeName = $subject->{'get'.ucfirst($this->property)}();
+
+        if (!$placeName) {
+            return new Marking();
+        }
+
+        return new Marking(array($placeName => 1));
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function setMarking($subject, Marking $marking, array $context = array())
+    {
+        $subject->{'set'.ucfirst($this->property)}(key($marking->getPlaces()), $context);
+    }
+}
